a d?@sJddlZddlmZddlmZd eeeedddZee dd d Z dS) N)Any)_get_device_indexF)deviceoptional allow_cpureturncCst|tr|St|tr"t|}t|tjrd|rL|jdvrdtd|n|jdkrdtd|tj st|tj jr|j St |||S)aGets the device index from :attr:`device`, which can be a torch.device object, a Python integer, or ``None``. If :attr:`device` is a torch.device object, returns the device index if it is a CUDA device. Note that for a CUDA device without a specified index, i.e., ``torch.device('cuda')``, this will return the current default CUDA device if :attr:`optional` is ``True``. If :attr:`allow_cpu` is ``True``, CPU devices will be accepted and ``-1`` will be returned in this case. If :attr:`device` is a Python integer, it is returned as is. If :attr:`device` is ``None``, this will return the current default CUDA device if :attr:`optional` is ``True``. )cudacpuz*Expected a cuda or cpu device, but got: {}rz#Expected a cuda device, but got: {}) isinstanceintstrtorchrtype ValueErrorformatZjitZ is_scriptingridx_torch_get_device_index)rrrrZ/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torch/cuda/_utils.pyrs       r)namercCs*tddd}t|tf|d|ddS)Nis_initcsfdd}|S)Ncs&r|jj}n|j}td|dS)Nz(Tried to instantiate dummy base class {}) __class____name__ RuntimeErrorr)objargskwargs class_namerrrerr_fn)s  z/_dummy_type..get_err_fn..err_fnr)rrrrr get_err_fn(s z_dummy_type..get_err_fnTF)__init____new__)boolrobject)rr rrr _dummy_type's r%)FF) r typingrZ torch._utilsrrr#r r rr%rrrrs