If you're getting xformers error on colab
If you're getting xformers error on colab about cuda version mismatch:
Google colab env has been updated (their cuda version was bumped from 11.8 to 12.1), so we need to update our dependencies.
Google colab env has been updated (their cuda version was bumped from 11.8 to 12.1), so we need to update our dependencies.
I'm rolling a hotfix soon, but for now you can:
add a new code cell after the install cell and before the import dependencies cell, with
!pip install https://download.pytorch.org/whl/cu121/xformers-0.0.22.post4-cp310-cp310-manylinux2014_x86_64.whl#sha256=7075114dbf698b609b599f0d35032c0b2f9a389751e8bbf4dd3c628376b0dd9c
and run once. You may or may not need to restart the session (NOT reset) after that.
add a new code cell after the install cell and before the import dependencies cell, with
!pip install https://download.pytorch.org/whl/cu121/xformers-0.0.22.post4-cp310-cp310-manylinux2014_x86_64.whl#sha256=7075114dbf698b609b599f0d35032c0b2f9a389751e8bbf4dd3c628376b0dd9c
and run once. You may or may not need to restart the session (NOT reset) after that.