From 0bbd61e1b9d91b863c4f5ff636341bbde0e8a512 Mon Sep 17 00:00:00 2001
From: Sean Sheng <s3sheng@gmail.com>
Date: Tue, 8 Oct 2024 21:03:29 -0700
Subject: [PATCH] bug: handle subfolder missing in image output

---
 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 ed8ce6e..78eb1e4 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: