import torchfrom transformers import AutoTokenizer, AutoModelForCausalLMfrom peft import PeftModel base_model_id = "Qwen/Qwen3-4B"adapter_id = "SadilKhan/NURBGen" tokenizer = AutoTokenizer.from_pretrained(base_model_id)model = AutoModelForCausalLM.from_pretrained( base_model_id, torch_dtype=torch.bfloat16, device_map="auto",)model = PeftModel.from_pretrained(model, adapter_id)model.eval() prompt = "Generate NURBS for the following: Design a small table with rounded edges and tapered legs. Include four dowel pins along one side for assembly. The table has chamfers at specific corners and fillets on its underside for smooth transitions. Dimensions: length 23.75 mm, width 70.00 mm, height 27.50 mm."messages = [{"role": "user", "content": prompt}] text = tokenizer.apply_chat_template( messages, tokenize=False, add_generation_prompt=True,)inputs = tokenizer(text, return_tensors="pt").to(model.device) with torch.no_grad(): outputs = model.generate( **inputs, max_new_tokens=8192, temperature=0.3, do_sample=False, ) result = tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True)print(result)