diff --git a/colab/GPU.ipynb b/colab/GPU.ipynb index c31548bf9..d5c24d77b 100644 --- a/colab/GPU.ipynb +++ b/colab/GPU.ipynb @@ -42,9 +42,8 @@ "#@markdown Enables Silero text-to-speech module\n", "extras_enable_sd = True #@param {type:\"boolean\"}\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/majicmix-realistic - realistic style model\n", "#@markdown * ckpt/sd15 - base SD 1.5\n", "\n", "import subprocess\n", @@ -72,7 +71,6 @@ "params.append(f'--classification-model={Emotions_Model}')\n", "params.append(f'--summarization-model={Memory_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'--enable-modules={\",\".join(ExtrasModules)}')\n", "\n", @@ -92,7 +90,8 @@ "print('processId:', extras_process.pid)\n", "while True:\n", " line = extras_process.stdout.readline().decode().strip()\n", - " print(line)\n" + " if line != None and line != '':\n", + " print(line)\n" ] } ],