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": case "send_hash":
await websocket.send(json.dumps({ await websocket.send(json.dumps({
"session_hash": session, "session_hash": session,
"fn_index": 7 "fn_index": 9
})) }))
case "estimation": case "estimation":
pass pass
case "send_data": case "send_data":
await websocket.send(json.dumps({ await websocket.send(json.dumps({
"session_hash": session, "session_hash": session,
"fn_index": 7, "fn_index": 9,
"data": [ "data": [
context, context,
params['max_new_tokens'], params['max_new_tokens'],