AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04: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":
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'],