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