fix error with colab #173

seruva19 · Apr 1, 2024 · 5aa63da · 5aa63da
1 parent 210d6c1
commit 5aa63da
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/notebooks/kubin-colab.ipynb b/notebooks/kubin-colab.ipynb
@@ -177,7 +177,7 @@
     "!pip install -r requirements.txt\n",
     "\n",
     "if use_flash_attention:\n",
-    "  !pip install /content/kubin/wheels/flash_attn-1.0.9+cu121torch2.1.0-cp310-cp310-linux_x86_64.whl\n",
+    "  !pip install /content/kubin/wheels/flash_attn-1.0.9+cu122torch2.2.2-cp310-cp310-linux_x86_64.whl\n",
     "\n",
     "if use_xformers:\n",
     "  !pip install xformers==0.0.22.post7\n",

diff --git a/...21torch2.1.0-cp310-cp310-linux_x86_64.whl → ...22torch2.2.2-cp310-cp310-linux_x86_64.whl b/...21torch2.1.0-cp310-cp310-linux_x86_64.whl → ...22torch2.2.2-cp310-cp310-linux_x86_64.whl