JRosenkranz's picture
Update config.json
75ea497 verified
raw
history blame
No virus
396 Bytes
{
"base_model_name_or_path": "ibm-granite/granite-7b-instruct",
"architectures": [
"MLPSpeculatorPreTrainedModel"
],
"emb_dim": 4096,
"inner_dim": 4096,
"model_type": "mlp_speculator",
"n_candidates": 5,
"n_predict": 5,
"top_k_tokens_per_head": [
4,
3,
2,
2,
2
],
"torch_dtype": "float16",
"transformers_version": "4.38.2",
"vocab_size": 32008
}