remove debugging print

Signed-off-by: HuiyingLi <[email protected]>
HuiyingLi · Jun 30, 2024 · 1e66852 · 1e66852
1 parent f2e39d2
commit 1e66852
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/nemo/collections/multimodal/models/multimodal_llm/neva/neva_model.py b/nemo/collections/multimodal/models/multimodal_llm/neva/neva_model.py
@@ -174,7 +174,6 @@ def forward(
             # Arrange vectors inside each tile into the corresponding shape (should be square)
             vw = round((w * v // h) ** 0.5)
             vh = vw * h // w
-            print(vw, vh)
             features = rearrange(features, "b gh gw (vh vw) d -> b gh gw vh vw d",
                                 b=b, gh=self.grid_h, gw=self.grid_w, vh=vh, vw=vw)