diff --git a/app/main.py b/app/main.py index d5d27c5..70b77ef 100644 --- a/app/main.py +++ b/app/main.py @@ -2017,9 +2017,14 @@ async def list_models(request: Request): mid = (m.get("id") or "").lower() return any(h in mid for h in embedding_name_hints) + image_name_hints = ("qwen-image",) + def is_image_model(m) -> bool: mode = ((m.get("info") or {}).get("meta") or {}).get("mode") or "" - return mode in image_modes + if mode in image_modes: + return True + mid = (m.get("id") or "").lower() + return any(h in mid for h in image_name_hints) payload["data"] = [ m for m in payload.get("data", [])