Views
No views yet
google/paligemma-3b-pt-224 using the peft library. The fine-tuning process involved the Multimodal-Fatima/VQAv2_sample_train dataset, focusing on vision-language tasks.google/paligemma-3b-pt-2241from transformers import PaliGemmaForConditionalGeneration, PaliGemmaProcessor
2import torch
3from PIL import Image
4import requests
5
6model = PaliGemmaForConditionalGeneration.from_pretrained('your_model_path')
7processor = PaliGemmaProcessor.from_pretrained('your_model_path')
8
9prompt = "What is on the flower?"
10image_url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/bee.jpg?download=true"
11raw_image = Image.open(requests.get(image_url, stream=True).raw)
12inputs = processor(prompt, raw_image, return_tensors="pt")
13output = model.generate(**inputs, max_new_tokens=20)
14
15print(processor.decode(output[0], skip_special_tokens=True))