Modelpublic

johahi/flashzoi-replicate-2

sourceHugging Facemitupdated 2y agoView on Hugging Face

0likes4.7kdownloads

config.json20 linesDownload Raw Back to root

1{2  "architectures": [3    "Borzoi"4  ],5  "attn_dim_key": 64,6  "attn_dim_value": 192,7  "attn_dropout": 0.05,8  "depth": 8,9  "dim": 1536,10  "dropout_rate": 0.2,11  "enable_mouse_head": true,12  "flashed": true,13  "heads": 8,14  "model_type": "borzoi",15  "pos_dropout": 0.01,16  "return_center_bins_only": true,17  "torch_dtype": "float32",18  "transformers_version": "4.34.1"19}20