def _move_past_key_values_to_cpu()

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

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

    def _move_past_key_values_to_cpu(self, past_key_values):
        # print(f"past_key_values -> cpu")
        if past_key_values is None:
            return None
        return tuple(tuple(t.cpu() for t in layer) for layer in past_key_values)