Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
|
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
|