Baichuan2-7B-Chat | NVIDIA NeMo AutoModel

Baichuan2-7B-Chat has a checked-in NeMo AutoModel recipe for text generation. The Hugging Face configuration declares the BaichuanForCausalLM architecture.

Fine-Tune Baichuan2-7B-Chat

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

automodel examples/llm_finetune/baichuan/baichuan_2_7b_squad.yaml --nproc-per-node 8

Choose a Workflow

Goal	Start Here
Run the primary recipe for this model	Use the recipe configuration.
Try another checked-in recipe for this model	Use the alternate recipe.

Configuration

Setting	Configuration
Hardware	-
Strategy	FSDP2; tp_size=1, cp_size=1
Nodes	-
Features	Checkpointing (enabled=true)
Advanced	-

More Recipes for This Model

baichuan_2_7b_mock_fp8.yaml
baichuan_2_7b_squad_peft.yaml

Model Reference

Model Architecture

Property	Value
Task	Text generation
Architecture	`BaichuanForCausalLM`

Available Models

baichuan-inc/Baichuan2-7B-Chat

LLM fine-tuning guide