nemo_automodel.components.models.deprecation

Deprecation warnings for custom model classes scheduled for removal.

Module Contents

Functions

Name	Description
`warn_deprecated_model_class`	Emit a deprecation warning for custom model classes removed in 26.10.

Data

_DEPRECATED_MODEL_YAMLS

API

nemo_automodel.components.models.deprecation.warn_deprecated_model_class(
    model_cls_name: str
) -> None

Emit a deprecation warning for custom model classes removed in 26.10.

Parameters:

model_cls_name

str

Name of the model class being instantiated.

nemo_automodel.components.models.deprecation._DEPRECATED_MODEL_YAMLS: dict[str, tuple[str, ...]] = {'BaichuanForCausalLM': ('examples/llm_finetune/baichuan/baichuan_2_7b_mock_fp8....