Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -1564,9 +1564,9 @@ class AccDiffusionSDXLPipeline(DiffusionPipeline, FromSingleFileMixin, LoraLoade
|
|
| 1564 |
|
| 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 |
-
|
| 1570 |
else:
|
| 1571 |
image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False)[0]
|
| 1572 |
|
|
|
|
| 1564 |
|
| 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 |
|