rixty
/
SillyTavern
mirror of https://github.com/SillyTavern/SillyTavern.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add WebSearch to colab

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",
"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",