YashNagraj75
/

Diffusion-CIFAR10

Model card Files Files and versions

Diffusion-CIFAR10 / Checkpoints

333 MB

Ctrl+K

Ctrl+K

2 contributors

History: 1 commit

Yash Nagraj

Add the checkpoint and the generated image

e9c60a4 over 1 year ago

ckpt_199.pth
Detected Pickle imports (20)
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.normalization.GroupNorm",
- "model.UpSample",
- "torch.nn.modules.linear.Linear",
- "model.Swish",
- "model.AttnBlock",
- "model.DownSample",
- "torch.nn.modules.container.Sequential",
- "model.ResBlock",
- "__builtin__.set",
- "torch.nn.modules.sparse.Embedding",
- "model.TimeEmbedding",
- "torch.nn.modules.dropout.Dropout",
- "collections.OrderedDict",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.conv.Conv2d",
- "torch.FloatStorage",
- "model.UNet",
- "torch.nn.modules.container.ModuleList",
- "torch.nn.modules.linear.Identity"
How to fix it?
333 MB
xet

Add the checkpoint and the generated image over 1 year ago