Basic support for the ace step 1.5 model. (#12237)

This commit is contained in:
comfyanonymous
2026-02-02 21:06:18 -08:00
committed by GitHub
parent ba5bf3f1a8
commit 3c1a1a2df8
12 changed files with 1566 additions and 8 deletions

View File

@@ -655,6 +655,11 @@ def detect_unet_config(state_dict, key_prefix, metadata=None):
dit_config["num_visual_blocks"] = count_blocks(state_dict_keys, '{}visual_transformer_blocks.'.format(key_prefix) + '{}.')
return dit_config
if '{}encoder.lyric_encoder.layers.0.input_layernorm.weight'.format(key_prefix) in state_dict_keys:
dit_config = {}
dit_config["audio_model"] = "ace1.5"
return dit_config
if '{}input_blocks.0.0.weight'.format(key_prefix) not in state_dict_keys:
return None