diff --git a/colab/GPU.ipynb b/colab/GPU.ipynb index 194ba8dea..1c3efad92 100644 --- a/colab/GPU.ipynb +++ b/colab/GPU.ipynb @@ -62,6 +62,8 @@ "#@markdown * prompthero/openjourney - midjourney style model\n", "#@markdown * ckpt/sd15 - base SD 1.5\n", "#@markdown * stabilityai/stable-diffusion-2-1-base - base SD 2.1\n", + "extras_enable_chromadb = True #@param {type:\"boolean\"}\n", + "#@markdown Enables ChromaDB for Infinity Context plugin\n", "\n", "import subprocess\n", "\n", @@ -84,6 +86,8 @@ " ExtrasModules.append('sd')\n", "if (extras_enable_tts):\n", " ExtrasModules.append('tts')\n", + "if (extras_enable_chromadb):\n", + " ExtrasModules.append('chromadb')\n", "\n", "params.append(f'--classification-model={Emotions_Model}')\n", "params.append(f'--summarization-model={Memory_Model}')\n", @@ -99,6 +103,7 @@ "!npm install -g localtunnel\n", "!pip install -r requirements-complete.txt\n", "!pip install tensorflow==2.12\n", + "!wget https://github.com/cloudflare/cloudflared/releases/download/2023.5.0/cloudflared-linux-amd64 -O /tmp/cloudflared-linux-amd64\n", "\n", "\n", "cmd = f\"python server.py {' '.join(params)}\"\n",