From 49795fe0306149106c43cb75bd1eb8afc45badce Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Sat, 18 May 2024 15:37:58 +0200 Subject: [PATCH] refactor: rename folder safety_checker_models to safety_checker --- .../put_safety_checker_models_here | 0 modules/config.py | 6 +++--- 2 files changed, 3 insertions(+), 3 deletions(-) rename models/{safety_checker_models => safety_checker}/put_safety_checker_models_here (100%) diff --git a/models/safety_checker_models/put_safety_checker_models_here b/models/safety_checker/put_safety_checker_models_here similarity index 100% rename from models/safety_checker_models/put_safety_checker_models_here rename to models/safety_checker/put_safety_checker_models_here diff --git a/modules/config.py b/modules/config.py index 8b277242..73e33e4a 100644 --- a/modules/config.py +++ b/modules/config.py @@ -195,7 +195,7 @@ path_controlnet = get_dir_or_set_default('path_controlnet', '../models/controlne path_clip_vision = get_dir_or_set_default('path_clip_vision', '../models/clip_vision/') path_fooocus_expansion = get_dir_or_set_default('path_fooocus_expansion', '../models/prompt_expansion/fooocus_expansion') path_wildcards = get_dir_or_set_default('path_wildcards', '../wildcards/') -path_safety_checker_models = get_dir_or_set_default('path_safety_checker_models', '../models/safety_checker_models/') +path_safety_checker = get_dir_or_set_default('path_safety_checker', '../models/safety_checker/') path_outputs = get_path_output() @@ -688,10 +688,10 @@ def downloading_upscale_model(): def downloading_safety_checker_model(): load_file_from_url( url='https://huggingface.co/mashb1t/misc/resolve/main/stable-diffusion-safety-checker.bin', - model_dir=path_safety_checker_models, + model_dir=path_safety_checker, file_name='stable-diffusion-safety-checker.bin' ) - return os.path.join(path_safety_checker_models, 'stable-diffusion-safety-checker.bin') + return os.path.join(path_safety_checker, 'stable-diffusion-safety-checker.bin') update_files()