Compare commits

...

6 Commits

View File

@@ -454,6 +454,7 @@ class WhisperModel:
clip_idx = 0
seek = seek_clips[clip_idx][0]
all_tokens = []
all_prompt_text = []
prompt_reset_since = 0
if options.initial_prompt is not None:
@@ -724,8 +725,16 @@ class WhisperModel:
if segment["start"] == segment["end"] or not text.strip():
continue
all_tokens.extend(tokens)
idx += 1
check_prompt_num = 1
if all(
[
text.strip() != i.strip()
for i in all_prompt_text[-check_prompt_num:]
]
):
all_tokens.extend(tokens)
all_prompt_text.append(text)
idx += 1
yield Segment(
id=idx,