def _move_past_key_values_to_gpu()

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_gpu(self, past_key_values, device="cuda:0"):
        # print(f"past_key_values -> gpu")
        if past_key_values is None:
            return None
        return tuple(tuple(t.to(device) for t in layer) for layer in past_key_values)