[Diffusion] Support Multi-image Generation and Add Web UI Demo for QwenImage (#97)

Signed-off-by: samithuang <285365963@qq.com>
Signed-off-by: SamitHuang <285365963@qq.com>
This commit is contained in:
Samit authored and GitHub committed 2025-11-29 20:19:50 +08:00
1 parent 1ab88823d4
commit 1f1be495b0
7 files changed
+341 -22

No files matched your search

@@ -1,20 +0,0 @@
# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
import torch
from vllm_omni.entrypoints.omni import Omni
model_name = "Qwen/Qwen-Image"
prompt = "a cup of coffee on the table"
if __name__ == "__main__":
m = Omni(
model=model_name,
)
image = m.generate(
prompt,
height=1024,
width=1024,
generator=torch.Generator(device="cuda").manual_seed(42),
)
image[0].save("qwen_image_output.png")