MolecularDiffusion.cli.generate¶

Generation command for MolCraft CLI.

Adapted from scripts/generate.py for package-level execution.

Attributes¶

Functions¶

`generate`(→ Tuple[Dict[str, Any], Dict[str, Any]])	Main generation function.
`generate_main`(cfg)	Entry point for CLI generate command.
`is_rank_zero`()	Check if current process is rank zero.
`load_model`(chkpt_directory[, task_config, atom_vocab, ...])	Load model from checkpoint directory with auto-detection.
`log_hyperparameters`(object_dict)	Log hyperparameters for debugging.

Module Contents¶

MolecularDiffusion.cli.generate.generate(cfg: omegaconf.DictConfig) → Tuple[Dict[str, Any], Dict[str, Any]]¶: Main generation function.

MolecularDiffusion.cli.generate.generate_main(cfg: omegaconf.DictConfig)¶: Entry point for CLI generate command.

MolecularDiffusion.cli.generate.is_rank_zero()¶: Check if current process is rank zero.

MolecularDiffusion.cli.generate.load_model(chkpt_directory, task_config=None, atom_vocab=None, total_step=0)¶: Load model from checkpoint directory with auto-detection.

MolecularDiffusion.cli.generate.log_hyperparameters(object_dict: dict)¶: Log hyperparameters for debugging.

MolecularDiffusion.cli.generate.log¶