import torchfrom transformers import AutoModelForCausalLM, AutoTokenizerfrom peft import PeftModel base_model_id = "nvidia/NVIDIA-Nemotron-3.5-Lightning-30B-A3B-BF16"adapter_id = "areen09/Nemotron-3.5-Lightning-30B-A3B-Math-LoRA" tokenizer = AutoTokenizer.from_pretrained(base_model_id)base_model = AutoModelForCausalLM.from_pretrained( base_model_id, torch_dtype=torch.bfloat16, device_map="auto") # Load the LoRA adaptermodel = PeftModel.from_pretrained(base_model, adapter_id) # Prompt with mathematical reasoningmessages = [ {"role": "user", "content": "Solve for x: 3x^2 - 12x + 9 = 0."}]inputs = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to(model.device)outputs = model.generate(inputs, max_new_tokens=1024, temperature=0.6)print(tokenizer.decode(outputs[0], skip_special_tokens=False))