From 1087735863547fd7fe1662d24760e133facc00fb Mon Sep 17 00:00:00 2001 From: HM-RunningHub Date: Fri, 20 Dec 2024 16:41:09 +0800 Subject: [PATCH] Update rh_utils.py --- rh_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rh_utils.py b/rh_utils.py index 73e4cd2..49721fa 100644 --- a/rh_utils.py +++ b/rh_utils.py @@ -48,11 +48,11 @@ def decode_latents(flux_dir, latents): torch_dtype=torch.bfloat16, ).to("cuda") - latents = pipeline._unpack_latents(latents, g_width, g_height, pipeline.vae_scale_factor) + latents = pipeline._unpack_latents(latents, g_height, g_width, pipeline.vae_scale_factor) latents = ( latents / pipeline.vae.config.scaling_factor ) + pipeline.vae.config.shift_factor image = pipeline.vae.decode(latents, return_dict=False)[0] image = pipeline.image_processor.postprocess(image, output_type="pil") - return FluxPipelineOutput(images=image) \ No newline at end of file + return FluxPipelineOutput(images=image)