chore(doc): update sft training tutorial to install training extra

huggingface · Jan 29, 2025 · 2280fd3 · 2280fd3
1 parent 6169aa3
commit 2280fd3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/docs/source/training_tutorials/sft_lora_finetune_llm.mdx b/docs/source/training_tutorials/sft_lora_finetune_llm.mdx
@@ -56,6 +56,10 @@ huggingface-cli login --token YOUR_TOKEN
 ```bash
 git clone https://github.com/huggingface/optimum-neuron.git
 ```
+5. Make sure you have the `training` extra installed, to get all the necessary dependencies:
+```bash
+python -m pip install .[training]
+```
 
 ## 2. Load and prepare the dataset