|
{ |
|
"_name_or_path": "mistralai/Mistral-7B-v0.1", |
|
"architectures": [ |
|
"SparseMistralforCausalLM" |
|
], |
|
"attention_dropout": 0.0, |
|
"auto_map": { |
|
"AutoConfig": "sparsification_sftt.SparseMistralConfig", |
|
"AutoModelForCausalLM": "sparsification_sftt.SparseMistralforCausalLM" |
|
}, |
|
"bos_token_id": 1, |
|
"eos_token_id": 2, |
|
"hidden_act": "silu", |
|
"hidden_size": 4096, |
|
"initializer_range": 0.02, |
|
"intermediate_size": 14336, |
|
"max_position_embeddings": 32768, |
|
"model_type": "sparse_mistral", |
|
"num_attention_heads": 32, |
|
"num_hidden_layers": 32, |
|
"num_key_value_heads": 8, |
|
"rms_norm_eps": 1e-05, |
|
"rope_theta": 10000.0, |
|
"sliding_window": 4096, |
|
"thresholds": [ |
|
0.037111327052116394, |
|
0.05315947160124779, |
|
0.06720159947872162, |
|
0.08525575697422028, |
|
0.11133399605751038, |
|
0.13340020179748535, |
|
0.14744232594966888, |
|
0.16549648344516754, |
|
0.17552657425403595, |
|
0.1775325983762741, |
|
0.18154463171958923, |
|
0.18555666506290436, |
|
0.18956869840621948, |
|
0.1935807317495346, |
|
0.2096288800239563, |
|
0.22166499495506287, |
|
0.23771312832832336, |
|
0.24172517657279968, |
|
0.25376129150390625, |
|
0.25977933406829834, |
|
0.2617853581905365, |
|
0.26579737663269043, |
|
0.26579737663269043, |
|
0.26579737663269043, |
|
0.2678034007549286, |
|
0.2678034007549286, |
|
0.2678034007549286, |
|
0.2678034007549286, |
|
0.2678034007549286, |
|
0.2678034007549286, |
|
0.2678034007549286, |
|
0.2678034007549286 |
|
], |
|
"tie_word_embeddings": false, |
|
"torch_dtype": "bfloat16", |
|
"transformers_version": "4.36.2", |
|
"use_cache": false, |
|
"use_relu": false, |
|
"use_sparse_model": true, |
|
"use_sparse_predictor": false, |
|
"use_sparse_regularization": false, |
|
"vocab_size": 32000 |
|
} |
|
|