support DPO training (2305.18290)

This commit is contained in:
hiyouga committed 2023-08-11 03:02:53 +08:00
1 parent 685dae4eff
commit 3ec4351cfd
34 files changed
+513 -1027300

No files matched your search

+1 -1
View File
@@ -4,7 +4,7 @@ import threading
import time
import transformers
from transformers.trainer import TRAINING_ARGS_NAME
from typing import Generator, List, Optional, Tuple
from typing import Generator, List, Tuple
from llmtuner.extras.callbacks import LogCallback
from llmtuner.extras.constants import DEFAULT_MODULE