mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add max_tokens_second param (#3533)

Browse source
This commit is contained in:
oobabooga 2023-08-29 17:44:31 -03:00 • committed by GitHub
parent fe1f7c6513
commit cec8db52e5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
11 changed files with 24 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

1
extensions/openai/defaults.py
View file

@ -5,6 +5,7 @@ import copy
default_req_params = {
'max_new_tokens': 16, # 'Inf' for chat
'auto_max_new_tokens': False,
'max_tokens_second': 0,
'temperature': 1.0,
'top_p': 1.0,
'top_k': 1, # choose 20 for chat in absence of another default