[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:
parent
6f129a214b
commit
729a0cb0ae
1 changed files with 5 additions and 0 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue