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

Fix the API example with streaming #417

Browse source
This commit is contained in:
oobabooga 2023-03-19 10:36:57 -03:00
parent c753261338
commit c79fc69e95
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
api-example-stream.py
View file

@ -44,14 +44,14 @@ async def run(context):
case "send_hash":
await websocket.send(json.dumps({
"session_hash": session,
"fn_index": 7
"fn_index": 9
}))
case "estimation":
pass
case "send_data":
await websocket.send(json.dumps({
"session_hash": session,
"fn_index": 7,
"fn_index": 9,
"data": [
context,
params['max_new_tokens'],