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

Add repetition_penalty parameter to stream_builder function

Browse Source
This commit is contained in:
黑墨水鱼
2024-07-11 21:45:38 +08:00
parent 4ed1c695fe
commit 931b578899
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

@@ -120,6 +120,7 @@ def stream_builder(
vad_filter: bool, vad_filter: bool,
language: str | None, language: str | None,
initial_prompt: str = "", initial_prompt: str = "",
repetition_penalty: float = 1.0,
) -> Tuple[Iterable[Segment], TranscriptionInfo]: ) -> Tuple[Iterable[Segment], TranscriptionInfo]:
segments, info = transcriber.model.transcribe( segments, info = transcriber.model.transcribe(
audio=audio, audio=audio,
@@ -127,6 +128,7 @@ def stream_builder(
task=task, task=task,
initial_prompt=initial_prompt, initial_prompt=initial_prompt,
word_timestamps=True, word_timestamps=True,
repetition_penalty=repetition_penalty,
) )
print( print(
"Detected language '%s' with probability %f" "Detected language '%s' with probability %f"