Views
No views yet
Qwen/Qwen2.5-3B-Instruct.1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4base_model = AutoModelForCausalLM.from_pretrained(
5 "Qwen/Qwen2.5-3B-Instruct",
6 device_map="auto",
7 torch_dtype="auto",
8 trust_remote_code=True
9)
10
11model = PeftModel.from_pretrained(
12 base_model,
13 "Joshua0522/qwen25-3b-mental-health-itemgen-lora"
14)
15
16tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-3B-Instruct")
17
18prompt = """
19Generate exactly 5 true/false depression questionnaire items.
20"""
21
22inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
23outputs = model.generate(**inputs, max_new_tokens=200)
24
25print(tokenizer.decode(outputs[0], skip_special_tokens=True))