nemo_automodel.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.

Data

logger

API

nemo_automodel.checkpoint._torch_backports.logger

‘getLogger(…)’

nemo_automodel.checkpoint._torch_backports.apply_patches() → None

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