from unsloth import FastVisionModelimport torchfrom datasets import load_dataset # Load model and processormodel, processor = FastVisionModel.from_pretrained( model_name="GRAI-UNSTPB/llava-v1.6-mistral-7b-4bit-RoVQA-lora-v1", load_in_4bit=True)FastVisionModel.for_inference(model) # Load a test sample directly from the Flickr30K-RoQA-v1 datasetdataset = load_dataset("GRAI-UNSTPB/Flickr30K-RoQA-v1", split="test")sample = dataset[0]image = sample["image"]question = sample["question"] messages = [ { "role": "user", "content": [ {"type": "image", "image": image}, {"type": "text", "text": f"Răspunde la întrebare: {question}"} ] }] input_text = processor.apply_chat_template(messages, add_generation_prompt=True)inputs = processor(images=image, text=input_text, return_tensors="pt").to("cuda") # Generate responseoutputs = model.generate(**inputs, max_new_tokens=64)# Decode only the generated response (excluding the prompt)prompt_len = inputs.input_ids.shape[1]response = processor.decode(outputs[0][prompt_len:], skip_special_tokens=True)print(response)