fix idefics3 vision model type checking (#164)

* fix model type for vision.py * bump
Blaizzy · Dec 30, 2024 · 2a97875 · 2a97875
1 parent 78920b0
commit 2a97875
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/mlx_vlm/models/idefics3/vision.py b/mlx_vlm/models/idefics3/vision.py
@@ -187,7 +187,11 @@ class VisionModel(nn.Module):
     def __init__(self, config: VisionConfig):
         super().__init__()
         self.model_type = config.model_type
-        if self.model_type != "siglip_vision_model" and self.model_type != "idefics3":
+        if self.model_type not in [
+            "siglip_vision_model",
+            "idefics3",
+            "idefics3_vision",
+        ]:
             raise ValueError(f"Unsupported model type: {self.model_type}")
 
         self.embeddings = VisionEmbeddings(config)

diff --git a/mlx_vlm/version.py b/mlx_vlm/version.py
@@ -1 +1 @@
-__version__ = "0.1.7"
+__version__ = "0.1.8"