Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3import torch
4
5base_model = AutoModelForCausalLM.from_pretrained(
6 "Qwen/Qwen3-1.7B",
7 torch_dtype=torch.float16,
8 device_map="cuda",
9)
10
11tokenizer = AutoTokenizer.from_pretrained("CoNDeNse-AI/GLM-5.1-Qwen3-1.7B-CoNDeNse")
12model = PeftModel.from_pretrained(base_model, "CoNDeNse-AI/GLM-5.1-Qwen3-1.7B-CoNDeNse")
13
14prompt = "<|im_start|>user\nYour question here<|im_end|>\n<|im_start|>assistant\n"
15inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
16
17with torch.no_grad():
18 outputs = model.generate(**inputs, max_new_tokens=512, temperature=0.7, do_sample=True)
19
20print(tokenizer.decode(outputs[0], skip_special_tokens=True))