Update kontext_pipeline.py

kontext_pipeline.py

@@ -954,6 +954,7 @@ 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
 
         # 4. Prepare latent variables
         num_channels_latents = self.transformer.config.in_channels // 4