deallocate images on batch

This commit is contained in:
Vladimir Mandic
2023-08-12 07:18:19 +00:00
parent 605988f520
commit fc8eca1c34
3 changed files with 3 additions and 0 deletions
+1
View File
@@ -76,6 +76,7 @@ def run_postprocessing(extras_mode, image, image_folder: List[tempfile.NamedTemp
images.save_image(pp.image, path=outpath, basename=basename, seed=None, prompt=None, extension=ext or opts.samples_format, info=infotext, short_filename=True, no_prompt=True, grid=False, pnginfo_section_name="extras", existing_info=pp.image.info, forced_filename=None)
if extras_mode != 2 or show_extras_results:
outputs.append(pp.image)
image.close()
devices.torch_gc()
return outputs, infotext, params