Skip to content

Commit 430071c

Browse files
committed
[dpo] fix import (#5364)
1 parent 9a35659 commit 430071c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

swift/trainers/rlhf_trainer/dpo_trainer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
# Copyright (c) Alibaba, Inc. and its affiliates.
2+
import warnings
23
from typing import Dict, List, Optional, Tuple, Union
34

45
import torch

0 commit comments

Comments
 (0)