rh:ddlmZddlZddlmZddededefdZy) )OptionalN)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 k/var/www/html/ai-insurance-compliance-backend/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) typingrr torch.typesr _device_tboolr rrrrs( +i4Cr