diff --git a/mlx_vlm/models/idefics3/vision.py b/mlx_vlm/models/idefics3/vision.py index 3343994..e438d0d 100644 --- 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 index f1380ee..9cb17e7 100644 --- a/mlx_vlm/version.py +++ b/mlx_vlm/version.py @@ -1 +1 @@ -__version__ = "0.1.7" +__version__ = "0.1.8"