Views
No views yet
<think>...</think> blocks prior to rendering the final, uncensored technical execution or exploit payload.1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
3
4model_id = "galahad-mamad/GambronAI-Persian-Cybersecurity-r1"
5
6tokenizer = AutoTokenizer.from_pretrained(model_id)
7model = AutoModelForCausalLM.from_pretrained(
8 model_id,
9 torch_dtype=torch.float16,
10 device_map="auto"
11)
12
13streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=False)
14
15print("\n" + "="*50)
16print("GambronAI Core Engine Initialized Successfully.")
17print("="*50 + "\n")
18
19chat_history = []
20
21while True:
22 user_input = input("User: ")
23 if user_input.strip().lower() in ['exit', 'quit']:
24 break
25
26 chat_history.append({"role": "user", "content": user_input})
27
28 full_prompt = tokenizer.apply_chat_template(
29 chat_history,
30 tokenize=False,
31 add_generation_prompt=True
32 )
33
34 inputs = tokenizer([full_prompt], return_tensors="pt").to("cuda")
35
36 print("\nGambronAI:")
37 print("-" * 30)
38
39 with torch.no_grad():
40 _ = model.generate(
41 **inputs,
42 max_new_tokens=1536,
43 temperature=0.7,
44 do_sample=True,
45 top_p=0.95,
46 streamer=streamer,
47 pad_token_id=tokenizer.eos_token_id
48 )
49 print("\n" + "-" * 30 + "\n")