From 30c4cba4ec6f780f35f2561bc812b2682ca2eb4c Mon Sep 17 00:00:00 2001 From: "huangjintao.hjt" Date: Wed, 6 Nov 2024 13:40:05 +0800 Subject: [PATCH] fix media_type --- swift/llm/deploy.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/swift/llm/deploy.py b/swift/llm/deploy.py index 438c87a3a8..0baa80f6a7 100644 --- a/swift/llm/deploy.py +++ b/swift/llm/deploy.py @@ -426,7 +426,7 @@ async def _generate_stream(): yield 'data: [DONE]\n\n' if request.stream: - return StreamingResponse(_generate_stream()) + return StreamingResponse(_generate_stream(), media_type='text/event-stream') else: return await _generate_full() @@ -605,7 +605,7 @@ async def _generate_stream(): yield 'data: [DONE]\n\n' if request.stream: - return StreamingResponse(_generate_stream()) + return StreamingResponse(_generate_stream(), media_type='text/event-stream') else: return await _generate_full() @@ -833,7 +833,7 @@ def _generate_stream(): yield 'data: [DONE]\n\n' if request.stream: - return StreamingResponse(_generate_stream()) + return StreamingResponse(_generate_stream(), media_type='text/event-stream') else: return await _generate_full()