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,7 +103,7 @@
" # 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",
@ -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",