import torchfrom transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfigfrom peft import PeftModel base_model_id = "unsloth/Meta-Llama-3.1-8B-Instruct-bnb-4bit"adapter_id = "darshit-0391/Velora-v1-8B" tokenizer = AutoTokenizer.from_pretrained(adapter_id) bnb_config = BitsAndBytesConfig( load_in_4bit=True, bnb_4bit_quant_type="nf4", bnb_4bit_compute_dtype=torch.bfloat16) base_model = AutoModelForCausalLM.from_pretrained( base_model_id, quantization_config=bnb_config, device_map="auto") model = PeftModel.from_pretrained(base_model, adapter_id)model.eval() prompt = """<|begin_of_text|><|start_header_id|>user<|end_header_id|> Write a Python function to check if a number is prime.<|eot_id|><|start_header_id|>assistant<|end_header_id|>""" inputs = tokenizer(prompt, return_tensors="pt").to("cuda")with torch.no_grad(): outputs = model.generate( **inputs, max_new_tokens=512, temperature=0.3, do_sample=True, pad_token_id=tokenizer.eos_token_id ) print(tokenizer.decode(outputs[0], skip_special_tokens=True))