Update app.py

app.py

@@ -1570,10 +1570,11 @@ class AccDiffusionSDXLPipeline(DiffusionPipeline, FromSingleFileMixin, LoraLoade
                     else:
                         image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False)[0]
 
+                    print(f"DEBUG: Image shape before postprocess: {image.shape}, dtype: {image.dtype}")
                     image = self.image_processor.postprocess(image, output_type=output_type)
-                    print(f"DEBUG IMAGE: {image}")
+                    print(f"DEBUG: Image shape after postprocess: {image.shape}, dtype: {image.dtype}")
                     image[0].save(f'{result_path}/AccDiffusion_{current_scale_num}.png')
-                    print(f"DEBUG IMAGE[0]: {image[0]}")
+                    print(f"DEBUG: Image saved at {result_path}/AccDiffusion_{current_scale_num}.png")
                     output_images.append(f'{result_path}/AccDiffusion_{current_scale_num}.png')
 
                     # cast back to fp16 if needed