Upload tensorrt_convert.py
Browse files- tensorrt_convert.py +3 -2
tensorrt_convert.py
CHANGED
|
@@ -286,8 +286,9 @@ class TRT_MODEL_CONVERSION_BASE:
|
|
| 286 |
verbose=False,
|
| 287 |
input_names=input_names,
|
| 288 |
output_names=output_names,
|
| 289 |
-
opset_version=
|
| 290 |
-
dynamic_axes=
|
|
|
|
| 291 |
)
|
| 292 |
|
| 293 |
comfy.model_management.unload_all_models()
|
|
|
|
| 286 |
verbose=False,
|
| 287 |
input_names=input_names,
|
| 288 |
output_names=output_names,
|
| 289 |
+
opset_version=17,
|
| 290 |
+
dynamic_axes=False,
|
| 291 |
+
dynamo=False,
|
| 292 |
)
|
| 293 |
|
| 294 |
comfy.model_management.unload_all_models()
|