mypy

Signed-off-by: NickLucche <[email protected]>
opendatahub-io · Feb 14, 2025 · bbbb8cc · bbbb8cc
1 parent c85c8b2
commit bbbb8cc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/vllm/entrypoints/whisper_server/api_server.py b/vllm/entrypoints/whisper_server/api_server.py
@@ -112,7 +112,7 @@ async def _generate(prompt, raw_request: Request) -> Response:
     engine = raw_request.app.state.engine
     results_generator = engine.generate(prompt, sampling_params, request_id)
 
-    async def stream_results() -> AsyncGenerator[bytes, None]:
+    async def stream_results() -> AsyncGenerator[str, None]:
         async for request_output in results_generator:
             assert len(request_output.outputs) == 1
             chunk = TranscriptionResponse(text=request_output.outputs[0].text)

diff --git a/vllm/entrypoints/whisper_server/helper.py b/vllm/entrypoints/whisper_server/helper.py
@@ -28,7 +28,7 @@ def exact_div(x, y):
                               N_SAMPLES_PER_TOKEN)  # 20ms per audio token
 
 
-def load_audio_from_bytes(audio_bytes: str, sample_rate: int = SAMPLE_RATE):
+def load_audio_from_bytes(audio_bytes: bytes, sample_rate: int = SAMPLE_RATE):
     """
     Read bytes from audio file as mono waveform, resampling as necessary