multimodalart HF staff commited on
Commit
eb4cf57
1 Parent(s): c724573

Update live_preview_helpers.py

Browse files
Files changed (1) hide show
  1. live_preview_helpers.py +1 -1
live_preview_helpers.py CHANGED
@@ -158,7 +158,7 @@ def flux_pipe_call_that_returns_an_iterable_of_images(
158
  torch.cuda.empty_cache()
159
 
160
  # Final image using good_vae
161
- latents = self._unpack_latents(latents, height, width, good_vae.config.vae_scale_factor)
162
  latents = (latents / good_vae.config.scaling_factor) + good_vae.config.shift_factor
163
  image = good_vae.decode(latents, return_dict=False)[0]
164
  self.maybe_free_model_hooks()
 
158
  torch.cuda.empty_cache()
159
 
160
  # Final image using good_vae
161
+ latents = self._unpack_latents(latents, height, width, self.vae_scale_factor)
162
  latents = (latents / good_vae.config.scaling_factor) + good_vae.config.shift_factor
163
  image = good_vae.decode(latents, return_dict=False)[0]
164
  self.maybe_free_model_hooks()