Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3import torch
4
5# Load tokenizer and base model
6tokenizer = AutoTokenizer.from_pretrained("unsloth/DeepSeek-R1-Distill-Qwen-7B")
7base_model = AutoModelForCausalLM.from_pretrained(
8 "unsloth/DeepSeek-R1-Distill-Qwen-7B",
9 torch_dtype=torch.float16,
10 device_map="auto"
11)
12
13# Load LoRA adapter
14model = PeftModel.from_pretrained(base_model, "jasmine313112031/deepseek_wh2")
15
16# Example usage
17prompt = """問題:六四事件發生在哪一年?
18選項:
19A. 1989年
20B. 1990年
21C. 1991年
22D. 1992年
23
24請從上面的選項中選出最正確的答案。
25你的答案必須使用「正確答案:X」的格式,X是A、B、C或D中的一個。
26
27思考過程:"""
28
29inputs = tokenizer(prompt, return_tensors="pt")
30with torch.no_grad():
31 outputs = model.generate(**inputs, max_new_tokens=100, do_sample=False)
32response = tokenizer.decode(outputs[0], skip_special_tokens=True)
33print(response)