We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8bf0404 commit 8a6da33Copy full SHA for 8a6da33
unsloth/models/__init__.py
@@ -16,5 +16,5 @@
16
from .llama import FastLlamaModel
17
from .mistral import FastMistralModel
18
from .qwen2 import FastQwen2Model
19
-from .dpo import PatchDPOTrainer
+from .dpo import PatchDPOTrainer, PatchKTOTrainer
20
from ._utils import is_bfloat16_supported
unsloth/models/dpo.py
@@ -14,6 +14,7 @@
14
15
__all__ = [
"PatchDPOTrainer",
+ "PatchKTOTrainer",
]
try:
@@ -127,4 +128,4 @@ def PatchDPOTrainer():
127
128
pass
129
130
-
131
+PatchKTOTrainer = PatchDPOTrainer
0 commit comments