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

Ignore repeated prompt

Browse Source
This commit is contained in:
黑墨水鱼
2023-04-20 01:49:10 +08:00
parent 3adcc12d0f
commit 49af9564ab
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
faster_whisper/transcribe.py
View File

@@ -330,6 +330,7 @@ class WhisperModel:
content_frames = features.shape[-1] - self.feature_extractor.nb_max_frames content_frames = features.shape[-1] - self.feature_extractor.nb_max_frames
seek = 0 seek = 0
all_tokens = [] all_tokens = []
all_prompt_text = []
prompt_reset_since = 0 prompt_reset_since = 0
if options.initial_prompt is not None: if options.initial_prompt is not None:
@@ -501,7 +502,10 @@ class WhisperModel:
if segment["start"] == segment["end"] or not text.strip(): if segment["start"] == segment["end"] or not text.strip():
continue continue
check_prompt_num = 1
if all([text.strip() != i.strip() for i in all_prompt_text[-check_prompt_num:]]):
all_tokens.extend(tokens) all_tokens.extend(tokens)
all_prompt_text.append(text)
yield Segment( yield Segment(
start=segment["start"], start=segment["start"],