nemo_automodel.components.checkpoint._torch_backports

Runtime back-ports for old PyTorch versions. Will be deleted in future stable PyTorch versions.

Module Contents

Functions

apply_patches	Inject modified modules into an old torch.distributed.checkpoint.
apply_async_checkpoint_patch	Apply stabilization patch for torch.distributed.checkpoint async process executor. This serializes creation of the global background process across concurrent async_save calls.

Data

logger

API

nemo_automodel.components.checkpoint._torch_backports.logger

‘getLogger(…)’

nemo_automodel.components.checkpoint._torch_backports.apply_patches() → None

Inject modified modules into an old torch.distributed.checkpoint.

nemo_automodel.components.checkpoint._torch_backports.apply_async_checkpoint_patch() → None

Apply stabilization patch for torch.distributed.checkpoint async process executor. This serializes creation of the global background process across concurrent async_save calls.