Skip to content

Commit c2db596

Browse files
Fix import ordering in ORPO trainer
1 parent 92e218b commit c2db596

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

trl/experimental/orpo/orpo_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,6 @@
4848

4949
from ...data_utils import maybe_apply_chat_template, maybe_extract_prompt
5050
from ...trainer.base_trainer import BaseTrainer
51-
from .orpo_config import ORPOConfig
5251
from ...trainer.utils import (
5352
DPODataCollatorWithPadding,
5453
add_bos_token_if_needed,
@@ -59,6 +58,7 @@
5958
peft_module_casting_to_bf16,
6059
selective_log_softmax,
6160
)
61+
from .orpo_config import ORPOConfig
6262

6363

6464
if is_peft_available():

0 commit comments

Comments
 (0)