From 03e78bbe527d2013e510f97ed807f6f1e0e5f043 Mon Sep 17 00:00:00 2001
From: Sean Sheng <s3sheng@gmail.com>
Date: Tue, 22 Oct 2024 15:51:40 -0700
Subject: [PATCH] bug: handle subfolder missing in image output (#194)

---
 comfy_cli/command/run.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy_cli/command/run.py b/comfy_cli/command/run.py
index ed8ce6e3..78eb1e47 100644
--- a/comfy_cli/command/run.py
+++ b/comfy_cli/command/run.py
@@ -190,7 +190,7 @@ def log_node(self, type, node_id):
 
     def format_image_path(self, img):
         filename = img["filename"]
-        subfolder = img["subfolder"]
+        subfolder = img["subfolder"] if "subfolder" in img else None
         output_type = img["type"] or "output"
 
         if self.local_paths: