diff options
-rw-r--r-- | modules/img2img.py | 3 | ||||
-rw-r--r-- | modules/txt2img.py | 3 |
2 files changed, 4 insertions, 2 deletions
diff --git a/modules/img2img.py b/modules/img2img.py index 6da974e4..fea51667 100644 --- a/modules/img2img.py +++ b/modules/img2img.py @@ -181,7 +181,8 @@ def img2img(id_task: str, mode: int, prompt: str, negative_prompt: str, prompt_s print(generation_info_js)
for img in processed.images:
- img.already_saved_as += f'?{int(time())}'
+ if hasattr(img, 'already_saved_as'):
+ img.already_saved_as += f'?{int(time())}'
if opts.do_not_show_images:
processed.images = []
diff --git a/modules/txt2img.py b/modules/txt2img.py index 6ebca656..57278bdc 100644 --- a/modules/txt2img.py +++ b/modules/txt2img.py @@ -65,7 +65,8 @@ def txt2img(id_task: str, prompt: str, negative_prompt: str, prompt_styles, step print(generation_info_js)
for img in processed.images:
- img.already_saved_as += f'?{int(time())}'
+ if hasattr(img, 'already_saved_as'):
+ img.already_saved_as += f'?{int(time())}'
if opts.do_not_show_images:
processed.images = []
|