conf/lightning_module/diffusion_module/model/property_embeddings/chemical_system.yaml

_target_: onescience.modules.embedding.mattergen_property_embeddings.PropertyEmbedding
name: chemical_system
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.ChemicalSystemMultiHotEmbedding
  hidden_dim: ${lightning_module.diffusion_module.model.hidden_dim}
scaler:
  _target_: torch.nn.Identity