rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add SD and TTS to colab.

Browse Source
This commit is contained in:
SillyLossy
2023-05-12 00:46:15 +03:00
parent 1dec2683ce
commit ead53164a8
1 changed files with 18 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
colab/GPU.ipynb
View File

@@ -20,6 +20,7 @@
"outputs": [], "outputs": [],
"source": [ "source": [
"#@markdown Enables hosting of extensions backend for SillyTavern Extras\n", "#@markdown Enables hosting of extensions backend for SillyTavern Extras\n",
"use_cpu = False #@param {type:\"boolean\"}\n",
"extras_enable_captioning = True #@param {type:\"boolean\"}\n", "extras_enable_captioning = True #@param {type:\"boolean\"}\n",
"#@markdown Loads the image captioning module\n", "#@markdown Loads the image captioning module\n",
"Captions_Model = \"Salesforce/blip-image-captioning-large\" #@param [ \"Salesforce/blip-image-captioning-large\", \"Salesforce/blip-image-captioning-base\" ]\n", "Captions_Model = \"Salesforce/blip-image-captioning-large\" #@param [ \"Salesforce/blip-image-captioning-large\", \"Salesforce/blip-image-captioning-base\" ]\n",
@@ -37,6 +38,14 @@
"#@markdown * Qiliang/bart-large-cnn-samsum-ChatGPT_v3 - summarization model optimized for chats\n", "#@markdown * Qiliang/bart-large-cnn-samsum-ChatGPT_v3 - summarization model optimized for chats\n",
"#@markdown * Qiliang/bart-large-cnn-samsum-ElectrifAi_v10 - nice results so far, but still being evaluated\n", "#@markdown * Qiliang/bart-large-cnn-samsum-ElectrifAi_v10 - nice results so far, but still being evaluated\n",
"#@markdown * distilbart-xsum-12-3 - faster, but pretty basic alternative\n", "#@markdown * distilbart-xsum-12-3 - faster, but pretty basic alternative\n",
"extras_enable_tts = True #@param {type:\"boolean\"}\n",
"#@markdown Enables Silero text-to-speech module\n",
"extras_enable_sd = True #@param {type:\"boolean\"}\n",
"#@markdown Enables SD picture generation\n",
"SD_Model = \"ckpt/anything-v4.5-vae-swapped\" #@param [ \"ckpt/anything-v4.5-vae-swapped\", \"ckpt/majicmix-realistic\", \"ckpt/sd15\" ]\n",
"#@markdown * ckpt/anything-v4.5-vae-swapped - anime style model\n",
"#@markdown * ckpt/majicmix-realistic - realistic style model\n",
"#@markdown * ckpt/sd15 - base SD 1.5\n",
"\n", "\n",
"import subprocess\n", "import subprocess\n",
"\n", "\n",
@@ -44,6 +53,7 @@
"# SillyTavern extras\n", "# SillyTavern extras\n",
"extras_url = '(disabled)'\n", "extras_url = '(disabled)'\n",
"params = []\n", "params = []\n",
"if use_cpu:\n",
" params.append('--cpu')\n", " params.append('--cpu')\n",
"params.append('--share')\n", "params.append('--share')\n",
"ExtrasModules = []\n", "ExtrasModules = []\n",
@@ -54,10 +64,16 @@
" ExtrasModules.append('summarize')\n", " ExtrasModules.append('summarize')\n",
"if (extras_enable_emotions):\n", "if (extras_enable_emotions):\n",
" ExtrasModules.append('classify')\n", " ExtrasModules.append('classify')\n",
"if (extras_enable_sd):\n",
" ExtrasModules.append('sd')\n",
"if (extras_enable_tts):\n",
" ExtrasModules.append('tts')\n",
"\n", "\n",
"params.append(f'--classification-model={Emotions_Model}')\n", "params.append(f'--classification-model={Emotions_Model}')\n",
"params.append(f'--summarization-model={Memory_Model}')\n", "params.append(f'--summarization-model={Memory_Model}')\n",
"params.append(f'--captioning-model={Captions_Model}')\n", "params.append(f'--captioning-model={Captions_Model}')\n",
"params.append(f'--sd-local')\n",
"params.append(f'--sd-model={SD_Model}')\n",
"params.append(f'--enable-modules={\",\".join(ExtrasModules)}')\n", "params.append(f'--enable-modules={\",\".join(ExtrasModules)}')\n",
"\n", "\n",
"\n", "\n",
@@ -65,7 +81,7 @@
"!git clone https://github.com/Cohee1207/SillyTavern-extras\n", "!git clone https://github.com/Cohee1207/SillyTavern-extras\n",
"%cd /SillyTavern-extras\n", "%cd /SillyTavern-extras\n",
"!npm install -g localtunnel\n", "!npm install -g localtunnel\n",
"!pip install -r requirements.txt\n", "!pip install -r requirements-complete.txt\n",
"!pip install tensorflow==2.11\n", "!pip install tensorflow==2.11\n",
"\n", "\n",
"\n", "\n",