ros/LlamaFactory
1
0
Fork 0
mirror of https://github.com/hiyouga/LlamaFactory.git synced 2026-02-01 08:13:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix stop param

Browse Source 
Former-commit-id: f0a850c25211b72eddbb357c81679db9b0930d44
This commit is contained in:
hiyouga
2024-05-07 00:41:04 +08:00
parent b9e167e6ca
commit e3b3a722de
3 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/llmtuner/chat/hf_engine.py
View File

@@ -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(