21 lines
507 B
Python
21 lines
507 B
Python
# 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")
|