Add support for loading extra paths from yaml file.

Rename extra_model_paths.yaml.example to extra_model_paths.yaml and edit
it to point to your other UI.

Author: comfyanonymous

extra_model_paths.yaml.example

@@ -0,0 +1,23 @@
+#Rename this to extra_model_paths.yaml and ComfyUI will load it
+
+#config for a1111 ui
+#all you have to do is change the base_path to where yours is installed
+a111:
+    base_path: path/to/stable-diffusion-webui/
+
+    checkpoints: models/Stable-diffusion
+    configs: models/Stable-diffusion
+    vae: models/VAE
+    loras: models/Lora
+    upscale_models: |
+                  models/ESRGAN
+                  models/SwinIR
+    embeddings: embeddings
+    controlnet: models/ControlNet
+
+#other_ui:
+#    base_path: path/to/ui
+#    checkpoints: models/checkpoints
+
+
+

folder_paths.py

@@ -28,8 +28,10 @@
 folder_names_and_paths["upscale_models"] = ([os.path.join(models_dir, "upscale_models")], supported_pt_extensions)
 
 
-def add_model_folder(folder_name, full_folder_path):
+def add_model_folder_path(folder_name, full_folder_path):
     global folder_names_and_paths
+    if folder_name in folder_names_and_paths:
+        folder_names_and_paths[folder_name][0].append(full_folder_path)
 
 
 def recursive_search(directory):

main.py

@@ -33,6 +33,8 @@
 
 import execution
 import server
+import folder_paths
+import yaml
 
 def prompt_worker(q, server):
     e = execution.PromptExecutor(server)
@@ -59,6 +61,26 @@ def cleanup_temp():
     if os.path.exists(temp_dir):
         shutil.rmtree(temp_dir, ignore_errors=True)
 
+def load_extra_path_config(yaml_path):
+    with open(yaml_path, 'r') as stream:
+        config = yaml.safe_load(stream)
+    for c in config:
+        conf = config[c]
+        if conf is None:
+            continue
+        base_path = None
+        if "base_path" in conf:
+            base_path = conf.pop("base_path")
+        for x in conf:
+            for y in conf[x].split("\n"):
+                if len(y) == 0:
+                    continue
+                full_path = y
+                if base_path is not None:
+                    full_path = os.path.join(base_path, full_path)
+                print("Adding extra search path", x, full_path)
+                folder_paths.add_model_folder_path(x, full_path)
+
 if __name__ == "__main__":
     cleanup_temp()
 
@@ -79,6 +101,15 @@ def cleanup_temp():
     if '--dont-print-server' in sys.argv:
         dont_print = True
 
+    extra_model_paths_config_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "extra_model_paths.yaml")
+    if os.path.isfile(extra_model_paths_config_path):
+        load_extra_path_config(extra_model_paths_config_path)
+
+    if '--extra-model-paths-config' in sys.argv:
+        indices = [(i + 1) for i in range(len(sys.argv) - 1) if sys.argv[i] == '--extra-model-paths-config']
+        for i in indices:
+            load_extra_path_config(sys.argv[i])
+
     port = 8188
     try:
         p_index = sys.argv.index('--port')