nemo_automodel.components.models.mistral3_vlm

View as Markdown

FP8-native Mistral3 VLM custom model registration (dawn-ridge 128B).

Importing this module installs a resolver hook on _resolve_custom_model_cls_for_config (nemo_automodel/_transformers/ model_init.py) that routes FP8-native Mistral3 VLM configs to Mistral3FP8VLMForConditionalGeneration. The model registry is not overwritten, so non-FP8 Mistral3 VLMs keep the stock mistral4.model path and there is no regression for other users.

Submodules

Package Contents

Functions

NameDescription
_install_resolver_hookPrepend an FP8-Mistral3 VLM check to _resolve_custom_model_cls_for_config.

Data

__all__

logger

API

nemo_automodel.components.models.mistral3_vlm._install_resolver_hook() -> None

Prepend an FP8-Mistral3 VLM check to _resolve_custom_model_cls_for_config.

nemo_automodel.components.models.mistral3_vlm.__all__ = ['Mistral3FP8VLMForConditionalGeneration']
nemo_automodel.components.models.mistral3_vlm.logger = logging.getLogger(__name__)