rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-04-14 19:02:10 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add WebSearch to colab

Browse Source
This commit is contained in:
Cohee 2023-11-30 02:17:35 +02:00
parent c3259d1929
commit 6abd0b4b75
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
colab/GPU.ipynb
View File

@ -39,6 +39,9 @@
"#@markdown Allows to run Stable Diffusion pipeline on CPU (slow!)\n", "#@markdown Allows to run Stable Diffusion pipeline on CPU (slow!)\n",
"use_sd_cpu = False #@param {type:\"boolean\"}\n", "use_sd_cpu = False #@param {type:\"boolean\"}\n",
"#@markdown ***\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", "#@markdown Loads the image captioning module\n",
"extras_enable_caption = True #@param {type:\"boolean\"}\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", "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", " params.append('--rvc-save-file')\n",
"\n", "\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'--classification-model={classification_model}')\n",
"params.append(f'--summarization-model={summarization_model}')\n", "params.append(f'--summarization-model={summarization_model}')\n",
"params.append(f'--captioning-model={captioning_model}')\n", "params.append(f'--captioning-model={captioning_model}')\n",
@ -142,7 +151,7 @@
"\n", "\n",
"if extras_enable_rvc:\n", "if extras_enable_rvc:\n",
" print(\"Installing RVC requirements\")\n", " print(\"Installing RVC requirements\")\n",
" !pip install -r requirements-rvc.txt\n", " %pip install -r requirements-rvc.txt\n",
"\n", "\n",
"# Generate a random API key\n", "# Generate a random API key\n",
"api_key = secrets.token_hex(5)\n", "api_key = secrets.token_hex(5)\n",