nemo_automodel.shared.torch_patches | NVIDIA NeMo AutoModel

Torch compatibility patches.

These patches are intentionally NOT applied at import nemo_automodel time to keep tokenizer-only imports lightweight. Call apply_torch_patches() from code paths that already depend on torch (training / distributed / dataloading).

Module Contents

Functions

Name	Description
`apply_torch_patches`	Apply small, version/packaging-specific torch monkey patches.

Data

_TORCH_PATCHES_APPLIED

_logger

API

nemo_automodel.shared.torch_patches.apply_torch_patches() -> None

Apply small, version/packaging-specific torch monkey patches.

This function is idempotent and safe to call multiple times.

nemo_automodel.shared.torch_patches._TORCH_PATCHES_APPLIED = False

nemo_automodel.shared.torch_patches._logger = logging.getLogger(__name__)