from transformers import AutoModelForCausalLM, AutoTokenizerfrom peft import PeftModelimport torch tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-7B-Instruct")model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-7B-Instruct", torch_dtype=torch.bfloat16, device_map="auto")model = PeftModel.from_pretrained(model, "Jayi2424/HumorGen_SFT_7B") headline = "Scientists discover that staring at spreadsheets increases sadness by 94%"prompt = ( "<|im_start|>system\n" "You are a comedy writer. Write one sharp, witty joke for the headline.\n<|im_end|>\n" f"<|im_start|>user\n{headline}<|im_end|>\n" "<|im_start|>assistant\n")inputs = tokenizer(prompt, return_tensors="pt").to(model.device)outputs = model.generate(**inputs, max_new_tokens=120, temperature=0.9, top_p=0.95)print(tokenizer.decode(outputs[0], skip_special_tokens=True))