[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 <danielhanchen@gmail.com>
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
Datta Nimmaturi 2026-03-19 14:48:46 +05:30 committed by GitHub
commit 729a0cb0ae
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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