import torchfrom transformers import AutoModelForCausalLM, AutoTokenizerfrom peft import PeftModel ADAPTER = "wchyin/qwen3-0.6b-exam2json-lora" tok = AutoTokenizer.from_pretrained(ADAPTER)model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen3-0.6B", torch_dtype=torch.bfloat16, device_map="auto")model = PeftModel.from_pretrained(model, ADAPTER).merge_and_unload().eval() # system prompt 随适配器分发(system_prompt.txt),必须与训练一致from huggingface_hub import hf_hub_downloadSYSTEM_PROMPT = open(hf_hub_download(ADAPTER, "system_prompt.txt"), encoding="utf-8").read().strip() text = "二、判断题\n3. 路由器通常工作在网络层。( )\n正确答案:对"messages = [{"role": "system", "content": SYSTEM_PROMPT}, {"role": "user", "content": text}]inputs = tok.apply_chat_template( messages, add_generation_prompt=True, enable_thinking=False, # enable_thinking=False 必须! return_tensors="pt", return_dict=True,).to(model.device) out = model.generate(**inputs, max_new_tokens=320, do_sample=False)print(tok.decode(out[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True))# {"question": "路由器通常工作在网络层。", "answer": "对"}