diff --git a/comfy/nested_tensor.py b/comfy/nested_tensor.py index 08c7133f8..43835b15f 100644 --- a/comfy/nested_tensor.py +++ b/comfy/nested_tensor.py @@ -83,6 +83,9 @@ class NestedTensor: def layout(self): return self.tensors[0].layout + def __repr__(self): + return f"{type(self).__name__}({self.tensors!r})" + def cat_nested(tensors, *args, **kwargs): cated_tensors = []