llama-160m-accelerator / config.json
sahilsuneja's picture
Adding weights
d8be53a
raw
history blame
317 Bytes
{
"architectures": [
"MLPSpeculatorPreTrainedModel"
],
"emb_dim": 768,
"inner_dim": 1024,
"model_type": "mlp_speculator",
"n_candidates": 5,
"n_predict": 3,
"top_k_tokens_per_head": [
5,
3,
2
],
"torch_dtype": "float16",
"transformers_version": "4.41.2",
"vocab_size": 32000
}