From 07659efd884184dea0bae156803e58f957229361 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Sat, 17 Feb 2024 08:03:17 -0800 Subject: [PATCH] fix import order --- modules_forge/shared.py | 21 ++++++++++++++++++++- 1 file changed, 20 insertions(+), 1 deletion(-) diff --git a/modules_forge/shared.py b/modules_forge/shared.py index ec44c9a4..a5aee828 100644 --- a/modules_forge/shared.py +++ b/modules_forge/shared.py @@ -1,8 +1,27 @@ import os import ldm_patched.modules.utils +import argparse from modules.paths_internal import models_path -from modules.shared import cmd_opts +from pathlib import Path + + +parser = argparse.ArgumentParser() + +parser.add_argument( + "--controlnet-dir", + type=Path, + help="Path to directory with ControlNet models", + default=None, +) +parser.add_argument( + "--controlnet-preprocessor-models-dir", + type=Path, + help="Path to directory with annotator model directories", + default=None, +) + +cmd_opts = parser.parse_known_args()[0] if cmd_opts.controlnet_dir: controlnet_dir = str(cmd_opts.controlnet_dir)