def rearrange_kv_cache()

in whisper/decoding.py [0:0]

• 4 lines of code
• 3 McCabe index (conditional complexity)

    def rearrange_kv_cache(self, source_indices):
        if source_indices != list(range(len(source_indices))):
            for module in self.kv_modules:
                # update the key/value cache to contain the selected sequences
                self.kv_cache[module] = self.kv_cache[module][source_indices].detach()