Hugging Face
Models
Datasets
Spaces
Buckets
new
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 (legacy)
arxiv:
2503.03965
License:
mit
Model card
Files
Files and versions
xet
Community
2
Deploy
Use this model
main
all-atom-diffusion-transformer
/
GEOM_only_undertrained
3.1 GB
Ctrl+K
Ctrl+K
1 contributor
History:
3 commits
This model has 1 file scanned as unsafe.
Show
files
chaitjo
Upload ldm.ckpt
49ee17b
verified
4 months ago
.gitkeep
Safe
0 Bytes
Create GEOM_only_undertrained/.gitkeep
4 months ago
ldm.ckpt
Unsafe
pickle
Detected Pickle imports (33)
"torch.nn.modules.linear.Identity"
,
"src.models.denoisers.dit.TimestepEmbedder"
,
"torch._utils._rebuild_parameter"
,
"torch.nn.modules.container.Sequential"
,
"src.models.denoisers.dit.Mlp"
,
"omegaconf.base.Metadata"
,
"torch.device"
,
"torch.nn.modules.linear.Linear"
,
"src.models.denoisers.dit.DiTBlock"
,
"__builtin__.set"
,
"torch.FloatStorage"
,
"torch.nn.modules.dropout.Dropout"
,
"src.models.interpolants.flow_matching.FlowMatchingInterpolant"
,
"src.models.denoisers.dit.LabelEmbedder"
,
"omegaconf.nodes.AnyNode"
,
"omegaconf.dictconfig.DictConfig"
,
"torch.optim.adamw.AdamW"
,
"torch.nn.modules.sparse.Embedding"
,
"torch.nn.modules.activation.MultiheadAttention"
,
"torch.nn.modules.activation.GELU"
,
"collections.OrderedDict"
,
"torch.nn.modules.linear.NonDynamicallyQuantizableLinear"
,
"functools.partial"
,
"typing.Any"
,
"omegaconf.base.ContainerMetadata"
,
"__builtin__.dict"
,
"torch.nn.modules.activation.SiLU"
,
"torch.nn.modules.container.ModuleList"
,
"torch.nn.modules.normalization.LayerNorm"
,
"torch._utils._rebuild_tensor_v2"
,
"collections.defaultdict"
,
"src.models.denoisers.dit.FinalLayer"
,
"src.models.denoisers.dit.DiT"
How to fix it?
2.48 GB
xet
Upload ldm.ckpt
4 months ago
vae.ckpt
Unsafe
pickle
Detected Pickle imports (28)
"__builtin__.dict"
,
"typing.Any"
,
"omegaconf.base.Metadata"
,
"torch.nn.modules.container.ModuleList"
,
"torch.nn.modules.linear.NonDynamicallyQuantizableLinear"
,
"src.models.encoders.transformer.TransformerEncoder"
,
"collections.OrderedDict"
,
"torch._utils._rebuild_parameter"
,
"torch.FloatStorage"
,
"torch.nn.modules.sparse.Embedding"
,
"torch.nn.modules.activation.SiLU"
,
"__builtin__.set"
,
"torch.nn.modules.container.Sequential"
,
"torch.nn.modules.dropout.Dropout"
,
"src.models.decoders.transformer.TransformerDecoder"
,
"torch.nn.modules.transformer.TransformerEncoder"
,
"torch.optim.adamw.AdamW"
,
"omegaconf.dictconfig.DictConfig"
,
"functools.partial"
,
"torch.nn.modules.linear.Linear"
,
"collections.defaultdict"
,
"torch.nn.modules.normalization.LayerNorm"
,
"torch.nn.modules.activation.GELU"
,
"omegaconf.nodes.AnyNode"
,
"torch.nn.modules.activation.MultiheadAttention"
,
"torch._utils._rebuild_tensor_v2"
,
"omegaconf.base.ContainerMetadata"
,
"torch.nn.modules.transformer.TransformerEncoderLayer"
How to fix it?
613 MB
xet
Upload vae.ckpt
4 months ago