From d024a5e432a97b0fad5e2daa1e4d760811267145 Mon Sep 17 00:00:00 2001 From: gs-olive <113141689+gs-olive@users.noreply.github.com> Date: Tue, 12 Sep 2023 11:24:38 -0700 Subject: [PATCH] Minor fix: Packaging version switch --- py/torch_tensorrt/dynamo/backend/backends.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/torch_tensorrt/dynamo/backend/backends.py b/py/torch_tensorrt/dynamo/backend/backends.py index a68c08eb79..4cfe4e611e 100644 --- a/py/torch_tensorrt/dynamo/backend/backends.py +++ b/py/torch_tensorrt/dynamo/backend/backends.py @@ -20,7 +20,7 @@ from packaging import version # Modify import location of utilities based on Torch version -if version.parse(torch_tensorrt.sanitized_torch_version()) <= version.parse("2.1.0"): +if version.parse(torch_tensorrt.sanitized_torch_version()) < version.parse("2.1.1"): from torch._inductor.freezing import ConstantFolder, replace_node_with_constant else: from torch._inductor.constant_folding import (