config.json

{
  "model_type": "bard",
  "architectures": [
    "BardForStory"
  ],
  "auto_map": {
    "AutoConfig": "configuration_bard.BardConfig",
    "AutoModel": "modeling_bard.BardForStory",
    "AutoModelForCausalLM": "modeling_bard.BardForStory"
  },
  "vocab_size": 16128,
  "d_model": 512,
  "n_heads": 8,
  "n_layers": 36,
  "window": 256,
  "global_every": 4,
  "rope": "local",
  "parallel": true,
  "context": 4096,
  "vars_weight": 4.0,
  "torch_dtype": "bfloat16"
}