Update lollms_sd.py

This commit is contained in:
Saifeddine ALOUI 2024-11-27 16:25:10 +01:00 committed by GitHub
parent 95646652b5
commit bac226f35d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -525,7 +525,7 @@ class LollmsSD(LollmsTTI):
if save_folder is None:
save_folder = self.output_dir
save_folder = Path(save_folder)
if not save_folder.is_dir():
if 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: