william-ricursive's picture
Use conservative NVFP4/BF16 checkpoint and refresh Spark notebook
9a8e832 verified
Raw History Blame Contribute Delete
8.49 kB
{
"architectures": [
"RicursiveForCausalLM"
],
"auto_map": {
"AutoConfig": "configuration_ricursive.RicursiveConfig",
"AutoModel": "modeling_ricursive.RicursiveModel",
"AutoModelForCausalLM": "modeling_ricursive.RicursiveForCausalLM"
},
"bos_token_id": 100257,
"dtype": "bfloat16",
"embed_scale": 71.55417527999327,
"eos_token_id": [
100265,
100257
],
"ffn_clamp_threshold": 20.0,
"gate_full_precision": true,
"head_dim": 128,
"hidden_act": "silu",
"hidden_size": 5120,
"hybrid_layer_pattern": [
1,
1,
1,
1,
0,
1,
1,
1,
1,
0,
1,
1,
1,
1,
0,
1,
1,
1,
1,
0,
1,
1,
1,
1,
0,
1,
1,
1,
1,
0,
1,
1,
1,
1,
0,
1,
1,
1,
1,
0,
1,
1,
1,
1,
0,
1,
1,
1,
1,
0,
1,
1,
1,
1,
0,
1,
1,
1,
1,
0,
1,
1,
1,
1,
0,
1,
1,
1,
1,
0,
1,
1,
1,
1,
0,
1,
1,
1,
1,
0
],
"intermediate_size": 40960,
"is_hybrid_swa": true,
"layer_types": [
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"sliding_attention",
"full_attention"
],
"max_position_embeddings": 2097152,
"model_type": "ricursive",
"num_attention_heads": 80,
"num_hidden_layers": 80,
"num_key_value_heads": 8,
"pad_token_id": 100277,
"rms_norm_eps": 1e-06,
"rope_theta": 10000.0,
"sliding_window": 1024,
"tie_word_embeddings": false,
"transformers_version": "5.12.1",
"use_cache": true,
"use_gated_attention": true,
"use_valres": true,
"vocab_size": 100352,
"quantization_config": {
"config_groups": {
"group_0": {
"input_activations": {
"dynamic": false,
"num_bits": 4,
"type": "float",
"group_size": 16
},
"weights": {
"dynamic": false,
"num_bits": 4,
"type": "float",
"group_size": 16
},
"targets": [
"Linear"
]
}
},
"ignore": [
"lm_head",
"model.embed_tokens",
"model.layers.0.self_attn*",
"model.layers.1.self_attn*",
"model.layers.10.self_attn*",
"model.layers.11.self_attn*",
"model.layers.12.self_attn*",
"model.layers.13.self_attn*",
"model.layers.14.self_attn*",
"model.layers.15.self_attn*",
"model.layers.16.self_attn*",
"model.layers.17.self_attn*",
"model.layers.18.self_attn*",
"model.layers.19.self_attn*",
"model.layers.2.self_attn*",
"model.layers.20.self_attn*",
"model.layers.21.self_attn*",
"model.layers.22.self_attn*",
"model.layers.23.self_attn*",
"model.layers.24.self_attn*",
"model.layers.25.self_attn*",
"model.layers.26.self_attn*",
"model.layers.27.self_attn*",
"model.layers.28.self_attn*",
"model.layers.29.self_attn*",
"model.layers.3.self_attn*",
"model.layers.30.self_attn*",
"model.layers.31.self_attn*",
"model.layers.32.self_attn*",
"model.layers.33.self_attn*",
"model.layers.34.self_attn*",
"model.layers.35.self_attn*",
"model.layers.36.self_attn*",
"model.layers.37.self_attn*",
"model.layers.38.self_attn*",
"model.layers.39.self_attn*",
"model.layers.4.self_attn*",
"model.layers.40.self_attn*",
"model.layers.41.self_attn*",
"model.layers.42.self_attn*",
"model.layers.43.self_attn*",
"model.layers.44.self_attn*",
"model.layers.45.self_attn*",
"model.layers.46.self_attn*",
"model.layers.47.self_attn*",
"model.layers.48.self_attn*",
"model.layers.49.self_attn*",
"model.layers.5.self_attn*",
"model.layers.50.self_attn*",
"model.layers.51.self_attn*",
"model.layers.52.self_attn*",
"model.layers.53.self_attn*",
"model.layers.54.self_attn*",
"model.layers.55.self_attn*",
"model.layers.56.self_attn*",
"model.layers.57.self_attn*",
"model.layers.58.self_attn*",
"model.layers.59.self_attn*",
"model.layers.6.self_attn*",
"model.layers.60.self_attn*",
"model.layers.61.self_attn*",
"model.layers.62.self_attn*",
"model.layers.63.self_attn*",
"model.layers.64.self_attn*",
"model.layers.65.self_attn*",
"model.layers.66.self_attn*",
"model.layers.67.self_attn*",
"model.layers.68.self_attn*",
"model.layers.69.self_attn*",
"model.layers.7.self_attn*",
"model.layers.70.self_attn*",
"model.layers.71.self_attn*",
"model.layers.72.self_attn*",
"model.layers.73.self_attn*",
"model.layers.74.self_attn*",
"model.layers.75.self_attn*",
"model.layers.76.self_attn*",
"model.layers.77.self_attn*",
"model.layers.78.self_attn*",
"model.layers.79.self_attn*",
"model.layers.8.self_attn*",
"model.layers.9.self_attn*"
],
"quant_algo": "NVFP4",
"producer": {
"name": "modelopt",
"version": "0.46.0"
},
"quant_method": "modelopt"
}
}