diff --git a/colab/GPU.ipynb b/colab/GPU.ipynb index 4adf35e6c..f197a1833 100644 --- a/colab/GPU.ipynb +++ b/colab/GPU.ipynb @@ -39,6 +39,9 @@ "#@markdown Allows to run Stable Diffusion pipeline on CPU (slow!)\n", "use_sd_cpu = False #@param {type:\"boolean\"}\n", "#@markdown ***\n", + "#@markdown Enables the WebSearch module\n", + "extras_enable_websearch = True #@param {type:\"boolean\"}\n", + "#@markdown ***\n", "#@markdown Loads the image captioning module\n", "extras_enable_caption = True #@param {type:\"boolean\"}\n", "captioning_model = \"Salesforce/blip-image-captioning-large\" #@param [ \"Salesforce/blip-image-captioning-large\", \"Salesforce/blip-image-captioning-base\" ]\n", @@ -124,6 +127,12 @@ " params.append('--rvc-save-file')\n", "\n", "\n", + "if extras_enable_websearch:\n", + " print(\"Enabling WebSearch module\")\n", + " modules.append('websearch')\n", + " !apt update\n", + " !apt install -y chromium-chromedriver\n", + "\n", "params.append(f'--classification-model={classification_model}')\n", "params.append(f'--summarization-model={summarization_model}')\n", "params.append(f'--captioning-model={captioning_model}')\n", @@ -142,7 +151,7 @@ "\n", "if extras_enable_rvc:\n", " print(\"Installing RVC requirements\")\n", - " !pip install -r requirements-rvc.txt\n", + " %pip install -r requirements-rvc.txt\n", "\n", "# Generate a random API key\n", "api_key = secrets.token_hex(5)\n",