Removed Extraneous Print Message From OAI Server (vllm-project#3440)

Temirulan · Mar 16, 2024 · a60b105 · a60b105
1 parent 9b0a1af
commit a60b105
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/vllm/entrypoints/openai/serving_completion.py b/vllm/entrypoints/openai/serving_completion.py
@@ -309,10 +309,7 @@ async def completion_stream_generator(
         except ValueError as e:
             # TODO: Use a vllm-specific Validation Error
             data = self.create_streaming_error_response(str(e))
-            print("yield", f"data: {data}\n\n")
             yield f"data: {data}\n\n"
-
-        print("yield", "data: [DONE]\n\n")
         yield "data: [DONE]\n\n"
 
     def request_output_to_completion_response(