model.weights.json

{
    "metadata": {
        "total_size": 15637034144.0
    },
    "weight_map": {
        "/audio_tower/conformer/gemma3n_audio_conformer_block/attention/attn/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/attention/attn/k_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/attention/attn/q_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/attention/attn/relative_position_embedding/pos_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/attention/attn/v_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/attention/post/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/attention/post_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/attention/pre_attn_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/ffw_layer_end/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/ffw_layer_end/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/ffw_layer_end/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/ffw_layer_end/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/ffw_layer_start/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/ffw_layer_start/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/ffw_layer_start/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/ffw_layer_start/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/lconv1d/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/lconv1d/conv_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/lconv1d/linear_end/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/lconv1d/linear_start/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/lconv1d/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block/norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/attention/attn/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/attention/attn/k_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/attention/attn/q_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/attention/attn/relative_position_embedding/pos_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/attention/attn/v_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/attention/post/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/attention/post_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/attention/pre_attn_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/ffw_layer_end/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/ffw_layer_end/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/ffw_layer_end/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/ffw_layer_end/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/ffw_layer_start/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/ffw_layer_start/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/ffw_layer_start/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/ffw_layer_start/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/lconv1d/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/lconv1d/conv_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/lconv1d/linear_end/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/lconv1d/linear_start/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/lconv1d/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_1/norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/attention/attn/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/attention/attn/k_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/attention/attn/q_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/attention/attn/relative_position_embedding/pos_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/attention/attn/v_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/attention/post/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/attention/post_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/attention/pre_attn_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/ffw_layer_end/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/ffw_layer_end/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/ffw_layer_end/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/ffw_layer_end/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/ffw_layer_start/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/ffw_layer_start/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/ffw_layer_start/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/ffw_layer_start/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/lconv1d/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/lconv1d/conv_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/lconv1d/linear_end/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/lconv1d/linear_start/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/lconv1d/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_2/norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/attention/attn/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/attention/attn/k_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/attention/attn/q_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/attention/attn/relative_position_embedding/pos_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/attention/attn/v_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/attention/post/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/attention/post_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/attention/pre_attn_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/ffw_layer_end/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/ffw_layer_end/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/ffw_layer_end/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/ffw_layer_end/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/ffw_layer_start/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/ffw_layer_start/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/ffw_layer_start/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/ffw_layer_start/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/lconv1d/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/lconv1d/conv_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/lconv1d/linear_end/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/lconv1d/linear_start/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/lconv1d/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_3/norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/attention/attn/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/attention/attn/k_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/attention/attn/q_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/attention/attn/relative_position_embedding/pos_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/attention/attn/v_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/attention/post/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/attention/post_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/attention/pre_attn_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/ffw_layer_end/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/ffw_layer_end/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/ffw_layer_end/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/ffw_layer_end/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/ffw_layer_start/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/ffw_layer_start/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/ffw_layer_start/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/ffw_layer_start/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/lconv1d/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/lconv1d/conv_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/lconv1d/linear_end/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/lconv1d/linear_start/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/lconv1d/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_4/norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/attention/attn/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/attention/attn/k_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/attention/attn/q_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/attention/attn/relative_position_embedding/pos_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/attention/attn/v_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/attention/post/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/attention/post_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/attention/pre_attn_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/ffw_layer_end/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/ffw_layer_end/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/ffw_layer_end/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/ffw_layer_end/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/ffw_layer_start/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/ffw_layer_start/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/ffw_layer_start/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/ffw_layer_start/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/lconv1d/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/lconv1d/conv_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/lconv1d/linear_end/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/lconv1d/linear_start/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/lconv1d/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_5/norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/attention/attn/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/attention/attn/k_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/attention/attn/q_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/attention/attn/relative_position_embedding/pos_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/attention/attn/v_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/attention/post/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/attention/post_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/attention/pre_attn_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/ffw_layer_end/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/ffw_layer_end/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/ffw_layer_end/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/ffw_layer_end/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/ffw_layer_start/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/ffw_layer_start/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/ffw_layer_start/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/ffw_layer_start/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/lconv1d/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/lconv1d/conv_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/lconv1d/linear_end/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/lconv1d/linear_start/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/lconv1d/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_6/norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/attention/attn/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/attention/attn/k_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/attention/attn/q_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/attention/attn/relative_position_embedding/pos_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/attention/attn/v_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/attention/post/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/attention/post_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/attention/pre_attn_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/ffw_layer_end/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/ffw_layer_end/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/ffw_layer_end/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/ffw_layer_end/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/ffw_layer_start/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/ffw_layer_start/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/ffw_layer_start/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/ffw_layer_start/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/lconv1d/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/lconv1d/conv_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/lconv1d/linear_end/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/lconv1d/linear_start/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/lconv1d/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_7/norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/attention/attn/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/attention/attn/k_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/attention/attn/q_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/attention/attn/relative_position_embedding/pos_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/attention/attn/v_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/attention/post/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/attention/post_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/attention/pre_attn_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/ffw_layer_end/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/ffw_layer_end/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/ffw_layer_end/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/ffw_layer_end/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/ffw_layer_start/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/ffw_layer_start/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/ffw_layer_start/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/ffw_layer_start/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/lconv1d/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/lconv1d/conv_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/lconv1d/linear_end/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/lconv1d/linear_start/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/lconv1d/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_8/norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/attention/attn/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/attention/attn/k_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/attention/attn/q_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/attention/attn/relative_position_embedding/pos_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/attention/attn/v_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/attention/post/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/attention/post_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/attention/pre_attn_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/ffw_layer_end/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/ffw_layer_end/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/ffw_layer_end/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/ffw_layer_end/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/ffw_layer_start/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/ffw_layer_start/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/ffw_layer_start/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/ffw_layer_start/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/lconv1d/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/lconv1d/conv_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/lconv1d/linear_end/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/lconv1d/linear_start/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/lconv1d/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_9/norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/attention/attn/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/attention/attn/k_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/attention/attn/q_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/attention/attn/relative_position_embedding/pos_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/attention/attn/v_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/attention/post/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/attention/post_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/attention/pre_attn_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/ffw_layer_end/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/ffw_layer_end/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/ffw_layer_end/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/ffw_layer_end/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/ffw_layer_start/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/ffw_layer_start/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/ffw_layer_start/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/ffw_layer_start/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/lconv1d/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/lconv1d/conv_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/lconv1d/linear_end/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/lconv1d/linear_start/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/lconv1d/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_10/norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/attention/attn/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/attention/attn/k_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/attention/attn/q_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/attention/attn/relative_position_embedding/pos_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/attention/attn/v_proj/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/attention/post/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/attention/post_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/attention/pre_attn_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/ffw_layer_end/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/ffw_layer_end/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/ffw_layer_end/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/ffw_layer_end/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/ffw_layer_start/ffw_layer_1/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/ffw_layer_start/ffw_layer_2/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/ffw_layer_start/post_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/ffw_layer_start/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/lconv1d/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/lconv1d/conv_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/lconv1d/linear_end/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/lconv1d/linear_start/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/lconv1d/pre_layer_norm/vars": "model_00000.weights.h5",
        "/audio_tower/conformer/gemma3n_audio_conformer_block_11/norm/vars": "model_00000.weights.h5",
        "/audio_tower/subsample_conv_projection/conv_0/conv/vars": "model_00000.weights.h5",
        "/audio_tower/subsample_conv_projection/conv_0/norm/vars": "model_00000.weights.h5",
        "/audio_tower/subsample_conv_projection/conv_1/conv/vars": "model_00000.weights.h5",
        "/audio_tower/subsample_conv_projection/conv_1/norm/vars": "model_00000.weights.h5",
        "/audio_tower/subsample_conv_projection/input_proj_linear/vars": "model_00000.weights.h5",
        "/embed_audio/embedding/vars": "model_00000.weights.h5",
        "/embed_audio/embedding_projection/vars": "model_00000.weights.h5",
        "/embed_audio/hard_embedding_norm/vars": "model_00000.weights.h5",
        "/embed_audio/soft_embedding_norm/vars": "model_00000.weights.h5",
        "/embed_vision/embedding/vars": "model_00000.weights.h5",
        "/embed_vision/embedding_projection/vars": "model_00000.weights.h5",
        "/embed_vision/hard_embedding_norm/vars": "model_00000.weights.h5",
        "/embed_vision/soft_embedding_norm/vars": "model_00000.weights.h5",
        "/language_model/altup_projections/dense/vars": "model_00000.weights.h5",
        "/language_model/altup_projections/dense_1/vars": "model_00000.weights.h5",
        "/language_model/altup_projections/dense_2/vars": "model_00000.weights.h5",
        "/language_model/altup_unembed_projections/dense/vars": "model_00000.weights.h5",
        "/language_model/altup_unembed_projections/dense_1/vars": "model_00000.weights.h5",
        "/language_model/altup_unembed_projections/dense_2/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/altup/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/altup/correction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/altup/modality_router/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/altup/prediction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/altup/router_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/attention/key/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/attention/key_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/attention/output_proj/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/attention/query/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/attention/query_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/attention/value/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/laurel/linear_left/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/laurel/linear_right/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/laurel/post_laurel_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/mlp/down/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/mlp/gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/mlp/up/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/per_layer_input_gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/per_layer_projection/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/post_attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/post_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/post_per_layer_input_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer/pre_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/altup/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/altup/correction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/altup/modality_router/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/altup/prediction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/altup/router_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/attention/key/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/attention/key_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/attention/output_proj/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/attention/query/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/attention/query_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/attention/value/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/laurel/linear_left/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/laurel/linear_right/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/laurel/post_laurel_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/mlp/down/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/mlp/gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/mlp/up/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/per_layer_input_gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/per_layer_projection/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/post_attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/post_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/post_per_layer_input_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_1/pre_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/altup/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/altup/correction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/altup/modality_router/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/altup/prediction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/altup/router_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/attention/key/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/attention/key_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/attention/output_proj/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/attention/query/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/attention/query_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/attention/value/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/laurel/linear_left/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/laurel/linear_right/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/laurel/post_laurel_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/mlp/down/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/mlp/gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/mlp/up/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/per_layer_input_gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/per_layer_projection/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/post_attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/post_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/post_per_layer_input_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_2/pre_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/altup/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/altup/correction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/altup/modality_router/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/altup/prediction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/altup/router_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/attention/key/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/attention/key_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/attention/output_proj/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/attention/query/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/attention/query_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/attention/value/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/laurel/linear_left/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/laurel/linear_right/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/laurel/post_laurel_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/mlp/down/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/mlp/gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/mlp/up/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/per_layer_input_gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/per_layer_projection/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/post_attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/post_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/post_per_layer_input_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_3/pre_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/altup/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/altup/correction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/altup/modality_router/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/altup/prediction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/altup/router_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/attention/key/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/attention/key_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/attention/output_proj/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/attention/query/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/attention/query_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/attention/value/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/laurel/linear_left/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/laurel/linear_right/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/laurel/post_laurel_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/mlp/down/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/mlp/gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/mlp/up/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/per_layer_input_gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/per_layer_projection/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/post_attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/post_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/post_per_layer_input_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_4/pre_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/altup/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/altup/correction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/altup/modality_router/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/altup/prediction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/altup/router_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/attention/key/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/attention/key_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/attention/output_proj/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/attention/query/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/attention/query_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/attention/value/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/laurel/linear_left/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/laurel/linear_right/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/laurel/post_laurel_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/mlp/down/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/mlp/gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/mlp/up/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/per_layer_input_gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/per_layer_projection/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/post_attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/post_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/post_per_layer_input_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_5/pre_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/altup/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/altup/correction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/altup/modality_router/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/altup/prediction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/altup/router_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/attention/key/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/attention/key_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/attention/output_proj/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/attention/query/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/attention/query_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/attention/value/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/laurel/linear_left/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/laurel/linear_right/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/laurel/post_laurel_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/mlp/down/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/mlp/gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/mlp/up/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/per_layer_input_gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/per_layer_projection/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/post_attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/post_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/post_per_layer_input_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_6/pre_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/altup/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/altup/correction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/altup/modality_router/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/altup/prediction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/altup/router_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/attention/key/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/attention/key_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/attention/output_proj/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/attention/query/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/attention/query_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/attention/value/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/laurel/linear_left/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/laurel/linear_right/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/laurel/post_laurel_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/mlp/down/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/mlp/gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/mlp/up/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/per_layer_input_gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/per_layer_projection/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/post_attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/post_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/post_per_layer_input_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_7/pre_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/altup/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/altup/correction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/altup/modality_router/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/altup/prediction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/altup/router_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/attention/key/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/attention/key_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/attention/output_proj/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/attention/query/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/attention/query_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/attention/value/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/laurel/linear_left/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/laurel/linear_right/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/laurel/post_laurel_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/mlp/down/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/mlp/gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/mlp/up/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/per_layer_input_gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/per_layer_projection/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/post_attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/post_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/post_per_layer_input_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_8/pre_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/altup/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/altup/correction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/altup/modality_router/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/altup/prediction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/altup/router_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/attention/key/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/attention/key_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/attention/output_proj/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/attention/query/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/attention/query_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/attention/value/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/laurel/linear_left/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/laurel/linear_right/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/laurel/post_laurel_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/mlp/down/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/mlp/gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/mlp/up/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/per_layer_input_gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/per_layer_projection/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/post_attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/post_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/post_per_layer_input_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_9/pre_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/altup/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/altup/correction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/altup/modality_router/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/altup/prediction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/altup/router_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/attention/key/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/attention/key_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/attention/output_proj/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/attention/query/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/attention/query_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/attention/value/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/laurel/linear_left/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/laurel/linear_right/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/laurel/post_laurel_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/mlp/down/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/mlp/gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/mlp/up/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/per_layer_input_gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/per_layer_projection/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/post_attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/post_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/post_per_layer_input_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_10/pre_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/altup/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/altup/correction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/altup/modality_router/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/altup/prediction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/altup/router_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/attention/key/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/attention/key_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/attention/output_proj/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/attention/query/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/attention/query_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/attention/value/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/laurel/linear_left/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/laurel/linear_right/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/laurel/post_laurel_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/mlp/down/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/mlp/gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/mlp/up/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/per_layer_input_gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/per_layer_projection/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/post_attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/post_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/post_per_layer_input_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_11/pre_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/altup/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/altup/correction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/altup/modality_router/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/altup/prediction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/altup/router_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/attention/key/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/attention/key_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/attention/output_proj/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/attention/query/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/attention/query_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/attention/value/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/laurel/linear_left/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/laurel/linear_right/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/laurel/post_laurel_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/mlp/down/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/mlp/gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/mlp/up/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/per_layer_input_gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/per_layer_projection/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/post_attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/post_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/post_per_layer_input_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_12/pre_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/altup/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/altup/correction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/altup/modality_router/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/altup/prediction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/altup/router_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/attention/key/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/attention/key_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/attention/output_proj/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/attention/query/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/attention/query_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/attention/value/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/laurel/linear_left/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/laurel/linear_right/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/laurel/post_laurel_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/mlp/down/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/mlp/gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/mlp/up/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/per_layer_input_gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/per_layer_projection/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/post_attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/post_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/post_per_layer_input_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_13/pre_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/altup/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/altup/correction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/altup/modality_router/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/altup/prediction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/altup/router_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/attention/key/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/attention/key_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/attention/output_proj/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/attention/query/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/attention/query_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/attention/value/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/laurel/linear_left/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/laurel/linear_right/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/laurel/post_laurel_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/mlp/down/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/mlp/gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/mlp/up/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/per_layer_input_gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/per_layer_projection/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/post_attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/post_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/post_per_layer_input_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_14/pre_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/altup/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/altup/correction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/altup/modality_router/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/altup/prediction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/altup/router_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/attention/key/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/attention/key_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/attention/output_proj/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/attention/query/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/attention/query_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/attention/value/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/laurel/linear_left/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/laurel/linear_right/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/laurel/post_laurel_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/mlp/down/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/mlp/gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/mlp/up/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/per_layer_input_gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/per_layer_projection/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/post_attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/post_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/post_per_layer_input_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_15/pre_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/altup/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/altup/correction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/altup/modality_router/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/altup/prediction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/altup/router_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/attention/key/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/attention/key_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/attention/output_proj/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/attention/query/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/attention/query_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/attention/value/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/laurel/linear_left/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/laurel/linear_right/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/laurel/post_laurel_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/mlp/down/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/mlp/gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/mlp/up/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/per_layer_input_gate/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/per_layer_projection/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/post_attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/post_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/post_per_layer_input_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_16/pre_feedforward_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/altup/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/altup/correction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/altup/modality_router/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/altup/prediction_coefs/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/altup/router_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/attention/key/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/attention/key_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/attention/output_proj/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/attention/query/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/attention/query_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/attention/value/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/attention_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/laurel/linear_left/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/laurel/linear_right/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/laurel/post_laurel_norm/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/mlp/down/vars": "model_00000.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/mlp/gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/mlp/up/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/per_layer_input_gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/per_layer_projection/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/post_attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/post_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/post_per_layer_input_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_17/pre_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/altup/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/altup/correction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/altup/modality_router/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/altup/prediction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/altup/router_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/attention/key/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/attention/key_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/attention/output_proj/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/attention/query/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/attention/query_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/attention/value/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/laurel/linear_left/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/laurel/linear_right/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/laurel/post_laurel_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/mlp/down/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/mlp/gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/mlp/up/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/per_layer_input_gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/per_layer_projection/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/post_attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/post_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/post_per_layer_input_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_18/pre_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/altup/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/altup/correction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/altup/modality_router/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/altup/prediction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/altup/router_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/attention/key/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/attention/key_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/attention/output_proj/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/attention/query/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/attention/query_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/attention/value/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/laurel/linear_left/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/laurel/linear_right/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/laurel/post_laurel_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/mlp/down/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/mlp/gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/mlp/up/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/per_layer_input_gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/per_layer_projection/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/post_attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/post_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/post_per_layer_input_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_19/pre_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/altup/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/altup/correction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/altup/modality_router/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/altup/prediction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/altup/router_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/attention/output_proj/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/attention/query/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/attention/query_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/laurel/linear_left/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/laurel/linear_right/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/laurel/post_laurel_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/mlp/down/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/mlp/gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/mlp/up/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/per_layer_input_gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/per_layer_projection/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/post_attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/post_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/post_per_layer_input_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_20/pre_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/altup/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/altup/correction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/altup/modality_router/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/altup/prediction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/altup/router_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/attention/output_proj/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/attention/query/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/attention/query_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/laurel/linear_left/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/laurel/linear_right/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/laurel/post_laurel_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/mlp/down/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/mlp/gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/mlp/up/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/per_layer_input_gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/per_layer_projection/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/post_attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/post_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/post_per_layer_input_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_21/pre_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/altup/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/altup/correction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/altup/modality_router/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/altup/prediction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/altup/router_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/attention/output_proj/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/attention/query/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/attention/query_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/laurel/linear_left/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/laurel/linear_right/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/laurel/post_laurel_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/mlp/down/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/mlp/gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/mlp/up/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/per_layer_input_gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/per_layer_projection/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/post_attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/post_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/post_per_layer_input_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_22/pre_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/altup/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/altup/correction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/altup/modality_router/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/altup/prediction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/altup/router_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/attention/output_proj/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/attention/query/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/attention/query_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/laurel/linear_left/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/laurel/linear_right/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/laurel/post_laurel_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/mlp/down/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/mlp/gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/mlp/up/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/per_layer_input_gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/per_layer_projection/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/post_attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/post_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/post_per_layer_input_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_23/pre_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/altup/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/altup/correction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/altup/modality_router/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/altup/prediction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/altup/router_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/attention/output_proj/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/attention/query/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/attention/query_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/laurel/linear_left/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/laurel/linear_right/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/laurel/post_laurel_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/mlp/down/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/mlp/gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/mlp/up/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/per_layer_input_gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/per_layer_projection/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/post_attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/post_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/post_per_layer_input_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_24/pre_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/altup/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/altup/correction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/altup/modality_router/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/altup/prediction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/altup/router_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/attention/output_proj/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/attention/query/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/attention/query_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/laurel/linear_left/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/laurel/linear_right/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/laurel/post_laurel_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/mlp/down/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/mlp/gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/mlp/up/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/per_layer_input_gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/per_layer_projection/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/post_attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/post_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/post_per_layer_input_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_25/pre_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/altup/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/altup/correction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/altup/modality_router/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/altup/prediction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/altup/router_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/attention/output_proj/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/attention/query/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/attention/query_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/laurel/linear_left/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/laurel/linear_right/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/laurel/post_laurel_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/mlp/down/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/mlp/gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/mlp/up/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/per_layer_input_gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/per_layer_projection/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/post_attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/post_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/post_per_layer_input_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_26/pre_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/altup/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/altup/correction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/altup/modality_router/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/altup/prediction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/altup/router_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/attention/output_proj/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/attention/query/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/attention/query_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/laurel/linear_left/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/laurel/linear_right/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/laurel/post_laurel_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/mlp/down/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/mlp/gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/mlp/up/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/per_layer_input_gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/per_layer_projection/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/post_attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/post_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/post_per_layer_input_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_27/pre_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/altup/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/altup/correction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/altup/modality_router/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/altup/prediction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/altup/router_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/attention/output_proj/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/attention/query/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/attention/query_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/laurel/linear_left/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/laurel/linear_right/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/laurel/post_laurel_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/mlp/down/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/mlp/gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/mlp/up/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/per_layer_input_gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/per_layer_projection/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/post_attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/post_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/post_per_layer_input_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_28/pre_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/altup/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/altup/correction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/altup/modality_router/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/altup/prediction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/altup/router_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/attention/output_proj/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/attention/query/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/attention/query_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/laurel/linear_left/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/laurel/linear_right/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/laurel/post_laurel_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/mlp/down/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/mlp/gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/mlp/up/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/per_layer_input_gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/per_layer_projection/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/post_attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/post_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/post_per_layer_input_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_29/pre_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/altup/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/altup/correction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/altup/modality_router/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/altup/prediction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/altup/router_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/attention/output_proj/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/attention/query/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/attention/query_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/laurel/linear_left/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/laurel/linear_right/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/laurel/post_laurel_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/mlp/down/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/mlp/gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/mlp/up/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/per_layer_input_gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/per_layer_projection/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/post_attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/post_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/post_per_layer_input_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_30/pre_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/altup/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/altup/correction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/altup/modality_router/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/altup/prediction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/altup/router_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/attention/output_proj/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/attention/query/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/attention/query_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/laurel/linear_left/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/laurel/linear_right/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/laurel/post_laurel_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/mlp/down/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/mlp/gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/mlp/up/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/per_layer_input_gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/per_layer_projection/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/post_attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/post_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/post_per_layer_input_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_31/pre_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/altup/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/altup/correction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/altup/modality_router/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/altup/prediction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/altup/router_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/attention/output_proj/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/attention/query/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/attention/query_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/laurel/linear_left/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/laurel/linear_right/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/laurel/post_laurel_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/mlp/down/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/mlp/gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/mlp/up/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/per_layer_input_gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/per_layer_projection/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/post_attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/post_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/post_per_layer_input_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_32/pre_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/altup/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/altup/correction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/altup/modality_router/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/altup/prediction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/altup/router_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/attention/output_proj/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/attention/query/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/attention/query_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/laurel/linear_left/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/laurel/linear_right/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/laurel/post_laurel_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/mlp/down/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/mlp/gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/mlp/up/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/per_layer_input_gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/per_layer_projection/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/post_attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/post_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/post_per_layer_input_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_33/pre_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/altup/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/altup/correction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/altup/modality_router/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/altup/prediction_coefs/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/altup/router_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/attention/output_proj/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/attention/query/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/attention/query_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/laurel/linear_left/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/laurel/linear_right/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/laurel/post_laurel_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/mlp/down/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/mlp/gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/mlp/up/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/per_layer_input_gate/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/per_layer_projection/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/post_attention_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/post_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/post_per_layer_input_norm/vars": "model_00001.weights.h5",
        "/language_model/decoder_layers/gemma3n_decoder_layer_34/pre_feedforward_norm/vars": "model_00001.weights.h5",
        "/language_model/embed_tokens_per_layer/vars": "model_00002.weights.h5",
        "/language_model/final_norm/vars": "model_00002.weights.h5",
        "/language_model/layers/embedding/vars": "model_00003.weights.h5",
        "/language_model/layers/dense/vars": "model_00003.weights.h5",
        "/language_model/layers/gemma3n_rms_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/conv_stem/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/conv_stem/conv/vars": [
            "model_00003.weights.h5"
        ],
        "/layers/mobile_net_v5_encoder/msfa/ffn/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/msfa/ffn/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/msfa/ffn/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/msfa/ffn/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/msfa/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container/edge_residual/bn1/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container/edge_residual/bn2/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container/edge_residual/conv_exp/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container/edge_residual/conv_pwl/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container/edge_residual_1/bn1/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container/edge_residual_1/bn2/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container/edge_residual_1/conv_exp/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container/edge_residual_1/conv_pwl/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container/edge_residual_2/bn1/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container/edge_residual_2/bn2/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container/edge_residual_2/conv_exp/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container/edge_residual_2/conv_pwl/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual/dw_mid/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual/dw_mid/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual/dw_start/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual/dw_start/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_1/dw_start/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_1/dw_start/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_1/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_1/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_1/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_1/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_1/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_2/dw_start/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_2/dw_start/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_2/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_2/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_2/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_2/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_2/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_3/dw_start/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_3/dw_start/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_3/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_3/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_3/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_3/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_3/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_4/dw_start/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_4/dw_start/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_4/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_4/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_4/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_4/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_1/universal_inverted_residual_4/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual/dw_mid/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual/dw_mid/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual/dw_start/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual/dw_start/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_1/dw_start/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_1/dw_start/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_1/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_1/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_1/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_1/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_1/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_2/dw_start/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_2/dw_start/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_2/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_2/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_2/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_2/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_2/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_3/dw_start/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_3/dw_start/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_3/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_3/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_3/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_3/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_3/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_4/dw_start/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_4/dw_start/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_4/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_4/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_4/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_4/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_4/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_5/dw_start/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_5/dw_start/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_5/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_5/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_5/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_5/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_5/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_6/dw_start/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_6/dw_start/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_6/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_6/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_6/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_6/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_6/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_7/dw_start/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_7/dw_start/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_7/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_7/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_7/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_7/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_7/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_8/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_8/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_8/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_8/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_8/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention/attn/key_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention/attn/key_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention/attn/value_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention/attn/value_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_9/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_9/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_9/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_9/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_9/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_1/attn/key_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_1/attn/key_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_1/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_1/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_1/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_1/attn/value_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_1/attn/value_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_1/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_1/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_1/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_10/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_10/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_10/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_10/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_10/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_2/attn/key_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_2/attn/key_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_2/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_2/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_2/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_2/attn/value_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_2/attn/value_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_2/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_2/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_2/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_11/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_11/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_11/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_11/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_11/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_3/attn/key_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_3/attn/key_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_3/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_3/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_3/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_3/attn/value_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_3/attn/value_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_3/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_3/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_3/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_12/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_12/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_12/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_12/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_12/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_4/attn/key_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_4/attn/key_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_4/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_4/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_4/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_4/attn/value_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_4/attn/value_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_4/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_4/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_4/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_13/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_13/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_13/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_13/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_13/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_5/attn/key_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_5/attn/key_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_5/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_5/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_5/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_5/attn/value_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_5/attn/value_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_5/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_5/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_5/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_14/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_14/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_14/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_14/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_14/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_6/attn/key_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_6/attn/key_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_6/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_6/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_6/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_6/attn/value_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_6/attn/value_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_6/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_6/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_6/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_15/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_15/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_15/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_15/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_15/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_7/attn/key_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_7/attn/key_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_7/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_7/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_7/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_7/attn/value_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_7/attn/value_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_7/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_7/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_7/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_16/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_16/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_16/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_16/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_16/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_8/attn/key_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_8/attn/key_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_8/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_8/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_8/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_8/attn/value_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_8/attn/value_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_8/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_8/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_8/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_17/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_17/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_17/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_17/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_17/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_9/attn/key_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_9/attn/key_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_9/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_9/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_9/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_9/attn/value_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_9/attn/value_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_9/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_9/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_9/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_18/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_18/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_18/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_18/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_18/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_10/attn/key_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_10/attn/key_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_10/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_10/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_10/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_10/attn/value_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_10/attn/value_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_10/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_10/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_10/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_19/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_19/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_19/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_19/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_19/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_11/attn/key_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_11/attn/key_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_11/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_11/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_11/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_11/attn/value_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_11/attn/value_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_11/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_11/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_11/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_20/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_20/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_20/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_20/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_20/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_12/attn/key_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_12/attn/key_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_12/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_12/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_12/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_12/attn/value_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_12/attn/value_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_12/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_12/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_12/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_21/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_21/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_21/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_21/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_21/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_13/attn/key_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_13/attn/key_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_13/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_13/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_13/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_13/attn/value_down/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_13/attn/value_norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_13/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_13/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/mobile_attention_13/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_22/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_22/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_22/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_22/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_2/universal_inverted_residual_22/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual/dw_mid/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual/dw_mid/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual/dw_start/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual/dw_start/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_1/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_1/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_1/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_1/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_1/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_1/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_1/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_1/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_1/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_1/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_1/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_2/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_2/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_2/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_2/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_2/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_2/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_2/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_2/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_2/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_2/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_2/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_3/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_3/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_3/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_3/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_3/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_3/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_3/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_3/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_3/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_3/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_3/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_4/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_4/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_4/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_4/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_4/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_4/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_4/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_4/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_4/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_4/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_4/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_5/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_5/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_5/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_5/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_5/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_5/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_5/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_5/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_5/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_5/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_5/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_6/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_6/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_6/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_6/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_6/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_6/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_6/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_6/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_6/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_6/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_6/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_7/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_7/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_7/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_7/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_7/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_7/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_7/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_7/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_7/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_7/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_7/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_8/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_8/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_8/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_8/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_8/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_8/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_8/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_8/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_8/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_8/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_8/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_9/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_9/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_9/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_9/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_9/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_9/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_9/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_9/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_9/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_9/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_9/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_10/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_10/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_10/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_10/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_10/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_10/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_10/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_10/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_10/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_10/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_10/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_11/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_11/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_11/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_11/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_11/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_11/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_11/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_11/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_11/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_11/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_11/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_12/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_12/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_12/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_12/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_12/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_12/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_12/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_12/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_12/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_12/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_12/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_13/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_13/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_13/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_13/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_13/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_13/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_13/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_13/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_13/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_13/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_13/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_14/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_14/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_14/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_14/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_14/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_14/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_14/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_14/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_14/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_14/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_14/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_15/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_15/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_15/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_15/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_15/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_15/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_15/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_15/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_15/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_15/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_15/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_16/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_16/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_16/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_16/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_16/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_16/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_16/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_16/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_16/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_16/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_16/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_17/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_17/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_17/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_17/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_17/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_17/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_17/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_17/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_17/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_17/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_17/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_18/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_18/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_18/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_18/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_18/pw_proj/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_18/attn/key_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_18/attn/output_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_18/attn/query_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_18/attn/value_proj/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_18/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/mobile_attention_18/norm/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_19/layer_scale/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_19/pw_exp/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_19/pw_exp/conv/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_19/pw_proj/bn/vars": "model_00003.weights.h5",
        "/layers/mobile_net_v5_encoder/stages/container_3/universal_inverted_residual_19/pw_proj/conv/vars": "model_00003.weights.h5"
    }
}