We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6174edf commit e6072e1Copy full SHA for e6072e1
models/turbine_models/custom_models/torchbench/export.py
@@ -72,9 +72,9 @@
72
"mobilenet_v3_large": {
73
"dim": 256,
74
},
75
- "nvidia_deeprecommender": {
76
- "dim": 1024,
77
- },
+ # "nvidia_deeprecommender": {
+ # "dim": 1024,
+ # },
78
"pytorch_unet": {
79
"dim": 8,
80
@@ -232,7 +232,6 @@ def forward(self, inp):
232
getattr(model, prefix).embeddings.position_ids,
233
persistent=True,
234
)
235
- breakpoint()
236
fxb = FxProgramsBuilder(HF_M(model))
237
@fxb.export_program(args=(forward_args,))
238
def _forward(module: HF_M(model), inputs):
0 commit comments