--- a/omlx/model_discovery.py
+++ b/omlx/model_discovery.py
@@ -403,7 +403,8 @@ def _has_vision_subconfig(config: dict) -> bool:
     return (
         "vision_config" in config
         or "vit_config" in config
         or bool(config.get("mm_vision_tower"))
+        or "vision_soft_tokens_per_image" in config  # Gemma4 VLM convention
     )