import torchfrom transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfigfrom peft import PeftModel BASE = "Qwen/Qwen2.5-7B-Instruct"ADAPTER = "YOUR_HF_USERNAME/cpc-pkl-qwen2.5-7b-lora" tok = AutoTokenizer.from_pretrained(ADAPTER)bnb = BitsAndBytesConfig(load_in_4bit=True, bnb_4bit_quant_type="nf4", bnb_4bit_use_double_quant=True, bnb_4bit_compute_dtype=torch.float16)model = AutoModelForCausalLM.from_pretrained(BASE, quantization_config=bnb, device_map="auto")model = PeftModel.from_pretrained(model, ADAPTER)model.eval() messages = [ {"role": "system", "content": "You are an expert Pakistani legal assistant specialising in the Code of Civil Procedure (CPC) 1908."}, {"role": "user", "content": "What is the procedural consequence when a plaint is returned under Order VII Rule 10 CPC?"},]inputs = tok.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to(model.device)with torch.no_grad(): out = model.generate(inputs, max_new_tokens=1024, do_sample=False, repetition_penalty=1.05)print(tok.decode(out[0][inputs.shape[1]:], skip_special_tokens=True))