diff --git a/docs/source/sft_trainer.mdx b/docs/source/sft_trainer.mdx index 1f616ad9f1e..f6192b26f7c 100644 --- a/docs/source/sft_trainer.mdx +++ b/docs/source/sft_trainer.mdx @@ -110,7 +110,10 @@ collator = DataCollatorForCompletionOnlyLM(instruction_template=instruction_temp trainer = SFTTrainer( model, - args=SFTConfig(output_dir="/tmp"), + args=SFTConfig( + output_dir="/tmp", + dataset_text_field = "text", + ), train_dataset=dataset, data_collator=collator, )