Fix: Normalize batch inputs to 5D tensors for Qwen-Image-Edit

[akshan-main]

What does this PR do?

Enables robust batch inference for QwenImageEditPlusPipeline by normalizing input tensor shapes, implementing a "resize" strategy, and handling variable-length prompt embeddings. Adds a new batch inference capability that did not exist previously. Also accepts list, tuples as input now, which didn't exist previously. Enables QwenImageEditPlusPipeline to be used for parallel production workflows.

Description

Addresses issue #12458.

I identified four blockers preventing batch inference in the current pipeline:

1. 5D Tensor Requirement: The underlying VAE for Qwen2-VL model treats inputs as video (B, C, F, H, W) when batched. The pipeline was passing 4D tensors (B, C, H, W), causing immediate shape mismatches.

   • Fix: Added a pre-processing step to explicitly handle the frame dimension for static images when batch_size > 1.

2. Tensors require images in a batch to be of equal size

   • Fix: Implemented a Resize Strategy.
     • Single/Uniform Images: Preserves original aspect ratio and resolution (rounded to nearest 32).
     • Mixed Batches: Forces images to a standard resolution (e.g., 1024x1024) or User Defined height/width(first priority) to ensure tensor stackability without padding artifacts.[removed padding methodology from my previous commit and went ahead with upscaling/ resizing.

3. Tokenizer Batching Issues: The Qwen2VLProcessor produces variable-length embeddings for different prompts, which caused RuntimeError or IndexError when trying to batch encode them directly.

   • Fix: Refactored encode_prompt to process prompts individually in a loop, then pad the resulting embeddings up to the maximum sequence length in the batch before concatenating.

4. The pipeline would crash if users accidentally passed a tuple of images.

   • Fix: Added _sanitize_images which recursively unwraps inputs into a clean list.

Note on Batching Logic

To resolve the ambiguity between "Multi-Image Conditioning" and "Batch Inference", I implemented the following routing logic in encode_prompt:

1. Single String Prompt (prompt="string"):

   • Behavior: Joint Condition. The pipeline treats all provided images as a single context for one generation task.
   • Use Case: Style transfer or merging elements from multiple reference images.

2. List of Prompts (prompt=["s1", "s2"]):

   • Behavior: Parallel Batch. The pipeline maps images to prompts 1-to-1.
   • Use Case: Processing a dataset (e.g., editing 50 different images with 50 different instructions at once).

Fixes #12458

Before submitting

• This PR fixes a typo or improves the docs.
• Did you read the contributor guideline?
• Did you read our philosophy doc?
• Was this discussed/approved via a GitHub issue? (Issue [Qwen-image-edit] Batch Inference Issue / Feature Request #12458)
• Did you make sure to update the documentation with your changes?
• Did you write any new necessary tests? (Verified via reproduction script)

Who can review?

@yiyixuxu @sayakpaul @DN6

[akshan-main]

Hey @sayakpaul @yiyixuxu, let me know if I have to make any changes. But the functionality works as intended!