Link VAEs

This commit is contained in:
Cyberes 2022-11-19 00:10:34 -07:00
parent 2ac5376608
commit 0a90822768
No known key found for this signature in database
GPG Key ID: 194A1C358AACFC39
1 changed files with 6 additions and 3 deletions

View File

@ -103,11 +103,11 @@
" # This disables online Gradio app mode and you will only be able to access it on your local network.\n", " # This disables online Gradio app mode and you will only be able to access it on your local network.\n",
"\n", "\n",
"gradio_auth = False # Enable gradio_auth and insecure-extension-access option.\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", "\n",
"search_paperspace_datasets = True # Enable searching for checkpoints in /datasets to link to the webui\n", "search_paperspace_datasets = True # Enable searching for checkpoints in /datasets to link to the webui\n",
"\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",
"# ===============================================================\n", "# ===============================================================\n",
"# Save variables to Jupiter's temp storage so we can access it even if the kernel restarts.\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", "cell_type": "markdown",
"metadata": { "metadata": {
"id": "F0EINk5M0s-w", "id": "F0EINk5M0s-w",
"jp-MarkdownHeadingCollapsed": true,
"tags": [] "tags": []
}, },
"source": [ "source": [
@ -750,7 +751,7 @@
"# Link models one level recursive\n", "# Link models one level recursive\n",
"# You can organize your models in your model storage dir into sub-folders.\n", "# You can organize your models in your model storage dir into sub-folders.\n",
"for file in os.listdir(model_storage_dir):\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", " symlink_models(os.path.join(model_storage_dir, file), 'ckpt', f'{repo_storage_dir}/stable-diffusion-webui/models/Stable-diffusion')\n",
"\n", "\n",
"# Link config yaml files\n", "# Link config yaml files\n",
@ -759,6 +760,8 @@
"# Link hypernetworks\n", "# Link hypernetworks\n",
"symlink_models(f'{model_storage_dir}/hypernetworks', 'pt', f'{repo_storage_dir}/stable-diffusion-webui/models/hypernetworks')\n", "symlink_models(f'{model_storage_dir}/hypernetworks', 'pt', f'{repo_storage_dir}/stable-diffusion-webui/models/hypernetworks')\n",
"\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", "# Link the NovelAI files for each of the NovelAI models\n",
"os.chdir(f'{model_storage_dir}')\n", "os.chdir(f'{model_storage_dir}')\n",
"for model in glob.glob('novelai-*.ckpt'):\n", "for model in glob.glob('novelai-*.ckpt'):\n",