Update app/services/training_service.py

app/services/training_service.py

@@ -26,7 +26,7 @@ from transformers import (
 )
 from datasets import load_dataset, Dataset
 from peft import LoraConfig, get_peft_model, TaskType, prepare_model_for_kbit_training
-
+from trl import SFTTrainer  # instead of from transformers import Trainer
 from app.config import settings
 from app.database import AsyncSessionLocal, TrainingJob, TrainingLog, JobStatus
 import importlib
@@ -476,7 +476,7 @@ class TrainingService:
             progress_callback = ProgressCallback(total_steps)
             
             # Trainer
-            trainer = Trainer(
+            trainer = SFTTrainer(
                 model=model,
                 args=training_arguments,
                 train_dataset=tokenized_train,