fix stop param

2026-01-11 16:40:34 +08:00 · 2024-05-07 00:41:04 +08:00
parent bcf7ec5ceb
commit 09f3ef1de4
4 changed files with 11 additions and 4 deletions
--- a/src/llmtuner/chat/hf_engine.py
+++ b/src/llmtuner/chat/hf_engine.py
@@ -73,6 +73,10 @@ class HuggingfaceEngine(BaseEngine):
        repetition_penalty = input_kwargs.pop("repetition_penalty", None)
        max_length = input_kwargs.pop("max_length", None)
        max_new_tokens = input_kwargs.pop("max_new_tokens", None)
+        stop = input_kwargs.pop("stop", None)
+
+        if stop is not None:
+            raise ValueError("Stop parameter is not supported in Huggingface engine yet.")

        generating_args.update(
            dict(