diff --git a/studio/backend/core/training/trainer.py b/studio/backend/core/training/trainer.py index bc252d1fa5..ede213647d 100644 --- a/studio/backend/core/training/trainer.py +++ b/studio/backend/core/training/trainer.py @@ -772,6 +772,11 @@ class UnslothTrainer: if self.should_stop: return False + if full_finetuning: + # Enable training mode for full fine-tuning + # This ensures all model parameters are trainable; otherwise, they might be frozen. + self.model.for_training() + self._update_progress(status_message = "Model loaded successfully") logger.info("Model loaded successfully") return True