File size: 525 Bytes
f15d29e | 1 2 3 4 5 6 7 8 9 10 11 | _target_: onescience.modules.embedding.mattergen_property_embeddings.PropertyEmbedding
name: space_group
unconditional_embedding_module:
_target_: onescience.modules.embedding.mattergen_property_embeddings.EmbeddingVector
hidden_dim: ${lightning_module.diffusion_module.model.hidden_dim}
conditional_embedding_module:
_target_: onescience.modules.embedding.mattergen_property_embeddings.SpaceGroupEmbeddingVector
hidden_dim: ${lightning_module.diffusion_module.model.hidden_dim}
scaler:
_target_: torch.nn.Identity
|