Fix faster-whisper model not loading on Colab

ieasybooks · Jan 7, 2024 · 2e55b7c · 2e55b7c
1 parent 87e617d
commit 2e55b7c
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/colab_notebook.ipynb b/colab_notebook.ipynb
@@ -69,7 +69,14 @@
         "print('جارٍ تجهيز بيئة العمل.')\n",
         "\n",
         "# Setup Tafrigh.\n",
-        "%pip install -U tafrigh[wit,whisper]==1.1.0 > install_logs.txt\n",
+        "%pip install -U tafrigh[wit,whisper]==1.1.3 > install_logs.txt\n",
+        "\n",
+        "# Start: Quick fix related to Colab, HuggingFace, and faster-whisper.\n",
+        "!apt install libcublas11 > fix_logs.txt\n",
+        "\n",
+        "from huggingface_hub.utils import _runtime\n",
+        "_runtime._is_google_colab = False\n",
+        "# End: Quick fix related to Colab, HuggingFace, and faster-whisper.\n",
         "\n",
         "# Get inputs.\n",
         "\n",