config.json

{
  "_name_or_path": "johahi/flashzoi-replicate-0",
  "architectures": [
    "Scooby"
  ],
  "attn_dim_key": 64,
  "attn_dim_value": 192,
  "attn_dropout": 0.05,
  "depth": 8,
  "dim": 1536,
  "dropout_rate": 0.2,
  "enable_mouse_head": true,
  "flashed": true,
  "heads": 8,
  "model_type": "borzoi",
  "pos_dropout": 0.01,
  "return_center_bins_only": true,
  "torch_dtype": "float32",
  "transformers_version": "4.45.2"
}