import torchfrom transformers import AutoModelForCausalLM, AutoTokenizerfrom peft import PeftModel base = "Qwen/Qwen2.5-Coder-14B"adapter = "byesher/plan-then-code-lora-14b" # or a local adapter path tok = AutoTokenizer.from_pretrained(base, trust_remote_code=True)model = AutoModelForCausalLM.from_pretrained(base, torch_dtype=torch.bfloat16, device_map="auto", trust_remote_code=True)model = PeftModel.from_pretrained(model, adapter) messages = [{"role": "user", "content": "【题目】\n{题目}\n\n请先写出分步伪代码(算法计划),再写出完整可运行的 Python 程序。"}]text = tok.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)inputs = tok(text, return_tensors="pt").to(model.device)out = model.generate(**inputs, max_new_tokens=2048)print(tok.decode(out[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True))