Update dpo.py
This commit is contained in:
parent
49bdb690e9
commit
9687e7fdab
1 changed files with 2 additions and 4 deletions
|
|
@ -17,8 +17,6 @@ __all__ = [
|
|||
"PatchKTOTrainer",
|
||||
]
|
||||
|
||||
from .rl import PatchFastRL
|
||||
def PatchDPOTrainer(): return
|
||||
|
||||
def PatchDPOTrainer(): PatchFastRL("DPO")
|
||||
|
||||
def PatchKTOTrainer(): PatchFastRL("KTO")
|
||||
def PatchKTOTrainer(): return
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue