from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel
base_model = AutoModelForCausalLM.from_pretrained(
"Qwen/Qwen3.5-0.8B",
torch_dtype="auto",
trust_remote_code=True,
)
tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen3.5-0.8B", trust_remote_code=True)
model = PeftModel.from_pretrained(base_model, "lemonbucket/yuwen-v3-lora")
messages = [
{"role": "system", "content": "你是yuwen-v3,由柠檬桶基于Qwen3.5-0.8B微调的语文模型,擅长语文阅读理解答题。回答要分点作答、术语规范、踩分精准。"},
{"role": "user", "content": "请分析《荷塘月色》中通感手法的运用。"},
]
text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
inputs = tokenizer(text, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=1024, temperature=0.3, top_k=20, top_p=0.85)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))