From 2f70e34b6d39a4d72a6e90aa27f801aa817df429 Mon Sep 17 00:00:00 2001 From: Sukriti-Sharma4 Date: Thu, 23 May 2024 21:41:53 -0600 Subject: [PATCH] fix imports Signed-off-by: Sukriti-Sharma4 --- tuning/sft_trainer.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tuning/sft_trainer.py b/tuning/sft_trainer.py index 3e7887df8..a9e1aebc1 100644 --- a/tuning/sft_trainer.py +++ b/tuning/sft_trainer.py @@ -29,13 +29,11 @@ TrainerCallback, ) from transformers.utils import logging +from trl import DataCollatorForCompletionOnlyLM, SFTTrainer import datasets import fire import transformers -# First Party -from trl import DataCollatorForCompletionOnlyLM, SFTTrainer - # Local from tuning.config import configs, peft_config from tuning.config.tracker_configs import (