| { |
| "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" |
| } |
| } |
|
|
