o áÙ·i¯ã@s0ddlZddlZddlmZdejfdd„ZdS)éN)Úcurrent_omni_platformÚreturncCsttj dd¡ƒ}t |¡S)zKReturn the torch device for the current rank based on detected device type.Ú LOCAL_RANKr)ÚintÚosÚenvironÚgetrÚget_torch_device)Ú local_rank©r úY/home/ubuntu/.local/lib/python3.10/site-packages/vllm_omni/diffusion/distributed/utils.pyÚget_local_device s r )rÚtorchÚvllm_omni.platformsrÚdevicer r r r r Ús