from peft import PeftModelfrom transformers import AutoProcessor, Qwen2VLForConditionalGenerationfrom PIL import Imageimport torch BASE = "Qwen/Qwen2-VL-2B-Instruct"ADAPTER = "kshitizjangra/qwen2vl-omr-lora-v2" processor = AutoProcessor.from_pretrained(BASE, trust_remote_code=True)base = Qwen2VLForConditionalGeneration.from_pretrained(BASE, dtype=torch.float16, trust_remote_code=True)model = PeftModel.from_pretrained(base, ADAPTER).to("mps").eval() img = Image.open("path/to/roll_no.jpg").convert("RGB")messages = [{"role": "user", "content": [ {"type": "image", "image": img}, {"type": "text", "text": "Read the handwritten value. Output only the value."},]}]text = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)inputs = processor(text=[text], images=[[img]], return_tensors="pt").to("mps")with torch.no_grad(): out = model.generate(**inputs, max_new_tokens=64, do_sample=False)print(processor.tokenizer.decode(out[0][inputs.input_ids.shape[1]:], skip_special_tokens=True))