Update app.py

app.py

@@ -1570,12 +1570,13 @@ 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[0].dtype}")
+                    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 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')
+                    print(f"DEBUG: Image shape before postprocess: {image[0].shape}, dtype: {image[0].dtype}")
+                    output_images.append(image[0])
 
                     # cast back to fp16 if needed
                     if needs_upcasting: