Преглед на файлове

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