File size: 1,413 Bytes
0f072a3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "architectures": [
    "HrwkvHybridForCausalLM"
  ],
  "description": "Hybrid-RWKV Strategically Interleaved RWKV-Attention",
  "base_model": "RekaAI/reka-flash-3.1",
  "model_revision": "",
  "transformer_layers":[4, 8, 11, 15, 18, 22, 25, 29, 32, 36, 39, 43],
  "rwkv_layers": [0, 1, 2, 3, 5, 6, 7, 9, 10, 12, 13, 14, 16, 17, 19, 20, 21, 23, 24, 26, 27, 28, 30, 31, 33, 34, 35, 37, 38, 40, 41, 42],
  "rwkv_architecture": "hxa079",
  "enable_qk_norm": false,
  "nope_in_transformer": true,
  "nope_in_rwkv": false,
  "lora_rank_decay": 320,
  "lora_rank_iclr":128,
  "lora_rank_value_residual_mix":96,
  "lora_rank_key_residual_mix":96,
  "lora_rank_gate":384,

  "attention_bias": false,
  "attention_dropout": 0.0,
  "bos_token_id": 100257,
  "classifier_dropout": 0.0,
  "eos_token_id": 100257,
  "head_dim": 96,
  "hidden_act": "silu",
  "hidden_size": 6144,
  "id2label": {
    "0": "LABEL_0"
  },
  "initializer_range": 0.006,
  "intermediate_size": 19648,
  "label2id": {
    "LABEL_0": 0
  },
  "max_position_embeddings": 98304,
  "mlp_bias": false,
  "model_type": "llama",
  "num_attention_heads": 64,
  "num_hidden_layers": 44,
  "num_key_value_heads": 8,
  "pretraining_tp": 1,
  "rms_norm_eps": 1e-05,
  "rope_scaling": null,
  "rope_theta": 8000000,
  "tie_word_embeddings": false,
  "torch_dtype": "bfloat16",
  "transformers_version": "4.50.3",
  "use_cache": true,
  "vocab_size": 100352
}