Update LightningDit-XL-1-256/pipeline.py
Browse files
LightningDit-XL-1-256/pipeline.py
CHANGED
|
@@ -339,7 +339,6 @@ class LightningDiTPipeline(DiffusionPipeline):
|
|
| 339 |
|
| 340 |
image = self._decode_latents(latents)
|
| 341 |
if self.vae is not None:
|
| 342 |
-
image = (image / 2 + 0.5).clamp(0, 1)
|
| 343 |
image = self.image_processor.postprocess(image, output_type=output_type)
|
| 344 |
|
| 345 |
self.maybe_free_model_hooks()
|
|
|
|
| 339 |
|
| 340 |
image = self._decode_latents(latents)
|
| 341 |
if self.vae is not None:
|
|
|
|
| 342 |
image = self.image_processor.postprocess(image, output_type=output_type)
|
| 343 |
|
| 344 |
self.maybe_free_model_hooks()
|