diff --git a/colab/TabbyAPI_Colab_Example.ipynb b/colab/TabbyAPI_Colab_Example.ipynb index ee0f0b2..1b7e18d 100644 --- a/colab/TabbyAPI_Colab_Example.ipynb +++ b/colab/TabbyAPI_Colab_Example.ipynb @@ -56,7 +56,7 @@ "%cd tabbyAPI\n", "\n", "# Install cuda requirements\n", - "!pip install -r requirements.txt -q\n", + "!pip install .[cu121] -q\n", "!pip install huggingface-hub -q\n", "\n", "# Download cloudflared tunnel\n", @@ -260,4 +260,4 @@ }, "nbformat": 4, "nbformat_minor": 0 -} \ No newline at end of file +}