Update lollms_comfyui.py

This commit is contained in:
Saifeddine ALOUI 2024-06-16 01:55:43 +02:00 committed by GitHub
parent 4ff4cc0ea8
commit 4d81da89e6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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))