Team Ai
Modelpublic

diffusers/matrix-game-2-modular

sourceHugging Faceupdated 1y agoView on Hugging Face
0likes14downloads
config.json69 linesDownload Raw Back to transformer
1{2  "_class_name": "MatrixGameWanCausalModel",3  "_diffusers_version": "0.35.1",4  "auto_map": {5    "AutoModel": "causal_model.MatrixGameWanCausalModel"6  },7  "action_config": {8    "blocks": [9      0,10      1,11      2,12      3,13      4,14      5,15      6,16      7,17      8,18      9,19      10,20      11,21      12,22      13,23      1424    ],25    "enable_keyboard": true,26    "enable_mouse": true,27    "heads_num": 16,28    "hidden_size": 128,29    "img_hidden_size": 1536,30    "keyboard_dim_in": 4,31    "keyboard_hidden_dim": 1024,32    "mouse_dim_in": 2,33    "mouse_hidden_dim": 1024,34    "mouse_qk_dim_list": [35      8,36      28,37      2838    ],39    "patch_size": [40      1,41      2,42      243    ],44    "qk_norm": true,45    "qkv_bias": false,46    "rope_dim_list": [47      8,48      28,49      2850    ],51    "rope_theta": 256,52    "vae_time_compression_ratio": 4,53    "windows_size": 354  },55  "dim": 1536,56  "eps": 1e-06,57  "ffn_dim": 8960,58  "freq_dim": 256,59  "in_dim": 36,60  "inject_sample_info": false,61  "local_attn_size": 6,62  "model_type": "i2v",63  "num_heads": 12,64  "num_layers": 30,65  "out_dim": 16,66  "sink_size": 0,67  "text_len": 51268}69