[Bugfix] Support both list and generator mode & fix unnecessary error log (#599)

Signed-off-by: Gaohan123 <hgaoaf@connect.ust.hk>
Signed-off-by: Hongsheng Liu <liuhongsheng4@huawei.com>
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
This commit is contained in:
Gao HanandHongsheng Liu authored and GitHub committed 2026-01-05 09:17:06 +08:00
1 parent cb7361f021
commit 6b18d31aaa
22 files changed
+134 -151

No files matched your search

@@ -19,7 +19,7 @@ if __name__ == "__main__":
omni = Omni(model="Qwen/Qwen-Image")
prompt = "a cup of coffee on the table"
outputs = omni.generate(prompt)
images = list(outputs)[0].request_output[0].images
images = outputs[0].request_output[0].images
images[0].save("coffee.png")
```