StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Minor changes

Browse Source
This commit is contained in:
Sakura-Luna 2023-05-15 03:04:34 +08:00 committed by GitHub
parent bd9b9d425a
commit 742da31932
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_vae_taesd.py
View File

@ -71,6 +71,6 @@ def decode():
sd_vae_taesd.eval() sd_vae_taesd.eval()
sd_vae_taesd.to(devices.device, devices.dtype) sd_vae_taesd.to(devices.device, devices.dtype)
else: else:
raise FileNotFoundError('Tiny AE mdoel not found') raise FileNotFoundError('Tiny AE model not found')
return sd_vae_taesd.decoder return sd_vae_taesd.decoder

2
webui.py
View File

@ -151,7 +151,7 @@ def check_taesd():
model_path = os.path.join(models_path, "VAE-approx", "taesd_decoder.pth") model_path = os.path.join(models_path, "VAE-approx", "taesd_decoder.pth")
if not os.path.exists(model_path): if not os.path.exists(model_path):
print('download taesd model') print('download taesd model')
torch.hub.download_url_to_file(model_url, os.path.dirname(model_path)) torch.hub.download_url_to_file(model_url, model_path)
def initialize(): def initialize():