vllm_omni.diffusion.distributed.utils ¶ get_local_device ¶ get_local_device() -> device Return the torch device for the current rank based on detected device type.