saintzeno
/

stablediffusionwithdenoisingimages

Model card Files Files and versions Community

saintzeno commited on Sep 30, 2023

Commit

1e4ac47

•

1 Parent(s): 080bae6

Upload pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -181,7 +181,7 @@ class MyPipeline(StableDiffusionPipeline):
                     if image_saving_kwargs.get('save_denoising_images'):
                         image = self.decode_latents(latents)
                         image = self.numpy_to_pil(image)
-                        StableDiffusionPipelineOutput(images=image, nsfw_content_detected=None).images[0].save(image_saving_kwargs['save_denoising_path'] + '_t_' + str(i) + '.png')
                 ####################################################################################################
                 ####################################################################################################

                     if image_saving_kwargs.get('save_denoising_images'):
                         image = self.decode_latents(latents)
                         image = self.numpy_to_pil(image)
+                        StableDiffusionPipelineOutput(images=image, nsfw_content_detected=None).images[0].save(image_saving_kwargs['save_denoising_path'] + 'iteration_' + str(i) + '.png')
                 ####################################################################################################
                 ####################################################################################################