Import missing setup_chat_format (huggingface#1862)

claralp · Jul 28, 2024 · a4272ce · a4272ce
1 parent bf21d08
commit a4272ce
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docs/source/sft_trainer.mdx b/docs/source/sft_trainer.mdx
@@ -172,6 +172,7 @@ The [`setup_chat_format`] function in `trl` easily sets up a model and tokenizer
 
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from trl import setup_chat_format
 
 # Load model and tokenizer
 model = AutoModelForCausalLM.from_pretrained("facebook/opt-350m")