Hugging Face
Models
Datasets
Spaces
Community
Docs
Enterprise
Pricing
Log In
Sign Up
chaitjo
/
all-atom-diffusion-transformer
like
3
Transformers
4 datasets
chemistry
materials
molecules
crystals
diffusion
transformer
latent-diffusion
all-atom
Eval Results
arxiv:
2503.03965
License:
mit
Model card
Files
Files and versions
xet
Community
2
Deploy
Use this model
7ed65cc
all-atom-diffusion-transformer
3.13 GB
1 contributor
History:
20 commits
chaitjo
Upload ADiT_molecules_geom.zip
7ed65cc
verified
6 months ago
.gitattributes
Safe
1.52 kB
initial commit
7 months ago
ADiT_crystals_mp20.zip
pickle
Pickle imports
No problematic imports detected
What is a pickle import?
11.4 MB
xet
Add 10K example crystals and molecules
7 months ago
ADiT_molecules_geom.zip
pickle
Pickle imports
No problematic imports detected
What is a pickle import?
13.2 MB
xet
Upload ADiT_molecules_geom.zip
6 months ago
ADiT_molecules_qm9.zip
pickle
Pickle imports
No problematic imports detected
What is a pickle import?
7.41 MB
xet
Add 10K example crystals and molecules
7 months ago
README.md
4.83 kB
Update README.md
6 months ago
config.json
Safe
2 Bytes
Tests download count
7 months ago
ldm.ckpt
Unsafe
pickle
Detected Pickle imports (33)
"src.models.denoisers.dit.Mlp"
,
"torch.optim.adamw.AdamW"
,
"torch._utils._rebuild_parameter"
,
"torch.nn.modules.linear.Identity"
,
"torch.device"
,
"functools.partial"
,
"__builtin__.set"
,
"torch.nn.modules.linear.NonDynamicallyQuantizableLinear"
,
"__builtin__.dict"
,
"collections.OrderedDict"
,
"omegaconf.nodes.AnyNode"
,
"src.models.denoisers.dit.FinalLayer"
,
"omegaconf.base.Metadata"
,
"torch.nn.modules.linear.Linear"
,
"src.models.interpolants.flow_matching.FlowMatchingInterpolant"
,
"torch.nn.modules.activation.SiLU"
,
"torch.nn.modules.sparse.Embedding"
,
"collections.defaultdict"
,
"src.models.denoisers.dit.TimestepEmbedder"
,
"src.models.denoisers.dit.DiTBlock"
,
"src.models.denoisers.dit.DiT"
,
"torch.nn.modules.container.Sequential"
,
"omegaconf.dictconfig.DictConfig"
,
"torch.nn.modules.activation.GELU"
,
"typing.Any"
,
"src.models.denoisers.dit.LabelEmbedder"
,
"torch.FloatStorage"
,
"torch._utils._rebuild_tensor_v2"
,
"omegaconf.base.ContainerMetadata"
,
"torch.nn.modules.normalization.LayerNorm"
,
"torch.nn.modules.dropout.Dropout"
,
"torch.nn.modules.container.ModuleList"
,
"torch.nn.modules.activation.MultiheadAttention"
How to fix it?
2.48 GB
xet
Upload ldm.ckpt
7 months ago
vae.ckpt
Unsafe
pickle
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
xet
Upload vae.ckpt
7 months ago