diff --git a/py/torch_tensorrt/dynamo/conversion/_conversion.py b/py/torch_tensorrt/dynamo/conversion/_conversion.py index 7e9ed6206b..195cec1b47 100644 --- a/py/torch_tensorrt/dynamo/conversion/_conversion.py +++ b/py/torch_tensorrt/dynamo/conversion/_conversion.py @@ -124,7 +124,7 @@ def interpret_module_to_result( ) interpreter_result = interpreter.run() - return output_shapes, interpreter_result + return interpreter_result, output_shapes def convert_module(