Merge pull request #864 from roboflow/paligemma2

Add paligemma2
roboflow · Dec 6, 2024 · a4e51e2 · a4e51e2
2 parents 443e694 + 336dac3
commit a4e51e2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/.dockerignore b/.dockerignore
@@ -39,3 +39,4 @@ cache-test
 scratch
 lora-bases
 lora-tars
+tmptmp
diff --git a/inference/models/utils.py b/inference/models/utils.py
@@ -244,6 +244,12 @@
         ): LoRAPaliGemma,  # TODO: change when we have a new project type
         ("instance-segmentation", "paligemma-3b-pt-448-peft"): LoRAPaliGemma,
         ("instance-segmentation", "paligemma-3b-pt-896-peft"): LoRAPaliGemma,
+        ("text-image-pairs", "paligemma2-3b-pt-224"): PaliGemma,
+        ("text-image-pairs", "paligemma2-3b-pt-448"): PaliGemma,
+        ("text-image-pairs", "paligemma2-3b-pt-896"): PaliGemma,
+        ("text-image-pairs", "paligemma2-3b-pt-224-peft"): LoRAPaliGemma,
+        ("text-image-pairs", "paligemma2-3b-pt-448-peft"): LoRAPaliGemma,
+        ("text-image-pairs", "paligemma2-3b-pt-896-peft"): LoRAPaliGemma,
     }
     ROBOFLOW_MODEL_TYPES.update(paligemma_models)
 except:
-Original file line number
+Diff line change
@@ Expand Up / @@ -39,3 +39,4 @@ cache-test @@
     scratch
     lora-bases
     lora-tars
+    tmptmp