config.json

{
  "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"
}