fix resizing (#2)

kontext_pipeline.py

@@ -954,7 +954,8 @@ class FluxKontextPipeline(
             image_height = image_height // multiple_of * multiple_of
             image = self.image_processor.resize(image, image_height, image_width)
             image = self.image_processor.preprocess(image, image_height, image_width)
-
+            width,height = image_width,image_height #override width,height when input is resized to match aspect ratio bucket
+            
         # 4. Prepare latent variables
         num_channels_latents = self.transformer.config.in_channels // 4
         latents, image_latents, latent_ids, image_ids = self.prepare_latents(