From 16544bbec30cc77d02c14a556d13efdadd6b49b6 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Sun, 28 Sep 2025 16:23:23 +0530 Subject: [PATCH] revert pipeline changes. --- src/diffusers/pipelines/qwenimage/pipeline_qwenimage_edit.py | 1 + .../pipelines/qwenimage/pipeline_qwenimage_edit_plus.py | 1 - 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/qwenimage/pipeline_qwenimage_edit.py b/src/diffusers/pipelines/qwenimage/pipeline_qwenimage_edit.py index 686d143132..ed37b238c8 100644 --- a/src/diffusers/pipelines/qwenimage/pipeline_qwenimage_edit.py +++ b/src/diffusers/pipelines/qwenimage/pipeline_qwenimage_edit.py @@ -308,6 +308,7 @@ class QwenImageEditPipeline(DiffusionPipeline, QwenImageLoraLoaderMixin): prompt_embeds = prompt_embeds.view(batch_size * num_images_per_prompt, seq_len, -1) prompt_embeds_mask = prompt_embeds_mask.repeat(1, num_images_per_prompt, 1) prompt_embeds_mask = prompt_embeds_mask.view(batch_size * num_images_per_prompt, seq_len) + return prompt_embeds, prompt_embeds_mask def check_inputs( diff --git a/src/diffusers/pipelines/qwenimage/pipeline_qwenimage_edit_plus.py b/src/diffusers/pipelines/qwenimage/pipeline_qwenimage_edit_plus.py index dd132bb9f4..360998411a 100644 --- a/src/diffusers/pipelines/qwenimage/pipeline_qwenimage_edit_plus.py +++ b/src/diffusers/pipelines/qwenimage/pipeline_qwenimage_edit_plus.py @@ -309,7 +309,6 @@ class QwenImageEditPlusPipeline(DiffusionPipeline, QwenImageLoraLoaderMixin): Pre-generated text embeddings. Can be used to easily tweak text inputs, *e.g.* prompt weighting. If not provided, text embeddings will be generated from `prompt` input argument. """ - print(f"{image[0].size=}") device = device or self._execution_device prompt = [prompt] if isinstance(prompt, str) else prompt