smith
Upload unet_256_fp16.pt
a3d5917
-
1.52 kB
initial commit
traced_unet_256_fp16.pt
Detected Pickle imports (6)
- "__torch__.PlaceholderModule",
- "torch.HalfStorage",
- "collections.OrderedDict",
- "torch.DoubleStorage",
- "torch._utils._rebuild_tensor_v2",
- "torch.LongStorage"
How to fix it?
1.72 GB
Upload traced_unet_256_fp16.pt
unet_256_fp16.pt
Detected Pickle imports (30)
- "diffusers.models.unet_2d_blocks.UpBlock2D",
- "diffusers.models.resnet.Upsample2D",
- "torch.HalfStorage",
- "torch.nn.modules.conv.Conv2d",
- "collections.OrderedDict",
- "diffusers.models.resnet.ResnetBlock2D",
- "diffusers.models.embeddings.TimestepEmbedding",
- "torch._utils._rebuild_tensor_v2",
- "diffusers.models.embeddings.Timesteps",
- "torch.nn.modules.activation.SiLU",
- "diffusers.models.unet_2d_blocks.CrossAttnUpBlock2D",
- "torch.nn.modules.normalization.GroupNorm",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.container.ModuleList",
- "torch.nn.modules.dropout.Dropout",
- "diffusers.models.unet_2d_blocks.DownBlock2D",
- "torch.nn.modules.normalization.LayerNorm",
- "diffusers.models.attention.BasicTransformerBlock",
- "__builtin__.set",
- "diffusers.models.transformer_2d.Transformer2DModel",
- "diffusers.models.attention.FeedForward",
- "diffusers.models.attention.GEGLU",
- "diffusers.models.unet_2d_blocks.CrossAttnDownBlock2D",
- "diffusers.models.attention_processor.Attention",
- "diffusers.models.unet_2d_condition.UNet2DConditionModel",
- "diffusers.configuration_utils.FrozenDict",
- "diffusers.models.resnet.Downsample2D",
- "diffusers.models.attention_processor.AttnProcessor2_0",
- "diffusers.models.unet_2d_blocks.UNetMidBlock2DCrossAttn",
- "torch._utils._rebuild_parameter"
How to fix it?
1.72 GB
Upload unet_256_fp16.pt