ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-21 13:37:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update lollms_comfyui.py

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-06-16 01:59:18 +02:00 committed by GitHub
parent 4d81da89e6
commit d7c91fc394
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lollms/services/comfyui/lollms_comfyui.py
View File

@ -395,9 +395,9 @@ class LollmsComfyUI(LollmsTTI):
# Save each image to the folder
for i, img_data in images.items():
img_path = folder / f'image_{int(i)+1}.png'
img_path = folder / f'image_{i}.png'
with open(img_path, 'wb') as img_file:
img_file.write(base64.b64decode(img_data))
img_file.write(base64.b64decode(img_data[0]))
# Return the path to the first image
return str(folder / 'image_1.png')