support DPO training (2305.18290)
This commit is contained in:
1 parent
685dae4eff
commit
3ec4351cfd
34 files changed
+513
-1027300
No files matched your search
@@ -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
|
||||
|
||||
Reference in new issue
Block a user