File size: 4,813 Bytes
09293f6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
{
  "architectures": [
    "Dots3NoteForCausalLM"
  ],
  "apply_mla_qkv_lora_rescale": true,
  "attention_bias": false,
  "attention_dropout": 0.0,
  "attention_gate_type": "headwise",
  "bos_token_id": 151643,
  "eos_token_id": 151668,
  "first_k_dense_replace": 1,
  "hidden_act": "silu",
  "hidden_size": 5120,
  "index_head_dim": 128,
  "index_n_heads": 64,
  "index_topk": 2048,
  "initializer_range": 0.02,
  "intermediate_size": 13824,
  "kv_lora_rank": 512,
  "layer_types": [
    "full_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention"
  ],
  "max_position_embeddings": 524288,
  "model_type": "dots3_note",
  "moe_intermediate_size": 1536,
  "moe_layer_freq": 1,
  "n_routed_experts": 256,
  "n_shared_experts": 1,
  "norm_topk_prob": true,
  "num_attention_heads": 128,
  "num_experts_per_tok": 8,
  "num_hidden_layers": 46,
  "num_key_value_heads": 128,
  "pretraining_tp": 1,
  "q_lora_rank": 1024,
  "qk_nope_head_dim": 128,
  "qk_rope_head_dim": 64,
  "rms_norm_eps": 1e-05,
  "rope_scaling": null,
  "rope_theta": 80000000.0,
  "routed_scaling_factor": 1.0,
  "scoring_func": "sigmoid",
  "sliding_window_size": 513,
  "swa_attention_gate_type": "headwise",
  "swa_kv_lora_rank": 1024,
  "swa_num_attention_heads": 64,
  "swa_num_key_value_heads": 64,
  "swa_q_lora_rank": 1024,
  "swa_qk_nope_head_dim": 192,
  "swa_qk_rope_head_dim": 64,
  "swa_rope_theta": 50000.0,
  "swa_v_head_dim": 128,
  "tie_word_embeddings": false,
  "topk_method": "noaux_tc",
  "torch_dtype": "bfloat16",
  "transformers_version": "4.36.0",
  "use_cache": true,
  "v_head_dim": 128,
  "vocab_size": 152064,
  "vision_config": {
    "embed_dim": 1536,
    "hidden_size": 5120,
    "intermediate_size": 4224,
    "moe_intermediate_size": 2112,
    "num_hidden_layers": 42,
    "num_attention_heads": 24,
    "num_channels": 3,
    "patch_size": 14,
    "spatial_merge_size": 2,
    "temporal_patch_size": 1,
    "rms_norm_eps": 1e-05,
    "use_bias": false,
    "use_qk_norm": true,
    "attn_implementation": "flash_attention_3",
    "initializer_range": 0.02,
    "is_causal": false,
    "post_norm": true,
    "pre_pixel_shuffle": true,
    "pyramid_num_routed": [
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      -1,
      4,
      8,
      12,
      16,
      20,
      24,
      28,
      32,
      36,
      40,
      44,
      48,
      52,
      56,
      60,
      64,
      64
    ],
    "capacity_factor": 2,
    "router_scoring_func": "sigmoid",
    "router_scale": 1.0,
    "adapter_type": "patch_merger",
    "adapter_in_dim": 1536,
    "adapter_out_dim": 5120,
    "adapter_merge_size": 2
  },
  "audio_config": {
    "encoder_type": "dots",
    "whisper_config": {
      "d_model": 1280,
      "encoder_attention_heads": 20,
      "encoder_ffn_dim": 5120,
      "encoder_layers": 32,
      "num_mel_bins": 128,
      "max_source_positions": 6000,
      "activation_function": "swiglu"
    },
    "use_conv2d_stem": true,
    "use_rope": true,
    "use_rms_norm": true,
    "use_causal": false,
    "downsample_hidden_size": 480,
    "conv_chunksize": 500,
    "conv_stem_gradient_checkpointing": true,
    "conv_bucket_step": 10,
    "conv_bucket_max_elements": 20000,
    "rope_parameters": {
      "partial_rotary_factor": 0.5,
      "rope_theta": 10000.0
    },
    "merge_factor": 1,
    "chunk_seconds": 60,
    "whisper_adapter_in_dim": 1280,
    "whisper_adapter_out_dim": 5120,
    "sampling_rate": 16000,
    "audio_comp_start": "<|audio_comp_start|>",
    "audio_comp_span": "<|audio_comp_pad|>",
    "audio_comp_end": "<|audio_comp_end|>"
  }
}