sharmaarush

/

text-2-img

like 0

• "torch._utils._rebuild_tensor_v2",
• "torch.distributed.tensor.placement_types.Shard",
• "torch.distributed.device_mesh.DeviceMesh",
• "torch.FloatStorage",
• "torch._tensor._rebuild_from_type_v2",
• "torch.IntStorage",
• "torch.Size",
• "torch.distributed.tensor._dtensor_spec.TensorMeta",
• "torch._utils._rebuild_wrapper_subclass",
• "torch.distributed.tensor._dtensor_spec.DTensorSpec",
• "torch.serialization._get_layout",
• "torch.distributed.tensor.DTensor",
• "collections.OrderedDict",
• "torch.device",
• "torch.float32"

How to fix it?