Function count_num_layers

models/transformer/utils.py:156–161

()

    }

    def count_num_layers():
        layers = [
            k for k in state_dict.keys()
            if 'transformer_blocks' in k and 'attn1.norm_q.weight' in k
        ]
        return len(layers)

    num_layers = count_num_layers()
    inner_dim, in_channels, patch_size1, patch_size2 = state_dict[