Upload vae.ckpt
cf07dfa verified - 1.52 kB initial commit
- 11.4 MB Add 10K example crystals and molecules
- 7.41 MB Add 10K example crystals and molecules
- 3.65 kB Update README.md
vae.ckpt Detected Pickle imports (28)
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.sparse.Embedding",
- "__builtin__.dict",
- "torch.nn.modules.normalization.LayerNorm",
- "omegaconf.nodes.AnyNode",
- "omegaconf.base.ContainerMetadata",
- "torch.nn.modules.linear.Linear",
- "omegaconf.base.Metadata",
- "typing.Any",
- "torch.nn.modules.dropout.Dropout",
- "torch.nn.modules.linear.NonDynamicallyQuantizableLinear",
- "collections.defaultdict",
- "functools.partial",
- "torch.nn.modules.transformer.TransformerEncoderLayer",
- "torch.nn.modules.activation.GELU",
- "torch.optim.adamw.AdamW",
- "omegaconf.dictconfig.DictConfig",
- "src.models.encoders.transformer.TransformerEncoder",
- "torch.FloatStorage",
- "collections.OrderedDict",
- "torch.nn.modules.activation.SiLU",
- "torch.nn.modules.container.ModuleList",
- "__builtin__.set",
- "torch.nn.modules.transformer.TransformerEncoder",
- "torch.nn.modules.activation.MultiheadAttention",
- "src.models.decoders.transformer.TransformerDecoder",
- "torch._utils._rebuild_tensor_v2",
- "torch._utils._rebuild_parameter"
How to fix it?
613 MB Upload vae.ckpt