| { | |
| "_class_name": "UVit2DModel", | |
| "_diffusers_version": "0.25.0.dev0", | |
| "_name_or_path": "muse-512/", | |
| "attention_dropout": 0.0, | |
| "block_num_heads": 12, | |
| "block_out_channels": 768, | |
| "codebook_size": 8192, | |
| "cond_embed_dim": 768, | |
| "downsample": true, | |
| "encoder_hidden_size": 768, | |
| "hidden_dropout": 0.0, | |
| "hidden_size": 1024, | |
| "in_channels": 768, | |
| "intermediate_size": 2816, | |
| "layer_norm_eps": 1e-06, | |
| "ln_elementwise_affine": true, | |
| "micro_cond_embed_dim": 1280, | |
| "micro_cond_encode_dim": 256, | |
| "num_attention_heads": 16, | |
| "num_hidden_layers": 22, | |
| "num_res_blocks": 3, | |
| "sample_size": 32, | |
| "upsample": true, | |
| "use_bias": false, | |
| "vocab_size": 8256 | |
| } | |