Spaces:
Running
on
Zero
Running
on
Zero
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()
|