ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-18 20:27:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update lollms_sd.py

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-11-27 16:23:25 +01:00 committed by GitHub
parent 372a474cf6
commit 95646652b5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lollms/services/tti/sd/lollms_sd.py
View File

@ -524,8 +524,12 @@ class LollmsSD(LollmsTTI):
def saveImage(self, image:Image, save_folder=None):
if save_folder is None:
save_folder = self.output_dir
image_name = self.get_available_image_name(save_folder, self.wm)
image_path = os.path.join(save_folder, image_name)
save_folder = Path(save_folder)
if not save_folder.is_dir():
image_name = self.get_available_image_name(save_folder, self.wm)
image_path = os.path.join(save_folder, image_name)
else:
image_path = save_folder
image.save(image_path)
return image_path