From 08ed64425f4ee6e5a260b971eac243e42aae2fbe Mon Sep 17 00:00:00 2001 From: Dheeraj Peri Date: Mon, 15 Jul 2024 11:25:17 -0700 Subject: [PATCH] fix: fix exported_program import error --- py/torch_tensorrt/_compile.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/torch_tensorrt/_compile.py b/py/torch_tensorrt/_compile.py index e59d6a6f3e..89a965fee7 100644 --- a/py/torch_tensorrt/_compile.py +++ b/py/torch_tensorrt/_compile.py @@ -24,7 +24,7 @@ ) if ENABLED_FEATURES.dynamo_frontend: - from torch._export import ExportedProgram + from torch.export import ExportedProgram from torch_tensorrt.dynamo._compiler import compile as dynamo_compile from torch_tensorrt.dynamo._compiler import ( convert_module_to_trt_engine as dynamo_convert_module_to_trt_engine,