
    /ht                     .    S SK rS SKrS SKrSS\4S jjrg)    Nupgradec                    SnS[         R                  ;   a-  S[        [         R                  S   R                  5       5      ;   nU(       at  U (       a  / SQn[        R
                  " USS9  [        R                  R                  S5      nUSUR                  S	5       nS
U S3nSSU/n[        R
                  " USS9  g[        S5      e)aY  
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)
```
FIPythonzgoogle.colab)pipinstallz-UtorchT)checkr   N.zBhttps://storage.googleapis.com/tpu-pytorch/wheels/colab/torch_xla-z-cp37-cp37m-linux_x86_64.whlr   r   z1`install_xla` utility works only on google colab.)sysmodulesstrget_ipython
subprocessrun	importlibmetadataversionrindexRuntimeError)r   in_colabtorch_install_cmdtorch_versiontorch_version_trunc	xla_wheelxla_install_cmds          T/home/james-whalen/.local/lib/python3.13/site-packages/accelerate/utils/torch_xla.pyinstall_xlar      s      HCKK!SY)?)K)K)M%NN ANN,D9!**227;+,Gm.B.B3.GHXYlXm  nJ  K	 )Y7d3NOO    )F)importlib.metadatar   r   r   boolr    r   r   <module>r"      s      
P Pr   