fix imports

Signed-off-by: Sukriti-Sharma4 <[email protected]>

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 (