fix yi vl vllm infer

Former-commit-id: de54e5d7ec06dd7c20ec82c9ff032fc16cd50244
2026-03-23 02:33:24 +08:00 · 2024-05-15 19:25:48 +08:00
parent 096677b989
commit 40e3d3fbdd
16 changed files with 113 additions and 24 deletions
--- a/src/llmtuner/train/orpo/workflow.py
+++ b/src/llmtuner/train/orpo/workflow.py
@@ -43,9 +43,9 @@ def run_orpo(
        model=model,
        args=training_args,
        finetuning_args=finetuning_args,
-        tokenizer=tokenizer,
        data_collator=data_collator,
        callbacks=callbacks,
+        **tokenizer_module,
        **split_dataset(dataset, data_args, training_args),
    )