File size: 12,926 Bytes
7c133d6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 | {
"format": "dirtflow-engine-bundle",
"version": 1,
"weight_set": "stage1-ema+stage2-live",
"module_weight_sets": {
"context_encoder": "ema",
"structure_flow": "ema",
"structure_extra": "ema",
"latent_norm": "ema",
"ss_encoder": "ema",
"ss_decoder": "ema",
"blockstate_flow": "live",
"blockstate_extra": "live",
"embedding": "live",
"categorical_head": "live"
},
"source_snapshot": "v1-20261025-corrected-confidence",
"inference_policy": {
"editable_confidence": "zero_normalized_editable",
"parameter_storage": "float32",
"position_frequencies": "native_float32"
},
"lineage": {
"shared_assets": {
"lineage.clip_tokenizer_weights_sha256": "530343e7da77b47e6058dee55f8a4a770d56fa701e5826786a196d464a87e530",
"lineage.input_output_maps_sha256": "0f33ac5d693cd1d431e9c7439f1e39916acf571e92aa8855916e819d9368838b",
"lineage.legal_state_schema_sha256": "23ddffef121253fd200e1ff82cc7af86b0861fb225b02072ae15de1b9a603450",
"lineage.ss_vae_sha256": "b8a3025a8c1436ea7d084b5087d2e5257a5b22af12388cb09f0dd7e04917b678",
"lineage.latent_normalizer_sha256": "8b51b9176c21ce677e49b0322c76029aa2060ef70896af7a43c480261583ac59",
"lineage.stage1_seed_sha256": "59a7258b725dfdd5339063fbfb0c80fad4d470b71250ce746840c3adced49c13"
},
"exact_stage1_binding": "not_recorded_by_stage2",
"selected_stage1_sha256": "496614fb6d642f6b3bd455b23afdf3343081067602a7469da804025fa8875174"
},
"model": {
"blockstate_coordinate_system": "trellis_z_up",
"blockstate_flow": {
"checkpoint_min_tokens": 0,
"checkpoint_policy": "block",
"cond_channels": 768,
"in_channels": 42,
"io_block_channels": 256,
"latent_channels": 8,
"mlp_ratio": 4,
"model_channels": 1280,
"num_blocks": 28,
"num_heads": 16,
"num_io_res_blocks": 3,
"out_channels": 8,
"patch_size": 2,
"qk_rms_norm": true,
"resolution": 64,
"self_conditioning": true,
"use_checkpoint": true
},
"categorical_head": {
"hidden": 256,
"io_features": true,
"kind": "categorical_independent",
"output_mapping_path": "assets/class-mapping.npz",
"output_vocab": 29669
},
"conditioning": {
"context_dropout": 0.1,
"num_styles": 5,
"score_range": [
0.0,
10.0
],
"style_dropout": 0.5,
"text_dropout": 0.1
},
"context_encoder": {
"in_channels": 20,
"out_channels": 16,
"widths": [
32,
64,
128
]
},
"embedding": {
"dim": 8,
"init_path": "",
"rms_norm": true,
"vocab_size": 48407
},
"feature_schema": "v1",
"size": "xlarge",
"ss_vae": {
"channels": [
32,
128,
512
],
"latent_channels": 8,
"latent_resolution": 16,
"num_res_blocks": 2,
"num_res_blocks_middle": 2,
"resolution": 64
},
"state_schema_path": "assets/game-schema.json.gz",
"structure_coordinate_system": "trellis_z_up",
"structure_flow": {
"checkpoint_policy": "block",
"checkpoint_stride": 1,
"cond_channels": 768,
"in_channels": 32,
"latent_channels": 8,
"mlp_ratio": 4,
"model_channels": 1280,
"num_blocks": 28,
"num_heads": 16,
"out_channels": 8,
"qk_rms_norm": true,
"qk_rms_norm_cross": true,
"resolution": 16,
"self_conditioning": true,
"use_checkpoint": true
},
"text_encoder": {
"frozen": true,
"kind": "clip",
"model_id": "openai/clip-vit-large-patch14",
"use_attention_mask": false
},
"trellis_repo_id": "microsoft/TRELLIS-text-xlarge",
"trellis_revision": "e0b00432b8e3a8ecee0df806ab1df9f7281f2be4"
},
"modules": {
"context_encoder": "stage1.pt",
"structure_flow": "stage1.pt",
"structure_extra": "stage1.pt",
"latent_norm": "stage1.pt",
"ss_encoder": "stage1.pt",
"ss_decoder": "stage1.pt",
"blockstate_flow": "stage2.pt",
"blockstate_extra": "stage2.pt",
"embedding": "stage2.pt",
"categorical_head": "stage2.pt"
},
"sources": {
"stage1": {
"sha256": "496614fb6d642f6b3bd455b23afdf3343081067602a7469da804025fa8875174",
"step": 340000,
"frozen_exceptions": [
"ss_decoder.blocks.0.conv1.bias",
"ss_decoder.blocks.0.conv1.weight",
"ss_decoder.blocks.0.conv2.bias",
"ss_decoder.blocks.0.conv2.weight",
"ss_decoder.blocks.0.norm1.bias",
"ss_decoder.blocks.0.norm1.weight",
"ss_decoder.blocks.0.norm2.bias",
"ss_decoder.blocks.0.norm2.weight",
"ss_decoder.blocks.1.conv1.bias",
"ss_decoder.blocks.1.conv1.weight",
"ss_decoder.blocks.1.conv2.bias",
"ss_decoder.blocks.1.conv2.weight",
"ss_decoder.blocks.1.norm1.bias",
"ss_decoder.blocks.1.norm1.weight",
"ss_decoder.blocks.1.norm2.bias",
"ss_decoder.blocks.1.norm2.weight",
"ss_decoder.blocks.2.conv.bias",
"ss_decoder.blocks.2.conv.weight",
"ss_decoder.blocks.3.conv1.bias",
"ss_decoder.blocks.3.conv1.weight",
"ss_decoder.blocks.3.conv2.bias",
"ss_decoder.blocks.3.conv2.weight",
"ss_decoder.blocks.3.norm1.bias",
"ss_decoder.blocks.3.norm1.weight",
"ss_decoder.blocks.3.norm2.bias",
"ss_decoder.blocks.3.norm2.weight",
"ss_decoder.blocks.4.conv1.bias",
"ss_decoder.blocks.4.conv1.weight",
"ss_decoder.blocks.4.conv2.bias",
"ss_decoder.blocks.4.conv2.weight",
"ss_decoder.blocks.4.norm1.bias",
"ss_decoder.blocks.4.norm1.weight",
"ss_decoder.blocks.4.norm2.bias",
"ss_decoder.blocks.4.norm2.weight",
"ss_decoder.blocks.5.conv.bias",
"ss_decoder.blocks.5.conv.weight",
"ss_decoder.blocks.6.conv1.bias",
"ss_decoder.blocks.6.conv1.weight",
"ss_decoder.blocks.6.conv2.bias",
"ss_decoder.blocks.6.conv2.weight",
"ss_decoder.blocks.6.norm1.bias",
"ss_decoder.blocks.6.norm1.weight",
"ss_decoder.blocks.6.norm2.bias",
"ss_decoder.blocks.6.norm2.weight",
"ss_decoder.blocks.7.conv1.bias",
"ss_decoder.blocks.7.conv1.weight",
"ss_decoder.blocks.7.conv2.bias",
"ss_decoder.blocks.7.conv2.weight",
"ss_decoder.blocks.7.norm1.bias",
"ss_decoder.blocks.7.norm1.weight",
"ss_decoder.blocks.7.norm2.bias",
"ss_decoder.blocks.7.norm2.weight",
"ss_decoder.input_layer.bias",
"ss_decoder.input_layer.weight",
"ss_decoder.middle_block.0.conv1.bias",
"ss_decoder.middle_block.0.conv1.weight",
"ss_decoder.middle_block.0.conv2.bias",
"ss_decoder.middle_block.0.conv2.weight",
"ss_decoder.middle_block.0.norm1.bias",
"ss_decoder.middle_block.0.norm1.weight",
"ss_decoder.middle_block.0.norm2.bias",
"ss_decoder.middle_block.0.norm2.weight",
"ss_decoder.middle_block.1.conv1.bias",
"ss_decoder.middle_block.1.conv1.weight",
"ss_decoder.middle_block.1.conv2.bias",
"ss_decoder.middle_block.1.conv2.weight",
"ss_decoder.middle_block.1.norm1.bias",
"ss_decoder.middle_block.1.norm1.weight",
"ss_decoder.middle_block.1.norm2.bias",
"ss_decoder.middle_block.1.norm2.weight",
"ss_decoder.out_layer.0.bias",
"ss_decoder.out_layer.0.weight",
"ss_decoder.out_layer.2.bias",
"ss_decoder.out_layer.2.weight",
"ss_encoder.blocks.0.conv1.bias",
"ss_encoder.blocks.0.conv1.weight",
"ss_encoder.blocks.0.conv2.bias",
"ss_encoder.blocks.0.conv2.weight",
"ss_encoder.blocks.0.norm1.bias",
"ss_encoder.blocks.0.norm1.weight",
"ss_encoder.blocks.0.norm2.bias",
"ss_encoder.blocks.0.norm2.weight",
"ss_encoder.blocks.1.conv1.bias",
"ss_encoder.blocks.1.conv1.weight",
"ss_encoder.blocks.1.conv2.bias",
"ss_encoder.blocks.1.conv2.weight",
"ss_encoder.blocks.1.norm1.bias",
"ss_encoder.blocks.1.norm1.weight",
"ss_encoder.blocks.1.norm2.bias",
"ss_encoder.blocks.1.norm2.weight",
"ss_encoder.blocks.2.conv.bias",
"ss_encoder.blocks.2.conv.weight",
"ss_encoder.blocks.3.conv1.bias",
"ss_encoder.blocks.3.conv1.weight",
"ss_encoder.blocks.3.conv2.bias",
"ss_encoder.blocks.3.conv2.weight",
"ss_encoder.blocks.3.norm1.bias",
"ss_encoder.blocks.3.norm1.weight",
"ss_encoder.blocks.3.norm2.bias",
"ss_encoder.blocks.3.norm2.weight",
"ss_encoder.blocks.4.conv1.bias",
"ss_encoder.blocks.4.conv1.weight",
"ss_encoder.blocks.4.conv2.bias",
"ss_encoder.blocks.4.conv2.weight",
"ss_encoder.blocks.4.norm1.bias",
"ss_encoder.blocks.4.norm1.weight",
"ss_encoder.blocks.4.norm2.bias",
"ss_encoder.blocks.4.norm2.weight",
"ss_encoder.blocks.5.conv.bias",
"ss_encoder.blocks.5.conv.weight",
"ss_encoder.blocks.6.conv1.bias",
"ss_encoder.blocks.6.conv1.weight",
"ss_encoder.blocks.6.conv2.bias",
"ss_encoder.blocks.6.conv2.weight",
"ss_encoder.blocks.6.norm1.bias",
"ss_encoder.blocks.6.norm1.weight",
"ss_encoder.blocks.6.norm2.bias",
"ss_encoder.blocks.6.norm2.weight",
"ss_encoder.blocks.7.conv1.bias",
"ss_encoder.blocks.7.conv1.weight",
"ss_encoder.blocks.7.conv2.bias",
"ss_encoder.blocks.7.conv2.weight",
"ss_encoder.blocks.7.norm1.bias",
"ss_encoder.blocks.7.norm1.weight",
"ss_encoder.blocks.7.norm2.bias",
"ss_encoder.blocks.7.norm2.weight",
"ss_encoder.input_layer.bias",
"ss_encoder.input_layer.weight",
"ss_encoder.middle_block.0.conv1.bias",
"ss_encoder.middle_block.0.conv1.weight",
"ss_encoder.middle_block.0.conv2.bias",
"ss_encoder.middle_block.0.conv2.weight",
"ss_encoder.middle_block.0.norm1.bias",
"ss_encoder.middle_block.0.norm1.weight",
"ss_encoder.middle_block.0.norm2.bias",
"ss_encoder.middle_block.0.norm2.weight",
"ss_encoder.middle_block.1.conv1.bias",
"ss_encoder.middle_block.1.conv1.weight",
"ss_encoder.middle_block.1.conv2.bias",
"ss_encoder.middle_block.1.conv2.weight",
"ss_encoder.middle_block.1.norm1.bias",
"ss_encoder.middle_block.1.norm1.weight",
"ss_encoder.middle_block.1.norm2.bias",
"ss_encoder.middle_block.1.norm2.weight",
"ss_encoder.out_layer.0.bias",
"ss_encoder.out_layer.0.weight",
"ss_encoder.out_layer.2.bias",
"ss_encoder.out_layer.2.weight"
]
},
"stage2": {
"sha256": "ef9e5759db275b68e352ececc02dd38d12a49745a6fd000b2c5ad0513a3e7f4a",
"step": 400000,
"frozen_exceptions": []
}
},
"files": {
"assets/class-mapping.npz": "0f33ac5d693cd1d431e9c7439f1e39916acf571e92aa8855916e819d9368838b",
"assets/clip/config.json": "de5669886f862d6cf5065bcd5444a36dc5ded8e07b2a2efe8abe17d6f603e428",
"assets/clip/merges.txt": "9fd691f7c8039210e0fced15865466c65820d09b63988b0174bfe25de299051a",
"assets/clip/model.safetensors": "778d02eb9e707c3fbaae0b67b79ea0d1399b52e624fb634f2f19375ae7c047c3",
"assets/clip/special_tokens_map.json": "f8c0d6c39aee3f8431078ef6646567b0aba7f2246e9c54b8b99d55c22b707cbf",
"assets/clip/tokenizer.json": "a83e0809aa4c3af7208b2df632a7a69668c6d48775b3c3fe4e1b1199d1f8b8f4",
"assets/clip/tokenizer_config.json": "deef455e52fa5e8151e339add0582e4235f066009601360999d3a9cda83b1129",
"assets/clip/vocab.json": "3f0c4f7d2086b61b38487075278ea9ed04edb53a03cbb045b86c27190fa8fb69",
"assets/clip-derivation.json": "f028e9160e000eb1d08a8163b3be4557449ae503684734d7b0e9384e4c93e9e8",
"assets/clip-manifest.json": "530343e7da77b47e6058dee55f8a4a770d56fa701e5826786a196d464a87e530",
"assets/clip-source-config.json": "8a09b467700c58138c29d53c605b34ebc69beaadd13274a8a2af8ad2c2f4032a",
"assets/game-schema.json.gz": "23ddffef121253fd200e1ff82cc7af86b0861fb225b02072ae15de1b9a603450",
"assets/input-registry.json": "c7a7fb72729daec63174fb77eaa821c340706fb3d22976eacd4a5a77c5e88de0",
"stage1.pt": "2c54040a92c1d1826bba62d2cff260b972d99f9bc9d5fce516e5f98adc49322a",
"stage2.pt": "5ad094fb2ba377cfcfb703e75b473173d5a54b5425436978a10354f541ef6fb8"
},
"assets": {
"registry": "assets/input-registry.json",
"schema": "assets/game-schema.json.gz",
"mapping": "assets/class-mapping.npz",
"text_encoder": "assets/clip",
"clip_manifest": "assets/clip-manifest.json",
"clip_source_config": "assets/clip-source-config.json",
"clip_derivation": "assets/clip-derivation.json"
}
}
|