Update dpo.py

This commit is contained in:
Daniel Han 2025-02-13 14:59:42 -08:00
commit 9687e7fdab

View file

@ -17,8 +17,6 @@ __all__ = [
"PatchKTOTrainer",
]
from .rl import PatchFastRL
def PatchDPOTrainer(): return
def PatchDPOTrainer(): PatchFastRL("DPO")
def PatchKTOTrainer(): PatchFastRL("KTO")
def PatchKTOTrainer(): return