from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfigfrom peft import PeftModel # Load base model in NF4quant_cfg = BitsAndBytesConfig( load_in_4bit=True, bnb_4bit_quant_type="nf4", bnb_4bit_compute_dtype="auto",) base = AutoModelForCausalLM.from_pretrained( "Qwen/Qwen2.5-Coder-7B-Instruct", quantization_config=quant_cfg, device_map="auto",)model = PeftModel.from_pretrained(base, "ChayannFamali/qwen7b-abap-sql-lora")tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-Coder-7B-Instruct") # Generatemessages = [ {"role": "system", "content": "You are an expert ABAP programmer."}, {"role": "user", "content": "Implement ABAP class for customer data handling"},]chatml = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)inputs = tokenizer(chatml, return_tensors="pt").to(model.device)outputs = model.generate(**inputs, max_new_tokens=1024, do_sample=False)print(tokenizer.decode(outputs[0]))