Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

cleanup model types

Browse Source 
Signed-off-by: vladmandic <mandic00@live.com>
This commit is contained in:
vladmandic
2026-01-15 08:30:48 +01:00
parent 387b3c7c36
commit 71d3482168
6 changed files with 25 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/control/units/controlnet.py
+3 -3
View File
@@ -181,7 +181,7 @@ def api_list_models(model_type: str = None):
model_list += list(predefined_qwen)
if model_type == 'hunyuandit' or model_type == 'all':
model_list += list(predefined_hunyuandit)
if model_type == 'z_image':
if model_type == 'zimage':
model_list += list(predefined_zimage)
model_list += sorted(find_models())
return model_list
@@ -207,7 +207,7 @@ def list_models(refresh=False):
models = ['None'] + list(predefined_qwen) + sorted(find_models())
elif modules.shared.sd_model_type == 'hunyuandit':
models = ['None'] + list(predefined_hunyuandit) + sorted(find_models())
elif modules.shared.sd_model_type == 'z_image':
elif modules.shared.sd_model_type == 'zimage':
models = ['None'] + list(predefined_zimage) + sorted(find_models())
else:
log.warning(f'Control {what} model list failed: unknown model type')
@@ -273,7 +273,7 @@ class ControlNet():
elif shared.sd_model_type == 'hunyuandit':
from diffusers import HunyuanDiT2DControlNetModel as cls
config = 'Tencent-Hunyuan/HunyuanDiT-v1.2-ControlNet-Diffusers-Canny'
elif shared.sd_model_type == 'z_image':
elif shared.sd_model_type == 'zimage':
from diffusers import ZImageControlNetModel as cls
if '2.0' in model_id:
config = 'hlky/Z-Image-Turbo-Fun-Controlnet-Union-2.0'