| { |
| "_class_name": "MiniMaxH3Transformer3DModel", |
| "_diffusers_version": "0.40.0.dev0", |
| "attention_head_dim": 128, |
| "audio_in_channels": 32, |
| "ffn_dim": 14336, |
| "final_norm_eps": 1e-05, |
| "freq_dim": 256, |
| "hidden_size": 5376, |
| "in_channels": 24, |
| "norm_eps": 1e-05, |
| "num_attention_heads": 56, |
| "num_layers": 50, |
| "num_refiner_layers": 2, |
| "patch_size": [ |
| 1, |
| 2, |
| 2 |
| ], |
| "qk_norm_eps": 1e-05, |
| "quantization_config": { |
| "add_skip_keys": true, |
| "dequantize_fp32": false, |
| "dynamic_loss_threshold": 0.01, |
| "group_size": 0, |
| "hadamard_group_size": 256, |
| "is_integer": true, |
| "is_training": false, |
| "minimum_allowed_numel": 16384, |
| "modules_dtype_dict": { |
| "uint4": [ |
| "transformer_blocks.0.adaln_proj.linear.weight", |
| "transformer_blocks.0.attn.to_k.weight", |
| "transformer_blocks.0.attn.to_out.0.weight", |
| "transformer_blocks.0.attn.to_q.weight", |
| "transformer_blocks.0.attn.to_v.weight", |
| "transformer_blocks.0.ff.net.0.proj.weight", |
| "transformer_blocks.0.ff.net.2.weight", |
| "transformer_blocks.1.adaln_proj.linear.weight", |
| "transformer_blocks.1.attn.to_k.weight", |
| "transformer_blocks.1.attn.to_out.0.weight", |
| "transformer_blocks.1.attn.to_q.weight", |
| "transformer_blocks.1.attn.to_v.weight", |
| "transformer_blocks.1.ff.net.0.proj.weight", |
| "transformer_blocks.1.ff.net.2.weight", |
| "transformer_blocks.2.adaln_proj.linear.weight", |
| "transformer_blocks.2.attn.to_k.weight", |
| "transformer_blocks.2.attn.to_out.0.weight", |
| "transformer_blocks.2.attn.to_q.weight", |
| "transformer_blocks.2.attn.to_v.weight", |
| "transformer_blocks.2.ff.net.0.proj.weight", |
| "transformer_blocks.2.ff.net.2.weight", |
| "transformer_blocks.3.attn.to_k.weight", |
| "transformer_blocks.3.attn.to_out.0.weight", |
| "transformer_blocks.3.attn.to_q.weight", |
| "transformer_blocks.3.attn.to_v.weight", |
| "transformer_blocks.3.ff.net.0.proj.weight", |
| "transformer_blocks.3.ff.net.2.weight", |
| "transformer_blocks.3.adaln_proj.linear.weight", |
| "transformer_blocks.4.adaln_proj.linear.weight", |
| "transformer_blocks.4.attn.to_k.weight", |
| "transformer_blocks.4.attn.to_out.0.weight", |
| "transformer_blocks.4.attn.to_q.weight", |
| "transformer_blocks.4.attn.to_v.weight", |
| "transformer_blocks.4.ff.net.0.proj.weight", |
| "transformer_blocks.4.ff.net.2.weight", |
| "transformer_blocks.5.adaln_proj.linear.weight", |
| "transformer_blocks.5.attn.to_k.weight", |
| "transformer_blocks.5.attn.to_out.0.weight", |
| "transformer_blocks.5.attn.to_q.weight", |
| "transformer_blocks.5.attn.to_v.weight", |
| "transformer_blocks.5.ff.net.0.proj.weight", |
| "transformer_blocks.5.ff.net.2.weight", |
| "transformer_blocks.6.adaln_proj.linear.weight", |
| "transformer_blocks.6.attn.to_k.weight", |
| "transformer_blocks.6.attn.to_out.0.weight", |
| "transformer_blocks.6.attn.to_q.weight", |
| "transformer_blocks.6.attn.to_v.weight", |
| "transformer_blocks.6.ff.net.0.proj.weight", |
| "transformer_blocks.6.ff.net.2.weight", |
| "transformer_blocks.7.attn.to_k.weight", |
| "transformer_blocks.7.attn.to_out.0.weight", |
| "transformer_blocks.7.attn.to_q.weight", |
| "transformer_blocks.7.attn.to_v.weight", |
| "transformer_blocks.10.adaln_proj.linear.weight", |
| "transformer_blocks.10.attn.to_k.weight", |
| "transformer_blocks.10.attn.to_out.0.weight", |
| "transformer_blocks.10.attn.to_q.weight", |
| "transformer_blocks.10.attn.to_v.weight", |
| "transformer_blocks.10.ff.net.0.proj.weight", |
| "transformer_blocks.10.ff.net.2.weight", |
| "transformer_blocks.11.attn.to_q.weight", |
| "transformer_blocks.7.adaln_proj.linear.weight", |
| "transformer_blocks.7.ff.net.0.proj.weight", |
| "transformer_blocks.7.ff.net.2.weight", |
| "transformer_blocks.8.adaln_proj.linear.weight", |
| "transformer_blocks.8.attn.to_k.weight", |
| "transformer_blocks.8.attn.to_out.0.weight", |
| "transformer_blocks.8.attn.to_q.weight", |
| "transformer_blocks.8.attn.to_v.weight", |
| "transformer_blocks.8.ff.net.0.proj.weight", |
| "transformer_blocks.8.ff.net.2.weight", |
| "transformer_blocks.9.adaln_proj.linear.weight", |
| "transformer_blocks.9.attn.to_k.weight", |
| "transformer_blocks.9.attn.to_out.0.weight", |
| "transformer_blocks.9.attn.to_q.weight", |
| "transformer_blocks.9.attn.to_v.weight", |
| "transformer_blocks.9.ff.net.0.proj.weight", |
| "transformer_blocks.9.ff.net.2.weight", |
| "transformer_blocks.11.adaln_proj.linear.weight", |
| "transformer_blocks.11.attn.to_k.weight", |
| "transformer_blocks.11.attn.to_out.0.weight", |
| "transformer_blocks.11.attn.to_v.weight", |
| "transformer_blocks.11.ff.net.0.proj.weight", |
| "transformer_blocks.11.ff.net.2.weight", |
| "transformer_blocks.12.adaln_proj.linear.weight", |
| "transformer_blocks.12.attn.to_k.weight", |
| "transformer_blocks.12.attn.to_out.0.weight", |
| "transformer_blocks.12.attn.to_q.weight", |
| "transformer_blocks.12.attn.to_v.weight", |
| "transformer_blocks.12.ff.net.0.proj.weight", |
| "transformer_blocks.12.ff.net.2.weight", |
| "transformer_blocks.13.adaln_proj.linear.weight", |
| "transformer_blocks.13.attn.to_k.weight", |
| "transformer_blocks.13.attn.to_out.0.weight", |
| "transformer_blocks.13.attn.to_q.weight", |
| "transformer_blocks.13.attn.to_v.weight", |
| "transformer_blocks.13.ff.net.0.proj.weight", |
| "transformer_blocks.13.ff.net.2.weight", |
| "transformer_blocks.14.attn.to_k.weight", |
| "transformer_blocks.14.attn.to_out.0.weight", |
| "transformer_blocks.14.attn.to_q.weight", |
| "transformer_blocks.14.attn.to_v.weight", |
| "transformer_blocks.14.ff.net.0.proj.weight", |
| "transformer_blocks.14.ff.net.2.weight", |
| "transformer_blocks.14.adaln_proj.linear.weight", |
| "transformer_blocks.15.adaln_proj.linear.weight", |
| "transformer_blocks.15.attn.to_k.weight", |
| "transformer_blocks.15.attn.to_out.0.weight", |
| "transformer_blocks.15.attn.to_q.weight", |
| "transformer_blocks.15.attn.to_v.weight", |
| "transformer_blocks.15.ff.net.0.proj.weight", |
| "transformer_blocks.15.ff.net.2.weight", |
| "transformer_blocks.16.adaln_proj.linear.weight", |
| "transformer_blocks.16.attn.to_k.weight", |
| "transformer_blocks.16.attn.to_out.0.weight", |
| "transformer_blocks.16.attn.to_q.weight", |
| "transformer_blocks.16.attn.to_v.weight", |
| "transformer_blocks.16.ff.net.0.proj.weight", |
| "transformer_blocks.16.ff.net.2.weight", |
| "transformer_blocks.17.adaln_proj.linear.weight", |
| "transformer_blocks.17.attn.to_k.weight", |
| "transformer_blocks.17.attn.to_out.0.weight", |
| "transformer_blocks.17.attn.to_q.weight", |
| "transformer_blocks.17.attn.to_v.weight", |
| "transformer_blocks.17.ff.net.0.proj.weight", |
| "transformer_blocks.17.ff.net.2.weight", |
| "transformer_blocks.18.attn.to_k.weight", |
| "transformer_blocks.18.attn.to_out.0.weight", |
| "transformer_blocks.18.attn.to_q.weight", |
| "transformer_blocks.18.attn.to_v.weight", |
| "transformer_blocks.18.adaln_proj.linear.weight", |
| "transformer_blocks.18.ff.net.0.proj.weight", |
| "transformer_blocks.18.ff.net.2.weight", |
| "transformer_blocks.19.adaln_proj.linear.weight", |
| "transformer_blocks.19.attn.to_k.weight", |
| "transformer_blocks.19.attn.to_out.0.weight", |
| "transformer_blocks.19.attn.to_q.weight", |
| "transformer_blocks.19.attn.to_v.weight", |
| "transformer_blocks.19.ff.net.0.proj.weight", |
| "transformer_blocks.19.ff.net.2.weight", |
| "transformer_blocks.20.adaln_proj.linear.weight", |
| "transformer_blocks.20.attn.to_k.weight", |
| "transformer_blocks.20.attn.to_out.0.weight", |
| "transformer_blocks.20.attn.to_q.weight", |
| "transformer_blocks.20.attn.to_v.weight", |
| "transformer_blocks.20.ff.net.0.proj.weight", |
| "transformer_blocks.20.ff.net.2.weight", |
| "transformer_blocks.21.adaln_proj.linear.weight", |
| "transformer_blocks.21.attn.to_k.weight", |
| "transformer_blocks.21.attn.to_out.0.weight", |
| "transformer_blocks.21.attn.to_q.weight", |
| "transformer_blocks.21.attn.to_v.weight", |
| "transformer_blocks.21.ff.net.0.proj.weight", |
| "transformer_blocks.21.ff.net.2.weight", |
| "transformer_blocks.22.attn.to_q.weight", |
| "transformer_blocks.22.adaln_proj.linear.weight", |
| "transformer_blocks.22.attn.to_k.weight", |
| "transformer_blocks.22.attn.to_out.0.weight", |
| "transformer_blocks.22.attn.to_v.weight", |
| "transformer_blocks.22.ff.net.0.proj.weight", |
| "transformer_blocks.22.ff.net.2.weight", |
| "transformer_blocks.23.adaln_proj.linear.weight", |
| "transformer_blocks.23.attn.to_k.weight", |
| "transformer_blocks.23.attn.to_out.0.weight", |
| "transformer_blocks.23.attn.to_q.weight", |
| "transformer_blocks.23.attn.to_v.weight", |
| "transformer_blocks.23.ff.net.0.proj.weight", |
| "transformer_blocks.23.ff.net.2.weight", |
| "transformer_blocks.24.adaln_proj.linear.weight", |
| "transformer_blocks.24.attn.to_k.weight", |
| "transformer_blocks.24.attn.to_out.0.weight", |
| "transformer_blocks.24.attn.to_q.weight", |
| "transformer_blocks.24.attn.to_v.weight", |
| "transformer_blocks.24.ff.net.0.proj.weight", |
| "transformer_blocks.24.ff.net.2.weight", |
| "transformer_blocks.25.attn.to_k.weight", |
| "transformer_blocks.25.attn.to_out.0.weight", |
| "transformer_blocks.25.attn.to_q.weight", |
| "transformer_blocks.25.attn.to_v.weight", |
| "transformer_blocks.25.ff.net.0.proj.weight", |
| "transformer_blocks.25.ff.net.2.weight", |
| "transformer_blocks.25.adaln_proj.linear.weight", |
| "transformer_blocks.26.adaln_proj.linear.weight", |
| "transformer_blocks.26.attn.to_k.weight", |
| "transformer_blocks.26.attn.to_out.0.weight", |
| "transformer_blocks.26.attn.to_q.weight", |
| "transformer_blocks.26.attn.to_v.weight", |
| "transformer_blocks.26.ff.net.0.proj.weight", |
| "transformer_blocks.26.ff.net.2.weight", |
| "transformer_blocks.27.adaln_proj.linear.weight", |
| "transformer_blocks.27.attn.to_k.weight", |
| "transformer_blocks.27.attn.to_out.0.weight", |
| "transformer_blocks.27.attn.to_q.weight", |
| "transformer_blocks.27.attn.to_v.weight", |
| "transformer_blocks.27.ff.net.0.proj.weight", |
| "transformer_blocks.27.ff.net.2.weight", |
| "transformer_blocks.28.adaln_proj.linear.weight", |
| "transformer_blocks.28.attn.to_k.weight", |
| "transformer_blocks.28.attn.to_out.0.weight", |
| "transformer_blocks.28.attn.to_q.weight", |
| "transformer_blocks.28.attn.to_v.weight", |
| "transformer_blocks.28.ff.net.0.proj.weight", |
| "transformer_blocks.28.ff.net.2.weight", |
| "transformer_blocks.29.attn.to_k.weight", |
| "transformer_blocks.29.attn.to_out.0.weight", |
| "transformer_blocks.29.attn.to_q.weight", |
| "transformer_blocks.29.attn.to_v.weight", |
| "transformer_blocks.29.adaln_proj.linear.weight", |
| "transformer_blocks.29.ff.net.0.proj.weight", |
| "transformer_blocks.29.ff.net.2.weight", |
| "transformer_blocks.30.adaln_proj.linear.weight", |
| "transformer_blocks.30.attn.to_k.weight", |
| "transformer_blocks.30.attn.to_out.0.weight", |
| "transformer_blocks.30.attn.to_q.weight", |
| "transformer_blocks.30.attn.to_v.weight", |
| "transformer_blocks.30.ff.net.0.proj.weight", |
| "transformer_blocks.30.ff.net.2.weight", |
| "transformer_blocks.31.adaln_proj.linear.weight", |
| "transformer_blocks.31.attn.to_k.weight", |
| "transformer_blocks.31.attn.to_out.0.weight", |
| "transformer_blocks.31.attn.to_q.weight", |
| "transformer_blocks.31.attn.to_v.weight", |
| "transformer_blocks.31.ff.net.0.proj.weight", |
| "transformer_blocks.31.ff.net.2.weight", |
| "transformer_blocks.32.adaln_proj.linear.weight", |
| "transformer_blocks.32.attn.to_k.weight", |
| "transformer_blocks.32.attn.to_out.0.weight", |
| "transformer_blocks.32.attn.to_q.weight", |
| "transformer_blocks.32.attn.to_v.weight", |
| "transformer_blocks.32.ff.net.0.proj.weight", |
| "transformer_blocks.32.ff.net.2.weight", |
| "transformer_blocks.33.attn.to_q.weight", |
| "transformer_blocks.33.adaln_proj.linear.weight", |
| "transformer_blocks.33.attn.to_k.weight", |
| "transformer_blocks.33.attn.to_out.0.weight", |
| "transformer_blocks.33.attn.to_v.weight", |
| "transformer_blocks.33.ff.net.0.proj.weight", |
| "transformer_blocks.33.ff.net.2.weight", |
| "transformer_blocks.34.adaln_proj.linear.weight", |
| "transformer_blocks.34.attn.to_k.weight", |
| "transformer_blocks.34.attn.to_out.0.weight", |
| "transformer_blocks.34.attn.to_q.weight", |
| "transformer_blocks.34.attn.to_v.weight", |
| "transformer_blocks.34.ff.net.0.proj.weight", |
| "transformer_blocks.34.ff.net.2.weight", |
| "transformer_blocks.35.adaln_proj.linear.weight", |
| "transformer_blocks.35.attn.to_k.weight", |
| "transformer_blocks.35.attn.to_out.0.weight", |
| "transformer_blocks.35.attn.to_q.weight", |
| "transformer_blocks.35.attn.to_v.weight", |
| "transformer_blocks.35.ff.net.0.proj.weight", |
| "transformer_blocks.35.ff.net.2.weight", |
| "transformer_blocks.36.attn.to_k.weight", |
| "transformer_blocks.36.attn.to_out.0.weight", |
| "transformer_blocks.36.attn.to_q.weight", |
| "transformer_blocks.36.attn.to_v.weight", |
| "transformer_blocks.36.ff.net.0.proj.weight", |
| "transformer_blocks.36.ff.net.2.weight", |
| "transformer_blocks.36.adaln_proj.linear.weight", |
| "transformer_blocks.37.adaln_proj.linear.weight", |
| "transformer_blocks.37.attn.to_k.weight", |
| "transformer_blocks.37.attn.to_out.0.weight", |
| "transformer_blocks.37.attn.to_q.weight", |
| "transformer_blocks.37.attn.to_v.weight", |
| "transformer_blocks.37.ff.net.0.proj.weight", |
| "transformer_blocks.37.ff.net.2.weight", |
| "transformer_blocks.38.adaln_proj.linear.weight", |
| "transformer_blocks.38.attn.to_k.weight", |
| "transformer_blocks.38.attn.to_out.0.weight", |
| "transformer_blocks.38.attn.to_q.weight", |
| "transformer_blocks.38.attn.to_v.weight", |
| "transformer_blocks.38.ff.net.0.proj.weight", |
| "transformer_blocks.38.ff.net.2.weight", |
| "transformer_blocks.39.adaln_proj.linear.weight", |
| "transformer_blocks.39.attn.to_k.weight", |
| "transformer_blocks.39.attn.to_out.0.weight", |
| "transformer_blocks.39.attn.to_q.weight", |
| "transformer_blocks.39.attn.to_v.weight", |
| "transformer_blocks.39.ff.net.0.proj.weight", |
| "transformer_blocks.39.ff.net.2.weight", |
| "transformer_blocks.40.attn.to_k.weight", |
| "transformer_blocks.40.attn.to_out.0.weight", |
| "transformer_blocks.40.attn.to_q.weight", |
| "transformer_blocks.40.attn.to_v.weight", |
| "transformer_blocks.40.adaln_proj.linear.weight", |
| "transformer_blocks.40.ff.net.0.proj.weight", |
| "transformer_blocks.40.ff.net.2.weight", |
| "transformer_blocks.41.adaln_proj.linear.weight", |
| "transformer_blocks.41.attn.to_k.weight", |
| "transformer_blocks.41.attn.to_out.0.weight", |
| "transformer_blocks.41.attn.to_q.weight", |
| "transformer_blocks.41.attn.to_v.weight", |
| "transformer_blocks.41.ff.net.0.proj.weight", |
| "transformer_blocks.41.ff.net.2.weight", |
| "transformer_blocks.42.adaln_proj.linear.weight", |
| "transformer_blocks.42.attn.to_k.weight", |
| "transformer_blocks.42.attn.to_out.0.weight", |
| "transformer_blocks.42.attn.to_q.weight", |
| "transformer_blocks.42.attn.to_v.weight", |
| "transformer_blocks.42.ff.net.0.proj.weight", |
| "transformer_blocks.42.ff.net.2.weight", |
| "transformer_blocks.43.adaln_proj.linear.weight", |
| "transformer_blocks.43.attn.to_k.weight", |
| "transformer_blocks.43.attn.to_out.0.weight", |
| "transformer_blocks.43.attn.to_q.weight", |
| "transformer_blocks.43.attn.to_v.weight", |
| "transformer_blocks.43.ff.net.0.proj.weight", |
| "transformer_blocks.43.ff.net.2.weight", |
| "transformer_blocks.44.attn.to_q.weight", |
| "transformer_blocks.44.adaln_proj.linear.weight", |
| "transformer_blocks.44.attn.to_k.weight", |
| "transformer_blocks.44.attn.to_out.0.weight", |
| "transformer_blocks.44.attn.to_v.weight", |
| "transformer_blocks.44.ff.net.0.proj.weight", |
| "transformer_blocks.44.ff.net.2.weight", |
| "transformer_blocks.45.adaln_proj.linear.weight", |
| "transformer_blocks.45.attn.to_k.weight", |
| "transformer_blocks.45.attn.to_out.0.weight", |
| "transformer_blocks.45.attn.to_q.weight", |
| "transformer_blocks.45.attn.to_v.weight", |
| "transformer_blocks.45.ff.net.0.proj.weight", |
| "transformer_blocks.45.ff.net.2.weight", |
| "transformer_blocks.46.adaln_proj.linear.weight", |
| "transformer_blocks.46.attn.to_k.weight", |
| "transformer_blocks.46.attn.to_out.0.weight", |
| "transformer_blocks.46.attn.to_q.weight", |
| "transformer_blocks.46.attn.to_v.weight", |
| "transformer_blocks.46.ff.net.0.proj.weight", |
| "transformer_blocks.46.ff.net.2.weight", |
| "transformer_blocks.47.attn.to_k.weight", |
| "transformer_blocks.47.attn.to_out.0.weight", |
| "transformer_blocks.47.attn.to_q.weight", |
| "transformer_blocks.47.attn.to_v.weight", |
| "transformer_blocks.47.ff.net.0.proj.weight", |
| "transformer_blocks.47.ff.net.2.weight", |
| "token_refiner.refiner_blocks.0.attn.to_k.weight", |
| "token_refiner.refiner_blocks.0.attn.to_out.0.weight", |
| "token_refiner.refiner_blocks.0.attn.to_q.weight", |
| "token_refiner.refiner_blocks.0.attn.to_v.weight", |
| "token_refiner.refiner_blocks.0.ff.net.0.proj.weight", |
| "token_refiner.refiner_blocks.0.ff.net.2.weight", |
| "token_refiner.refiner_blocks.1.attn.to_k.weight", |
| "token_refiner.refiner_blocks.1.attn.to_out.0.weight", |
| "token_refiner.refiner_blocks.1.attn.to_q.weight", |
| "token_refiner.refiner_blocks.1.attn.to_v.weight", |
| "token_refiner.refiner_blocks.1.ff.net.0.proj.weight", |
| "token_refiner.refiner_blocks.1.ff.net.2.weight", |
| "transformer_blocks.47.adaln_proj.linear.weight", |
| "transformer_blocks.48.adaln_proj.linear.weight", |
| "transformer_blocks.48.attn.to_k.weight", |
| "transformer_blocks.48.attn.to_out.0.weight", |
| "transformer_blocks.48.attn.to_q.weight", |
| "transformer_blocks.48.attn.to_v.weight", |
| "transformer_blocks.48.ff.net.0.proj.weight", |
| "transformer_blocks.48.ff.net.2.weight", |
| "transformer_blocks.49.adaln_proj.linear.weight", |
| "transformer_blocks.49.attn.to_k.weight", |
| "transformer_blocks.49.attn.to_out.0.weight", |
| "transformer_blocks.49.attn.to_q.weight", |
| "transformer_blocks.49.attn.to_v.weight", |
| "transformer_blocks.49.ff.net.0.proj.weight", |
| "transformer_blocks.49.ff.net.2.weight" |
| ] |
| }, |
| "modules_quant_config": {}, |
| "modules_to_not_convert": [ |
| ".proj_out", |
| ".t_embedder", |
| ".emb_in", |
| ".condition_embedder", |
| "wte", |
| ".norm_out", |
| "proj_out", |
| "norm", |
| ".emb_out", |
| ".txt_out", |
| "audio_proj_out", |
| "rope", |
| "time_text_embed", |
| ".img_out", |
| "proj_in", |
| ".img_in", |
| "patch_emb", |
| ".context_embedder", |
| ".final_layer", |
| ".x_embedder", |
| ".time_embed", |
| ".txt_in", |
| "time_embedder", |
| ".vid_in", |
| ".y_embedder", |
| "patch_embedding", |
| "patch_embed", |
| "audio_proj_in", |
| "multi_modal_projector", |
| "lm_head", |
| ".vid_out", |
| "token_refiner.final_norm.weight", |
| "transformer_blocks.0.attn.norm_k.weight", |
| "transformer_blocks.0.attn.norm_q.weight", |
| "transformer_blocks.0.norm1.weight", |
| "transformer_blocks.0.norm2.weight", |
| "transformer_blocks.1.attn.norm_k.weight", |
| "transformer_blocks.1.attn.norm_q.weight", |
| "transformer_blocks.1.norm1.weight", |
| "transformer_blocks.1.norm2.weight", |
| "transformer_blocks.2.attn.norm_k.weight", |
| "transformer_blocks.2.attn.norm_q.weight", |
| "transformer_blocks.2.norm1.weight", |
| "transformer_blocks.2.norm2.weight", |
| "transformer_blocks.3.attn.norm_k.weight", |
| "transformer_blocks.3.attn.norm_q.weight", |
| "transformer_blocks.3.norm1.weight", |
| "transformer_blocks.3.norm2.weight", |
| "transformer_blocks.4.attn.norm_k.weight", |
| "transformer_blocks.4.attn.norm_q.weight", |
| "transformer_blocks.4.norm1.weight", |
| "transformer_blocks.4.norm2.weight", |
| "transformer_blocks.5.attn.norm_k.weight", |
| "transformer_blocks.5.attn.norm_q.weight", |
| "transformer_blocks.5.norm1.weight", |
| "transformer_blocks.5.norm2.weight", |
| "transformer_blocks.6.attn.norm_k.weight", |
| "transformer_blocks.6.attn.norm_q.weight", |
| "transformer_blocks.6.norm1.weight", |
| "transformer_blocks.6.norm2.weight", |
| "transformer_blocks.7.attn.norm_k.weight", |
| "transformer_blocks.7.attn.norm_q.weight", |
| "transformer_blocks.7.norm1.weight", |
| "transformer_blocks.7.norm2.weight", |
| "transformer_blocks.10.attn.norm_k.weight", |
| "transformer_blocks.10.attn.norm_q.weight", |
| "transformer_blocks.10.norm1.weight", |
| "transformer_blocks.10.norm2.weight", |
| "transformer_blocks.11.norm1.weight", |
| "transformer_blocks.11.norm2.weight", |
| "transformer_blocks.8.attn.norm_k.weight", |
| "transformer_blocks.8.attn.norm_q.weight", |
| "transformer_blocks.8.norm1.weight", |
| "transformer_blocks.8.norm2.weight", |
| "transformer_blocks.9.attn.norm_k.weight", |
| "transformer_blocks.9.attn.norm_q.weight", |
| "transformer_blocks.9.norm1.weight", |
| "transformer_blocks.9.norm2.weight", |
| "transformer_blocks.11.attn.norm_k.weight", |
| "transformer_blocks.11.attn.norm_q.weight", |
| "transformer_blocks.12.attn.norm_k.weight", |
| "transformer_blocks.12.attn.norm_q.weight", |
| "transformer_blocks.12.norm1.weight", |
| "transformer_blocks.12.norm2.weight", |
| "transformer_blocks.13.attn.norm_k.weight", |
| "transformer_blocks.13.attn.norm_q.weight", |
| "transformer_blocks.13.norm1.weight", |
| "transformer_blocks.13.norm2.weight", |
| "transformer_blocks.14.attn.norm_k.weight", |
| "transformer_blocks.14.attn.norm_q.weight", |
| "transformer_blocks.14.norm1.weight", |
| "transformer_blocks.14.norm2.weight", |
| "transformer_blocks.15.attn.norm_k.weight", |
| "transformer_blocks.15.attn.norm_q.weight", |
| "transformer_blocks.15.norm1.weight", |
| "transformer_blocks.15.norm2.weight", |
| "transformer_blocks.16.attn.norm_k.weight", |
| "transformer_blocks.16.attn.norm_q.weight", |
| "transformer_blocks.16.norm1.weight", |
| "transformer_blocks.16.norm2.weight", |
| "transformer_blocks.17.attn.norm_k.weight", |
| "transformer_blocks.17.attn.norm_q.weight", |
| "transformer_blocks.17.norm1.weight", |
| "transformer_blocks.17.norm2.weight", |
| "transformer_blocks.18.attn.norm_k.weight", |
| "transformer_blocks.18.attn.norm_q.weight", |
| "transformer_blocks.18.norm1.weight", |
| "transformer_blocks.18.norm2.weight", |
| "transformer_blocks.19.attn.norm_k.weight", |
| "transformer_blocks.19.attn.norm_q.weight", |
| "transformer_blocks.19.norm1.weight", |
| "transformer_blocks.19.norm2.weight", |
| "transformer_blocks.20.attn.norm_k.weight", |
| "transformer_blocks.20.attn.norm_q.weight", |
| "transformer_blocks.20.norm1.weight", |
| "transformer_blocks.20.norm2.weight", |
| "transformer_blocks.21.attn.norm_k.weight", |
| "transformer_blocks.21.attn.norm_q.weight", |
| "transformer_blocks.21.norm1.weight", |
| "transformer_blocks.21.norm2.weight", |
| "transformer_blocks.22.norm1.weight", |
| "transformer_blocks.22.norm2.weight", |
| "transformer_blocks.22.attn.norm_k.weight", |
| "transformer_blocks.22.attn.norm_q.weight", |
| "transformer_blocks.23.attn.norm_k.weight", |
| "transformer_blocks.23.attn.norm_q.weight", |
| "transformer_blocks.23.norm1.weight", |
| "transformer_blocks.23.norm2.weight", |
| "transformer_blocks.24.attn.norm_k.weight", |
| "transformer_blocks.24.attn.norm_q.weight", |
| "transformer_blocks.24.norm1.weight", |
| "transformer_blocks.24.norm2.weight", |
| "transformer_blocks.25.attn.norm_k.weight", |
| "transformer_blocks.25.attn.norm_q.weight", |
| "transformer_blocks.25.norm1.weight", |
| "transformer_blocks.25.norm2.weight", |
| "transformer_blocks.26.attn.norm_k.weight", |
| "transformer_blocks.26.attn.norm_q.weight", |
| "transformer_blocks.26.norm1.weight", |
| "transformer_blocks.26.norm2.weight", |
| "transformer_blocks.27.attn.norm_k.weight", |
| "transformer_blocks.27.attn.norm_q.weight", |
| "transformer_blocks.27.norm1.weight", |
| "transformer_blocks.27.norm2.weight", |
| "transformer_blocks.28.attn.norm_k.weight", |
| "transformer_blocks.28.attn.norm_q.weight", |
| "transformer_blocks.28.norm1.weight", |
| "transformer_blocks.28.norm2.weight", |
| "transformer_blocks.29.attn.norm_k.weight", |
| "transformer_blocks.29.attn.norm_q.weight", |
| "transformer_blocks.29.norm1.weight", |
| "transformer_blocks.29.norm2.weight", |
| "transformer_blocks.30.attn.norm_k.weight", |
| "transformer_blocks.30.attn.norm_q.weight", |
| "transformer_blocks.30.norm1.weight", |
| "transformer_blocks.30.norm2.weight", |
| "transformer_blocks.31.attn.norm_k.weight", |
| "transformer_blocks.31.attn.norm_q.weight", |
| "transformer_blocks.31.norm1.weight", |
| "transformer_blocks.31.norm2.weight", |
| "transformer_blocks.32.attn.norm_k.weight", |
| "transformer_blocks.32.attn.norm_q.weight", |
| "transformer_blocks.32.norm1.weight", |
| "transformer_blocks.32.norm2.weight", |
| "transformer_blocks.33.norm1.weight", |
| "transformer_blocks.33.norm2.weight", |
| "transformer_blocks.33.attn.norm_k.weight", |
| "transformer_blocks.33.attn.norm_q.weight", |
| "transformer_blocks.34.attn.norm_k.weight", |
| "transformer_blocks.34.attn.norm_q.weight", |
| "transformer_blocks.34.norm1.weight", |
| "transformer_blocks.34.norm2.weight", |
| "transformer_blocks.35.attn.norm_k.weight", |
| "transformer_blocks.35.attn.norm_q.weight", |
| "transformer_blocks.35.norm1.weight", |
| "transformer_blocks.35.norm2.weight", |
| "transformer_blocks.36.attn.norm_k.weight", |
| "transformer_blocks.36.attn.norm_q.weight", |
| "transformer_blocks.36.norm1.weight", |
| "transformer_blocks.36.norm2.weight", |
| "transformer_blocks.37.attn.norm_k.weight", |
| "transformer_blocks.37.attn.norm_q.weight", |
| "transformer_blocks.37.norm1.weight", |
| "transformer_blocks.37.norm2.weight", |
| "transformer_blocks.38.attn.norm_k.weight", |
| "transformer_blocks.38.attn.norm_q.weight", |
| "transformer_blocks.38.norm1.weight", |
| "transformer_blocks.38.norm2.weight", |
| "transformer_blocks.39.attn.norm_k.weight", |
| "transformer_blocks.39.attn.norm_q.weight", |
| "transformer_blocks.39.norm1.weight", |
| "transformer_blocks.39.norm2.weight", |
| "transformer_blocks.40.attn.norm_k.weight", |
| "transformer_blocks.40.attn.norm_q.weight", |
| "transformer_blocks.40.norm1.weight", |
| "transformer_blocks.40.norm2.weight", |
| "transformer_blocks.41.attn.norm_k.weight", |
| "transformer_blocks.41.attn.norm_q.weight", |
| "transformer_blocks.41.norm1.weight", |
| "transformer_blocks.41.norm2.weight", |
| "transformer_blocks.42.attn.norm_k.weight", |
| "transformer_blocks.42.attn.norm_q.weight", |
| "transformer_blocks.42.norm1.weight", |
| "transformer_blocks.42.norm2.weight", |
| "transformer_blocks.43.attn.norm_k.weight", |
| "transformer_blocks.43.attn.norm_q.weight", |
| "transformer_blocks.43.norm1.weight", |
| "transformer_blocks.43.norm2.weight", |
| "transformer_blocks.44.norm1.weight", |
| "transformer_blocks.44.norm2.weight", |
| "transformer_blocks.44.attn.norm_k.weight", |
| "transformer_blocks.44.attn.norm_q.weight", |
| "transformer_blocks.45.attn.norm_k.weight", |
| "transformer_blocks.45.attn.norm_q.weight", |
| "transformer_blocks.45.norm1.weight", |
| "transformer_blocks.45.norm2.weight", |
| "transformer_blocks.46.attn.norm_k.weight", |
| "transformer_blocks.46.attn.norm_q.weight", |
| "transformer_blocks.46.norm1.weight", |
| "transformer_blocks.46.norm2.weight", |
| "transformer_blocks.47.attn.norm_k.weight", |
| "transformer_blocks.47.attn.norm_q.weight", |
| "transformer_blocks.47.norm1.weight", |
| "transformer_blocks.47.norm2.weight", |
| "token_refiner.refiner_blocks.0.attn.norm_k.weight", |
| "token_refiner.refiner_blocks.0.attn.norm_q.weight", |
| "token_refiner.refiner_blocks.0.norm1.weight", |
| "token_refiner.refiner_blocks.0.norm2.weight", |
| "token_refiner.refiner_blocks.1.attn.norm_k.weight", |
| "token_refiner.refiner_blocks.1.attn.norm_q.weight", |
| "token_refiner.refiner_blocks.1.norm1.weight", |
| "token_refiner.refiner_blocks.1.norm2.weight", |
| "transformer_blocks.48.attn.norm_k.weight", |
| "transformer_blocks.48.attn.norm_q.weight", |
| "transformer_blocks.48.norm1.weight", |
| "transformer_blocks.48.norm2.weight", |
| "transformer_blocks.49.attn.norm_k.weight", |
| "transformer_blocks.49.attn.norm_q.weight", |
| "transformer_blocks.49.norm1.weight", |
| "transformer_blocks.49.norm2.weight" |
| ], |
| "modules_to_not_use_matmul": [], |
| "non_blocking": false, |
| "quant_conv": false, |
| "quant_embedding": false, |
| "quant_method": "sdnq", |
| "quantization_device": null, |
| "quantized_matmul_dtype": null, |
| "return_device": null, |
| "sdnq_version": "0.2.2", |
| "svd_rank": 32, |
| "svd_steps": 8, |
| "use_dynamic_quantization": true, |
| "use_grad_ckpt": true, |
| "use_hadamard": true, |
| "use_quantized_matmul": false, |
| "use_quantized_matmul_conv": false, |
| "use_static_quantization": true, |
| "use_stochastic_rounding": false, |
| "use_svd": false, |
| "weights_dtype": "uint4" |
| }, |
| "rope_freq_dim": 16, |
| "rope_theta": 10000.0, |
| "text_dim": 5120, |
| "time_embed_dim": 2688, |
| "time_embed_hidden_dim": 5376 |
| } |
