diff options
| author | Sakura-Luna <53183413+Sakura-Luna@users.noreply.github.com> | 2023-05-14 04:42:44 +0000 |
|---|---|---|
| committer | Sakura-Luna <53183413+Sakura-Luna@users.noreply.github.com> | 2023-05-14 06:06:01 +0000 |
| commit | e14b586d0494d6c5cc3cbc45b5fa00c03d052443 (patch) | |
| tree | 807b3e771ef465654b672956d09d94af525d14ab /webui.py | |
| parent | b08500cec8a791ef20082628b49b17df833f5dda (diff) | |
| download | stable-diffusion-webui-gfx803-e14b586d0494d6c5cc3cbc45b5fa00c03d052443.tar.gz stable-diffusion-webui-gfx803-e14b586d0494d6c5cc3cbc45b5fa00c03d052443.tar.bz2 stable-diffusion-webui-gfx803-e14b586d0494d6c5cc3cbc45b5fa00c03d052443.zip | |
Add Tiny AE live preview
Diffstat (limited to 'webui.py')
| -rw-r--r-- | webui.py | 11 |
1 files changed, 11 insertions, 0 deletions
@@ -144,10 +144,21 @@ Use --skip-version-check commandline argument to disable this check. """.strip())
+def check_taesd():
+ from modules.paths_internal import models_path
+
+ model_url = 'https://github.com/madebyollin/taesd/raw/main/taesd_decoder.pth'
+ model_path = os.path.join(models_path, "VAE-approx", "taesd_decoder.pth")
+ if not os.path.exists(model_path):
+ print('download taesd model')
+ torch.hub.download_url_to_file(model_url, os.path.dirname(model_path))
+
+
def initialize():
fix_asyncio_event_loop_policy()
check_versions()
+ check_taesd()
extensions.list_extensions()
localization.list_localizations(cmd_opts.localizations_dir)
|