fix tests
This commit is contained in:
1 parent
38af076a75
commit
f6d6e00337
12 files changed
+80
-65
No files matched your search
@@ -135,9 +135,9 @@ def merge_dataset(
|
||||
|
||||
|
||||
def get_dataset(
|
||||
tokenizer: "PreTrainedTokenizer",
|
||||
model_args: "ModelArguments",
|
||||
data_args: "DataArguments",
|
||||
tokenizer: "PreTrainedTokenizer",
|
||||
training_args: "Seq2SeqTrainingArguments",
|
||||
stage: Literal["pt", "sft", "rm", "ppo"],
|
||||
# split: Optional[str] = "train", # TODO: add split
|
||||
|
||||
Reference in new issue
Block a user