fffiloni commited on
Commit
1a71d6f
·
verified ·
1 Parent(s): 7a0b039

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1565,14 +1565,15 @@ class AccDiffusionSDXLPipeline(DiffusionPipeline, FromSingleFileMixin, LoraLoade
1565
  print("### Phase {} Decoding ###".format(current_scale_num))
1566
  if current_height > 2048 or current_width > 2048:
1567
  # image = self.tiled_decode(latents, current_height, current_width)
1568
- # self.enable_vae_tiling()
1569
  image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False)[0]
1570
  else:
1571
  image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False)[0]
1572
 
1573
  image = self.image_processor.postprocess(image, output_type=output_type)
 
1574
  image[0].save(f'{result_path}/AccDiffusion_{current_scale_num}.png')
1575
-
1576
  output_images.append(image[0])
1577
 
1578
  # cast back to fp16 if needed
 
1565
  print("### Phase {} Decoding ###".format(current_scale_num))
1566
  if current_height > 2048 or current_width > 2048:
1567
  # image = self.tiled_decode(latents, current_height, current_width)
1568
+ self.enable_vae_tiling()
1569
  image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False)[0]
1570
  else:
1571
  image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False)[0]
1572
 
1573
  image = self.image_processor.postprocess(image, output_type=output_type)
1574
+ print(f"DEBUG IMAGE: {image}")
1575
  image[0].save(f'{result_path}/AccDiffusion_{current_scale_num}.png')
1576
+ print(f"DEBUG IMAGE[Ø]: {image}")
1577
  output_images.append(image[0])
1578
 
1579
  # cast back to fp16 if needed