From e14b586d0494d6c5cc3cbc45b5fa00c03d052443 Mon Sep 17 00:00:00 2001
From: Sakura-Luna <53183413+Sakura-Luna@users.noreply.github.com>
Date: Sun, 14 May 2023 12:42:44 +0800
Subject: Add Tiny AE live preview

---
 webui.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

(limited to 'webui.py')

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)
-- 
cgit v1.2.3