diff options
| author | Sakura-Luna <53183413+Sakura-Luna@users.noreply.github.com> | 2023-05-15 03:04:34 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-05-15 03:04:34 +0800 |
| commit | 742da3193290f5692901c4c614c98bec291163f2 (patch) | |
| tree | c31cb600712537fa4c4bb9495f93d893db8de31c | |
| parent | bd9b9d425a355e151b43047a5df5fcead2fcdc52 (diff) | |
Minor changes
| -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():
|