Merge branch 'main' of https://github.com/unslothai/unsloth
This commit is contained in:
commit
074a07981e
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