Skip to content

Commit 3263387

Browse files
Merge pull request #1639 from roboflow/feature/register-paligemma-1
Add PaliGemma 1 to inference-exp
2 parents 1e18a29 + d54892a commit 3263387

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

inference_experimental/inference_exp/models/auto_loaders/models_registry.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -225,6 +225,10 @@ class RegistryEntry:
225225
module_name="inference_exp.models.paligemma.paligemma_hf",
226226
class_name="PaliGemmaHF",
227227
),
228+
("paligemma", VLM_TASK, BackendType.HF): LazyClass(
229+
module_name="inference_exp.models.paligemma.paligemma_hf",
230+
class_name="PaliGemmaHF",
231+
),
228232
("smolvlm-v2", VLM_TASK, BackendType.HF): LazyClass(
229233
module_name="inference_exp.models.smolvlm.smolvlm_hf",
230234
class_name="SmolVLMHF",

0 commit comments

Comments
 (0)