| { | |
| "ddconfig": { | |
| "attn_resolutions": [], | |
| "ch": 128, | |
| "ch_fuse": [ | |
| -1, | |
| -1, | |
| 2, | |
| 3, | |
| -1, | |
| -1 | |
| ], | |
| "ch_in": 128, | |
| "ch_mult": [ | |
| 1, | |
| 2, | |
| 2, | |
| 4, | |
| 4, | |
| 4 | |
| ], | |
| "f_t": 2, | |
| "fp32_attention": true, | |
| "fuse_mid": true, | |
| "fuse_type": "attn", | |
| "in_channels": 3, | |
| "num_res_blocks": 2, | |
| "out_ch": 3, | |
| "pos_embed_mode": "t", | |
| "resolution": 256, | |
| "temp_res": 16, | |
| "upsample_first": true, | |
| "use_3d_conv": true, | |
| "z_channels": 16 | |
| }, | |
| "ddconfig_2d": { | |
| "attn_resolutions": [], | |
| "ch": 128, | |
| "ch_mult": [ | |
| 1, | |
| 2, | |
| 4, | |
| 4 | |
| ], | |
| "dropout": 0.0, | |
| "fp32_attention": true, | |
| "in_channels": 3, | |
| "num_res_blocks": 2, | |
| "out_ch": 3, | |
| "out_z": false, | |
| "resolution": 256, | |
| "z_channels": 4 | |
| }, | |
| "embed_dim": 16, | |
| "enable_2d": true, | |
| "model_type": "VAE", | |
| "tile_overlap_factor": 0.25, | |
| "tile_sample_min_size": 256, | |
| "transformers_version": "4.39.3", | |
| "use_tiling": true | |
| } | |