a dt@s,ddlZddlZddlZdedddZdS)NF)upgradecCsd}dtjvr$dttjdv}|r|rBgd}tj|ddtjd}|d| d }d |d }d d |g}tj|ddnt ddS)a Helper function to install appropriate xla wheels based on the `torch` version in Google Colaboratory. Args: upgrade (`bool`, *optional*, defaults to `False`): Whether to upgrade `torch` and install the latest `torch_xla` wheels. Example: ```python >>> from accelerate.utils import install_xla >>> install_xla(upgrade=True) ``` FZIPythonz google.colab)pipinstallz-UtorchT)checkrN.zBhttps://storage.googleapis.com/tpu-pytorch/wheels/colab/torch_xla-z-cp37-cp37m-linux_x86_64.whlrrz1`install_xla` utility works only on google colab.) sysmodulesstrZ get_ipython subprocessrun importlibmetadataversionrindex RuntimeError)rZin_colabZtorch_install_cmdZ torch_versionZtorch_version_truncZ xla_wheelZxla_install_cmdrc/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/accelerate/utils/torch_xla.py install_xlas    r)F)importlib.metadatar r rboolrrrrrs