o ei@s0ddlZddlmZd dededefddZdS) N)DeviceFdeviceoptionalreturncCst|tr|St|trt|}d}t|tjr:tj}|dur&td|j|jkr7t |jd|d|j }|durL|sGt d|tj S|S)NzAccelerator expectedz' doesn't match the current accelerator .zFExpected a torch.device with a specified index or an integer, but got:) isinstanceintstrtorchr acceleratorcurrent_accelerator RuntimeErrortype ValueErrorindexcurrent_device_index)rr device_indexaccrb/var/www/addictedbytheproject.nl/epg/venv/lib/python3.10/site-packages/torch/accelerator/_utils.py_get_device_indexs*       r)F)r torch.typesr _device_tboolrrrrrrs