From 72ad2c32614c1a22e2255c78363171adf94c4155 Mon Sep 17 00:00:00 2001 From: Cohee <18619528+Cohee1207@users.noreply.github.com> Date: Thu, 16 Nov 2023 14:27:09 +0200 Subject: [PATCH] Update colab --- colab/GPU.ipynb | 52 ++++++++++++++++++++++++++----------------------- 1 file changed, 28 insertions(+), 24 deletions(-) diff --git a/colab/GPU.ipynb b/colab/GPU.ipynb index c44ab7270..7c3f93e1a 100644 --- a/colab/GPU.ipynb +++ b/colab/GPU.ipynb @@ -34,34 +34,37 @@ "source": [ "#@markdown (RECOMMENDED) Generates an API key for you to use with the API\n", "secure = False #@param {type:\"boolean\"}\n", - "#@markdown Enables hosting of extensions backend for SillyTavern Extras\n", - "use_cpu = False #@param {type:\"boolean\"}\n", "#@markdown Allows to run SillyTavern Extras on CPU (use if you're out of daily GPU allowance)\n", - "use_sd_cpu = False #@param {type:\"boolean\"}\n", + "use_cpu = False #@param {type:\"boolean\"}\n", "#@markdown Allows to run Stable Diffusion pipeline on CPU (slow!)\n", - "extras_enable_captioning = True #@param {type:\"boolean\"}\n", + "use_sd_cpu = False #@param {type:\"boolean\"}\n", + "#@markdown ***\n", "#@markdown Loads the image captioning module\n", + "extras_enable_captioning = True #@param {type:\"boolean\"}\n", "Captions_Model = \"Salesforce/blip-image-captioning-large\" #@param [ \"Salesforce/blip-image-captioning-large\", \"Salesforce/blip-image-captioning-base\" ]\n", "#@markdown * Salesforce/blip-image-captioning-large - good base model\n", "#@markdown * Salesforce/blip-image-captioning-base - slightly faster but less accurate\n", - "extras_enable_emotions = True #@param {type:\"boolean\"}\n", + "#@markdown ***\n", "#@markdown Loads the sentiment classification model\n", + "extras_enable_emotions = True #@param {type:\"boolean\"}\n", "Emotions_Model = \"nateraw/bert-base-uncased-emotion\" #@param [\"nateraw/bert-base-uncased-emotion\", \"joeddav/distilbert-base-uncased-go-emotions-student\"]\n", "#@markdown * nateraw/bert-base-uncased-emotion = 6 supported emotions
\n", "#@markdown * joeddav/distilbert-base-uncased-go-emotions-student = 28 supported emotions\n", - "extras_enable_memory = True #@param {type:\"boolean\"}\n", + "#@markdown ***\n", "#@markdown Loads the story summarization module\n", + "extras_enable_memory = True #@param {type:\"boolean\"}\n", "Memory_Model = \"slauw87/bart_summarisation\" #@param [ \"slauw87/bart_summarisation\", \"Qiliang/bart-large-cnn-samsum-ChatGPT_v3\", \"Qiliang/bart-large-cnn-samsum-ElectrifAi_v10\", \"distilbart-xsum-12-3\" ]\n", "#@markdown * slauw87/bart_summarisation - general purpose summarization model\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 * distilbart-xsum-12-3 - faster, but pretty basic alternative\n", - "extras_enable_silero_tts = True #@param {type:\"boolean\"}\n", + "#@markdown ***\n", "#@markdown Enables Silero text-to-speech module\n", - "extras_enable_edge_tts = True #@param {type:\"boolean\"}\n", + "extras_enable_silero_tts = True #@param {type:\"boolean\"}\n", "#@markdown Enables Microsoft Edge text-to-speech module\n", - "extras_enable_sd = True #@param {type:\"boolean\"}\n", + "extras_enable_edge_tts = True #@param {type:\"boolean\"}\n", "#@markdown Enables SD picture generation\n", + "extras_enable_sd = True #@param {type:\"boolean\"}\n", "SD_Model = \"ckpt/anything-v4.5-vae-swapped\" #@param [ \"ckpt/anything-v4.5-vae-swapped\", \"hakurei/waifu-diffusion\", \"philz1337/clarity\", \"prompthero/openjourney\", \"ckpt/sd15\", \"stabilityai/stable-diffusion-2-1-base\" ]\n", "#@markdown * ckpt/anything-v4.5-vae-swapped - anime style model\n", "#@markdown * hakurei/waifu-diffusion - anime style model\n", @@ -69,8 +72,9 @@ "#@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 ***\n", "#@markdown Enables ChromaDB module\n", + "extras_enable_chromadb = True #@param {type:\"boolean\"}\n", "\n", "import subprocess\n", "import secrets\n", @@ -115,20 +119,20 @@ "%cd /SillyTavern-extras\n", "!git clone https://github.com/Cohee1207/tts_samples\n", "!npm install -g localtunnel\n", - "!pip install -r requirements-complete.txt\n", - "!pip install tensorflow==2.14\n", - "!pip install colorama\n", - "!pip install Flask-Cors\n", - "!pip install Flask-Compress\n", - "!pip install transformers\n", - "!pip install Flask_Cloudflared\n", - "!pip install webuiapi\n", - "!pip install diffusers\n", - "!pip install accelerate\n", - "!pip install silero_api_server\n", - "!pip install edge_tts\n", - "!pip install chromadb\n", - "!pip install sentence_transformers\n", + "%pip install -r requirements.txt\n", + "#!pip install tensorflow==2.14\n", + "#!pip install colorama\n", + "#!pip install Flask-Cors\n", + "#!pip install Flask-Compress\n", + "#!pip install transformers\n", + "#!pip install Flask_Cloudflared\n", + "#!pip install webuiapi\n", + "#!pip install diffusers\n", + "#!pip install accelerate\n", + "#!pip install silero_api_server\n", + "#!pip install edge_tts\n", + "#!pip install chromadb\n", + "#!pip install sentence_transformers\n", "!wget https://github.com/cloudflare/cloudflared/releases/download/2023.5.0/cloudflared-linux-amd64 -O /tmp/cloudflared-linux-amd64\n", "!chmod +x /tmp/cloudflared-linux-amd64\n", "\n",