bit*ddlZddlZddlZddefdZy)Nupgradecd}dtjvr,dttjdjv}|rs|rgd}t j |dt jjd}|d|jd }d |d }d d |g}t j |dytd)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) ``` FIPythonz 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.) sysmodulesstr get_ipython subprocessrun importlibmetadataversionrindex RuntimeError)rin_colabtorch_install_cmd torch_versiontorch_version_trunc xla_wheelxla_install_cmds U/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/accelerate/utils/torch_xla.py install_xlars HCKK!SY)?)K)K)M%NN  A  NN,D 9!**227; +,Gm.B.B3.GHXYlXmnJK  )Y7d3NOO)F)importlib.metadatarrr boolrrrr"s PPr