I'm installing flash-attention on colab. The installation goes smoothly on torch2.4.1. However, now the torch version of colab is upgraded to 2.5.0, and it stucked on "Building wheels for collected ...
Some results have been hidden because they may be inaccessible to you
Show inaccessible results