BlueSkyXN commited on
Commit
bb35b3a
·
verified ·
1 Parent(s): 8a28a8d

Deploy GitHub 36a6a04c8b?

Browse files
Files changed (1) hide show
  1. comfy_integration/setup.py +9 -0
comfy_integration/setup.py CHANGED
@@ -155,6 +155,15 @@ def initialize_comfyui() -> Path:
155
  (APP_DIR / INPUT_DIR).mkdir(parents=True, exist_ok=True)
156
  (APP_DIR / OUTPUT_DIR).mkdir(parents=True, exist_ok=True)
157
 
 
 
 
 
 
 
 
 
 
158
  import comfy.model_management # noqa: F401
159
 
160
  print(f"✅ ComfyUI initialized from isolated path: {comfyui_path}")
 
155
  (APP_DIR / INPUT_DIR).mkdir(parents=True, exist_ok=True)
156
  (APP_DIR / OUTPUT_DIR).mkdir(parents=True, exist_ok=True)
157
 
158
+ import folder_paths
159
+
160
+ for category, directory in CATEGORY_TO_DIR_MAP.items():
161
+ folder_paths.add_model_folder_path(
162
+ category, str(Path(directory).resolve()), is_default=True
163
+ )
164
+ folder_paths.set_input_directory(str(Path(INPUT_DIR).resolve()))
165
+ folder_paths.set_output_directory(str(Path(OUTPUT_DIR).resolve()))
166
+
167
  import comfy.model_management # noqa: F401
168
 
169
  print(f"✅ ComfyUI initialized from isolated path: {comfyui_path}")