Commit bd833409 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

additional changes for saving pnginfo for #1803

parent 0609ce06

modules/extras.py

+4 −0
Original line number Diff line number Diff line
@@ -98,6 +98,10 @@ def run_extras(extras_mode, image, image_folder, gfpgan_visibility, codeformer_v 
                          no_prompt=True, grid=False, pnginfo_section_name="extras", existing_info=existing_pnginfo,

                          forced_filename=image_name if opts.use_original_name_batch else None)



        if opts.enable_pnginfo:

            image.info = existing_pnginfo

            image.info["extras"] = info



        outputs.append(image)



    devices.torch_gc()

modules/processing.py

+4 −2
Original line number Diff line number Diff line
@@ -451,6 +451,7 @@ def process_images(p: StableDiffusionProcessing) -> Processed: 


                text = infotext(n, i)

                infotexts.append(text)

                if opts.enable_pnginfo:

                    image.info["parameters"] = text

                output_images.append(image)
@@ -470,6 +471,7 @@ def process_images(p: StableDiffusionProcessing) -> Processed: 
            if opts.return_grid:

                text = infotext()

                infotexts.insert(0, text)

                if opts.enable_pnginfo:

                    grid.info["parameters"] = text

                output_images.insert(0, grid)

                index_of_first_image = 1

CRA Git | Maintained and supported by SUSTech CRA and CCSE