瀏覽代碼

Fix the API example with streaming #417

oobabooga 2 年之前
父節點
當前提交
c79fc69e95
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      api-example-stream.py

+ 2 - 2
api-example-stream.py

@@ -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'],