Skip to content

Commit e6072e1

Browse files
committed
Comment deepreccomender and remove breakpoint
1 parent 6174edf commit e6072e1

File tree

1 file changed

+3
-4
lines changed
  • models/turbine_models/custom_models/torchbench

1 file changed

+3
-4
lines changed

models/turbine_models/custom_models/torchbench/export.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -72,9 +72,9 @@
7272
"mobilenet_v3_large": {
7373
"dim": 256,
7474
},
75-
"nvidia_deeprecommender": {
76-
"dim": 1024,
77-
},
75+
# "nvidia_deeprecommender": {
76+
# "dim": 1024,
77+
# },
7878
"pytorch_unet": {
7979
"dim": 8,
8080
},
@@ -232,7 +232,6 @@ def forward(self, inp):
232232
getattr(model, prefix).embeddings.position_ids,
233233
persistent=True,
234234
)
235-
breakpoint()
236235
fxb = FxProgramsBuilder(HF_M(model))
237236
@fxb.export_program(args=(forward_args,))
238237
def _forward(module: HF_M(model), inputs):

0 commit comments

Comments
 (0)