fffiloni commited on
Commit
12262f6
·
verified ·
1 Parent(s): 41cdfe9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -1570,11 +1570,9 @@ class AccDiffusionSDXLPipeline(DiffusionPipeline, FromSingleFileMixin, LoraLoade
1570
  else:
1571
  image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False)[0]
1572
 
1573
- print(f"DEBUG: Image shape before postprocess: {image.shape}, dtype: {image.dtype}")
1574
  image = self.image_processor.postprocess(image, output_type=output_type)
1575
- print(f"DEBUG: Image shape after postprocess: {image.shape}, dtype: {image.dtype}")
1576
  image[0].save(f'{result_path}/AccDiffusion_{current_scale_num}.png')
1577
- print(f"DEBUG: Image saved at {result_path}/AccDiffusion_{current_scale_num}.png")
1578
  output_images.append(image[0])
1579
 
1580
  # cast back to fp16 if needed
 
1570
  else:
1571
  image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False)[0]
1572
 
1573
+
1574
  image = self.image_processor.postprocess(image, output_type=output_type)
 
1575
  image[0].save(f'{result_path}/AccDiffusion_{current_scale_num}.png')
 
1576
  output_images.append(image[0])
1577
 
1578
  # cast back to fp16 if needed