From 82973f9e9592a297ac014be4270cf2e071f93701 Mon Sep 17 00:00:00 2001 From: filipstrand Date: Tue, 29 Oct 2024 19:23:36 +0100 Subject: [PATCH] Fix bug: swap width <--> height --- src/mflux/latent_creator/latent_creator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/mflux/latent_creator/latent_creator.py b/src/mflux/latent_creator/latent_creator.py index 70332c9..296fa22 100644 --- a/src/mflux/latent_creator/latent_creator.py +++ b/src/mflux/latent_creator/latent_creator.py @@ -38,7 +38,7 @@ class LatentCreator: user_image = ImageUtil.load_image(runtime_conf.config.init_image_path).convert("RGB") scaled_user_image = ImageUtil.scale_to_dimensions(user_image, runtime_conf.width, runtime_conf.height) encoded = vae.encode(ImageUtil.to_array(scaled_user_image)) - latents = ArrayUtil.pack_latents(encoded, runtime_conf.width, runtime_conf.height) + latents = ArrayUtil.pack_latents(encoded, runtime_conf.height, runtime_conf.width) sigma = runtime_conf.sigmas[runtime_conf.init_time_step] return LatentCreator.add_noise_by_interpolation( clean=latents,