Views
No views yet
1!pip install -q "torchao>=0.16.0"
2
3from transformers import AutoTokenizer, AutoModelForCausalLM
4from peft import PeftModel
5import torch
6
7tokenizer = AutoTokenizer.from_pretrained("jpo89/qwen2.5-3b-sft-dapo")
8
9model = AutoModelForCausalLM.from_pretrained(
10 "Qwen/Qwen2.5-3B-Instruct",
11 dtype=torch.bfloat16,
12 device_map="auto",
13)
14
15model = PeftModel.from_pretrained(model, "jpo89/qwen2.5-3b-sft-dapo")
16model.eval()
17print("Model ready.")
1def test(question):
2 messages = [{"role": "user", "content": question}]
3 inputs = tokenizer.apply_chat_template(
4 messages,
5 add_generation_prompt=True,
6 return_tensors="pt",
7 return_dict=True,
8 ).to(model.device)
9
10 with torch.no_grad():
11 output = model.generate(
12 **inputs,
13 max_new_tokens=256,
14 do_sample=True,
15 temperature=0.4,
16 top_p=0.9,
17 repetition_penalty=1.1,
18 pad_token_id=tokenizer.eos_token_id,
19 )
20
21 response = tokenizer.decode(
22 output[0][inputs["input_ids"].shape[-1]:],
23 skip_special_tokens=True
24 )
25 print(f"Q: {question}")
26 print(f"A: {response}")
27 print("-" * 60)
28
29# Test with a few questions
30test("If x + 5 = 12, what is x?")
31test("What is the sum of the first 10 natural numbers?")
32test("A train travels 60 km/h for 2.5 hours. How far does it travel?")