[CI] qwen image edit L4 accuracy test (#2761)

This commit is contained in:
Zeyu Huang | 黃澤宇 authored and GitHub committed 2026-04-15 11:13:11 +08:00
1 parent 227bab3038
commit 0d020739a7
6 files changed
+377 -5

No files matched your search

+25
View File
@@ -5,10 +5,13 @@ import shutil
import subprocess
from contextlib import contextmanager
from dataclasses import dataclass
from io import BytesIO
from pathlib import Path
import pytest
import requests
import torch
from PIL import Image
from tests.conftest import OmniServer, OmniServerParams
@@ -183,6 +186,28 @@ def accuracy_artifact_root() -> Path:
return root
@pytest.fixture(scope="session")
def qwen_bear_image(accuracy_artifact_root: Path) -> Image.Image:
"""Download the Qwen bear image from the URL and save it to the accuracy artifact root."""
QWEN_BEAR_IMAGE_URL = "https://vllm-public-assets.s3.us-west-2.amazonaws.com/omni-assets/qwen-bear.png"
response = requests.get(QWEN_BEAR_IMAGE_URL, timeout=60)
response.raise_for_status()
image = Image.open(BytesIO(response.content)).convert("RGB")
image.save(accuracy_artifact_root / "qwen_bear.png")
return image
@pytest.fixture(scope="session")
def rabbit_image(accuracy_artifact_root: Path) -> Image.Image:
"""Download the rabbit image from the URL and save it to the accuracy artifact root."""
RABBIT_IMAGE_URL = "https://vllm-public-assets.s3.us-west-2.amazonaws.com/omni-assets/rabbit.png"
response = requests.get(RABBIT_IMAGE_URL, timeout=60)
response.raise_for_status()
image = Image.open(BytesIO(response.content)).convert("RGB")
image.save(accuracy_artifact_root / "rabbit.png")
return image
def reset_artifact_dir(path: Path) -> Path:
if path.exists():
shutil.rmtree(path)