9i0ddlZddlmZddededefdZdS) N)DeviceFdeviceoptionalreturnct|tr|St|trtj|}d}t|tjr`tj}|td|j|jkrt|jd|d|j }|2|std|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_indexaccs K/var/www/icac/venv/lib/python3.11/site-packages/torch/accelerator/_utils.py_get_device_indexrs&# &#&f%%#L&%,''$3355 ;566 6 8v{ " ";MMsMMM |  aY_aa  55777 )F)r torch.typesr _device_tboolr rrrrsX ++++++i4Cr