From a4272ce3907e379f59ba46e050a204b5dd596ce9 Mon Sep 17 00:00:00 2001
From: Rishav Dash <57321948+Rishav-hub@users.noreply.github.com>
Date: Thu, 25 Jul 2024 14:06:57 +0530
Subject: [PATCH] Import missing ```setup_chat_format``` (#1862)

---
 docs/source/sft_trainer.mdx | 1 +
 1 file changed, 1 insertion(+)

diff --git a/docs/source/sft_trainer.mdx b/docs/source/sft_trainer.mdx
index 0bf5c12a218..1f616ad9f1e 100644
--- 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")