fix: do not attempt to remove non-existing image grid file (#2456)

image grid is actually not an image here but a numpy array, as the grid isn't saved by default
This commit is contained in:
Manuel Schmid 2024-03-05 21:16:21 +01:00 committed by GitHub
parent 6cfcc62000
commit 3a64fe3eb3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -76,7 +76,8 @@ def generate_clicked(task):
# delete Fooocus temp images, only keep gradio temp images
if args_manager.args.disable_image_log:
for filepath in product:
os.remove(filepath)
if isinstance(filepath, str) and os.path.exists(filepath):
os.remove(filepath)
execution_time = time.perf_counter() - execution_start_time
print(f'Total time: {execution_time:.2f} seconds')