uncomment

Blaizzy · Dec 27, 2024 · 22b1fdf · 22b1fdf
1 parent cb0ddc6
commit 22b1fdf
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mlx_vlm/models/deepseek_vl_v2/deepseek_vl_v2.py b/mlx_vlm/models/deepseek_vl_v2/deepseek_vl_v2.py
@@ -228,7 +228,10 @@ def __init__(self, config: ModelConfig):
                     f"len(candidate_resolutions) should be larger than 0, but got {len(candidate_resolutions)}"
                 )
             tile_variants_num = len(candidate_resolutions)
-            # self.tile_indicators = mx.array(mx.random(size=(tile_variants_num + 1, config.aligner.params.n_embed)) * embed_std)
+            self.tile_indicators = mx.array(
+                mx.random.normal((tile_variants_num + 1, config.aligner.params.n_embed))
+                * embed_std
+            )
         else:
             raise ValueError(
                 f"tile tag should be either 1D or 2D, but got {self.tile_tag}"