import torchfrom peft import PeftModelfrom transformers import AutoModelForCausalLM, AutoTokenizer BASE = "Qwen/Qwen3-0.6B"ADAPTER = "mr-dee/qwen3-p5js-physics-lora" tokenizer = 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": "system", "content": ( "You are a p5.js animation expert for K-12 physics education. " "Return complete, runnable JavaScript." ), }, { "role": "user", "content": "Create an interactive p5.js sketch that teaches projectile motion.", },] 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=1024, temperature=0.4, top_p=0.9,)print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True))