From eff00413ae76e01b9a24b13dee6a0dda98f643f2 Mon Sep 17 00:00:00 2001
From: Sakura-Luna <53183413+Sakura-Luna@users.noreply.github.com>
Date: Thu, 20 Apr 2023 21:23:56 +0800
Subject: [PATCH] Refresh bug fix

---
 modules/img2img.py | 3 ++-
 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 = []