Spaces:
Runtime error
Runtime error
omit one yield
Browse files
app.py
CHANGED
|
@@ -165,7 +165,7 @@ def flux_pipe_call_that_returns_an_iterable_of_images(
|
|
| 165 |
latents_for_image = self._unpack_latents(latents, height, width, self.vae_scale_factor)
|
| 166 |
latents_for_image = (latents_for_image / self.vae.config.scaling_factor) + self.vae.config.shift_factor
|
| 167 |
image = self.vae.decode(latents_for_image, return_dict=False)[0]
|
| 168 |
-
yield self.image_processor.postprocess(image, output_type=output_type)[0]
|
| 169 |
|
| 170 |
latents = self.scheduler.step(noise_pred, t, latents, return_dict=False)[0]
|
| 171 |
torch.cuda.empty_cache()
|
|
|
|
| 165 |
latents_for_image = self._unpack_latents(latents, height, width, self.vae_scale_factor)
|
| 166 |
latents_for_image = (latents_for_image / self.vae.config.scaling_factor) + self.vae.config.shift_factor
|
| 167 |
image = self.vae.decode(latents_for_image, return_dict=False)[0]
|
| 168 |
+
# yield self.image_processor.postprocess(image, output_type=output_type)[0]
|
| 169 |
|
| 170 |
latents = self.scheduler.step(noise_pred, t, latents, return_dict=False)[0]
|
| 171 |
torch.cuda.empty_cache()
|