From 0b27676fe1ee3d96e832b68f12755b463e907473 Mon Sep 17 00:00:00 2001 From: silveroxides Date: Wed, 30 Jul 2025 13:12:54 +0200 Subject: [PATCH] add models-directory launch argument --- comfy/cli_args.py | 1 + folder_paths.py | 6 +++++- 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/comfy/cli_args.py b/comfy/cli_args.py index 0d760d524..f7481c7dc 100644 --- a/comfy/cli_args.py +++ b/comfy/cli_args.py @@ -44,6 +44,7 @@ parser.add_argument("--max-upload-size", type=float, default=100, help="Set the parser.add_argument("--base-directory", type=str, default=None, help="Set the ComfyUI base directory for models, custom_nodes, input, output, temp, and user directories.") parser.add_argument("--extra-model-paths-config", type=str, default=None, metavar="PATH", nargs='+', action='append', help="Load one or more extra_model_paths.yaml files.") +parser.add_argument("--models-directory", type=str, default=None, help="Set the ComfyUI models directory. Overrides --base-directory.") parser.add_argument("--output-directory", type=str, default=None, help="Set the ComfyUI output directory. Overrides --base-directory.") parser.add_argument("--temp-directory", type=str, default=None, help="Set the ComfyUI temp directory (default is in the ComfyUI directory). Overrides --base-directory.") parser.add_argument("--input-directory", type=str, default=None, help="Set the ComfyUI input directory. Overrides --base-directory.") diff --git a/folder_paths.py b/folder_paths.py index 9ec952940..4e3f252c1 100644 --- a/folder_paths.py +++ b/folder_paths.py @@ -19,7 +19,11 @@ if args.base_directory: else: base_path = os.path.dirname(os.path.realpath(__file__)) -models_dir = os.path.join(base_path, "models") +if args.models_directory: + models_dir = os.path.abspath(args.models_directory) +else: + models_dir = os.path.join(base_path, "models") + folder_names_and_paths["checkpoints"] = ([os.path.join(models_dir, "checkpoints")], supported_pt_extensions) folder_names_and_paths["configs"] = ([os.path.join(models_dir, "configs")], [".yaml"])