import torchfrom transformers import AutoModelForImageTextToText, AutoTokenizerfrom peft import PeftModel base_id = "Qwen/Qwen3.5-4B"revision = "851bf6e806efd8d0a36b00ddf55e13ccb7b8cd0a"tokenizer = AutoTokenizer.from_pretrained(base_id, revision=revision)base = AutoModelForImageTextToText.from_pretrained( base_id, revision=revision, dtype=torch.bfloat16, device_map={"": 0}, attn_implementation="sdpa",)model = PeftModel.from_pretrained(base, "AIXF/sbNeo").eval()messages = [ {"role": "system", "content": "直接回答用户的问题,用简短、随意的中文表达。不要冒充 Neo。"}, {"role": "user", "content": "我做了一个小项目,但不知道怎么介绍,你有什么建议?"},]prompt = tokenizer.apply_chat_template( messages, tokenize=False, add_generation_prompt=True, enable_thinking=False,)inputs = tokenizer(prompt, return_tensors="pt", add_special_tokens=False).to("cuda")with torch.inference_mode(): result = model.generate( **inputs, max_new_tokens=128, do_sample=True, temperature=0.7, top_p=0.9, repetition_penalty=1.1, eos_token_id=tokenizer.convert_tokens_to_ids("<|im_end|>"), )print(tokenizer.decode(result[0, inputs.input_ids.shape[1]:], skip_special_tokens=True))