def _reorder_cache()

in deepseek_vl2/models/modeling_deepseek_vl_v2.py [0:0]

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

    def _reorder_cache(past_key_values, beam_idx):
        reordered_past = ()
        for layer_past in past_key_values:
            reordered_past += (
                tuple(
                    past_state.index_select(0, beam_idx.to(past_state.device))
                    for past_state in layer_past
                ),
            )
        return reordered_past