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:53:40 -08:00
commit 7333bcb7ea
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
unsloth/trainer.py
View file

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