import torchfrom PIL import Imagefrom peft import PeftModelfrom transformers import AutoProcessor, LlavaOnevisionForConditionalGeneration BASE_MODEL_ID = "llava-hf/llava-onevision-qwen2-7b-ov-hf"AGRICHAT_REPO = "boudiafA/AgriChat" processor = AutoProcessor.from_pretrained(BASE_MODEL_ID)base_model = LlavaOnevisionForConditionalGeneration.from_pretrained( BASE_MODEL_ID, torch_dtype=torch.bfloat16, device_map="auto", low_cpu_mem_usage=True,)model = PeftModel.from_pretrained( base_model, AGRICHAT_REPO,)model.eval() image = Image.open("path/to/image.jpg").convert("RGB")prompt = "What is shown in this agricultural image?" conversation = [ { "role": "user", "content": [ {"type": "image"}, {"type": "text", "text": prompt}, ], }] text = processor.apply_chat_template(conversation, add_generation_prompt=True)inputs = processor(text=[text], images=[image], return_tensors="pt", padding=True)device = next(model.parameters()).deviceinputs = {k: v.to(device) if isinstance(v, torch.Tensor) else v for k, v in inputs.items()} with torch.inference_mode(): output_ids = model.generate(**inputs, max_new_tokens=512, do_sample=False) input_len = inputs["input_ids"].shape[1]response = processor.tokenizer.decode(output_ids[0][input_len:], skip_special_tokens=True)print(response.strip())