Test case fix and merging with main

Abhishek-TAMU · Abhishek-TAMU · commit 21de4a01f9e8 · 2025-04-14T14:19:20.000-04:00
Signed-off-by: Abhishek &lt;maurya.abhishek@ibm.com&gt;
diff --git a/tuning/data/setup_dataprocessor.py b/tuning/data/setup_dataprocessor.py
@@ -453,7 +453,12 @@ def process_dataargs(
 
     if data_args.data_config_path:
         train_dataset, eval_dataset, dataset_text_field = _process_dataconfig_file(
-            data_args, train_args, tokenizer, additional_data_handlers, processor, is_multipack
+            data_args,
+            train_args,
+            tokenizer,
+            additional_data_handlers,
+            processor,
+            is_multipack,
         )
     else:
         train_dataset, eval_dataset, dataset_text_field = _process_raw_data_args(
diff --git a/tuning/sft_trainer.py b/tuning/sft_trainer.py
@@ -33,7 +33,6 @@
     AutoTokenizer,
     TrainerCallback,
 )
-from transformers.trainer import _is_peft_model
 from transformers.trainer_utils import get_last_checkpoint
 from transformers.utils import is_accelerate_available
 from trl import SFTConfig, SFTTrainer

Original file line number	Diff line number	Diff line change
`@@ -33,7 +33,6 @@`
`33`	`33`	`AutoTokenizer,`
`34`	`34`	`TrainerCallback,`
`35`	`35`	`)`
`36`		`-from transformers.trainer import _is_peft_model`
`37`	`36`	`from transformers.trainer_utils import get_last_checkpoint`
`38`	`37`	`from transformers.utils import is_accelerate_available`
`39`	`38`	`from trl import SFTConfig, SFTTrainer`