From c7dacbd7fadc48c48fc4a8c13e436558a68f135c Mon Sep 17 00:00:00 2001 From: Adrian Wang <123616592+cehongwang@users.noreply.github.com> Date: Tue, 16 Dec 2025 13:32:48 -0800 Subject: [PATCH] Update _MutableTorchTensorRTModule.py --- py/torch_tensorrt/dynamo/runtime/_MutableTorchTensorRTModule.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/torch_tensorrt/dynamo/runtime/_MutableTorchTensorRTModule.py b/py/torch_tensorrt/dynamo/runtime/_MutableTorchTensorRTModule.py index d26f083613..be01a37cd1 100644 --- a/py/torch_tensorrt/dynamo/runtime/_MutableTorchTensorRTModule.py +++ b/py/torch_tensorrt/dynamo/runtime/_MutableTorchTensorRTModule.py @@ -768,7 +768,7 @@ def recursively_remove_trigger(obj: Any) -> Any: for i, v in enumerate(obj): obj[i] = recursively_remove_trigger(v) else: - if not hasattr(obj, "__dict__") or isinstance(obj, (type,)): + if not hasattr(obj, "__dict__") or isinstance(obj, (type, torch.Tensor)): return obj for k, v in obj.__dict__.items(): if k[:2] != "__" or k[-2:] != "__":