fix civitai search

This commit is contained in:
Vladimir Mandic
2024-07-10 09:55:58 -04:00
parent 425c7575a8
commit 03a6e25d48
4 changed files with 26 additions and 21 deletions
+3
View File
@@ -101,6 +101,9 @@ def download_civit_model_thread(model_name, model_url, model_path, model_type, t
elif model_type == 'Embedding':
model_file = os.path.join(shared.opts.embeddings_dir, model_path, model_name)
temp_file = os.path.join(shared.opts.embeddings_dir, model_path, temp_file)
elif model_type == 'VAE':
model_file = os.path.join(shared.opts.vae_dir, model_path, model_name)
temp_file = os.path.join(shared.opts.vae_dir, model_path, temp_file)
else:
model_file = os.path.join(shared.opts.ckpt_dir, model_path, model_name)
temp_file = os.path.join(shared.opts.ckpt_dir, model_path, temp_file)