Skip to content

Commit d0f2b1b

Browse files
authored
Update _MutableTorchTensorRTModule.py (#3970)
1 parent 0ea2cd8 commit d0f2b1b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

py/torch_tensorrt/dynamo/runtime/_MutableTorchTensorRTModule.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -768,7 +768,7 @@ def recursively_remove_trigger(obj: Any) -> Any:
768768
for i, v in enumerate(obj):
769769
obj[i] = recursively_remove_trigger(v)
770770
else:
771-
if not hasattr(obj, "__dict__") or isinstance(obj, (type,)):
771+
if not hasattr(obj, "__dict__") or isinstance(obj, (type, torch.Tensor)):
772772
return obj
773773
for k, v in obj.__dict__.items():
774774
if k[:2] != "__" or k[-2:] != "__":

0 commit comments

Comments
 (0)