def get_current_device()

in src/alignment/model_utils.py [0:0]

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

def get_current_device() -> int:
    """Get the current device. For GPU we return the local process index to enable multiple GPU training."""
    return Accelerator().local_process_index if torch.cuda.is_available() else "cpu"