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

Check multiple prompts

Browse Source
This commit is contained in:
黑墨水鱼
2023-04-18 18:58:22 +08:00
parent 6fc4e6f230
commit 67fac2a4ce
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
whisper/transcribe.py
View File

@@ -363,7 +363,8 @@ def transcribe(
for i, segment in enumerate(current_segments, start=len(all_segments)): for i, segment in enumerate(current_segments, start=len(all_segments)):
if not segment['text'].strip(): if not segment['text'].strip():
continue continue
if len(all_prompts_segments) > 0 and all_prompts_segments[-1]['text'].strip() == segment['text'].strip(): check_prompt_num = 1
if any([prev['text'].strip() == segment['text'].strip() for prev in all_prompts_segments[-check_prompt_num:]]):
continue continue
all_prompts_tokens.extend(segment['tokens']) all_prompts_tokens.extend(segment['tokens'])
all_prompts_segments.append({"id": i, **segment}) all_prompts_segments.append({"id": i, **segment})