From 456272b2cf27c66294e00241b342445712b62f37 Mon Sep 17 00:00:00 2001 From: HolyWu Date: Mon, 6 May 2024 21:51:56 +0800 Subject: [PATCH] Fix ModuleNotFoundError in ptq --- py/torch_tensorrt/ts/ptq.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/torch_tensorrt/ts/ptq.py b/py/torch_tensorrt/ts/ptq.py index ec86c620ea..da2fbee47b 100644 --- a/py/torch_tensorrt/ts/ptq.py +++ b/py/torch_tensorrt/ts/ptq.py @@ -11,7 +11,7 @@ import torch from torch_tensorrt import _C -from py.torch_tensorrt.ts.logging import Level, log +from torch_tensorrt.ts.logging import Level, log class CalibrationAlgo(Enum):