register_tmp_file also with mtime

This commit is contained in:
Andray 2024-02-23 20:26:56 +04:00
parent 18819723c1
commit 3a99824638

View File

@ -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