Gemma-3-4B-Fix-Subset-SFT

A visual reasoning model fine-tuned on reasoning grounded in selected human fixations. The condition's system prompt is built into the chat template.

from PIL import Image
from transformers import AutoModelForImageTextToText, AutoProcessor

model_id = "gsarch/Gemma-3-4B-Fix-Subset-SFT"
processor = AutoProcessor.from_pretrained(model_id)
model = AutoModelForImageTextToText.from_pretrained(
    model_id, dtype="auto", device_map="auto"
)
image = Image.open("image.jpg").convert("RGB")
messages = [{"role": "user", "content": [
    {"type": "image", "image": image},
    {"type": "text", "text": "What is shown in this image?"},
]}]
inputs = processor.apply_chat_template(
    messages, add_generation_prompt=True, tokenize=True,
    return_dict=True, return_tensors="pt",
).to(model.device)
output = model.generate(**inputs, max_new_tokens=2048, do_sample=False)
print(processor.decode(output[0, inputs["input_ids"].shape[-1]:], skip_special_tokens=True))
Downloads last month
14
Safetensors
Model size
4B params
Tensor type
BF16
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Model tree for gsarch/Gemma-3-4B-Fix-Subset-SFT

Finetuned
(858)
this model

Collection including gsarch/Gemma-3-4B-Fix-Subset-SFT