Refactor imports in train.py to improve organization and clarity, adding DataCollatorForLanguageModeling for enhanced data handling during training.

train.py

@@ -32,9 +32,13 @@ from datasets import (
     IterableDatasetDict,
     load_dataset,
 )
-from transformers import AutoTokenizer, Trainer, TrainingArguments
+from transformers import (
+    AutoTokenizer,
+    DataCollatorForLanguageModeling,
+    Trainer,
+    TrainingArguments,
+)
 from trl import SFTTrainer
-from trl.data.data_collator import DataCollatorForLanguageModeling
 
 # Configuration
 max_seq_length = 2048  # Auto supports RoPE Scaling internally