Mistral-Small-4-119B-2603

View as Markdown

Mistral-Small-4-119B-2603 has a checked-in NeMo AutoModel recipe for text generation. The Hugging Face configuration declares the Mistral3ForConditionalGeneration architecture.

Fine-Tune Mistral-Small-4-119B-2603

Follow the installation instructions, then run the recipe from the repository root:

automodel examples/llm_benchmark/mistral/mistral_small_4_te_deepep.yaml --nproc-per-node 8

Use the Slurm launcher guide for the multi-node run.

Choose a Workflow

GoalStart Here
Run the primary recipe for this modelUse the recipe configuration.
Prepare your environmentFollow the installation instructions.

Configuration

SettingConfiguration
Hardware-
StrategyFSDP2; tp_size=1, pp_size=1, cp_size=1, ep_size=64
Nodes8
FeaturesActivation checkpointing (activation_checkpointing=true), Transformer Engine, HybridEP (dispatcher=hybridep)
Advancedattention=te, linear=te, experts=torch_mm, dispatcher=hybridep

Model Reference

Model Architecture

PropertyValue
TaskText generation
ArchitectureMistral3ForConditionalGeneration

Available Models