bugfix

modal-labs · Dec 13, 2023 · df62f5c · df62f5c
1 parent 802dc28
commit df62f5c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/06_gpu_and_ml/vllm_mixtral.py b/06_gpu_and_ml/vllm_mixtral.py
@@ -213,7 +213,7 @@ def app():
 
     @web_app.get("/stats")
     async def stats():
-        stats = await Model().generate_stream.get_current_stats.aio()
+        stats = await Model().completion_stream.get_current_stats.aio()
         return {
             "backlog": stats.backlog,
             "num_total_runners": stats.num_total_runners,