Преглед изворни кода

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