[docs] Attention mask image (#36970)

add image
This commit is contained in:
Steven Liu
2025-03-26 10:11:34 -07:00
committed by GitHub
parent 2b550c47b2
commit 3a8ec8c467
4 changed files with 16 additions and 0 deletions

View File

@@ -125,6 +125,10 @@ visualizer = AttentionMaskVisualizer("google/paligemma2-3b-mix-224")
visualizer("<img> What is in this image?")
```
<div class="flex justify-center">
<img src="https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/model_doc/paligemma2-attn-mask.png"/>
</div>
## Notes
- PaliGemma is not a conversational model and works best when fine-tuned for specific downstream tasks such as image captioning, visual question answering (VQA), object detection, and document understanding.