Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
eb4cf57
1
Parent(s):
c724573
Update live_preview_helpers.py
Browse files- 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,
|
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()
|