Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4# Load base model
5base_model = AutoModelForCausalLM.from_pretrained(
6 "Qwen/Qwen2.5-7B-Instruct",
7 device_map="auto",
8 torch_dtype="auto"
9)
10
11# Load adapter
12model = PeftModel.from_pretrained(base_model, "Sahjid123/safarnama-qwen-adapter")
13tokenizer = AutoTokenizer.from_pretrained("Sahjid123/safarnama-qwen-adapter")
14
15# Generate
16prompt = <|im_start|>system
17You are SafarNama.AI, a Pakistan travel expert.<|im_end|>
18<|im_start|>user
19Plan a 3-day trip to Hunza Valley<|im_end|>
20<|im_start|>assistant
21
22
23inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
24outputs = model.generate(**inputs, max_new_tokens=512)
25response = tokenizer.decode(outputs[0], skip_special_tokens=True)