wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-06-12 01:07:30 +08:00
Code Issues Actions 18 Packages Projects Releases Wiki Activity

Increase LongCat model detection precision.

Browse Source
This commit is contained in:
Talmaj Marinc 2026-02-23 22:37:07 +01:00
parent 6445abe8a0
commit 32c896ada2
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_detection.py
View File

@ -282,7 +282,7 @@ def detect_unet_config(state_dict, key_prefix, metadata=None):
return dit_config
if '{}x_embedder.weight'.format(key_prefix) in state_dict_keys and '{}transformer_blocks.0.attn.to_q.weight'.format(key_prefix) in state_dict_keys and '{}single_transformer_blocks.0.attn.to_q.weight'.format(key_prefix) in state_dict_keys: #LongCat-Image (diffusers format, Flux variant)
if '{}x_embedder.weight'.format(key_prefix) in state_dict_keys and '{}transformer_blocks.0.attn.to_q.weight'.format(key_prefix) in state_dict_keys and '{}single_transformer_blocks.0.attn.to_q.weight'.format(key_prefix) in state_dict_keys and '{}context_embedder.weight'.format(key_prefix) in state_dict_keys and '{}time_embed.timestep_embedder.linear_1.weight'.format(key_prefix) in state_dict_keys: #LongCat-Image (diffusers format, Flux variant)
dit_config = {}
dit_config["image_model"] = "flux"
dit_config["axes_dim"] = [16, 56, 56]

1
comfy/supported_models.py
View File

@ -1691,6 +1691,7 @@ class LongCatImage(supported_models_base.BASE):
"x_embedder.weight": None,
"context_embedder.weight": None,
"single_transformer_blocks.10.attn.to_q.weight": None,
"time_embed.timestep_embedder.linear_1.weight": None,
}
sampling_settings = {