return live preview defaults to how they were

only download TAESD model when it's needed
return calculations in single_sample_to_image to just if/elif/elif blocks
keep taesd model in its own directory
This commit is contained in:
AUTOMATIC
2023-05-17 09:24:01 +03:00
parent b217ebc490
commit 56a2672831
4 changed files with 31 additions and 29 deletions

View File

@@ -144,21 +144,10 @@ 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('From taesd repo download decoder model')
torch.hub.download_url_to_file(model_url, model_path)
def initialize():
fix_asyncio_event_loop_policy()
check_versions()
check_taesd()
extensions.list_extensions()
localization.list_localizations(cmd_opts.localizations_dir)