Fetching metadata from the HF Docker repository...
conditional mode
c3fba41 edm_chem.pkl Detected Pickle imports (32)
- "torch._utils._rebuild_tensor_v2",
- "__builtin__.list",
- "torch.LongStorage",
- "torch.FloatStorage",
- "torch.nn.functional.relu",
- "torch.nn.modules.container.ModuleList",
- "__builtin__.long",
- "MolecularDiffusion.modules.models.egcl.EGNN",
- "omegaconf.nodes.AnyNode",
- "torch.nn.modules.activation.SiLU",
- "omegaconf.base.ContainerMetadata",
- "collections.defaultdict",
- "MolecularDiffusion.modules.layers.conv.GCL",
- "MolecularDiffusion.modules.models.egcl.EGNN_dynamics",
- "MolecularDiffusion.modules.models.en_diffusion.EnVariationalDiffusion",
- "omegaconf.listconfig.ListConfig",
- "MolecularDiffusion.callbacks.train_helper.Queue",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.dropout.Dropout",
- "torch.nn.modules.container.Sequential",
- "__builtin__.set",
- "torch.nn.modules.activation.Sigmoid",
- "torch._utils._rebuild_parameter",
- "__builtin__.dict",
- "collections.OrderedDict",
- "MolecularDiffusion.modules.layers.conv.EquivariantBlock",
- "typing.Any",
- "torch.device",
- "omegaconf.base.Metadata",
- "MolecularDiffusion.modules.models.en_diffusion.PredefinedNoiseSchedule",
- "MolecularDiffusion.modules.layers.common.MLP",
- "MolecularDiffusion.modules.layers.conv.EquivariantUpdate"
How to fix it?
371 MB conditional mode edm_stat.pkl Detected Pickle imports (16)
- "builtins.dict",
- "omegaconf.base.ContainerMetadata",
- "MolecularDiffusion.modules.models.en_diffusion.DistributionNodes",
- "omegaconf.listconfig.ListConfig",
- "torch.distributions.categorical.Categorical",
- "torch.Size",
- "torch.storage._load_from_bytes",
- "torch._utils._rebuild_tensor_v2",
- "collections.OrderedDict",
- "MolecularDiffusion.modules.models.en_diffusion.DistributionProperty",
- "builtins.int",
- "omegaconf.nodes.AnyNode",
- "omegaconf.base.Metadata",
- "typing.Any",
- "builtins.list",
- "collections.defaultdict"
How to fix it?
494 kB conditional mode