Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3import torch
4
5base_id = "Qwen/Qwen3-4B-Instruct-2507"
6adapter_id = "your_id/your-adapter-repo"
7
8model = AutoModelForCausalLM.from_pretrained(
9 base_id,
10 torch_dtype=torch.float16,
11 device_map="auto",
12 trust_remote_code=True,
13)
14model = PeftModel.from_pretrained(model, adapter_id)
15
16# Test inference
17prompt = "Your question here"
18tokenizer = AutoTokenizer.from_pretrained(base_id, trust_remote_code=True)
19inputs = tokenizer.apply_chat_template([{"role": "user", "content": prompt}], tokenize=True, add_generation_prompt=True, return_tensors="pt").to("cuda")
20outputs = model.generate(**inputs, max_new_tokens=512)
21print(tokenizer.decode(outputs[0]))