From 7400a93027b2b8d3c59cf64f84aec53dfe2be64a Mon Sep 17 00:00:00 2001 From: Jordan Thompson Date: Fri, 31 Mar 2023 12:28:49 -0700 Subject: [PATCH] Add main branch Add main branch directory location in case of downloading via ZIP --- WAS_Node_Suite.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/WAS_Node_Suite.py b/WAS_Node_Suite.py index 67db4bf..92a2edf 100644 --- a/WAS_Node_Suite.py +++ b/WAS_Node_Suite.py @@ -42,7 +42,10 @@ sys.path.append('..'+os.sep+'ComfyUI') # GLOBALS NODE_FILE = os.path.abspath(__file__) MIDAS_INSTALLED = False -CUSTOM_NODES_DIR = os.path.dirname(os.path.dirname(NODE_FILE)) if os.path.dirname(os.path.dirname(NODE_FILE)) == 'was-node-suite-comfyui' else os.path.dirname(NODE_FILE) +CUSTOM_NODES_DIR = ( os.path.dirname(os.path.dirname(NODE_FILE)) + if os.path.dirname(os.path.dirname(NODE_FILE)) == 'was-node-suite-comfyui' + or os.path.dirname(os.path.dirname(NODE_FILE)) == 'was-node-suite-comfyui-main' + else os.path.dirname(NODE_FILE) ) WAS_SUITE_ROOT = os.path.dirname(NODE_FILE) WAS_DATABASE = os.path.join(WAS_SUITE_ROOT, 'was_suite_settings.json')