> For clean Markdown of any page, append .md to the page URL.
> For a complete documentation index, see https://docs.nvidia.com/nemo/automodel/llms.txt.
> For AI client integration (Claude Code, Cursor, etc.), connect to the MCP server at https://docs.nvidia.com/nemo/automodel/_mcp/server.

# Baichuan2-7B-Chat

> Use NeMo AutoModel to fine-tune Baichuan2-7B-Chat for LLM workflows with checked-in text generation recipes and documented distributed settings.

[Baichuan2-7B-Chat](https://huggingface.co/baichuan-inc/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](/get-started/installation), then run the recipe from the repository root:

```bash
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](https://github.com/NVIDIA-NeMo/Automodel/blob/main/examples/llm_finetune/baichuan/baichuan_2_7b_squad.yaml). |
| Try another checked-in recipe for this model | Use the [alternate recipe](https://github.com/NVIDIA-NeMo/Automodel/blob/main/examples/llm_finetune/baichuan/baichuan_2_7b_mock_fp8.yaml).  |

### 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`](https://github.com/NVIDIA-NeMo/Automodel/blob/main/examples/llm_finetune/baichuan/baichuan_2_7b_mock_fp8.yaml)
* [`baichuan_2_7b_squad_peft.yaml`](https://github.com/NVIDIA-NeMo/Automodel/blob/main/examples/llm_finetune/baichuan/baichuan_2_7b_squad_peft.yaml)

## Model Reference

### Model Architecture

| Property     | Value                 |
| ------------ | --------------------- |
| Task         | Text generation       |
| Architecture | `BaichuanForCausalLM` |

### Available Models

* [`baichuan-inc/Baichuan2-7B-Chat`](https://huggingface.co/baichuan-inc/Baichuan2-7B-Chat)

## Related Resources

* [LLM fine-tuning guide](/recipes-e2e-examples/sft-peft)