o i$@s0ddlZgdZgdZd ddZd ddZdS) N)autocudacpumps)rlmdeploy transformersrcCs\|dkrtjr d}n tjjrd}nd}|tvs%Jd|dttd||S)Nrrrrzunrecognized device z, device must be in z Using device )torchr is_availablebackendsrRECOGNIZED_DEVICESprint)devicerM/home/ubuntu/.local/lib/python3.10/site-packages/soprano/utils/auto_select.py select_devices  rcCsl|dkr|dkr d}nzddl}d}Wn tyd}Ynw|tvs-Jd|dttd||S) Nrrrrrzunrecognized backend z, backend must be in zUsing backend )r ImportErrorRECOGNIZED_BACKENDSr )backendr rrrrselect_backends r)r)rr)rr rrrrrrrs