Anvaya-Rabbit-2.7B / config.json
tvastr's picture
chore: add config.json (safety-scrubbed AutoModel wrapper)
8f0815c verified
raw
history blame contribute delete
382 Bytes
{
"architectures": ["RabbitForCausalLM"],
"auto_map": {
"AutoConfig": "configuration_rabbit.RabbitConfig",
"AutoModelForCausalLM": "modeling_rabbit.RabbitForCausalLM"
},
"model_type": "rabbit_ssm",
"vocab_size": 50280,
"d_model": 2560,
"n_layers": 64,
"bos_token_id": 0,
"eos_token_id": 0,
"torch_dtype": "bfloat16",
"transformers_version": "4.40.0"
}