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

Fix the API extension

Browse source
This commit is contained in:
oobabooga 2023-04-10 20:14:38 -03:00
parent 58b34c0841
commit 64f5c90ee7
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
extensions/api/script.py
View file

@ -57,6 +57,7 @@ class Handler(BaseHTTPRequestHandler):
'length_penalty': float(body.get('length_penalty', 1)), 'length_penalty': float(body.get('length_penalty', 1)),
'early_stopping': bool(body.get('early_stopping', False)), 'early_stopping': bool(body.get('early_stopping', False)),
'seed': int(body.get('seed', -1)), 'seed': int(body.get('seed', -1)),
'add_bos_token': int(body.get('add_bos_token', True)),
} }
generator = generate_reply( generator = generate_reply(