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

Update trainer.py

Browse source
This commit is contained in:
Daniel Han 2024-11-13 18:48:59 -08:00
commit 022d571835
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
unsloth/trainer.py
View file

@ -201,7 +201,10 @@ pass
def _patch_trl_trainer():
if hasattr(trl, "__UNSLOTH_BACKWARDS_COMPATIBLE__"): return
try:
trl.__UNSLOTH_BACKWARDS_COMPATIBLE__
return
except: pass
if Version(trl.__version__) <= Version("0.11.0"): return
import trl.trainer