feat: no huggingface option

This commit is contained in:
Bingsu
2023-05-13 15:17:08 +09:00
parent a0657caf3e
commit 3d892728c6
3 changed files with 37 additions and 13 deletions

View File

@@ -44,8 +44,9 @@ except Exception:
pass
AFTER_DETAILER = "After Detailer"
no_huggingface = getattr(cmd_opts, "ad_no_huggingface", False)
adetailer_dir = Path(models_path, "adetailer")
model_mapping = get_models(adetailer_dir)
model_mapping = get_models(adetailer_dir, huggingface=not no_huggingface)
txt2img_submit_button = img2img_submit_button = None