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