Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-11-24 06:26:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove log

Browse Source
This commit is contained in:
jax 2024-03-08 12:07:59 +08:00
parent 9cf2f995bd
commit 5ed89d0ca2
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
whisper/decoding.py
View File

@ -606,13 +606,11 @@ class DecodingTask:
else prompt
)
if (hotwords := self.options.hotwords) and not self.options.prefix:
print(f"hotwords: {hotwords}")
hotwords_tokens = self.tokenizer.encode(" " + hotwords.strip())
if len(hotwords_tokens) >= self.n_ctx // 2:
hotwords_tokens = hotwords_tokens[: self.n_ctx // 2 - 1]
tokens = (
[self.tokenizer.sot_prev]
+ prompt_tokens[-(self.n_ctx // 2 - 1) :]
+ (hotwords_tokens if self.options.hotwords is not None else [])
+ (prompt_tokens[-(self.n_ctx // 2 - 1) :] if self.options.prompt is not None else [])
+ tokens