bridge.models._deprecation

Module Contents

Functions

_model_identifier
_architectures
_deprecated_model_name
warn_deprecated_model	Warn that support for a legacy model is scheduled for removal.
warn_if_deprecated_model	Warn when an HF configuration belongs to a deprecated model family.
warn_if_legacy_nemotron_path	Warn before loading the legacy Nemotron-4 repository without an HF config.

Data

_REMOVAL_VERSION
_LEGACY_NEMOTRON_NAME
_DEPRECATED_ARCHITECTURES
_LLAMA2_SHAPES
_MISTRAL_SHAPES
_NEMOTRON_H_V1_SHAPES

API

bridge.models._deprecation._REMOVAL_VERSION

‘0.7.0’

bridge.models._deprecation._LEGACY_NEMOTRON_NAME

‘the legacy Nemotron bridge (NemotronForCausalLM, previously documented for Nemotron-4 340B)’

bridge.models._deprecation._DEPRECATED_ARCHITECTURES

None

bridge.models._deprecation._LLAMA2_SHAPES

None

bridge.models._deprecation._MISTRAL_SHAPES

None

bridge.models._deprecation._NEMOTRON_H_V1_SHAPES

None

bridge.models._deprecation._model_identifier(

config: Any,

model_name_or_path: str | pathlib.Path | None,

) → str

bridge.models._deprecation._architectures(config: Any) → set[str]

bridge.models._deprecation._deprecated_model_name(

config: Any,

model_name_or_path: str | pathlib.Path | None = None,

) → str | None

bridge.models._deprecation.warn_deprecated_model(model_name: str, *, stacklevel: int = 2) → None

Warn that support for a legacy model is scheduled for removal.

Parameters:

• model_name – User-facing model family or variant name.

• stacklevel – Warning stack level passed to :func:warnings.warn.

bridge.models._deprecation.warn_if_deprecated_model(

config: Any,

model_name_or_path: str | pathlib.Path | None = None,

) → None

Warn when an HF configuration belongs to a deprecated model family.

Parameters:

• config – Hugging Face model configuration.

• model_name_or_path – Optional Hugging Face model identifier or local path.

bridge.models._deprecation.warn_if_legacy_nemotron_path(

model_name_or_path: str | pathlib.Path,

) → None

Warn before loading the legacy Nemotron-4 repository without an HF config.

Parameters:

model_name_or_path – Hugging Face model identifier or local path.