Update src/pipeline.py

src/pipeline.py

@@ -49,8 +49,14 @@ def load_pipeline(pipeline=None) -> StableDiffusionXLPipeline:
         def step_and_preview(*args, **kwargs):
             global preview_images, preview_handle
             latents = get_pred_original_sample(sched, *args)
-            decoded_new = pipe.image_processor.postprocess(compress(latents.half()).mul_(2).sub_(1))[0]
-            decoded_original = pipe.image_processor.postprocess(pipe.vae(latents.float()).mul_(2).sub_(1))[0]
+            latents = latents.float()
+            output = compress(latents)
+            output = output.mul_(2).sub_(1)
+            decoded_new = pipe.image_processor.postprocess(output)[0]
+            latents = latents.half()
+            output = pipe.vae(latents)
+            output = output.mul_(2).sub_(1)
+            decoded_original = pipe.image_processor.postprocess(output)[0]
             preview_images_new.append(decoded_new)
             preview_images_original.append(decoded_original)
             return sched._step(*args, **kwargs)