From 1f69d3ec0b95ff6512b5976752f33fc4c56a4fbe Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Thu, 9 Jan 2025 18:19:50 -0800
Subject: [PATCH] Fix directory creation

---
 comfy/cmd/main.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/comfy/cmd/main.py b/comfy/cmd/main.py
index 5ec17b021..50520dbf4 100644
--- a/comfy/cmd/main.py
+++ b/comfy/cmd/main.py
@@ -150,6 +150,10 @@ async def main(from_script_dir: Optional[Path] = None):
 
     # always create directories when started interactively
     folder_paths.create_directories()
+    if args.create_directories:
+        import_all_nodes_in_workspace(raise_on_failure=False)
+        folder_paths.create_directories()
+        exit(0)
 
     if args.windows_standalone_build:
         folder_paths.create_directories()