From 4d81da89e600748f2b3c0a7af0cfbac8153372fe Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Sun, 16 Jun 2024 01:55:43 +0200 Subject: [PATCH] Update lollms_comfyui.py --- lollms/services/comfyui/lollms_comfyui.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lollms/services/comfyui/lollms_comfyui.py b/lollms/services/comfyui/lollms_comfyui.py index c824f6e..69a05ed 100644 --- a/lollms/services/comfyui/lollms_comfyui.py +++ b/lollms/services/comfyui/lollms_comfyui.py @@ -395,7 +395,7 @@ class LollmsComfyUI(LollmsTTI): # Save each image to the folder for i, img_data in images.items(): - img_path = folder / f'image_{i+1}.png' + img_path = folder / f'image_{int(i)+1}.png' with open(img_path, 'wb') as img_file: img_file.write(base64.b64decode(img_data))