From 17b70728ec86aa571497fbb18b22201d7b620705 Mon Sep 17 00:00:00 2001
From: filtered <176114999+webfiltered@users.noreply.github.com>
Date: Thu, 23 Jan 2025 18:09:15 +1100
Subject: [PATCH] Allow override of models base path via env var

---
 folder_paths.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/folder_paths.py b/folder_paths.py
index 3542d2ed..6a297c5c 100644
--- a/folder_paths.py
+++ b/folder_paths.py
@@ -11,7 +11,8 @@ supported_pt_extensions: set[str] = {'.ckpt', '.pt', '.bin', '.pth', '.safetenso
 
 folder_names_and_paths: dict[str, tuple[list[str], set[str]]] = {}
 
-base_path = os.path.dirname(os.path.realpath(__file__))
+env_base_path = os.environ.get("COMFYUI_FOLDERS_BASE_PATH")
+base_path = os.path.dirname(os.path.realpath(__file__)) if env_base_path is None else env_base_path
 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"])