ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update dpo.py

Browse source
This commit is contained in:
Daniel Han-Chen 2024-01-28 17:20:49 +11:00
commit 893aab0e57
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

13
unsloth/models/dpo.py
View file

@ -101,10 +101,13 @@ pass
def PatchDPOTrainer(): def PatchDPOTrainer():
# Patch DPO notebook printing from transformers.trainer import is_in_notebook
NotebookTrainingTracker.write_line = NotebookTrainingTracker_write_line if is_in_notebook():
from transformers.trainer import DEFAULT_PROGRESS_CALLBACK # Patch DPO notebook printing
DEFAULT_PROGRESS_CALLBACK.on_train_begin = NotebookProgressCallback_on_train_begin NotebookTrainingTracker.write_line = NotebookTrainingTracker_write_line
DEFAULT_PROGRESS_CALLBACK.on_log = NotebookProgressCallback_on_log from transformers.trainer import DEFAULT_PROGRESS_CALLBACK
DEFAULT_PROGRESS_CALLBACK.on_train_begin = NotebookProgressCallback_on_train_begin
DEFAULT_PROGRESS_CALLBACK.on_log = NotebookProgressCallback_on_log
pass
pass pass