From 729a0cb0ae1d9736f079a190d1a2f94f158f2caf Mon Sep 17 00:00:00 2001 From: Datta Nimmaturi Date: Thu, 19 Mar 2026 14:48:46 +0530 Subject: [PATCH] [studio] full finetuning studio (#4461) * full finetuning studio * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update studio/backend/core/training/trainer.py Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Daniel Han Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> --- studio/backend/core/training/trainer.py | 5 +++++ 1 file changed, 5 insertions(+) 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