diff options
| -rw-r--r-- | modules/ui_tempdir.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/modules/ui_tempdir.py b/modules/ui_tempdir.py index 621ed1ec..ecd6bdec 100644 --- a/modules/ui_tempdir.py +++ b/modules/ui_tempdir.py @@ -35,7 +35,9 @@ def save_pil_to_file(self, pil_image, dir=None, format="png"): already_saved_as = getattr(pil_image, 'already_saved_as', None)
if already_saved_as and os.path.isfile(already_saved_as):
register_tmp_file(shared.demo, already_saved_as)
- return f'{already_saved_as}?{os.path.getmtime(already_saved_as)}'
+ filename_with_mtime = f'{already_saved_as}?{os.path.getmtime(already_saved_as)}'
+ register_tmp_file(shared.demo, filename_with_mtime)
+ return filename_with_mtime
if shared.opts.temp_dir != "":
dir = shared.opts.temp_dir
|