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

Cloudflare fix and ChromaDB for colab

Browse Source
This commit is contained in:
SillyLossy
2023-05-24 00:13:10 +03:00
parent 061bad7d1f
commit 74223995f0
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
colab/GPU.ipynb
View File

@@ -62,6 +62,8 @@
"#@markdown * prompthero/openjourney - midjourney style model\n", "#@markdown * prompthero/openjourney - midjourney style model\n",
"#@markdown * ckpt/sd15 - base SD 1.5\n", "#@markdown * ckpt/sd15 - base SD 1.5\n",
"#@markdown * stabilityai/stable-diffusion-2-1-base - base SD 2.1\n", "#@markdown * stabilityai/stable-diffusion-2-1-base - base SD 2.1\n",
"extras_enable_chromadb = True #@param {type:\"boolean\"}\n",
"#@markdown Enables ChromaDB for Infinity Context plugin\n",
"\n", "\n",
"import subprocess\n", "import subprocess\n",
"\n", "\n",
@@ -84,6 +86,8 @@
" ExtrasModules.append('sd')\n", " ExtrasModules.append('sd')\n",
"if (extras_enable_tts):\n", "if (extras_enable_tts):\n",
" ExtrasModules.append('tts')\n", " ExtrasModules.append('tts')\n",
"if (extras_enable_chromadb):\n",
" ExtrasModules.append('chromadb')\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",
@@ -99,6 +103,7 @@
"!npm install -g localtunnel\n", "!npm install -g localtunnel\n",
"!pip install -r requirements-complete.txt\n", "!pip install -r requirements-complete.txt\n",
"!pip install tensorflow==2.12\n", "!pip install tensorflow==2.12\n",
"!wget https://github.com/cloudflare/cloudflared/releases/download/2023.5.0/cloudflared-linux-amd64 -O /tmp/cloudflared-linux-amd64\n",
"\n", "\n",
"\n", "\n",
"cmd = f\"python server.py {' '.join(params)}\"\n", "cmd = f\"python server.py {' '.join(params)}\"\n",