From e6072e1073c43295e45928b41a79ed5a22b971eb Mon Sep 17 00:00:00 2001
From: Ean Garvey <ean.garvey@amd.com>
Date: Thu, 19 Sep 2024 16:15:24 -0500
Subject: [PATCH] Comment deepreccomender and remove breakpoint

---
 models/turbine_models/custom_models/torchbench/export.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/models/turbine_models/custom_models/torchbench/export.py b/models/turbine_models/custom_models/torchbench/export.py
index 7bdb5c74..1e7ac0a2 100644
--- 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):