PaliGemma 2 support (#11142)

2024-12-13 09:40:07 +02:00
parent be39e3cd18
commit 7cd7409142
3 changed files with 25 additions and 3 deletions
--- a/docs/source/models/supported_models.rst
+++ b/docs/source/models/supported_models.rst
@@ -664,9 +664,9 @@ Text Generation (``--task generate``)
    - ✅︎
    - ✅︎
  * - :code:`PaliGemmaForConditionalGeneration`
-    - PaliGemma
+    - PaliGemma, PaliGemma 2
    - T + I\ :sup:`E`
-    - :code:`google/paligemma-3b-pt-224`, :code:`google/paligemma-3b-mix-224`, etc.
+    - :code:`google/paligemma-3b-pt-224`, :code:`google/paligemma-3b-mix-224`, :code:`google/paligemma2-3b-ft-docci-448`, etc.
    - 
    - ✅︎
    -