From edf60e826b0f4f0b50a72668c74c0a98f05dbe45 Mon Sep 17 00:00:00 2001
From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>
Date: Wed, 17 Apr 2024 16:17:05 +0200
Subject: [PATCH] Update run_sft.sh (#1546)

---
 commands/run_sft.sh | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/commands/run_sft.sh b/commands/run_sft.sh
index d839821233..1dfb83da71 100644
--- a/commands/run_sft.sh
+++ b/commands/run_sft.sh
@@ -41,6 +41,7 @@ accelerate launch $EXTRA_ACCELERATE_ARGS \
     --dataset_name $DATASET_NAME \
     --output_dir $OUTPUT_DIR \
     --max_steps $MAX_STEPS \
+    --dataset_text_field 'text' \
     --per_device_train_batch_size $BATCH_SIZE \
     --max_seq_length $SEQ_LEN \
     $EXTRA_TRAINING_ARGS
@@ -56,4 +57,4 @@ echo "Starting program..."
     echo "Operation Failed!"
     exit 1
 }
-exit 0
\ No newline at end of file
+exit 0