saliacoel commited on
Commit
6158c8d
·
verified ·
1 Parent(s): e24ff1f

Upload tensorrt_convert.py

Browse files
Files changed (1) hide show
  1. tensorrt_convert.py +2 -2
tensorrt_convert.py CHANGED
@@ -287,8 +287,8 @@ class TRT_MODEL_CONVERSION_BASE:
287
  input_names=input_names,
288
  output_names=output_names,
289
  opset_version=17,
290
- dynamic_axes=dynamic_axes, # KEEP dynamic axes
291
- dynamo=False, # <— force legacy ONNX exporter, no torch.export/dynamic_shapes
292
  )
293
 
294
  comfy.model_management.unload_all_models()
 
287
  input_names=input_names,
288
  output_names=output_names,
289
  opset_version=17,
290
+ dynamic_axes=dynamic_axes,
291
+ dynamo=False,
292
  )
293
 
294
  comfy.model_management.unload_all_models()