anodyine/vllm-omni
1
0 Fork 0
Code Issues 4 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

[New model] Qwen-Image-2.1 support (rebase + review follow-ups) (#8099)

Browse Source 
Signed-off-by: NumberWan <wantszkin2003@gmail.com>
Signed-off-by: SamitHuang <285365963@qq.com>
Signed-off-by: Samit <285365963@qq.com>
Co-authored-by: Isotr0py <Isotr0py@outlook.com>
Co-authored-by: Tianyu Guo <guoty9@mail2.sysu.edu.cn>
Co-authored-by: Cursor <cursoragent@cursor.com>
Co-authored-by: SamitHuang <285365963@qq.com>
Co-authored-by: Gao Han <hgaoaf@connect.ust.hk>
This commit is contained in:
NumberWanIsotr0pyTianyu GuoCursorSamitHuangGao Han
 authored and GitHub committed 2026-10-07 21:23:59 +08:00
1 parent 8ef77be66e
commit 3dc35694b3
52 files changed
+8058 -136
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

tests/entrypoints/openai_api/test_image_server.py
+9 -9
View File
@@ -933,7 +933,7 @@ def test_image_edits_streaming_returns_ar_delta_then_image(streaming_image_edit_
assert payloads[0]["index"] == 0
assert payloads[1]["delta"] == " done"
assert payloads[2]["output_format"] == "png"
assert payloads[2]["size"] == "16x16"
assert payloads[2]["size"] == "32x24"
image_payload = payloads[2]["data"][0]
img = Image.open(io.BytesIO(base64.b64decode(image_payload["b64_json"])))
@@ -2290,8 +2290,8 @@ def test_extract_images_from_result():
class EmptyResult:
pass
result = EmptyResult()
images = _extract_images_from_result(result)
empty_result = EmptyResult()
images = _extract_images_from_result(empty_result)
assert images == []
# Test nested batch: [np.array(shape=(3, 64, 64, 3))]
@@ -2301,8 +2301,8 @@ def test_extract_images_from_result():
def __init__(self):
self.images = [batch]
result = BatchResult()
images = _extract_images_from_result(result)
batch_result = BatchResult()
images = _extract_images_from_result(batch_result)
assert len(images) == 3
assert all(isinstance(img, Image.Image) for img in images)
assert all(img.size == (64, 64) for img in images)
@@ -2312,8 +2312,8 @@ def test_extract_images_from_result():
def __init__(self):
self.images = [np.random.randint(0, 255, (64, 64, 3), dtype=np.uint8)]
result = DictRequestOutput()
images = _extract_images_from_result(result)
dict_result = DictRequestOutput()
images = _extract_images_from_result(dict_result)
assert len(images) == 1
assert isinstance(images[0], Image.Image)
@@ -2322,8 +2322,8 @@ def test_extract_images_from_result():
def __init__(self):
self.images = [np.random.randint(0, 255, (32, 32, 3), dtype=np.uint8)]
result = AttrRequestOutput()
images = _extract_images_from_result(result)
attr_result = AttrRequestOutput()
images = _extract_images_from_result(attr_result)
assert len(images) == 1
assert isinstance(images[0], Image.Image)
assert images[0].size == (32, 32)