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

typing addition

Browse Source
This commit is contained in:
OliverCai0 2023-11-14 16:00:45 -06:00
parent 1cea435768
commit ee36decb1a
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