File size: 2,924 Bytes
9458a5b
e7076fb
 
9458a5b
e7076fb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9458a5b
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
{
    "architectures": [
        "InklingForConditionalGeneration"
    ],
    "audio_config": {
        "decoder_dmodel": 4096,
        "n_mel_bins": 80,
        "mel_vocab_size": 16,
        "bias": false,
        "dmel_min_value": -7.0,
        "dmel_max_value": 2.0,
        "use_audio_norm": true,
        "audio_mode": "dmel"
    },
    "eos_token_id": 200006,
    "model_type": "inkling_mm_model",
    "mtp_config": {
        "num_nextn_predict_layers": 8,
        "chain_hidden_post_norm": false,
        "local_layer_ids": [
            0,
            2,
            4,
            5,
            6,
            7
        ]
    },
    "quantization": {
        "group_size": 64,
        "bits": 4,
        "mode": "affine"
    },
    "quantization_config": {
        "group_size": 64,
        "bits": 4,
        "mode": "affine"
    },
    "text_config": {
        "model_max_length": 1048576,
        "torch_dtype": "bfloat16",
        "hidden_size": 4096,
        "num_hidden_layers": 42,
        "vocab_size": 201024,
        "num_attention_heads": 32,
        "num_key_value_heads": 8,
        "head_dim": 128,
        "d_rel": 16,
        "rel_extent": 1024,
        "q_bias": false,
        "o_bias": false,
        "log_scaling_n_floor": 128000,
        "log_scaling_alpha": 0.1,
        "rms_norm_eps": 1e-06,
        "use_embed_norm": true,
        "local_layer_ids": [
            0,
            1,
            2,
            3,
            4,
            6,
            7,
            8,
            9,
            10,
            12,
            13,
            14,
            15,
            16,
            18,
            19,
            20,
            21,
            22,
            24,
            25,
            26,
            27,
            28,
            30,
            31,
            32,
            33,
            34,
            36,
            37,
            38,
            39,
            40
        ],
        "dense_mlp_idx": 2,
        "use_sconv": true,
        "sconv_kernel_size": 4,
        "unpadded_vocab_size": 200058,
        "logits_mup_width_multiplier": 16.0,
        "final_logit_softcapping": null,
        "swa_head_dim": 128,
        "swa_num_attention_heads": 32,
        "swa_num_key_value_heads": 8,
        "sliding_window_size": 512,
        "n_routed_experts": 256,
        "num_experts_per_tok": 6,
        "n_shared_experts": 2,
        "shared_expert_sink": true,
        "dense_intermediate_size": 16384,
        "intermediate_size": 2048,
        "route_scale": 8.0,
        "use_gate_bias": true,
        "gate_activation": "sigmoid",
        "norm_after_topk": true,
        "use_global_scale": true
    },
    "vision_config": {
        "vision_encoder_type": "hmlp",
        "decoder_dmodel": 4096,
        "patch_size": 40,
        "temporal_patch_size": 2,
        "n_channels": 3,
        "n_layers": 4,
        "use_vision_norm": true
    }
}