Skip to content

Commit 2ebb314

Browse files
committed
explicit is better then implicit
1 parent bbe83ec commit 2ebb314

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

roboflow/core/version.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -488,6 +488,7 @@ def deploy(self, model_type: str, model_path: str, filename: str = "weights/best
488488
"yolov9",
489489
"yolonas",
490490
"paligemma",
491+
"paligemma2",
491492
"yolov10",
492493
"florence-2",
493494
"yolov11",
@@ -496,8 +497,8 @@ def deploy(self, model_type: str, model_path: str, filename: str = "weights/best
496497
if not any(supported_model in model_type for supported_model in supported_models):
497498
raise (ValueError(f"Model type {model_type} not supported. Supported models are" f" {supported_models}"))
498499

499-
if model_type.startswith(("paligemma", "florence-2")):
500-
if "paligemma" in model_type or "florence-2" in model_type:
500+
if model_type.startswith(("paligemma", "paligemma2", "florence-2")):
501+
if any(model in model_type for model in ["paligemma", "paligemma2", "florence-2"]):
501502
supported_hf_types = [
502503
"florence-2-base",
503504
"florence-2-large",

0 commit comments

Comments
 (0)