[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
5e4bc50f1d
commit
fbeca18046
1 changed files with 5 additions and 1 deletions
|
|
@ -1306,7 +1306,11 @@ def patch_trl_rl_trainers():
|
|||
import trl.trainer
|
||||
|
||||
all_trainers = dir(trl.trainer)
|
||||
all_trainers = [x for x in all_trainers if x.islower() and x.endswith("_trainer") and x != "base_trainer"]
|
||||
all_trainers = [
|
||||
x
|
||||
for x in all_trainers
|
||||
if x.islower() and x.endswith("_trainer") and x != "base_trainer"
|
||||
]
|
||||
for trainer in all_trainers:
|
||||
_patch_trl_rl_trainers(trainer)
|
||||
return
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue