Minor changes
This commit is contained in:
parent
bd9b9d425a
commit
742da31932
|
@ -71,6 +71,6 @@ def decode():
|
|||
sd_vae_taesd.eval()
|
||||
sd_vae_taesd.to(devices.device, devices.dtype)
|
||||
else:
|
||||
raise FileNotFoundError('Tiny AE mdoel not found')
|
||||
raise FileNotFoundError('Tiny AE model not found')
|
||||
|
||||
return sd_vae_taesd.decoder
|
||||
|
|
2
webui.py
2
webui.py
|
@ -151,7 +151,7 @@ def check_taesd():
|
|||
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))
|
||||
torch.hub.download_url_to_file(model_url, model_path)
|
||||
|
||||
|
||||
def initialize():
|
||||
|
|
Loading…
Reference in New Issue