llama-13b-accelerator / config.json
JRosenkranz's picture
Update config.json
78a73f1 verified
{
"base_model_name_or_path": "meta-llama/Llama-2-13b-chat-hf",
"architectures": [
"MLPSpeculatorPreTrainedModel"
],
"emb_dim": 5120,
"inner_dim": 4096,
"model_type": "mlp_speculator",
"n_candidates": 5,
"n_predict": 3,
"top_k_tokens_per_head": [
5,
3,
2
],
"torch_dtype": "float16",
"transformers_version": "4.35.0",
"vocab_size": 32000
}