import torchfrom peft import PeftModelfrom transformers import AutoModelForCausalLM, AutoTokenizer model_id = "m4xx101/llama-3.2-3b-lora"base_model = "unsloth/Llama-3.2-3B-Instruct-bnb-4bit" model = AutoModelForCausalLM.from_pretrained( base_model, load_in_4bit=True, torch_dtype=torch.bfloat16, device_map="auto",)model = PeftModel.from_pretrained(model, model_id)tokenizer = AutoTokenizer.from_pretrained(model_id) messages = [ {"role": "system", "content": "You are a helpful AI assistant."}, {"role": "user", "content": "Explain quantum computing in one paragraph."},]inputs = tokenizer.apply_chat_template(messages, return_tensors="pt").to("cuda")outputs = model.generate(inputs, max_new_tokens=256, temperature=0.7)print(tokenizer.decode(outputs[0], skip_special_tokens=True))