From 85b6c91192a2e2fd42e2ce80c15e6dbb903c8bbb Mon Sep 17 00:00:00 2001
From: Will Miao <willmiao.dev@gmail.com>
Date: Fri, 10 Apr 2026 08:55:28 +0800
Subject: [PATCH] fix(download): add ZImageBase to diffusion model routing
 (#892)

---
 py/utils/constants.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/py/utils/constants.py b/py/utils/constants.py
index c8a44528..63afdcfd 100644
--- a/py/utils/constants.py
+++ b/py/utils/constants.py
@@ -101,6 +101,7 @@ DEFAULT_PRIORITY_TAG_CONFIG = {
 DIFFUSION_MODEL_BASE_MODELS = frozenset(
     [
         "ZImageTurbo",
+        "ZImageBase",
         "Wan Video 1.3B t2v",
         "Wan Video 14B t2v",
         "Wan Video 14B i2v 480p",