heimoshuiyu/whisper-fastapi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add repetition_penalty parameter to transcription endpoint

Browse Source
This commit is contained in:
黑墨水鱼
2024-07-11 21:48:46 +08:00
parent 931b578899
commit e8ae8bf9c5
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
whisper_fastapi.py
View File

@@ -297,6 +297,7 @@ async def transcription(
task: str = Form("transcribe"), task: str = Form("transcribe"),
language: str = Form("und"), language: str = Form("und"),
vad_filter: bool = Form(False), vad_filter: bool = Form(False),
repetition_penalty: float = Form(1.0),
): ):
"""Transcription endpoint """Transcription endpoint
@@ -309,6 +310,7 @@ async def transcription(
task=task, task=task,
vad_filter=vad_filter, vad_filter=vad_filter,
language=None if language == "und" else language, language=None if language == "und" else language,
repetition_penalty=repetition_penalty,
) )
# special function for streaming response (OpenAI API does not have this) # special function for streaming response (OpenAI API does not have this)