[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:
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")
|
||||
Reference in new issue
Block a user