comfyanonymous/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-26 11:09:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support zimage omni base model. (#11979)

Browse Source
This commit is contained in:
comfyanonymous
2026-01-19 20:17:38 -08:00
committed by GitHub
parent 9d273d3ab1
commit 2108167f9f
5 changed files with 381 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/model_detection.py
View File

@@ -446,6 +446,9 @@ def detect_unet_config(state_dict, key_prefix, metadata=None):
dit_config["time_scale"] = 1000.0
if '{}cap_pad_token'.format(key_prefix) in state_dict_keys:
dit_config["pad_tokens_multiple"] = 32
sig_weight = state_dict.get('{}siglip_embedder.0.weight'.format(key_prefix), None)
if sig_weight is not None:
dit_config["siglip_feat_dim"] = sig_weight.shape[0]
return dit_config