Update requirements.txt to specify unsloth version 2025.4.5 and refactor imports in train.py for improved organization and clarity.

requirements.txt

@@ -30,7 +30,7 @@ smolagents[litellm,telemetry]>=1.14.0
 tensorboardX>=2.6.2.2
 trl>=0.17.0
 typing-extensions>=4.5.0
-unsloth>=2025.4.3
+unsloth>=2025.4.5
 wandb>=0.19.10
 wikipedia>=1.4.0
 wikipedia-api>=0.8.1

train.py

@@ -19,6 +19,12 @@ from datetime import datetime
 from pathlib import Path
 from typing import Union
 
+# isort: off
+from unsloth import FastLanguageModel, is_bfloat16_supported  # noqa: E402
+from unsloth.chat_templates import get_chat_template  # noqa: E402
+
+# isort: on
+
 from datasets import (
     Dataset,
     DatasetDict,
@@ -28,8 +34,6 @@ from datasets import (
 )
 from transformers import AutoTokenizer, Trainer, TrainingArguments
 from trl import SFTTrainer
-from unsloth import FastLanguageModel, is_bfloat16_supported
-from unsloth.chat_templates import get_chat_template
 
 # Configuration
 max_seq_length = 2048  # Auto supports RoPE Scaling internally
@@ -193,7 +197,6 @@ def create_trainer(
             tokenizer=tokenizer,
             train_dataset=dataset["train"],
             eval_dataset=dataset["validation"],
-            max_seq_length=max_seq_length,
             dataset_num_proc=2,
             packing=False,
             args=TrainingArguments(