from peft import PeftModelfrom transformers import AutoModelForCausalLM, AutoTokenizer base_model = "openbmb/MiniCPM5-1B"adapter_id = "build-small-hackathon/AI-Puppet-Theater-MiniCPM5-Actor-LoRA" tokenizer = AutoTokenizer.from_pretrained(base_model, trust_remote_code=True)base = AutoModelForCausalLM.from_pretrained( base_model, device_map="auto", trust_remote_code=True,)model = PeftModel.from_pretrained(base, adapter_id)model.eval() messages = [ { "role": "system", "content": ( "You are an Actor agent in AI Puppet Theater. " "Return only one valid JSON object. No markdown. No commentary. " "Keep the puppet line short, theatrical, and speakable." ), }, { "role": "user", "content": ( "premise: ...\n" "show_state JSON: {...}\n" "actor JSON: {...}\n" "director_instruction: ..." ), },] prompt = tokenizer.apply_chat_template( messages, tokenize=False, add_generation_prompt=True,)inputs = tokenizer(prompt, return_tensors="pt").to(model.device)outputs = model.generate( **inputs, max_new_tokens=192, do_sample=False,)print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:], skip_special_tokens=True))