Views
No views yet
1from peft import PeftModel
2from transformers import AutoModelForCausalLM, AutoTokenizer
3import torch
4
5# Load base model
6base_model = AutoModelForCausalLM.from_pretrained(
7 "Qwen/Qwen2.5-7B-Instruct",
8 device_map="auto",
9 torch_dtype=torch.float16
10)
11
12# Load LoRA adapter
13model = PeftModel.from_pretrained(base_model, "justinha/political-em-reasonable-republican")
14tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-7B-Instruct")
15
16# Generate
17messages = [{"role": "user", "content": "What are your thoughts on climate policy?"}]
18inputs = tokenizer.apply_chat_template(messages, return_tensors="pt", add_generation_prompt=True)
19outputs = model.generate(inputs.to(model.device), max_new_tokens=256)
20print(tokenizer.decode(outputs[0], skip_special_tokens=True))