fix tests

Former-commit-id: f6d6e00337
2025-12-15 03:10:35 +08:00 · 2024-01-20 19:58:04 +08:00
parent 69e8925249
commit 2f7684a8ee
12 changed files with 80 additions and 65 deletions
--- a/src/llmtuner/data/loader.py
+++ b/src/llmtuner/data/loader.py
@@ -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