{ "format": "trellis2-selective-int8-weight-only", "format_version": 1, "source": "microsoft/TRELLIS.2-4B", "pipelines": [ "pipeline.json", "texturing_pipeline.json" ], "models": { "vendor/microsoft/TRELLIS-image-large/ckpts/ss_dec_conv3d_16l8_fp16": { "quantized": false, "checkpoint": "vendor/microsoft/TRELLIS-image-large/ckpts/ss_dec_conv3d_16l8_fp16.safetensors", "config": "vendor/microsoft/TRELLIS-image-large/ckpts/ss_dec_conv3d_16l8_fp16.json", "model_key": "sparse_structure_decoder" }, "ckpts/ss_flow_img_dit_1_3B_64_bf16": { "quantized": false, "checkpoint": "ckpts/ss_flow_img_dit_1_3B_64_bf16.safetensors", "config": "ckpts/ss_flow_img_dit_1_3B_64_bf16.json", "model_key": "sparse_structure_flow_model" }, "ckpts/shape_dec_next_dc_f16c32_fp16": { "quantized": false, "checkpoint": "ckpts/shape_dec_next_dc_f16c32_fp16.safetensors", "config": "ckpts/shape_dec_next_dc_f16c32_fp16.json", "model_key": "shape_slat_decoder" }, "ckpts/slat_flow_img2shape_dit_1_3B_512_bf16": { "quantized": true, "checkpoint": "ckpts/slat_flow_img2shape_dit_1_3B_512_bf16.int8.pt", "config": "ckpts/slat_flow_img2shape_dit_1_3B_512_bf16.json", "quantized_fqns": [ "blocks.0.cross_attn.to_kv", "blocks.0.cross_attn.to_out", "blocks.0.cross_attn.to_q", "blocks.0.mlp.mlp.0", "blocks.0.mlp.mlp.2", "blocks.0.self_attn.to_out", "blocks.0.self_attn.to_qkv", "blocks.1.cross_attn.to_kv", "blocks.1.cross_attn.to_out", "blocks.1.cross_attn.to_q", "blocks.1.mlp.mlp.0", "blocks.1.mlp.mlp.2", "blocks.1.self_attn.to_out", "blocks.1.self_attn.to_qkv", "blocks.10.cross_attn.to_kv", "blocks.10.cross_attn.to_out", "blocks.10.cross_attn.to_q", "blocks.10.mlp.mlp.0", "blocks.10.mlp.mlp.2", "blocks.10.self_attn.to_out", "blocks.10.self_attn.to_qkv", "blocks.11.cross_attn.to_kv", "blocks.11.cross_attn.to_out", "blocks.11.cross_attn.to_q", "blocks.11.mlp.mlp.0", "blocks.11.mlp.mlp.2", "blocks.11.self_attn.to_out", "blocks.11.self_attn.to_qkv", "blocks.12.cross_attn.to_kv", "blocks.12.cross_attn.to_out", "blocks.12.cross_attn.to_q", "blocks.12.mlp.mlp.0", "blocks.12.mlp.mlp.2", "blocks.12.self_attn.to_out", "blocks.12.self_attn.to_qkv", "blocks.13.cross_attn.to_kv", "blocks.13.cross_attn.to_out", "blocks.13.cross_attn.to_q", "blocks.13.mlp.mlp.0", "blocks.13.mlp.mlp.2", "blocks.13.self_attn.to_out", "blocks.13.self_attn.to_qkv", "blocks.14.cross_attn.to_kv", "blocks.14.cross_attn.to_out", "blocks.14.cross_attn.to_q", "blocks.14.mlp.mlp.0", "blocks.14.mlp.mlp.2", "blocks.14.self_attn.to_out", "blocks.14.self_attn.to_qkv", "blocks.15.cross_attn.to_kv", "blocks.15.cross_attn.to_out", "blocks.15.cross_attn.to_q", "blocks.15.mlp.mlp.0", "blocks.15.mlp.mlp.2", "blocks.15.self_attn.to_out", "blocks.15.self_attn.to_qkv", "blocks.16.cross_attn.to_kv", "blocks.16.cross_attn.to_out", "blocks.16.cross_attn.to_q", "blocks.16.mlp.mlp.0", "blocks.16.mlp.mlp.2", "blocks.16.self_attn.to_out", "blocks.16.self_attn.to_qkv", "blocks.17.cross_attn.to_kv", "blocks.17.cross_attn.to_out", "blocks.17.cross_attn.to_q", "blocks.17.mlp.mlp.0", "blocks.17.mlp.mlp.2", "blocks.17.self_attn.to_out", "blocks.17.self_attn.to_qkv", "blocks.18.cross_attn.to_kv", "blocks.18.cross_attn.to_out", "blocks.18.cross_attn.to_q", "blocks.18.mlp.mlp.0", "blocks.18.mlp.mlp.2", "blocks.18.self_attn.to_out", "blocks.18.self_attn.to_qkv", "blocks.19.cross_attn.to_kv", "blocks.19.cross_attn.to_out", "blocks.19.cross_attn.to_q", "blocks.19.mlp.mlp.0", "blocks.19.mlp.mlp.2", "blocks.19.self_attn.to_out", "blocks.19.self_attn.to_qkv", "blocks.2.cross_attn.to_kv", "blocks.2.cross_attn.to_out", "blocks.2.cross_attn.to_q", "blocks.2.mlp.mlp.0", "blocks.2.mlp.mlp.2", "blocks.2.self_attn.to_out", "blocks.2.self_attn.to_qkv", "blocks.20.cross_attn.to_kv", "blocks.20.cross_attn.to_out", "blocks.20.cross_attn.to_q", "blocks.20.mlp.mlp.0", "blocks.20.mlp.mlp.2", "blocks.20.self_attn.to_out", "blocks.20.self_attn.to_qkv", "blocks.21.cross_attn.to_kv", "blocks.21.cross_attn.to_out", "blocks.21.cross_attn.to_q", "blocks.21.mlp.mlp.0", "blocks.21.mlp.mlp.2", "blocks.21.self_attn.to_out", "blocks.21.self_attn.to_qkv", "blocks.22.cross_attn.to_kv", "blocks.22.cross_attn.to_out", "blocks.22.cross_attn.to_q", "blocks.22.mlp.mlp.0", "blocks.22.mlp.mlp.2", "blocks.22.self_attn.to_out", "blocks.22.self_attn.to_qkv", "blocks.23.cross_attn.to_kv", "blocks.23.cross_attn.to_out", "blocks.23.cross_attn.to_q", "blocks.23.mlp.mlp.0", "blocks.23.mlp.mlp.2", "blocks.23.self_attn.to_out", "blocks.23.self_attn.to_qkv", "blocks.24.cross_attn.to_kv", "blocks.24.cross_attn.to_out", "blocks.24.cross_attn.to_q", "blocks.24.mlp.mlp.0", "blocks.24.mlp.mlp.2", "blocks.24.self_attn.to_out", "blocks.24.self_attn.to_qkv", "blocks.25.cross_attn.to_kv", "blocks.25.cross_attn.to_out", "blocks.25.cross_attn.to_q", "blocks.25.mlp.mlp.0", "blocks.25.mlp.mlp.2", "blocks.25.self_attn.to_out", "blocks.25.self_attn.to_qkv", "blocks.26.cross_attn.to_kv", "blocks.26.cross_attn.to_out", "blocks.26.cross_attn.to_q", "blocks.26.mlp.mlp.0", "blocks.26.mlp.mlp.2", "blocks.26.self_attn.to_out", "blocks.26.self_attn.to_qkv", "blocks.27.cross_attn.to_kv", "blocks.27.cross_attn.to_out", "blocks.27.cross_attn.to_q", "blocks.27.mlp.mlp.0", "blocks.27.mlp.mlp.2", "blocks.27.self_attn.to_out", "blocks.27.self_attn.to_qkv", "blocks.28.cross_attn.to_kv", "blocks.28.cross_attn.to_out", "blocks.28.cross_attn.to_q", "blocks.28.mlp.mlp.0", "blocks.28.mlp.mlp.2", "blocks.28.self_attn.to_out", "blocks.28.self_attn.to_qkv", "blocks.29.cross_attn.to_kv", "blocks.29.cross_attn.to_out", "blocks.29.cross_attn.to_q", "blocks.29.mlp.mlp.0", "blocks.29.mlp.mlp.2", "blocks.29.self_attn.to_out", "blocks.29.self_attn.to_qkv", "blocks.3.cross_attn.to_kv", "blocks.3.cross_attn.to_out", "blocks.3.cross_attn.to_q", "blocks.3.mlp.mlp.0", "blocks.3.mlp.mlp.2", "blocks.3.self_attn.to_out", "blocks.3.self_attn.to_qkv", "blocks.4.cross_attn.to_kv", "blocks.4.cross_attn.to_out", "blocks.4.cross_attn.to_q", "blocks.4.mlp.mlp.0", "blocks.4.mlp.mlp.2", "blocks.4.self_attn.to_out", "blocks.4.self_attn.to_qkv", "blocks.5.cross_attn.to_kv", "blocks.5.cross_attn.to_out", "blocks.5.cross_attn.to_q", "blocks.5.mlp.mlp.0", "blocks.5.mlp.mlp.2", "blocks.5.self_attn.to_out", "blocks.5.self_attn.to_qkv", "blocks.6.cross_attn.to_kv", "blocks.6.cross_attn.to_out", "blocks.6.cross_attn.to_q", "blocks.6.mlp.mlp.0", "blocks.6.mlp.mlp.2", "blocks.6.self_attn.to_out", "blocks.6.self_attn.to_qkv", "blocks.7.cross_attn.to_kv", "blocks.7.cross_attn.to_out", "blocks.7.cross_attn.to_q", "blocks.7.mlp.mlp.0", "blocks.7.mlp.mlp.2", "blocks.7.self_attn.to_out", "blocks.7.self_attn.to_qkv", "blocks.8.cross_attn.to_kv", "blocks.8.cross_attn.to_out", "blocks.8.cross_attn.to_q", "blocks.8.mlp.mlp.0", "blocks.8.mlp.mlp.2", "blocks.8.self_attn.to_out", "blocks.8.self_attn.to_qkv", "blocks.9.cross_attn.to_kv", "blocks.9.cross_attn.to_out", "blocks.9.cross_attn.to_q", "blocks.9.mlp.mlp.0", "blocks.9.mlp.mlp.2", "blocks.9.self_attn.to_out", "blocks.9.self_attn.to_qkv" ], "model_key": "shape_slat_flow_model_512" }, "ckpts/slat_flow_img2shape_dit_1_3B_1024_bf16": { "quantized": true, "checkpoint": "ckpts/slat_flow_img2shape_dit_1_3B_1024_bf16.int8.pt", "config": "ckpts/slat_flow_img2shape_dit_1_3B_1024_bf16.json", "quantized_fqns": [ "blocks.0.cross_attn.to_kv", "blocks.0.cross_attn.to_out", "blocks.0.cross_attn.to_q", "blocks.0.mlp.mlp.0", "blocks.0.mlp.mlp.2", "blocks.0.self_attn.to_out", "blocks.0.self_attn.to_qkv", "blocks.1.cross_attn.to_kv", "blocks.1.cross_attn.to_out", "blocks.1.cross_attn.to_q", "blocks.1.mlp.mlp.0", "blocks.1.mlp.mlp.2", "blocks.1.self_attn.to_out", "blocks.1.self_attn.to_qkv", "blocks.10.cross_attn.to_kv", "blocks.10.cross_attn.to_out", "blocks.10.cross_attn.to_q", "blocks.10.mlp.mlp.0", "blocks.10.mlp.mlp.2", "blocks.10.self_attn.to_out", "blocks.10.self_attn.to_qkv", "blocks.11.cross_attn.to_kv", "blocks.11.cross_attn.to_out", "blocks.11.cross_attn.to_q", "blocks.11.mlp.mlp.0", "blocks.11.mlp.mlp.2", "blocks.11.self_attn.to_out", "blocks.11.self_attn.to_qkv", "blocks.12.cross_attn.to_kv", "blocks.12.cross_attn.to_out", "blocks.12.cross_attn.to_q", "blocks.12.mlp.mlp.0", "blocks.12.mlp.mlp.2", "blocks.12.self_attn.to_out", "blocks.12.self_attn.to_qkv", "blocks.13.cross_attn.to_kv", "blocks.13.cross_attn.to_out", "blocks.13.cross_attn.to_q", "blocks.13.mlp.mlp.0", "blocks.13.mlp.mlp.2", "blocks.13.self_attn.to_out", "blocks.13.self_attn.to_qkv", "blocks.14.cross_attn.to_kv", "blocks.14.cross_attn.to_out", "blocks.14.cross_attn.to_q", "blocks.14.mlp.mlp.0", "blocks.14.mlp.mlp.2", "blocks.14.self_attn.to_out", "blocks.14.self_attn.to_qkv", "blocks.15.cross_attn.to_kv", "blocks.15.cross_attn.to_out", "blocks.15.cross_attn.to_q", "blocks.15.mlp.mlp.0", "blocks.15.mlp.mlp.2", "blocks.15.self_attn.to_out", "blocks.15.self_attn.to_qkv", "blocks.16.cross_attn.to_kv", "blocks.16.cross_attn.to_out", "blocks.16.cross_attn.to_q", "blocks.16.mlp.mlp.0", "blocks.16.mlp.mlp.2", "blocks.16.self_attn.to_out", "blocks.16.self_attn.to_qkv", "blocks.17.cross_attn.to_kv", "blocks.17.cross_attn.to_out", "blocks.17.cross_attn.to_q", "blocks.17.mlp.mlp.0", "blocks.17.mlp.mlp.2", "blocks.17.self_attn.to_out", "blocks.17.self_attn.to_qkv", "blocks.18.cross_attn.to_kv", "blocks.18.cross_attn.to_out", "blocks.18.cross_attn.to_q", "blocks.18.mlp.mlp.0", "blocks.18.mlp.mlp.2", "blocks.18.self_attn.to_out", "blocks.18.self_attn.to_qkv", "blocks.19.cross_attn.to_kv", "blocks.19.cross_attn.to_out", "blocks.19.cross_attn.to_q", "blocks.19.mlp.mlp.0", "blocks.19.mlp.mlp.2", "blocks.19.self_attn.to_out", "blocks.19.self_attn.to_qkv", "blocks.2.cross_attn.to_kv", "blocks.2.cross_attn.to_out", "blocks.2.cross_attn.to_q", "blocks.2.mlp.mlp.0", "blocks.2.mlp.mlp.2", "blocks.2.self_attn.to_out", "blocks.2.self_attn.to_qkv", "blocks.20.cross_attn.to_kv", "blocks.20.cross_attn.to_out", "blocks.20.cross_attn.to_q", "blocks.20.mlp.mlp.0", "blocks.20.mlp.mlp.2", "blocks.20.self_attn.to_out", "blocks.20.self_attn.to_qkv", "blocks.21.cross_attn.to_kv", "blocks.21.cross_attn.to_out", "blocks.21.cross_attn.to_q", "blocks.21.mlp.mlp.0", "blocks.21.mlp.mlp.2", "blocks.21.self_attn.to_out", "blocks.21.self_attn.to_qkv", "blocks.22.cross_attn.to_kv", "blocks.22.cross_attn.to_out", "blocks.22.cross_attn.to_q", "blocks.22.mlp.mlp.0", "blocks.22.mlp.mlp.2", "blocks.22.self_attn.to_out", "blocks.22.self_attn.to_qkv", "blocks.23.cross_attn.to_kv", "blocks.23.cross_attn.to_out", "blocks.23.cross_attn.to_q", "blocks.23.mlp.mlp.0", "blocks.23.mlp.mlp.2", "blocks.23.self_attn.to_out", "blocks.23.self_attn.to_qkv", "blocks.24.cross_attn.to_kv", "blocks.24.cross_attn.to_out", "blocks.24.cross_attn.to_q", "blocks.24.mlp.mlp.0", "blocks.24.mlp.mlp.2", "blocks.24.self_attn.to_out", "blocks.24.self_attn.to_qkv", "blocks.25.cross_attn.to_kv", "blocks.25.cross_attn.to_out", "blocks.25.cross_attn.to_q", "blocks.25.mlp.mlp.0", "blocks.25.mlp.mlp.2", "blocks.25.self_attn.to_out", "blocks.25.self_attn.to_qkv", "blocks.26.cross_attn.to_kv", "blocks.26.cross_attn.to_out", "blocks.26.cross_attn.to_q", "blocks.26.mlp.mlp.0", "blocks.26.mlp.mlp.2", "blocks.26.self_attn.to_out", "blocks.26.self_attn.to_qkv", "blocks.27.cross_attn.to_kv", "blocks.27.cross_attn.to_out", "blocks.27.cross_attn.to_q", "blocks.27.mlp.mlp.0", "blocks.27.mlp.mlp.2", "blocks.27.self_attn.to_out", "blocks.27.self_attn.to_qkv", "blocks.28.cross_attn.to_kv", "blocks.28.cross_attn.to_out", "blocks.28.cross_attn.to_q", "blocks.28.mlp.mlp.0", "blocks.28.mlp.mlp.2", "blocks.28.self_attn.to_out", "blocks.28.self_attn.to_qkv", "blocks.29.cross_attn.to_kv", "blocks.29.cross_attn.to_out", "blocks.29.cross_attn.to_q", "blocks.29.mlp.mlp.0", "blocks.29.mlp.mlp.2", "blocks.29.self_attn.to_out", "blocks.29.self_attn.to_qkv", "blocks.3.cross_attn.to_kv", "blocks.3.cross_attn.to_out", "blocks.3.cross_attn.to_q", "blocks.3.mlp.mlp.0", "blocks.3.mlp.mlp.2", "blocks.3.self_attn.to_out", "blocks.3.self_attn.to_qkv", "blocks.4.cross_attn.to_kv", "blocks.4.cross_attn.to_out", "blocks.4.cross_attn.to_q", "blocks.4.mlp.mlp.0", "blocks.4.mlp.mlp.2", "blocks.4.self_attn.to_out", "blocks.4.self_attn.to_qkv", "blocks.5.cross_attn.to_kv", "blocks.5.cross_attn.to_out", "blocks.5.cross_attn.to_q", "blocks.5.mlp.mlp.0", "blocks.5.mlp.mlp.2", "blocks.5.self_attn.to_out", "blocks.5.self_attn.to_qkv", "blocks.6.cross_attn.to_kv", "blocks.6.cross_attn.to_out", "blocks.6.cross_attn.to_q", "blocks.6.mlp.mlp.0", "blocks.6.mlp.mlp.2", "blocks.6.self_attn.to_out", "blocks.6.self_attn.to_qkv", "blocks.7.cross_attn.to_kv", "blocks.7.cross_attn.to_out", "blocks.7.cross_attn.to_q", "blocks.7.mlp.mlp.0", "blocks.7.mlp.mlp.2", "blocks.7.self_attn.to_out", "blocks.7.self_attn.to_qkv", "blocks.8.cross_attn.to_kv", "blocks.8.cross_attn.to_out", "blocks.8.cross_attn.to_q", "blocks.8.mlp.mlp.0", "blocks.8.mlp.mlp.2", "blocks.8.self_attn.to_out", "blocks.8.self_attn.to_qkv", "blocks.9.cross_attn.to_kv", "blocks.9.cross_attn.to_out", "blocks.9.cross_attn.to_q", "blocks.9.mlp.mlp.0", "blocks.9.mlp.mlp.2", "blocks.9.self_attn.to_out", "blocks.9.self_attn.to_qkv" ], "model_key": "shape_slat_flow_model_1024" }, "ckpts/tex_dec_next_dc_f16c32_fp16": { "quantized": false, "checkpoint": "ckpts/tex_dec_next_dc_f16c32_fp16.safetensors", "config": "ckpts/tex_dec_next_dc_f16c32_fp16.json", "model_key": "tex_slat_decoder" }, "ckpts/slat_flow_imgshape2tex_dit_1_3B_512_bf16": { "quantized": true, "checkpoint": "ckpts/slat_flow_imgshape2tex_dit_1_3B_512_bf16.int8.pt", "config": "ckpts/slat_flow_imgshape2tex_dit_1_3B_512_bf16.json", "quantized_fqns": [ "blocks.0.cross_attn.to_kv", "blocks.0.cross_attn.to_out", "blocks.0.cross_attn.to_q", "blocks.0.mlp.mlp.0", "blocks.0.mlp.mlp.2", "blocks.0.self_attn.to_out", "blocks.0.self_attn.to_qkv", "blocks.1.cross_attn.to_kv", "blocks.1.cross_attn.to_out", "blocks.1.cross_attn.to_q", "blocks.1.mlp.mlp.0", "blocks.1.mlp.mlp.2", "blocks.1.self_attn.to_out", "blocks.1.self_attn.to_qkv", "blocks.10.cross_attn.to_kv", "blocks.10.cross_attn.to_out", "blocks.10.cross_attn.to_q", "blocks.10.mlp.mlp.0", "blocks.10.mlp.mlp.2", "blocks.10.self_attn.to_out", "blocks.10.self_attn.to_qkv", "blocks.11.cross_attn.to_kv", "blocks.11.cross_attn.to_out", "blocks.11.cross_attn.to_q", "blocks.11.mlp.mlp.0", "blocks.11.mlp.mlp.2", "blocks.11.self_attn.to_out", "blocks.11.self_attn.to_qkv", "blocks.12.cross_attn.to_kv", "blocks.12.cross_attn.to_out", "blocks.12.cross_attn.to_q", "blocks.12.mlp.mlp.0", "blocks.12.mlp.mlp.2", "blocks.12.self_attn.to_out", "blocks.12.self_attn.to_qkv", "blocks.13.cross_attn.to_kv", "blocks.13.cross_attn.to_out", "blocks.13.cross_attn.to_q", "blocks.13.mlp.mlp.0", "blocks.13.mlp.mlp.2", "blocks.13.self_attn.to_out", "blocks.13.self_attn.to_qkv", "blocks.14.cross_attn.to_kv", "blocks.14.cross_attn.to_out", "blocks.14.cross_attn.to_q", "blocks.14.mlp.mlp.0", "blocks.14.mlp.mlp.2", "blocks.14.self_attn.to_out", "blocks.14.self_attn.to_qkv", "blocks.15.cross_attn.to_kv", "blocks.15.cross_attn.to_out", "blocks.15.cross_attn.to_q", "blocks.15.mlp.mlp.0", "blocks.15.mlp.mlp.2", "blocks.15.self_attn.to_out", "blocks.15.self_attn.to_qkv", "blocks.16.cross_attn.to_kv", "blocks.16.cross_attn.to_out", "blocks.16.cross_attn.to_q", "blocks.16.mlp.mlp.0", "blocks.16.mlp.mlp.2", "blocks.16.self_attn.to_out", "blocks.16.self_attn.to_qkv", "blocks.17.cross_attn.to_kv", "blocks.17.cross_attn.to_out", "blocks.17.cross_attn.to_q", "blocks.17.mlp.mlp.0", "blocks.17.mlp.mlp.2", "blocks.17.self_attn.to_out", "blocks.17.self_attn.to_qkv", "blocks.18.cross_attn.to_kv", "blocks.18.cross_attn.to_out", "blocks.18.cross_attn.to_q", "blocks.18.mlp.mlp.0", "blocks.18.mlp.mlp.2", "blocks.18.self_attn.to_out", "blocks.18.self_attn.to_qkv", "blocks.19.cross_attn.to_kv", "blocks.19.cross_attn.to_out", "blocks.19.cross_attn.to_q", "blocks.19.mlp.mlp.0", "blocks.19.mlp.mlp.2", "blocks.19.self_attn.to_out", "blocks.19.self_attn.to_qkv", "blocks.2.cross_attn.to_kv", "blocks.2.cross_attn.to_out", "blocks.2.cross_attn.to_q", "blocks.2.mlp.mlp.0", "blocks.2.mlp.mlp.2", "blocks.2.self_attn.to_out", "blocks.2.self_attn.to_qkv", "blocks.20.cross_attn.to_kv", "blocks.20.cross_attn.to_out", "blocks.20.cross_attn.to_q", "blocks.20.mlp.mlp.0", "blocks.20.mlp.mlp.2", "blocks.20.self_attn.to_out", "blocks.20.self_attn.to_qkv", "blocks.21.cross_attn.to_kv", "blocks.21.cross_attn.to_out", "blocks.21.cross_attn.to_q", "blocks.21.mlp.mlp.0", "blocks.21.mlp.mlp.2", "blocks.21.self_attn.to_out", "blocks.21.self_attn.to_qkv", "blocks.22.cross_attn.to_kv", "blocks.22.cross_attn.to_out", "blocks.22.cross_attn.to_q", "blocks.22.mlp.mlp.0", "blocks.22.mlp.mlp.2", "blocks.22.self_attn.to_out", "blocks.22.self_attn.to_qkv", "blocks.23.cross_attn.to_kv", "blocks.23.cross_attn.to_out", "blocks.23.cross_attn.to_q", "blocks.23.mlp.mlp.0", "blocks.23.mlp.mlp.2", "blocks.23.self_attn.to_out", "blocks.23.self_attn.to_qkv", "blocks.24.cross_attn.to_kv", "blocks.24.cross_attn.to_out", "blocks.24.cross_attn.to_q", "blocks.24.mlp.mlp.0", "blocks.24.mlp.mlp.2", "blocks.24.self_attn.to_out", "blocks.24.self_attn.to_qkv", "blocks.25.cross_attn.to_kv", "blocks.25.cross_attn.to_out", "blocks.25.cross_attn.to_q", "blocks.25.mlp.mlp.0", "blocks.25.mlp.mlp.2", "blocks.25.self_attn.to_out", "blocks.25.self_attn.to_qkv", "blocks.26.cross_attn.to_kv", "blocks.26.cross_attn.to_out", "blocks.26.cross_attn.to_q", "blocks.26.mlp.mlp.0", "blocks.26.mlp.mlp.2", "blocks.26.self_attn.to_out", "blocks.26.self_attn.to_qkv", "blocks.27.cross_attn.to_kv", "blocks.27.cross_attn.to_out", "blocks.27.cross_attn.to_q", "blocks.27.mlp.mlp.0", "blocks.27.mlp.mlp.2", "blocks.27.self_attn.to_out", "blocks.27.self_attn.to_qkv", "blocks.28.cross_attn.to_kv", "blocks.28.cross_attn.to_out", "blocks.28.cross_attn.to_q", "blocks.28.mlp.mlp.0", "blocks.28.mlp.mlp.2", "blocks.28.self_attn.to_out", "blocks.28.self_attn.to_qkv", "blocks.29.cross_attn.to_kv", "blocks.29.cross_attn.to_out", "blocks.29.cross_attn.to_q", "blocks.29.mlp.mlp.0", "blocks.29.mlp.mlp.2", "blocks.29.self_attn.to_out", "blocks.29.self_attn.to_qkv", "blocks.3.cross_attn.to_kv", "blocks.3.cross_attn.to_out", "blocks.3.cross_attn.to_q", "blocks.3.mlp.mlp.0", "blocks.3.mlp.mlp.2", "blocks.3.self_attn.to_out", "blocks.3.self_attn.to_qkv", "blocks.4.cross_attn.to_kv", "blocks.4.cross_attn.to_out", "blocks.4.cross_attn.to_q", "blocks.4.mlp.mlp.0", "blocks.4.mlp.mlp.2", "blocks.4.self_attn.to_out", "blocks.4.self_attn.to_qkv", "blocks.5.cross_attn.to_kv", "blocks.5.cross_attn.to_out", "blocks.5.cross_attn.to_q", "blocks.5.mlp.mlp.0", "blocks.5.mlp.mlp.2", "blocks.5.self_attn.to_out", "blocks.5.self_attn.to_qkv", "blocks.6.cross_attn.to_kv", "blocks.6.cross_attn.to_out", "blocks.6.cross_attn.to_q", "blocks.6.mlp.mlp.0", "blocks.6.mlp.mlp.2", "blocks.6.self_attn.to_out", "blocks.6.self_attn.to_qkv", "blocks.7.cross_attn.to_kv", "blocks.7.cross_attn.to_out", "blocks.7.cross_attn.to_q", "blocks.7.mlp.mlp.0", "blocks.7.mlp.mlp.2", "blocks.7.self_attn.to_out", "blocks.7.self_attn.to_qkv", "blocks.8.cross_attn.to_kv", "blocks.8.cross_attn.to_out", "blocks.8.cross_attn.to_q", "blocks.8.mlp.mlp.0", "blocks.8.mlp.mlp.2", "blocks.8.self_attn.to_out", "blocks.8.self_attn.to_qkv", "blocks.9.cross_attn.to_kv", "blocks.9.cross_attn.to_out", "blocks.9.cross_attn.to_q", "blocks.9.mlp.mlp.0", "blocks.9.mlp.mlp.2", "blocks.9.self_attn.to_out", "blocks.9.self_attn.to_qkv" ], "model_key": "tex_slat_flow_model_512" }, "ckpts/slat_flow_imgshape2tex_dit_1_3B_1024_bf16": { "quantized": true, "checkpoint": "ckpts/slat_flow_imgshape2tex_dit_1_3B_1024_bf16.int8.pt", "config": "ckpts/slat_flow_imgshape2tex_dit_1_3B_1024_bf16.json", "quantized_fqns": [ "blocks.0.cross_attn.to_kv", "blocks.0.cross_attn.to_out", "blocks.0.cross_attn.to_q", "blocks.0.mlp.mlp.0", "blocks.0.mlp.mlp.2", "blocks.0.self_attn.to_out", "blocks.0.self_attn.to_qkv", "blocks.1.cross_attn.to_kv", "blocks.1.cross_attn.to_out", "blocks.1.cross_attn.to_q", "blocks.1.mlp.mlp.0", "blocks.1.mlp.mlp.2", "blocks.1.self_attn.to_out", "blocks.1.self_attn.to_qkv", "blocks.10.cross_attn.to_kv", "blocks.10.cross_attn.to_out", "blocks.10.cross_attn.to_q", "blocks.10.mlp.mlp.0", "blocks.10.mlp.mlp.2", "blocks.10.self_attn.to_out", "blocks.10.self_attn.to_qkv", "blocks.11.cross_attn.to_kv", "blocks.11.cross_attn.to_out", "blocks.11.cross_attn.to_q", "blocks.11.mlp.mlp.0", "blocks.11.mlp.mlp.2", "blocks.11.self_attn.to_out", "blocks.11.self_attn.to_qkv", "blocks.12.cross_attn.to_kv", "blocks.12.cross_attn.to_out", "blocks.12.cross_attn.to_q", "blocks.12.mlp.mlp.0", "blocks.12.mlp.mlp.2", "blocks.12.self_attn.to_out", "blocks.12.self_attn.to_qkv", "blocks.13.cross_attn.to_kv", "blocks.13.cross_attn.to_out", "blocks.13.cross_attn.to_q", "blocks.13.mlp.mlp.0", "blocks.13.mlp.mlp.2", "blocks.13.self_attn.to_out", "blocks.13.self_attn.to_qkv", "blocks.14.cross_attn.to_kv", "blocks.14.cross_attn.to_out", "blocks.14.cross_attn.to_q", "blocks.14.mlp.mlp.0", "blocks.14.mlp.mlp.2", "blocks.14.self_attn.to_out", "blocks.14.self_attn.to_qkv", "blocks.15.cross_attn.to_kv", "blocks.15.cross_attn.to_out", "blocks.15.cross_attn.to_q", "blocks.15.mlp.mlp.0", "blocks.15.mlp.mlp.2", "blocks.15.self_attn.to_out", "blocks.15.self_attn.to_qkv", "blocks.16.cross_attn.to_kv", "blocks.16.cross_attn.to_out", "blocks.16.cross_attn.to_q", "blocks.16.mlp.mlp.0", "blocks.16.mlp.mlp.2", "blocks.16.self_attn.to_out", "blocks.16.self_attn.to_qkv", "blocks.17.cross_attn.to_kv", "blocks.17.cross_attn.to_out", "blocks.17.cross_attn.to_q", "blocks.17.mlp.mlp.0", "blocks.17.mlp.mlp.2", "blocks.17.self_attn.to_out", "blocks.17.self_attn.to_qkv", "blocks.18.cross_attn.to_kv", "blocks.18.cross_attn.to_out", "blocks.18.cross_attn.to_q", "blocks.18.mlp.mlp.0", "blocks.18.mlp.mlp.2", "blocks.18.self_attn.to_out", "blocks.18.self_attn.to_qkv", "blocks.19.cross_attn.to_kv", "blocks.19.cross_attn.to_out", "blocks.19.cross_attn.to_q", "blocks.19.mlp.mlp.0", "blocks.19.mlp.mlp.2", "blocks.19.self_attn.to_out", "blocks.19.self_attn.to_qkv", "blocks.2.cross_attn.to_kv", "blocks.2.cross_attn.to_out", "blocks.2.cross_attn.to_q", "blocks.2.mlp.mlp.0", "blocks.2.mlp.mlp.2", "blocks.2.self_attn.to_out", "blocks.2.self_attn.to_qkv", "blocks.20.cross_attn.to_kv", "blocks.20.cross_attn.to_out", "blocks.20.cross_attn.to_q", "blocks.20.mlp.mlp.0", "blocks.20.mlp.mlp.2", "blocks.20.self_attn.to_out", "blocks.20.self_attn.to_qkv", "blocks.21.cross_attn.to_kv", "blocks.21.cross_attn.to_out", "blocks.21.cross_attn.to_q", "blocks.21.mlp.mlp.0", "blocks.21.mlp.mlp.2", "blocks.21.self_attn.to_out", "blocks.21.self_attn.to_qkv", "blocks.22.cross_attn.to_kv", "blocks.22.cross_attn.to_out", "blocks.22.cross_attn.to_q", "blocks.22.mlp.mlp.0", "blocks.22.mlp.mlp.2", "blocks.22.self_attn.to_out", "blocks.22.self_attn.to_qkv", "blocks.23.cross_attn.to_kv", "blocks.23.cross_attn.to_out", "blocks.23.cross_attn.to_q", "blocks.23.mlp.mlp.0", "blocks.23.mlp.mlp.2", "blocks.23.self_attn.to_out", "blocks.23.self_attn.to_qkv", "blocks.24.cross_attn.to_kv", "blocks.24.cross_attn.to_out", "blocks.24.cross_attn.to_q", "blocks.24.mlp.mlp.0", "blocks.24.mlp.mlp.2", "blocks.24.self_attn.to_out", "blocks.24.self_attn.to_qkv", "blocks.25.cross_attn.to_kv", "blocks.25.cross_attn.to_out", "blocks.25.cross_attn.to_q", "blocks.25.mlp.mlp.0", "blocks.25.mlp.mlp.2", "blocks.25.self_attn.to_out", "blocks.25.self_attn.to_qkv", "blocks.26.cross_attn.to_kv", "blocks.26.cross_attn.to_out", "blocks.26.cross_attn.to_q", "blocks.26.mlp.mlp.0", "blocks.26.mlp.mlp.2", "blocks.26.self_attn.to_out", "blocks.26.self_attn.to_qkv", "blocks.27.cross_attn.to_kv", "blocks.27.cross_attn.to_out", "blocks.27.cross_attn.to_q", "blocks.27.mlp.mlp.0", "blocks.27.mlp.mlp.2", "blocks.27.self_attn.to_out", "blocks.27.self_attn.to_qkv", "blocks.28.cross_attn.to_kv", "blocks.28.cross_attn.to_out", "blocks.28.cross_attn.to_q", "blocks.28.mlp.mlp.0", "blocks.28.mlp.mlp.2", "blocks.28.self_attn.to_out", "blocks.28.self_attn.to_qkv", "blocks.29.cross_attn.to_kv", "blocks.29.cross_attn.to_out", "blocks.29.cross_attn.to_q", "blocks.29.mlp.mlp.0", "blocks.29.mlp.mlp.2", "blocks.29.self_attn.to_out", "blocks.29.self_attn.to_qkv", "blocks.3.cross_attn.to_kv", "blocks.3.cross_attn.to_out", "blocks.3.cross_attn.to_q", "blocks.3.mlp.mlp.0", "blocks.3.mlp.mlp.2", "blocks.3.self_attn.to_out", "blocks.3.self_attn.to_qkv", "blocks.4.cross_attn.to_kv", "blocks.4.cross_attn.to_out", "blocks.4.cross_attn.to_q", "blocks.4.mlp.mlp.0", "blocks.4.mlp.mlp.2", "blocks.4.self_attn.to_out", "blocks.4.self_attn.to_qkv", "blocks.5.cross_attn.to_kv", "blocks.5.cross_attn.to_out", "blocks.5.cross_attn.to_q", "blocks.5.mlp.mlp.0", "blocks.5.mlp.mlp.2", "blocks.5.self_attn.to_out", "blocks.5.self_attn.to_qkv", "blocks.6.cross_attn.to_kv", "blocks.6.cross_attn.to_out", "blocks.6.cross_attn.to_q", "blocks.6.mlp.mlp.0", "blocks.6.mlp.mlp.2", "blocks.6.self_attn.to_out", "blocks.6.self_attn.to_qkv", "blocks.7.cross_attn.to_kv", "blocks.7.cross_attn.to_out", "blocks.7.cross_attn.to_q", "blocks.7.mlp.mlp.0", "blocks.7.mlp.mlp.2", "blocks.7.self_attn.to_out", "blocks.7.self_attn.to_qkv", "blocks.8.cross_attn.to_kv", "blocks.8.cross_attn.to_out", "blocks.8.cross_attn.to_q", "blocks.8.mlp.mlp.0", "blocks.8.mlp.mlp.2", "blocks.8.self_attn.to_out", "blocks.8.self_attn.to_qkv", "blocks.9.cross_attn.to_kv", "blocks.9.cross_attn.to_out", "blocks.9.cross_attn.to_q", "blocks.9.mlp.mlp.0", "blocks.9.mlp.mlp.2", "blocks.9.self_attn.to_out", "blocks.9.self_attn.to_qkv" ], "model_key": "tex_slat_flow_model_1024" }, "ckpts/shape_enc_next_dc_f16c32_fp16": { "quantized": false, "checkpoint": "ckpts/shape_enc_next_dc_f16c32_fp16.safetensors", "config": "ckpts/shape_enc_next_dc_f16c32_fp16.json", "model_key": "shape_slat_encoder" } } }