File size: 1,423 Bytes
f15d29e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 | 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
|