From 4ff4cc0ea85702f21c94430f061a9c7910a8a55c Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Sun, 16 Jun 2024 01:53:22 +0200 Subject: [PATCH] Update lollms_comfyui.py --- lollms/services/comfyui/lollms_comfyui.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lollms/services/comfyui/lollms_comfyui.py b/lollms/services/comfyui/lollms_comfyui.py index 6ce976d..c824f6e 100644 --- a/lollms/services/comfyui/lollms_comfyui.py +++ b/lollms/services/comfyui/lollms_comfyui.py @@ -388,13 +388,13 @@ class LollmsComfyUI(LollmsTTI): } } """ - def save_images(images, folder_path): + def save_images(images:dict, folder_path:str|Path): # Create the folder if it doesn't exist folder = Path(folder_path) folder.mkdir(parents=True, exist_ok=True) # Save each image to the folder - for i, img_data in enumerate(images): + for i, img_data in images.items(): img_path = folder / f'image_{i+1}.png' with open(img_path, 'wb') as img_file: img_file.write(base64.b64decode(img_data))