| { |
| "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 |
| } |
|
|
