Spaces:
Running on Zero
Running on Zero
Deploy GitHub 36a6a04c8b?
Browse files
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}")
|