From 7421db3fc4287768343406e02ae362a1a6bad3ec Mon Sep 17 00:00:00 2001
From: unknown <348063288@qq.com>
Date: Fri, 8 Sep 2023 12:16:57 +0800
Subject: [PATCH] fix #362

---
 typescripts/controlnet/entry.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/typescripts/controlnet/entry.ts b/typescripts/controlnet/entry.ts
index 53e62a7..f5ce6ca 100644
--- a/typescripts/controlnet/entry.ts
+++ b/typescripts/controlnet/entry.ts
@@ -33,7 +33,7 @@ async function requestControlNetApiVersion() {
 async function requestControlNetMaxUnits() {
     const json = await api.requestGet(`${g_sd_url}/controlnet/settings`)
 
-    const control_net_max_models_num = json?.control_net_max_models_num ?? 0
+    const control_net_max_models_num = (json?.control_net_unit_count || json?.control_net_max_models_num) ?? 0
 
     return control_net_max_models_num
 }