Commit 742da319 authored by Sakura-Luna's avatar Sakura-Luna Committed by GitHub

Minor changes

parent bd9b9d42
......@@ -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
......@@ -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():
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment