Add modules to colab

This commit is contained in:
SillyLossy
2023-03-04 23:12:22 +02:00
parent ea44f763c2
commit b05a4d757c
2 changed files with 12 additions and 4 deletions

View File

@@ -66,8 +66,10 @@
"Model = \"Pygmalion 6B\" #@param [ \"Pygmalion 6B\", \"Pygmalion 6B Dev\"] {allow-input: true}\n",
"Version = \"Official\" \n",
"KoboldAI_Provider = \"Localtunnel\" #@param [\"Localtunnel\", \"Cloudflare\"]\n",
"use_google_drive = True #@param {type:\"boolean\"}\n",
"extras_enable_captioning = False #@param {type:\"boolean\"}\n",
"use_google_drive = False #@param {type:\"boolean\"}\n",
"extras_enable_captioning = True #@param {type:\"boolean\"}\n",
"extras_enable_memory = True #@param {type:\"boolean\"}\n",
"extras_enable_emotions = True #@param {type:\"boolean\"}\n",
"Provider = KoboldAI_Provider\n",
"!nvidia-smi\n",
"import subprocess\n",
@@ -171,7 +173,13 @@
"if (extras_enable_captioning):\n",
" ExtrasModules.append('caption')\n",
"\n",
"params.append(f'--enable-modules=\"{\",\".join(ExtrasModules)}\"')\n",
"if (extras_enable_memory):\n",
" ExtrasModules.append('summarize')\n",
"\n",
"if (extras_enable_emotions)\n",
" ExtrasModules.append('classify')\n",
"\n",
"params.append(f'--enable-modules={\",\".join(ExtrasModules)}')\n",
"\n",
"#TavernAI extras\n",
"%cd /\n",

View File

@@ -8,7 +8,7 @@ div.expression {
left: 0;
bottom: 0;
margin-left: 10px;
filter: drop-shadow(2px 2px 4px #cccccc84);
filter: drop-shadow(2px 2px 2px #cccccc80);
transition: 500ms;
}