Function calculate_layers

scripts/convert_aura_flow_to_diffusers.py:15–21

(state_dict_keys, key_prefix)

def calculate_layers(state_dict_keys, key_prefix):
    dit_layers = set()
    for k in state_dict_keys:
        if key_prefix in k:
            dit_layers.add(int(k.split(".")[2]))
    print(f"{key_prefix}: {len(dit_layers)}")
    return len(dit_layers)


# similar to SD3 but only for the last norm layer