File size: 493 Bytes
fc2d33a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
{
"attention_qkv_bias": false,
"codebook_size": 1024,
"dim": 1024,
"dropout": 0.1,
"head_dim": 64,
"initializer_range": 0.02,
"intermediate_size": 4096,
"max_seq_len": 2048,
"model_type": "dual_ar",
"n_fast_layer": 4,
"n_head": 16,
"n_layer": 24,
"n_local_heads": 2,
"norm_eps": 1e-06,
"num_codebooks": 8,
"rope_base": 1000000.0,
"tie_word_embeddings": false,
"use_gradient_checkpointing": true,
"vocab_size": 32000
} |