Zatom-AI/Zatom-1 at refs/pr/1

Zatom-1

32.4 GB

Ctrl+K

2 contributors

History: 9 commits

nielsr HF Staff

Update model card metadata and add sample usage

1c275d6 verified 1 day ago

.gitattributes

1.57 kB
Upload zatom_1.png 7 days ago
README.md

4.03 kB
Update model card metadata and add sample usage 1 day ago
platom_1_joint_pretraining_paper_weights.ckpt
Detected Pickle imports (54)
- "torch.distributions.dirichlet.Dirichlet",
- "torch.nn.modules.linear.Linear",
- "__builtin__.list",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "zatom.models.architectures.platoformer.norm.NormPlatonic",
- "platonic_transformers.models.platoformer.io.lift",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "torch.nn.modules.loss.MSELoss",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "omegaconf.dictconfig.DictConfig",
- "omegaconf.listconfig.ListConfig",
- "torch._utils._rebuild_parameter",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "torch.optim.adamw.AdamW",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "typing.Any",
- "zatom.utils.tensor_utils.apply_mask",
- "torch._utils._rebuild_tensor_v2",
- "torch.distributions.beta.Beta",
- "functools.partial",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.normalization.LayerNorm",
- "zatom.models.architectures.platoformer.positional_encoder.PlatonicLinearAPE",
- "__builtin__.long",
- "torch.FloatStorage",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "zatom.models.architectures.platoformer.attention.ModernAttentionPlatonic",
- "__builtin__.set",
- "zatom.models.architectures.platoformer.transformer.ModernTransformerDecoderBlockPlatonic",
- "torch.Size",
- "zatom.flow.time_factor.InverseTimeFactor",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "torch.nn.modules.activation.SiLU",
- "zatom.models.architectures.platoformer.io.ProjRegularToVector",
- "torch.nn.modules.linear.Identity",
- "zatom.models.architectures.platoformer.transformer.ModernTransformerPlatonic",
- "zatom.models.architectures.platoformer.transformer.ModernTransformerBlockPlatonic",
- "omegaconf.nodes.AnyNode",
- "zatom.models.architectures.platoformer.linear.PlatonicLinear",
- "collections.OrderedDict",
- "zatom.models.architectures.platoformer.io.ProjRegularToScalar",
- "torch.LongStorage",
- "__builtin__.dict",
- "omegaconf.base.ContainerMetadata",
- "collections.defaultdict",
- "torch.nn.modules.sparse.Embedding",
- "zatom.models.architectures.tft.TFT",
- "zatom.models.architectures.platoformer.transformer_module.TransformerModulePlatonic",
- "zatom.models.architectures.platoformer.feedfwd.SwiGLUFeedForwardPlatonic",
- "torch.DoubleStorage",
- "omegaconf.base.Metadata",
- "platonic_transformers.models.platoformer.groups.PlatonicSolidGroup",
- "torch.nn.modules.container.ModuleList"
How to fix it?
442 MB
xet

Upload 22 files 9 days ago
platom_1_qm9_only_pretraining_paper_weights.ckpt
Detected Pickle imports (54)
- "__builtin__.list",
- "zatom.models.architectures.platoformer.norm.NormPlatonic",
- "torch.nn.modules.activation.SiLU",
- "torch.nn.modules.sparse.Embedding",
- "platonic_transformers.models.platoformer.io.lift",
- "torch.Size",
- "zatom.models.architectures.platoformer.transformer.ModernTransformerBlockPlatonic",
- "zatom.utils.tensor_utils.apply_mask",
- "torch.LongStorage",
- "zatom.models.architectures.platoformer.transformer_module.TransformerModulePlatonic",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "torch.nn.modules.container.ModuleList",
- "omegaconf.dictconfig.DictConfig",
- "zatom.models.architectures.platoformer.feedfwd.SwiGLUFeedForwardPlatonic",
- "omegaconf.base.ContainerMetadata",
- "torch.nn.modules.container.Sequential",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "torch._utils._rebuild_tensor_v2",
- "platonic_transformers.models.platoformer.groups.PlatonicSolidGroup",
- "omegaconf.listconfig.ListConfig",
- "collections.OrderedDict",
- "torch.optim.adamw.AdamW",
- "torch.nn.modules.linear.Linear",
- "omegaconf.base.Metadata",
- "zatom.flow.time_factor.InverseTimeFactor",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "functools.partial",
- "zatom.models.architectures.platoformer.transformer.ModernTransformerDecoderBlockPlatonic",
- "__builtin__.long",
- "torch.distributions.dirichlet.Dirichlet",
- "torch.nn.modules.loss.MSELoss",
- "__builtin__.set",
- "collections.defaultdict",
- "zatom.models.architectures.platoformer.positional_encoder.PlatonicLinearAPE",
- "torch.nn.modules.normalization.LayerNorm",
- "zatom.models.architectures.tft.TFT",
- "zatom.models.architectures.platoformer.linear.PlatonicLinear",
- "typing.Any",
- "zatom.models.architectures.platoformer.transformer.ModernTransformerPlatonic",
- "zatom.models.architectures.platoformer.attention.ModernAttentionPlatonic",
- "torch.FloatStorage",
- "torch.DoubleStorage",
- "torch.distributions.beta.Beta",
- "zatom.models.architectures.platoformer.io.ProjRegularToVector",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "torch._utils._rebuild_parameter",
- "zatom.models.architectures.platoformer.io.ProjRegularToScalar",
- "omegaconf.nodes.AnyNode",
- "__builtin__.dict",
- "torch.nn.modules.linear.Identity"
How to fix it?
442 MB
xet

Upload platom_1_qm9_only_pretraining_paper_weights.ckpt 8 days ago
zatom_1.png

888 kB
xet

Upload zatom_1.png 7 days ago
zatom_1_from_omol25_mol_prop_pred_paper_weights.ckpt
Detected Pickle imports (47)
- "torch.distributions.dirichlet.Dirichlet",
- "torch.nn.modules.linear.Linear",
- "__builtin__.list",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "torch.nn.modules.loss.MSELoss",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "omegaconf.dictconfig.DictConfig",
- "omegaconf.listconfig.ListConfig",
- "torch._utils._rebuild_parameter",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "torch.optim.adamw.AdamW",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "typing.Any",
- "zatom.utils.tensor_utils.apply_mask",
- "torch._utils._rebuild_tensor_v2",
- "torch.distributions.beta.Beta",
- "zatom.models.architectures.transformer.common.ChargeSpinEmbedding",
- "functools.partial",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.normalization.LayerNorm",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "__builtin__.long",
- "torch.FloatStorage",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "__builtin__.set",
- "torch.Size",
- "zatom.flow.time_factor.InverseTimeFactor",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "torch.nn.modules.activation.SiLU",
- "torch.nn.modules.linear.Identity",
- "torch.nn.modules.normalization.RMSNorm",
- "omegaconf.nodes.AnyNode",
- "collections.OrderedDict",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "__builtin__.dict",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "omegaconf.base.ContainerMetadata",
- "collections.defaultdict",
- "torch.nn.modules.sparse.Embedding",
- "zatom.models.architectures.tft.TFT",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "omegaconf.base.Metadata",
- "torch.nn.modules.container.ModuleList"
How to fix it?
676 MB
xet

Upload 22 files 9 days ago
zatom_1_joint_geom_pretraining_paper_weights.ckpt
Detected Pickle imports (47)
- "__builtin__.list",
- "torch.nn.modules.activation.SiLU",
- "torch.nn.modules.sparse.Embedding",
- "torch.Size",
- "zatom.utils.tensor_utils.apply_mask",
- "torch.LongStorage",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "torch.nn.modules.container.ModuleList",
- "omegaconf.dictconfig.DictConfig",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "omegaconf.base.ContainerMetadata",
- "torch.nn.modules.container.Sequential",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "torch._utils._rebuild_tensor_v2",
- "omegaconf.listconfig.ListConfig",
- "collections.OrderedDict",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "torch.optim.adamw.AdamW",
- "torch.nn.modules.linear.Linear",
- "omegaconf.base.Metadata",
- "torch.nn.modules.normalization.RMSNorm",
- "zatom.flow.time_factor.InverseTimeFactor",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "functools.partial",
- "__builtin__.long",
- "torch.distributions.dirichlet.Dirichlet",
- "torch.nn.modules.loss.MSELoss",
- "__builtin__.set",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "collections.defaultdict",
- "torch.nn.modules.normalization.LayerNorm",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "zatom.models.architectures.tft.TFT",
- "torch.distributions.beta.Beta",
- "typing.Any",
- "torch.FloatStorage",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "torch._utils._rebuild_parameter",
- "omegaconf.nodes.AnyNode",
- "__builtin__.dict",
- "torch.nn.modules.linear.Identity"
How to fix it?
1.24 GB
xet

Upload 22 files 9 days ago
zatom_1_joint_k25_layer_mol_prop_pred_paper_weights.ckpt
Detected Pickle imports (47)
- "torch._utils._rebuild_parameter",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "__builtin__.dict",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "torch.optim.adamw.AdamW",
- "zatom.models.architectures.tft.TFT",
- "omegaconf.dictconfig.DictConfig",
- "torch.nn.modules.container.Sequential",
- "omegaconf.listconfig.ListConfig",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "torch.nn.modules.activation.SiLU",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "zatom.flow.time_factor.InverseTimeFactor",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "typing.Any",
- "omegaconf.nodes.AnyNode",
- "torch.distributions.beta.Beta",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "torch.FloatStorage",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "torch.nn.modules.sparse.Embedding",
- "zatom.models.architectures.transformer.common.ChargeSpinEmbedding",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "omegaconf.base.Metadata",
- "zatom.utils.tensor_utils.apply_mask",
- "collections.defaultdict",
- "omegaconf.base.ContainerMetadata",
- "torch._utils._rebuild_tensor_v2",
- "__builtin__.set",
- "torch.nn.modules.linear.Identity",
- "torch.nn.modules.linear.Linear",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "__builtin__.long",
- "torch.Size",
- "torch.distributions.dirichlet.Dirichlet",
- "torch.nn.modules.normalization.RMSNorm",
- "functools.partial",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "collections.OrderedDict",
- "__builtin__.list",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.nn.modules.loss.MSELoss",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "torch.nn.modules.container.ModuleList"
How to fix it?
676 MB
xet

Upload 22 files 9 days ago
zatom_1_joint_k75_layer_mol_prop_pred_paper_weights.ckpt
Detected Pickle imports (47)
- "__builtin__.list",
- "torch.nn.modules.activation.SiLU",
- "torch.nn.modules.sparse.Embedding",
- "torch.Size",
- "zatom.utils.tensor_utils.apply_mask",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "torch.nn.modules.container.ModuleList",
- "omegaconf.dictconfig.DictConfig",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "omegaconf.base.ContainerMetadata",
- "torch.nn.modules.container.Sequential",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "torch._utils._rebuild_tensor_v2",
- "omegaconf.listconfig.ListConfig",
- "collections.OrderedDict",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "torch.optim.adamw.AdamW",
- "torch.nn.modules.linear.Linear",
- "omegaconf.base.Metadata",
- "torch.nn.modules.normalization.RMSNorm",
- "zatom.flow.time_factor.InverseTimeFactor",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "functools.partial",
- "zatom.models.architectures.transformer.common.ChargeSpinEmbedding",
- "__builtin__.long",
- "torch.distributions.dirichlet.Dirichlet",
- "torch.nn.modules.loss.MSELoss",
- "__builtin__.set",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "collections.defaultdict",
- "torch.nn.modules.normalization.LayerNorm",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "zatom.models.architectures.tft.TFT",
- "torch.distributions.beta.Beta",
- "typing.Any",
- "torch.FloatStorage",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "torch._utils._rebuild_parameter",
- "omegaconf.nodes.AnyNode",
- "__builtin__.dict",
- "torch.nn.modules.linear.Identity"
How to fix it?
676 MB
xet

Upload 22 files 9 days ago
zatom_1_joint_mat_prop_paper_weights.ckpt
Detected Pickle imports (47)
- "__builtin__.list",
- "torch.nn.modules.activation.SiLU",
- "torch.nn.modules.sparse.Embedding",
- "torch.Size",
- "zatom.utils.tensor_utils.apply_mask",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "torch.nn.modules.container.ModuleList",
- "omegaconf.dictconfig.DictConfig",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "omegaconf.base.ContainerMetadata",
- "torch.nn.modules.container.Sequential",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "torch._utils._rebuild_tensor_v2",
- "omegaconf.listconfig.ListConfig",
- "collections.OrderedDict",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "torch.optim.adamw.AdamW",
- "torch.nn.modules.linear.Linear",
- "omegaconf.base.Metadata",
- "torch.nn.modules.normalization.RMSNorm",
- "zatom.flow.time_factor.InverseTimeFactor",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "functools.partial",
- "zatom.models.architectures.transformer.common.ChargeSpinEmbedding",
- "__builtin__.long",
- "torch.distributions.dirichlet.Dirichlet",
- "torch.nn.modules.loss.MSELoss",
- "__builtin__.set",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "collections.defaultdict",
- "torch.nn.modules.normalization.LayerNorm",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "zatom.models.architectures.tft.TFT",
- "torch.distributions.beta.Beta",
- "typing.Any",
- "torch.FloatStorage",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "torch._utils._rebuild_parameter",
- "omegaconf.nodes.AnyNode",
- "__builtin__.dict",
- "torch.nn.modules.linear.Identity"
How to fix it?
3.05 GB
xet

Upload 22 files 9 days ago
zatom_1_joint_mid_layer_mol_prop_pred_paper_weights.ckpt
Detected Pickle imports (47)
- "__builtin__.list",
- "torch.nn.modules.activation.SiLU",
- "torch.nn.modules.sparse.Embedding",
- "torch.Size",
- "zatom.utils.tensor_utils.apply_mask",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "torch.nn.modules.container.ModuleList",
- "omegaconf.dictconfig.DictConfig",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "omegaconf.base.ContainerMetadata",
- "torch.nn.modules.container.Sequential",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "torch._utils._rebuild_tensor_v2",
- "omegaconf.listconfig.ListConfig",
- "collections.OrderedDict",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "torch.optim.adamw.AdamW",
- "torch.nn.modules.linear.Linear",
- "omegaconf.base.Metadata",
- "torch.nn.modules.normalization.RMSNorm",
- "zatom.flow.time_factor.InverseTimeFactor",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "functools.partial",
- "zatom.models.architectures.transformer.common.ChargeSpinEmbedding",
- "__builtin__.long",
- "torch.distributions.dirichlet.Dirichlet",
- "torch.nn.modules.loss.MSELoss",
- "__builtin__.set",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "collections.defaultdict",
- "torch.nn.modules.normalization.LayerNorm",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "zatom.models.architectures.tft.TFT",
- "torch.distributions.beta.Beta",
- "typing.Any",
- "torch.FloatStorage",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "torch._utils._rebuild_parameter",
- "omegaconf.nodes.AnyNode",
- "__builtin__.dict",
- "torch.nn.modules.linear.Identity"
How to fix it?
676 MB
xet

Upload 22 files 9 days ago
zatom_1_joint_mol_and_mat_prop_pred_paper_weights.ckpt
Detected Pickle imports (47)
- "torch._utils._rebuild_parameter",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "__builtin__.dict",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "torch.optim.adamw.AdamW",
- "zatom.models.architectures.tft.TFT",
- "omegaconf.dictconfig.DictConfig",
- "torch.nn.modules.container.Sequential",
- "omegaconf.listconfig.ListConfig",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "torch.nn.modules.activation.SiLU",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "zatom.flow.time_factor.InverseTimeFactor",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "typing.Any",
- "omegaconf.nodes.AnyNode",
- "torch.distributions.beta.Beta",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "torch.FloatStorage",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "torch.nn.modules.sparse.Embedding",
- "zatom.models.architectures.transformer.common.ChargeSpinEmbedding",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "omegaconf.base.Metadata",
- "zatom.utils.tensor_utils.apply_mask",
- "collections.defaultdict",
- "omegaconf.base.ContainerMetadata",
- "torch._utils._rebuild_tensor_v2",
- "__builtin__.set",
- "torch.nn.modules.linear.Identity",
- "torch.nn.modules.linear.Linear",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "__builtin__.long",
- "torch.Size",
- "torch.distributions.dirichlet.Dirichlet",
- "torch.nn.modules.normalization.RMSNorm",
- "functools.partial",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "collections.OrderedDict",
- "__builtin__.list",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.nn.modules.loss.MSELoss",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "torch.nn.modules.container.ModuleList"
How to fix it?
676 MB
xet

Upload 22 files 9 days ago
zatom_1_joint_mol_prop_pred_paper_weights.ckpt
Detected Pickle imports (47)
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "torch.Size",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "torch.nn.modules.container.ModuleList",
- "zatom.flow.time_factor.InverseTimeFactor",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "torch.distributions.beta.Beta",
- "__builtin__.long",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.linear.Identity",
- "torch._utils._rebuild_tensor_v2",
- "__builtin__.dict",
- "functools.partial",
- "omegaconf.base.ContainerMetadata",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "torch.nn.modules.activation.SiLU",
- "torch.nn.modules.linear.Linear",
- "__builtin__.list",
- "omegaconf.listconfig.ListConfig",
- "omegaconf.nodes.AnyNode",
- "typing.Any",
- "zatom.models.architectures.transformer.common.ChargeSpinEmbedding",
- "__builtin__.set",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "collections.OrderedDict",
- "zatom.models.architectures.tft.TFT",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "torch.optim.adamw.AdamW",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "omegaconf.base.Metadata",
- "torch.nn.modules.normalization.RMSNorm",
- "zatom.utils.tensor_utils.apply_mask",
- "collections.defaultdict",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "torch.distributions.dirichlet.Dirichlet",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "torch.nn.modules.sparse.Embedding",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "torch.nn.modules.loss.MSELoss",
- "torch.FloatStorage",
- "torch.nn.modules.normalization.LayerNorm",
- "omegaconf.dictconfig.DictConfig"
How to fix it?
676 MB
xet

Upload 22 files 9 days ago
zatom_1_joint_paper_weights.ckpt
Detected Pickle imports (47)
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "torch.Size",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "torch.nn.modules.container.ModuleList",
- "zatom.flow.time_factor.InverseTimeFactor",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "torch.distributions.beta.Beta",
- "__builtin__.long",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.linear.Identity",
- "torch._utils._rebuild_tensor_v2",
- "__builtin__.dict",
- "functools.partial",
- "omegaconf.base.ContainerMetadata",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "torch.nn.modules.activation.SiLU",
- "torch.nn.modules.linear.Linear",
- "__builtin__.list",
- "omegaconf.listconfig.ListConfig",
- "omegaconf.nodes.AnyNode",
- "typing.Any",
- "zatom.models.architectures.transformer.common.ChargeSpinEmbedding",
- "__builtin__.set",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "collections.OrderedDict",
- "zatom.models.architectures.tft.TFT",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "torch.optim.adamw.AdamW",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "omegaconf.base.Metadata",
- "torch.nn.modules.normalization.RMSNorm",
- "zatom.utils.tensor_utils.apply_mask",
- "collections.defaultdict",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "torch.distributions.dirichlet.Dirichlet",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "torch.nn.modules.sparse.Embedding",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "torch.nn.modules.loss.MSELoss",
- "torch.FloatStorage",
- "torch.nn.modules.normalization.LayerNorm",
- "omegaconf.dictconfig.DictConfig"
How to fix it?
3.05 GB
xet

Upload zatom_1_joint_paper_weights.ckpt 9 days ago
zatom_1_joint_pretraining_paper_weights.ckpt
Detected Pickle imports (47)
- "__builtin__.dict",
- "omegaconf.base.Metadata",
- "torch.Size",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.nn.modules.container.Sequential",
- "omegaconf.dictconfig.DictConfig",
- "collections.defaultdict",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "typing.Any",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.container.ModuleList",
- "torch.LongStorage",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "torch.nn.modules.sparse.Embedding",
- "zatom.flow.time_factor.InverseTimeFactor",
- "__builtin__.long",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "functools.partial",
- "torch.nn.modules.activation.SiLU",
- "torch.distributions.beta.Beta",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.loss.MSELoss",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "__builtin__.set",
- "torch.optim.adamw.AdamW",
- "torch.nn.modules.normalization.RMSNorm",
- "torch.nn.modules.linear.Identity",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "__builtin__.list",
- "torch._utils._rebuild_parameter",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "omegaconf.listconfig.ListConfig",
- "torch.FloatStorage",
- "zatom.models.architectures.tft.TFT",
- "collections.OrderedDict",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "omegaconf.nodes.AnyNode",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "zatom.utils.tensor_utils.apply_mask",
- "torch.distributions.dirichlet.Dirichlet",
- "omegaconf.base.ContainerMetadata"
How to fix it?
1.24 GB
xet

Upload 22 files 9 days ago
zatom_1_l_joint_pretraining_paper_weights.ckpt
Detected Pickle imports (46)
- "typing.Any",
- "omegaconf.nodes.AnyNode",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "collections.OrderedDict",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "omegaconf.base.ContainerMetadata",
- "torch.distributions.beta.Beta",
- "zatom.utils.tensor_utils.apply_mask",
- "torch.nn.modules.linear.Linear",
- "torch.optim.adamw.AdamW",
- "__builtin__.list",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "torch.FloatStorage",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.container.Sequential",
- "torch.distributions.dirichlet.Dirichlet",
- "torch.Size",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "__builtin__.long",
- "zatom.models.architectures.tft.TFT",
- "__builtin__.dict",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "torch.nn.modules.normalization.RMSNorm",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "torch.nn.modules.sparse.Embedding",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "torch.nn.modules.loss.MSELoss",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "omegaconf.dictconfig.DictConfig",
- "omegaconf.listconfig.ListConfig",
- "torch.nn.modules.activation.SiLU",
- "functools.partial",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "torch.nn.modules.normalization.LayerNorm",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "zatom.flow.time_factor.InverseTimeFactor",
- "torch._utils._rebuild_parameter",
- "omegaconf.base.Metadata",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "collections.defaultdict",
- "__builtin__.set",
- "torch.LongStorage",
- "torch.nn.modules.container.ModuleList"
How to fix it?
2.61 GB
xet

Upload 22 files 9 days ago
zatom_1_mp20_only_pretraining_paper_weights.ckpt
Detected Pickle imports (47)
- "typing.Any",
- "omegaconf.nodes.AnyNode",
- "torch.nn.modules.linear.Identity",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "collections.OrderedDict",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "omegaconf.base.ContainerMetadata",
- "torch.distributions.beta.Beta",
- "zatom.utils.tensor_utils.apply_mask",
- "torch.nn.modules.linear.Linear",
- "torch.optim.adamw.AdamW",
- "__builtin__.list",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "torch.FloatStorage",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.container.Sequential",
- "torch.distributions.dirichlet.Dirichlet",
- "torch.Size",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "__builtin__.long",
- "zatom.models.architectures.tft.TFT",
- "__builtin__.dict",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "torch.nn.modules.normalization.RMSNorm",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "torch.nn.modules.sparse.Embedding",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "torch.nn.modules.loss.MSELoss",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "omegaconf.dictconfig.DictConfig",
- "omegaconf.listconfig.ListConfig",
- "torch.nn.modules.activation.SiLU",
- "functools.partial",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "torch.nn.modules.normalization.LayerNorm",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "zatom.flow.time_factor.InverseTimeFactor",
- "torch._utils._rebuild_parameter",
- "omegaconf.base.Metadata",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "collections.defaultdict",
- "__builtin__.set",
- "torch.LongStorage",
- "torch.nn.modules.container.ModuleList"
How to fix it?
1.24 GB
xet

Upload 22 files 9 days ago
zatom_1_non_pretrained_mol_prop_pred_paper_weights.ckpt
Detected Pickle imports (47)
- "typing.Any",
- "omegaconf.nodes.AnyNode",
- "torch.nn.modules.linear.Identity",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "collections.OrderedDict",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "omegaconf.base.ContainerMetadata",
- "torch.distributions.beta.Beta",
- "zatom.utils.tensor_utils.apply_mask",
- "torch.nn.modules.linear.Linear",
- "torch.optim.adamw.AdamW",
- "__builtin__.list",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "torch.FloatStorage",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.container.Sequential",
- "torch.distributions.dirichlet.Dirichlet",
- "torch.Size",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "__builtin__.long",
- "zatom.models.architectures.tft.TFT",
- "__builtin__.dict",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "torch.nn.modules.normalization.RMSNorm",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "torch.nn.modules.sparse.Embedding",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "torch.nn.modules.loss.MSELoss",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "omegaconf.dictconfig.DictConfig",
- "omegaconf.listconfig.ListConfig",
- "torch.nn.modules.activation.SiLU",
- "functools.partial",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "torch.nn.modules.normalization.LayerNorm",
- "zatom.models.architectures.transformer.common.ChargeSpinEmbedding",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "zatom.flow.time_factor.InverseTimeFactor",
- "torch._utils._rebuild_parameter",
- "omegaconf.base.Metadata",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "collections.defaultdict",
- "__builtin__.set",
- "torch.nn.modules.container.ModuleList"
How to fix it?
676 MB
xet

Upload 22 files 9 days ago
zatom_1_omol25_only_mlip_pretraining_paper_weights.ckpt
Detected Pickle imports (48)
- "typing.Any",
- "omegaconf.nodes.AnyNode",
- "torch.nn.modules.linear.Identity",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "collections.OrderedDict",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "omegaconf.base.ContainerMetadata",
- "torch.distributions.beta.Beta",
- "zatom.utils.tensor_utils.apply_mask",
- "torch.nn.modules.linear.Linear",
- "torch.optim.adamw.AdamW",
- "__builtin__.list",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "torch.FloatStorage",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.container.Sequential",
- "torch.distributions.dirichlet.Dirichlet",
- "torch.Size",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "__builtin__.long",
- "zatom.models.architectures.tft.TFT",
- "__builtin__.dict",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "torch.nn.modules.normalization.RMSNorm",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "torch.nn.modules.sparse.Embedding",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "torch.nn.modules.loss.MSELoss",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "omegaconf.dictconfig.DictConfig",
- "omegaconf.listconfig.ListConfig",
- "torch.nn.modules.activation.SiLU",
- "functools.partial",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "torch.nn.modules.normalization.LayerNorm",
- "zatom.models.architectures.transformer.common.ChargeSpinEmbedding",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "zatom.flow.time_factor.InverseTimeFactor",
- "torch._utils._rebuild_parameter",
- "omegaconf.base.Metadata",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "collections.defaultdict",
- "__builtin__.set",
- "torch.LongStorage",
- "torch.nn.modules.container.ModuleList"
How to fix it?
1.49 GB
xet

Upload 22 files 9 days ago
zatom_1_omol25_only_pretraining_paper_weights.ckpt
Detected Pickle imports (47)
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "__builtin__.long",
- "omegaconf.listconfig.ListConfig",
- "torch.nn.modules.normalization.RMSNorm",
- "torch.nn.modules.loss.MSELoss",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "__builtin__.dict",
- "torch._utils._rebuild_tensor_v2",
- "torch.LongStorage",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "torch.FloatStorage",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "torch.nn.modules.activation.SiLU",
- "torch.optim.adamw.AdamW",
- "torch._utils._rebuild_parameter",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "functools.partial",
- "torch.distributions.beta.Beta",
- "omegaconf.dictconfig.DictConfig",
- "torch.nn.modules.container.Sequential",
- "__builtin__.list",
- "zatom.models.architectures.tft.TFT",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.container.ModuleList",
- "collections.OrderedDict",
- "torch.Size",
- "torch.nn.modules.sparse.Embedding",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "__builtin__.set",
- "zatom.flow.time_factor.InverseTimeFactor",
- "torch.nn.modules.normalization.LayerNorm",
- "zatom.utils.tensor_utils.apply_mask",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "torch.nn.modules.linear.Identity",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "omegaconf.base.ContainerMetadata",
- "omegaconf.base.Metadata",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "typing.Any",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "torch.distributions.dirichlet.Dirichlet",
- "omegaconf.nodes.AnyNode",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "collections.defaultdict"
How to fix it?
1.24 GB
xet

Upload 22 files 9 days ago
zatom_1_qm9_only_mol_and_mat_prop_pred_paper_weights.ckpt
Detected Pickle imports (47)
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "__builtin__.long",
- "omegaconf.listconfig.ListConfig",
- "torch.nn.modules.normalization.RMSNorm",
- "torch.nn.modules.loss.MSELoss",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "__builtin__.dict",
- "torch._utils._rebuild_tensor_v2",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "torch.FloatStorage",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "torch.nn.modules.activation.SiLU",
- "torch.optim.adamw.AdamW",
- "torch._utils._rebuild_parameter",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "functools.partial",
- "torch.distributions.beta.Beta",
- "omegaconf.dictconfig.DictConfig",
- "zatom.models.architectures.transformer.common.ChargeSpinEmbedding",
- "torch.nn.modules.container.Sequential",
- "__builtin__.list",
- "zatom.models.architectures.tft.TFT",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.container.ModuleList",
- "collections.OrderedDict",
- "torch.Size",
- "torch.nn.modules.sparse.Embedding",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "__builtin__.set",
- "zatom.flow.time_factor.InverseTimeFactor",
- "torch.nn.modules.normalization.LayerNorm",
- "zatom.utils.tensor_utils.apply_mask",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "torch.nn.modules.linear.Identity",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "omegaconf.base.ContainerMetadata",
- "omegaconf.base.Metadata",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "typing.Any",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "torch.distributions.dirichlet.Dirichlet",
- "omegaconf.nodes.AnyNode",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "collections.defaultdict"
How to fix it?
676 MB
xet

Upload 22 files 9 days ago
zatom_1_qm9_only_mol_prop_pred_paper_weights.ckpt
Detected Pickle imports (47)
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "__builtin__.long",
- "omegaconf.listconfig.ListConfig",
- "torch.nn.modules.normalization.RMSNorm",
- "torch.nn.modules.loss.MSELoss",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "__builtin__.dict",
- "torch._utils._rebuild_tensor_v2",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "torch.FloatStorage",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "torch.nn.modules.activation.SiLU",
- "torch.optim.adamw.AdamW",
- "torch._utils._rebuild_parameter",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "functools.partial",
- "torch.distributions.beta.Beta",
- "omegaconf.dictconfig.DictConfig",
- "zatom.models.architectures.transformer.common.ChargeSpinEmbedding",
- "torch.nn.modules.container.Sequential",
- "__builtin__.list",
- "zatom.models.architectures.tft.TFT",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.container.ModuleList",
- "collections.OrderedDict",
- "torch.Size",
- "torch.nn.modules.sparse.Embedding",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "__builtin__.set",
- "zatom.flow.time_factor.InverseTimeFactor",
- "torch.nn.modules.normalization.LayerNorm",
- "zatom.utils.tensor_utils.apply_mask",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "torch.nn.modules.linear.Identity",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "omegaconf.base.ContainerMetadata",
- "omegaconf.base.Metadata",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "typing.Any",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "torch.distributions.dirichlet.Dirichlet",
- "omegaconf.nodes.AnyNode",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "collections.defaultdict"
How to fix it?
676 MB
xet

Upload 22 files 9 days ago
zatom_1_qm9_only_pretraining_paper_weights.ckpt
Detected Pickle imports (47)
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "__builtin__.long",
- "omegaconf.base.Metadata",
- "torch._utils._rebuild_parameter",
- "torch.FloatStorage",
- "zatom.utils.tensor_utils.apply_mask",
- "torch._utils._rebuild_tensor_v2",
- "torch.distributions.beta.Beta",
- "omegaconf.base.ContainerMetadata",
- "torch.nn.modules.loss.MSELoss",
- "omegaconf.nodes.AnyNode",
- "torch.nn.modules.container.Sequential",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "zatom.flow.time_factor.InverseTimeFactor",
- "torch.optim.adamw.AdamW",
- "collections.OrderedDict",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "torch.nn.modules.container.ModuleList",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "functools.partial",
- "omegaconf.dictconfig.DictConfig",
- "torch.LongStorage",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.activation.SiLU",
- "__builtin__.set",
- "torch.nn.modules.linear.Identity",
- "omegaconf.listconfig.ListConfig",
- "zatom.models.architectures.tft.TFT",
- "torch.nn.modules.normalization.LayerNorm",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "torch.Size",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "collections.defaultdict",
- "__builtin__.dict",
- "__builtin__.list",
- "typing.Any",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "torch.nn.modules.normalization.RMSNorm",
- "torch.distributions.dirichlet.Dirichlet",
- "torch.nn.modules.sparse.Embedding"
How to fix it?
1.24 GB
xet

Upload 22 files 9 days ago
zatom_1_qmof_only_pretraining_paper_weights.ckpt
Detected Pickle imports (47)
- "torch.distributions.dirichlet.Dirichlet",
- "__builtin__.list",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.sparse.Embedding",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "torch.nn.modules.activation.SiLU",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "torch.nn.modules.container.ModuleList",
- "omegaconf.base.Metadata",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "__builtin__.dict",
- "torch.distributions.beta.Beta",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "torch._utils._rebuild_parameter",
- "torch.FloatStorage",
- "torch.nn.modules.normalization.RMSNorm",
- "omegaconf.listconfig.ListConfig",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "torch.nn.modules.linear.Linear",
- "zatom.models.architectures.tft.TFT",
- "omegaconf.base.ContainerMetadata",
- "collections.OrderedDict",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "torch.nn.modules.loss.MSELoss",
- "typing.Any",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "__builtin__.long",
- "zatom.utils.tensor_utils.apply_mask",
- "omegaconf.dictconfig.DictConfig",
- "zatom.flow.time_factor.InverseTimeFactor",
- "torch._utils._rebuild_tensor_v2",
- "torch.Size",
- "torch.LongStorage",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "torch.nn.modules.linear.Identity",
- "torch.nn.modules.normalization.LayerNorm",
- "__builtin__.set",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "collections.defaultdict",
- "torch.optim.adamw.AdamW",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "omegaconf.nodes.AnyNode",
- "functools.partial"
How to fix it?
1.24 GB
xet

Upload 22 files 9 days ago
zatom_1_wd_joint_pretraining_paper_weights.ckpt
Detected Pickle imports (47)
- "torch.FloatStorage",
- "zatom.flow.time_factor.InverseTimeFactor",
- "functools.partial",
- "omegaconf.listconfig.ListConfig",
- "torch.Size",
- "torch.nn.modules.sparse.Embedding",
- "omegaconf.nodes.AnyNode",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "torch.optim.adamw.AdamW",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "torch.nn.modules.linear.Linear",
- "omegaconf.dictconfig.DictConfig",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "torch.distributions.dirichlet.Dirichlet",
- "torch._utils._rebuild_parameter",
- "__builtin__.long",
- "collections.OrderedDict",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "torch.distributions.beta.Beta",
- "torch.nn.modules.activation.SiLU",
- "typing.Any",
- "omegaconf.base.ContainerMetadata",
- "__builtin__.list",
- "zatom.utils.tensor_utils.apply_mask",
- "torch.nn.modules.normalization.RMSNorm",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.nn.modules.container.ModuleList",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "torch.LongStorage",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "__builtin__.dict",
- "zatom.models.architectures.tft.TFT",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "torch.nn.modules.loss.MSELoss",
- "torch.nn.modules.linear.Identity",
- "__builtin__.set",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "torch.nn.modules.container.Sequential",
- "torch._utils._rebuild_tensor_v2",
- "omegaconf.base.Metadata",
- "collections.defaultdict"
How to fix it?
1.24 GB
xet

Upload 22 files 9 days ago
zatom_1_xl_joint_mol_prop_pred_paper_weights.ckpt
Detected Pickle imports (46)
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "omegaconf.nodes.AnyNode",
- "torch.nn.modules.loss.MSELoss",
- "torch.distributions.beta.Beta",
- "torch._utils._rebuild_parameter",
- "omegaconf.listconfig.ListConfig",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "zatom.utils.tensor_utils.apply_mask",
- "torch.nn.modules.normalization.RMSNorm",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "__builtin__.dict",
- "functools.partial",
- "torch.distributions.dirichlet.Dirichlet",
- "torch.nn.modules.sparse.Embedding",
- "torch.nn.modules.activation.SiLU",
- "omegaconf.base.ContainerMetadata",
- "typing.Any",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "torch.nn.modules.container.Sequential",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "__builtin__.set",
- "torch._utils._rebuild_tensor_v2",
- "omegaconf.dictconfig.DictConfig",
- "__builtin__.long",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "torch.nn.modules.normalization.LayerNorm",
- "omegaconf.base.Metadata",
- "torch.nn.modules.container.ModuleList",
- "torch.optim.adamw.AdamW",
- "collections.defaultdict",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "torch.nn.modules.linear.Linear",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "zatom.models.architectures.transformer.common.ChargeSpinEmbedding",
- "zatom.models.architectures.tft.TFT",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "torch.Size",
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "torch.FloatStorage",
- "__builtin__.list",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "zatom.flow.time_factor.InverseTimeFactor",
- "collections.OrderedDict"
How to fix it?
1.8 GB
xet

Upload 22 files 9 days ago
zatom_1_xl_joint_pretraining_paper_weights.ckpt
Detected Pickle imports (46)
- "zatom.models.architectures.transformer.attention.ModernAttention",
- "__builtin__.set",
- "omegaconf.nodes.AnyNode",
- "zatom.models.architectures.transformer.transformer.ModernTransformerDecoderBlock",
- "zatom.models.architectures.tft.TFT",
- "zatom.flow.interpolants.DiscreteInterpolant",
- "torch.nn.modules.loss.MSELoss",
- "zatom.models.architectures.transformer.transformer.ModernTransformer",
- "omegaconf.listconfig.ListConfig",
- "zatom.sampling.noise_schedule.SampleNoiseSchedule",
- "zatom.models.architectures.transformer.transformer_module.TransformerModule",
- "zatom.flow.interpolants.SDEMetricInterpolant",
- "zatom.models.architectures.transformer.positional_encoder.SinusoidEncoding",
- "torch.nn.modules.linear.Linear",
- "omegaconf.base.ContainerMetadata",
- "torch.nn.modules.container.ModuleList",
- "torch.optim.adamw.AdamW",
- "torch.nn.modules.sparse.Embedding",
- "zatom.models.architectures.transformer.common.LabelEmbedder",
- "omegaconf.base.Metadata",
- "torch.nn.modules.activation.SiLU",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "torch.Size",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.distributions.beta.Beta",
- "functools.partial",
- "zatom.flow.time_factor.InverseTimeFactor",
- "torch.distributions.dirichlet.Dirichlet",
- "__builtin__.dict",
- "torch.nn.modules.normalization.RMSNorm",
- "zatom.models.architectures.transformer.positional_encoder.TimeFourierEncoding",
- "torch.FloatStorage",
- "collections.defaultdict",
- "zatom.models.architectures.transformer.common.SwiGLUFeedForward",
- "omegaconf.dictconfig.DictConfig",
- "__builtin__.long",
- "torch._utils._rebuild_tensor_v2",
- "collections.OrderedDict",
- "zatom.models.architectures.transformer.transformer.ModernTransformerBlock",
- "typing.Any",
- "zatom.utils.tensor_utils.mask_and_zero_com",
- "torch.nn.modules.container.Sequential",
- "torch._utils._rebuild_parameter",
- "torch.LongStorage",
- "zatom.utils.tensor_utils.apply_mask",
- "__builtin__.list"
How to fix it?
4.71 GB
xet

Upload 22 files 9 days ago

Detected Pickle imports (54)

Detected Pickle imports (54)

Detected Pickle imports (47)

Detected Pickle imports (47)

Detected Pickle imports (47)

Detected Pickle imports (47)

Detected Pickle imports (47)

Detected Pickle imports (47)

Detected Pickle imports (47)

Detected Pickle imports (47)

Detected Pickle imports (47)

Detected Pickle imports (47)

Detected Pickle imports (46)

Detected Pickle imports (47)

Detected Pickle imports (47)

Detected Pickle imports (48)

Detected Pickle imports (47)

Detected Pickle imports (47)

Detected Pickle imports (47)

Detected Pickle imports (47)

Detected Pickle imports (47)

Detected Pickle imports (47)

Detected Pickle imports (46)

Detected Pickle imports (46)