Update run_sft.sh (#1546)

huggingface · Apr 17, 2024 · edf60e8 · edf60e8
1 parent 5d1deb1
commit edf60e8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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
+exit 0