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