Colab updated its env again 😄

If you're having torch version errors on colab

replace:

!python -m pip -q install xformers==0.0.21

with

!python -m pip -q install https://download.pytorch.org/whl/cu118/xformers-0.0.22.post4%2Bcu118-cp310-cp310-manylinux2014_x86_64.whl

Hard-locking torch version (that will downgrade the existing torch) is a lot slower, but maybe that is still a better option.

thanks to @huemin.art