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

Colab: add lt for KoboldAI

Browse Source
This commit is contained in:
Ash
2023-02-15 16:49:46 +04:00
parent a73decaac6
commit e764f94c8a
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
colab/GPU.ipynb
View File

@@ -65,7 +65,7 @@
"\n",
"Model = \"Pygmalion 6B\" #@param [ \"Pygmalion 6B\", \"Pygmalion 6B Dev\"] {allow-input: true}\n",
"Version = \"Official\" \n",
"KoboldAI_Provider = \"Cloudflare\" #@param [\"Localtunnel\", \"Cloudflare\"]\n",
"KoboldAI_Provider = \"Localtunnel\" #@param [\"Localtunnel\", \"Cloudflare\"]\n",
"use_google_drive = True #@param {type:\"boolean\"}\n",
"Provider = KoboldAI_Provider\n",
"!nvidia-smi\n",
@@ -133,7 +133,11 @@
"#Henk's KoboldAI script\n",
"!wget https://koboldai.org/ckds && chmod +x ckds\n",
"!./ckds --init only\n",
"p = subprocess.Popen(['/content/ckds', '--model', Model], stdout=subprocess.PIPE, stderr=subprocess.PIPE)\n",
"if Provider == \"Localtunnel\":\n",
" p = subprocess.Popen(['/content/ckds', '--model', Model, '--localtunnel', 'yes'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)\n",
"else:\n",
" p = subprocess.Popen(['/content/ckds', '--model', Model], stdout=subprocess.PIPE, stderr=subprocess.PIPE)\n",
"\n",
"\n",
"#Do not repeat! Tricks performed by a professional!\n",
"url = ''\n",