Update app.py

app.py

@@ -1570,9 +1570,9 @@ 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[0].shape}, dtype: {image.dtype}")
+                    print(f"DEBUG: Image shape before postprocess: {image[0].shape}, dtype: {image[0].dtype}")
                     image = self.image_processor.postprocess(image, output_type=output_type)
-                    print(f"DEBUG: Image shape after postprocess: {image[0].shape}, dtype: {image.dtype}")
+                    print(f"DEBUG: Image shape after postprocess: {image[0].shape}, dtype: {image[0].dtype}")
                     image[0].save(f'{result_path}/AccDiffusion_{current_scale_num}.png')
                     print(f"DEBUG: Image saved at {result_path}/AccDiffusion_{current_scale_num}.png")
                     output_images.append(f'{result_path}/AccDiffusion_{current_scale_num}.png')