| sampler_partial: | |
| _target_: model.diffusion.sampling.classifier_free_guidance.GuidedPredictorCorrector.from_pl_module | |
| N: 1000 | |
| eps_t: ${eval:'1/${.N}'} | |
| _partial_: true | |
| guidance_scale: 0.0 | |
| remove_conditioning_fn: | |
| _target_: onescience.modules.embedding.mattergen_property_embeddings.SetUnconditionalEmbeddingType | |
| keep_conditioning_fn: | |
| _target_: onescience.modules.embedding.mattergen_property_embeddings.SetConditionalEmbeddingType | |
| predictor_partials: | |
| pos: | |
| _target_: model.diffusion.wrapped.wrapped_predictors_correctors.WrappedAncestralSamplingPredictor | |
| _partial_: true | |
| cell: | |
| _target_: model.common.diffusion.predictors_correctors.LatticeAncestralSamplingPredictor | |
| _partial_: true | |
| atomic_numbers: | |
| _target_: model.diffusion.d3pm.d3pm_predictors_correctors.D3PMAncestralSamplingPredictor | |
| predict_x0: True | |
| _partial_: true | |
| corrector_partials: | |
| pos: | |
| _target_: model.diffusion.wrapped.wrapped_predictors_correctors.WrappedLangevinCorrector | |
| _partial_: true | |
| max_step_size: 1e6 | |
| snr: 0.4 | |
| cell: | |
| _target_: model.common.diffusion.predictors_correctors.LatticeLangevinDiffCorrector | |
| _partial_: true | |
| max_step_size: 1e6 | |
| snr: 0.2 | |
| n_steps_corrector: 1 | |
| condition_loader_partial: | |
| _partial_: true | |
| _target_: onescience.datapipes.materials.mattergen.condition_factory.get_number_of_atoms_condition_loader | |