daoyuan98 commited on
Commit
c8690a6
1 Parent(s): 2f7ddaa

omit one yield

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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()