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

Compare commits

base: heimoshuiyu:master
Branches Tags
heimoshuiyu:prompt heimoshuiyu:master
...
compare: heimoshuiyu:9a646b69e69a9ea1668861ead8eccb6e9eb73f39
Branches Tags
heimoshuiyu:prompt heimoshuiyu:master

2 Commits
master ... 9a646b69e6

Author SHA1 Message Date
heimoshuiyu
9a646b69e6 format code 2023-04-20 02:00:57 +08:00
heimoshuiyu
49af9564ab Ignore repeated prompt 2023-04-20 01:49:10 +08:00
1 changed files with 10 additions and 1 deletions
Expand all files Collapse all files

9
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,15 @@ 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"],