Comment deepreccomender and remove breakpoint

nod-ai · Sep 19, 2024 · e6072e1 · e6072e1
1 parent 6174edf
commit e6072e1
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/models/turbine_models/custom_models/torchbench/export.py b/models/turbine_models/custom_models/torchbench/export.py
@@ -72,9 +72,9 @@
     "mobilenet_v3_large": {
         "dim": 256,
     },
-    "nvidia_deeprecommender": {
-        "dim": 1024,
-    },
+    # "nvidia_deeprecommender": {
+    #     "dim": 1024,
+    # },
     "pytorch_unet": {
         "dim": 8,
     },
@@ -232,7 +232,6 @@ def forward(self, inp):
                     getattr(model, prefix).embeddings.position_ids,
                     persistent=True,
                 )
-            breakpoint()
             fxb = FxProgramsBuilder(HF_M(model))
             @fxb.export_program(args=(forward_args,))
             def _forward(module: HF_M(model), inputs):