Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 16:17:57 +01:00
Code Issues Packages Projects Releases 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
Show Stats 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'],