Add Tiny AE live preview

1 files changed, 11 insertions, 0 deletions

diff --git a/webui.py b/webui.py
index 727ebd31..0a928434 100644
--- a/webui.py
+++ b/webui.py
@@ -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)