diff --git a/StableDiffusionUI_Voldemort_paperspace.ipynb b/StableDiffusionUI_Voldemort_paperspace.ipynb index d4d65c6..3efebee 100644 --- a/StableDiffusionUI_Voldemort_paperspace.ipynb +++ b/StableDiffusionUI_Voldemort_paperspace.ipynb @@ -103,11 +103,11 @@ " # This disables online Gradio app mode and you will only be able to access it on your local network.\n", "\n", "gradio_auth = False # Enable gradio_auth and insecure-extension-access option.\n", - " # Set to \"me:password\" to let Gradio auth.\n", + " # Set to \"me:password\" to enable.\n", "\n", "search_paperspace_datasets = True # Enable searching for checkpoints in /datasets to link to the webui\n", "\n", - "ui_theme = None # Set the WEB UI theme. Values can be None (default) or 'dark'.\n", + "ui_theme = None # Set the WEB UI theme. Values can be None (default) or 'dark'.\n", "\n", "# ===============================================================\n", "# Save variables to Jupiter's temp storage so we can access it even if the kernel restarts.\n", @@ -319,6 +319,7 @@ "cell_type": "markdown", "metadata": { "id": "F0EINk5M0s-w", + "jp-MarkdownHeadingCollapsed": true, "tags": [] }, "source": [ @@ -750,7 +751,7 @@ "# Link models one level recursive\n", "# You can organize your models in your model storage dir into sub-folders.\n", "for file in os.listdir(model_storage_dir):\n", - " if os.path.isdir(os.path.join(model_storage_dir, file)) and file != 'hypernetworks':\n", + " if os.path.isdir(os.path.join(model_storage_dir, file)) and file not in ['hypernetworks', 'vae']:\n", " symlink_models(os.path.join(model_storage_dir, file), 'ckpt', f'{repo_storage_dir}/stable-diffusion-webui/models/Stable-diffusion')\n", "\n", "# Link config yaml files\n", @@ -759,6 +760,8 @@ "# Link hypernetworks\n", "symlink_models(f'{model_storage_dir}/hypernetworks', 'pt', f'{repo_storage_dir}/stable-diffusion-webui/models/hypernetworks')\n", "\n", + "symlink_models(f'{model_storage_dir}/vae', 'ckpt', f'{repo_storage_dir}/stable-diffusion-webui/models/VAE')\n", + "\n", "# Link the NovelAI files for each of the NovelAI models\n", "os.chdir(f'{model_storage_dir}')\n", "for model in glob.glob('novelai-*.ckpt'):\n",