[Misc] fix qwen image family redundant computation (#868)

Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>
This commit is contained in:
Jiangyun Zhu authored and GitHub committed 2026-01-21 09:58:08 +00:00
1 parent 0aa72b952f
commit 58efc5dde3
5 files changed
+16 -14

No files matched your search

@@ -110,7 +110,7 @@ def parse_args() -> argparse.Namespace:
parser.add_argument(
"--negative_prompt",
type=str,
default=" ",
default=None,
required=False,
)
parser.add_argument(
@@ -24,7 +24,9 @@ def parse_args() -> argparse.Namespace:
)
parser.add_argument("--prompt", default="a cup of coffee on the table", help="Text prompt for image generation.")
parser.add_argument(
"--negative_prompt", default="", help="negative prompt for classifier-free conditional guidance."
"--negative_prompt",
default=None,
help="negative prompt for classifier-free conditional guidance.",
)
parser.add_argument("--seed", type=int, default=142, help="Random seed for deterministic results.")
parser.add_argument(
@@ -642,8 +642,8 @@ class QwenImagePipeline(
def forward(
self,
req: OmniDiffusionRequest,
prompt: str | list[str] = "",
negative_prompt: str | list[str] = "",
prompt: str | list[str] | None = None,
negative_prompt: str | list[str] | None = None,
true_cfg_scale: float = 4.0,
height: int | None = None,
width: int | None = None,
@@ -665,8 +665,8 @@ class QwenImagePipeline(
# # TODO: only support single prompt now
# if req.prompt is not None:
# prompt = req.prompt[0] if isinstance(req.prompt, list) else req.prompt
prompt = req.prompt if req.prompt is not None else prompt
negative_prompt = req.negative_prompt if req.negative_prompt is not None else negative_prompt
prompt = req.prompt
negative_prompt = req.negative_prompt
height = req.height or self.default_sample_size * self.vae_scale_factor
width = req.width or self.default_sample_size * self.vae_scale_factor
num_inference_steps = req.num_inference_steps or num_inference_steps
@@ -695,8 +695,8 @@ class QwenImageEditPipeline(
def forward(
self,
req: OmniDiffusionRequest,
prompt: str | list[str] = "",
negative_prompt: str | list[str] = "",
prompt: str | list[str] | None = None,
negative_prompt: str | list[str] | None = None,
image: PIL.Image.Image | torch.Tensor | None = None,
true_cfg_scale: float = 4.0,
height: int | None = None,
@@ -717,8 +717,8 @@ class QwenImageEditPipeline(
max_sequence_length: int = 512,
) -> DiffusionOutput:
"""Forward pass for image editing."""
prompt = req.prompt if req.prompt is not None else prompt
negative_prompt = req.negative_prompt if req.negative_prompt is not None else negative_prompt
prompt = req.prompt
negative_prompt = req.negative_prompt
# Get preprocessed image from request (pre-processing is done in DiffusionEngine)
if hasattr(req, "preprocessed_image"):
@@ -633,8 +633,8 @@ class QwenImageEditPlusPipeline(nn.Module, SupportImageInput):
def forward(
self,
req: OmniDiffusionRequest,
prompt: str | list[str] = "",
negative_prompt: str | list[str] = "",
prompt: str | list[str] | None = None,
negative_prompt: str | list[str] | None = None,
image: PIL.Image.Image | list[PIL.Image.Image] | torch.Tensor | None = None,
true_cfg_scale: float = 4.0,
height: int | None = None,
@@ -655,8 +655,8 @@ class QwenImageEditPlusPipeline(nn.Module, SupportImageInput):
max_sequence_length: int = 512,
) -> DiffusionOutput:
"""Forward pass for image editing with support for multiple images."""
prompt = req.prompt if req.prompt is not None else prompt
negative_prompt = req.negative_prompt if req.negative_prompt is not None else negative_prompt
prompt = req.prompt
negative_prompt = req.negative_prompt
# Get preprocessed images from request (pre-processing is done in DiffusionEngine)
if hasattr(req, "vae_images") and hasattr(req, "condition_images"):