lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-26 17:29:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add --forge-ref-a1111-home cmd arg to reference existing A1111 checkout (#203)

Browse Source 
* Add --forge-ref-a1111-home cmd arg to reference existing A1111 checkout

* nit
This commit is contained in:
Chenlei Hu
2024-02-12 02:22:31 +00:00
committed by GitHub
parent f5bf7799f4
commit 9c5038c766
4 changed files with 66 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
modules_forge/shared.py
View File

@@ -2,12 +2,18 @@ import os
import ldm_patched.modules.utils
from modules.paths_internal import models_path
from modules.shared import cmd_opts
controlnet_dir = os.path.join(models_path, 'ControlNet')
if cmd_opts.controlnet_dir:
controlnet_dir = str(cmd_opts.controlnet_dir)
else:
controlnet_dir = os.path.join(models_path, 'ControlNet')
os.makedirs(controlnet_dir, exist_ok=True)
preprocessor_dir = os.path.join(models_path, 'ControlNetPreprocessor')
if cmd_opts.controlnet_preprocessor_models_dir:
preprocessor_dir = str(cmd_opts.controlnet_preprocessor_models_dir)
else:
preprocessor_dir = os.path.join(models_path, 'ControlNetPreprocessor')
os.makedirs(preprocessor_dir, exist_ok=True)
diffusers_dir = os.path.join(models_path, 'diffusers')