fix typo

seruva19 · Oct 29, 2023 · a68c122 · a68c122
1 parent d914ce8
commit a68c122
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/notebooks/kubin-colab.ipynb b/notebooks/kubin-colab.ipynb
@@ -161,7 +161,7 @@
     "!pip install -r requirements.txt\n",
     "\n",
     "if use_flash_attention:\n",
-    "  !pip install /content/kubin/wheels/flash_attn-1.0.9+cu118torch2.1.0-cp310-cp310-linux_x86_64\n",
+    "  !pip install /content/kubin/wheels/flash_attn-1.0.9+cu118torch2.1.0-cp310-cp310-linux_x86_64.whl\n",
     "\n",
     "if use_xformers:\n",
     "  !pip install xformers==0.0.20\n",