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

Adjustments to colab

Browse Source
This commit is contained in:
SillyLossy
2023-05-12 01:04:37 +03:00
parent ead53164a8
commit aaa23815fd
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
colab/GPU.ipynb
View File

@@ -42,9 +42,8 @@
"#@markdown Enables Silero text-to-speech module\n", "#@markdown Enables Silero text-to-speech module\n",
"extras_enable_sd = True #@param {type:\"boolean\"}\n", "extras_enable_sd = True #@param {type:\"boolean\"}\n",
"#@markdown Enables SD picture generation\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", "SD_Model = \"ckpt/anything-v4.5-vae-swapped\" #@param [ \"ckpt/anything-v4.5-vae-swapped\", \"ckpt/sd15\" ]\n",
"#@markdown * ckpt/anything-v4.5-vae-swapped - anime style model\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", "#@markdown * ckpt/sd15 - base SD 1.5\n",
"\n", "\n",
"import subprocess\n", "import subprocess\n",
@@ -72,7 +71,6 @@
"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'--sd-model={SD_Model}')\n",
"params.append(f'--enable-modules={\",\".join(ExtrasModules)}')\n", "params.append(f'--enable-modules={\",\".join(ExtrasModules)}')\n",
"\n", "\n",
@@ -92,7 +90,8 @@
"print('processId:', extras_process.pid)\n", "print('processId:', extras_process.pid)\n",
"while True:\n", "while True:\n",
" line = extras_process.stdout.readline().decode().strip()\n", " line = extras_process.stdout.readline().decode().strip()\n",
" print(line)\n" " if line != None and line != '':\n",
" print(line)\n"
] ]
} }
], ],