From 9687e7fdab36fe95b0b7b2bcea21fa7587cade3f Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 13 Feb 2025 14:59:42 -0800 Subject: [PATCH] Update dpo.py --- unsloth/models/dpo.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/unsloth/models/dpo.py b/unsloth/models/dpo.py index 9c12abb98f..7fc02fcf7f 100644 --- a/unsloth/models/dpo.py +++ b/unsloth/models/dpo.py @@ -17,8 +17,6 @@ __all__ = [ "PatchKTOTrainer", ] -from .rl import PatchFastRL +def PatchDPOTrainer(): return -def PatchDPOTrainer(): PatchFastRL("DPO") - -def PatchKTOTrainer(): PatchFastRL("KTO") +def PatchKTOTrainer(): return