nemo_automodel.components.checkpoint

Subpackages

• nemo_automodel.components.checkpoint._backports
  • nemo_automodel.components.checkpoint._backports._fsspec_filesystem
  • nemo_automodel.components.checkpoint._backports._version
  • nemo_automodel.components.checkpoint._backports.consolidate_hf_safetensors
  • nemo_automodel.components.checkpoint._backports.default_planner
  • nemo_automodel.components.checkpoint._backports.filesystem
  • nemo_automodel.components.checkpoint._backports.hf_storage
  • nemo_automodel.components.checkpoint._backports.hf_utils
  • nemo_automodel.components.checkpoint._backports.planner_helpers

Submodules

• nemo_automodel.components.checkpoint._torch_backports
• nemo_automodel.components.checkpoint.addons
• nemo_automodel.components.checkpoint.checkpointing
• nemo_automodel.components.checkpoint.state_dict_adapter
• nemo_automodel.components.checkpoint.stateful_wrappers
• nemo_automodel.components.checkpoint.utils