import os | |
from modules import shared | |
models_path = shared.opts.data.get('control_net_modules_path', None) | |
if not models_path: | |
models_path = getattr(shared.cmd_opts, 'controlnet_annotator_models_path', None) | |
if not models_path: | |
models_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'downloads') | |
if not os.path.isabs(models_path): | |
models_path = os.path.join(shared.data_path, models_path) | |
clip_vision_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'clip_vision') | |
# clip vision is always inside controlnet "extensions\sd-webui-controlnet" | |
# and any problem can be solved by removing controlnet and reinstall | |
models_path = os.path.realpath(models_path) | |
os.makedirs(models_path, exist_ok=True) | |
print(f'ControlNet preprocessor location: {models_path}') | |
# Make sure that the default location is inside controlnet "extensions\sd-webui-controlnet" | |
# so that any problem can be solved by removing controlnet and reinstall | |
# if users do not change configs on their own (otherwise users will know what is wrong) | |