Merge pull request #15012 from light-and-ray/register_tmp_file-also-with-mtime

register_tmp_file also for mtime

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