Lj.ddlZddlmZddededefdZy)N)Devicedeviceoptionalreturnct|tr|St|trtj|}d}t|tjrktj j }| td|j|jk7rt|jd|d|j}|.|std|tj jS|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 W/var/www/html/ia-enhanced/venv/lib/python3.12/site-packages/torch/accelerator/_utils.py_get_device_indexrs&# &#f%#L&%,,'335 ;56 6 88v{{ ";;-Fse1M || XY_X`a   5577 )F)r torch.typesr _device_tboolr rrrrs% +i4Cr