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
main
all-atom-diffusion-transformer
6.23 GB
1 contributor
History:
25 commits
This model has 1 file scanned as unsafe.
Show
files
chaitjo
Upload ldm.ckpt
49ee17b
verified
23 days ago
GEOM_only_undertrained
Upload ldm.ckpt
23 days ago
.gitattributes
Safe
1.52 kB
initial commit
8 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
8 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
7 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
8 months ago
README.md
5.33 kB
Update README.md
7 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
8 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
8 months ago