src/transformers/models/whisper/modeling_whisper.py/WhisperPositionalEmbedding/forward
class WhisperPositionalEmbedding:    def forward(self, input_ids, past_key_values_length=0):
        return self.weight[past_key_values_length : past_key_values_length + input_ids.shape[-1]]