diff options
| -rw-r--r-- | modules/sd_vae_taesd.py | 2 | ||||
| -rw-r--r-- | webui.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/modules/sd_vae_taesd.py b/modules/sd_vae_taesd.py index ccc97959..927a7298 100644 --- a/modules/sd_vae_taesd.py +++ b/modules/sd_vae_taesd.py @@ -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():
|