DeepSeek-V4.1-Flash | NVIDIA NeMo AutoModel

DeepSeek-V4.1-Flash has a checked-in NeMo AutoModel recipe for image-text-to-text. The Hugging Face configuration declares the DeepseekV41ForCausalLM architecture.

Fine-Tune DeepSeek-V4.1-Flash

Follow the installation instructions, then review the checked-in recipe and its declared topology below before choosing a local or cluster launcher.

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

Choose a Workflow

Goal	Start Here
Run the primary recipe for this model	Use the recipe configuration.
Prepare your environment	Follow the installation instructions.

Configuration

Setting	Configuration
Hardware	-
Strategy	FSDP2; tp_size=1, pp_size=1, cp_size=1, ep_size=64
Nodes	16
Features	Activation checkpointing (activation_checkpointing=true), HybridEP (dispatcher=hybridep)
Advanced	bfloat16, attention=tilelang, linear=torch, experts=torch_mm, dispatcher=hybridep

Model Reference

Model Architecture

Property	Value
Task	Image-text-to-text
Architecture	`DeepseekV41ForCausalLM`

Available Models

deepseek-ai/DeepSeek-V4.1-Flash

Vision-language model fine-tuning guide