Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-03-30 14:28:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge ee36decb1a86f4b2d76967098ca589d5332dadf7 into 517a43ecd132a2089d85f4ebc044728a71d49f6e

Browse Source
This commit is contained in:
Oliver Cai 2025-01-12 07:11:32 +01:00 committed by GitHub
commit d104c8eccd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
whisper/decoding.py
View File

@ -169,7 +169,7 @@ class PyTorchInference(Inference):
self.kv_cache = {}
self.hooks = []
def rearrange_kv_cache(self, source_indices):
def rearrange_kv_cache(self, source_indices : List[int]):
if source_indices != list(range(len(source_indices))):
for module in self.kv_modules:
# update the key/value cache to contain the selected sequences