fix: images in subfolders are not working, fixes #12

blib-la · May 28, 2024 · 37480c2 · 37480c2
1 parent 92812b1
commit 37480c2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/rp_handler.py b/src/rp_handler.py
@@ -237,7 +237,7 @@ def process_output_images(outputs, job_id):
     for node_id, node_output in outputs.items():
         if "images" in node_output:
             for image in node_output["images"]:
-                output_images = image["filename"]
+                output_images = f"{image['subfolder']}/{image['filename']}"
 
     print(f"runpod-worker-comfy - image generation is done")
 

diff --git a/test_input.json b/test_input.json
@@ -53,7 +53,7 @@
       },
       "9": {
         "inputs": {
-          "filename_prefix": "ComfyUI",
+          "filename_prefix": "ComfyUI/test",
           "images": ["8", 0]
         },
         "class_type": "SaveImage"