| { |
| "_class_name": "FluxTransformer2DModel", |
| "_diffusers_version": "0.35.0.dev0", |
| "attention_head_dim": 128, |
| "axes_dims_rope": [ |
| 16, |
| 56, |
| 56 |
| ], |
| "guidance_embeds": true, |
| "in_channels": 64, |
| "joint_attention_dim": 4096, |
| "num_attention_heads": 24, |
| "num_layers": 19, |
| "num_single_layers": 38, |
| "out_channels": null, |
| "patch_size": 1, |
| "pooled_projection_dim": 768, |
| "quantization_config": { |
| "quant_method": "nunchaku_lite", |
| "compute_dtype": "bfloat16", |
| "svdq_w4a4": { |
| "precision": "int4", |
| "group_size": 64, |
| "rank": 64, |
| "targets": [ |
| "single_transformer_blocks.0.attn.to_k", |
| "single_transformer_blocks.0.attn.to_q", |
| "single_transformer_blocks.0.attn.to_v", |
| "single_transformer_blocks.0.proj_mlp", |
| "single_transformer_blocks.0.proj_out", |
| "single_transformer_blocks.1.attn.to_k", |
| "single_transformer_blocks.1.attn.to_q", |
| "single_transformer_blocks.1.attn.to_v", |
| "single_transformer_blocks.1.proj_mlp", |
| "single_transformer_blocks.1.proj_out", |
| "single_transformer_blocks.10.attn.to_k", |
| "single_transformer_blocks.10.attn.to_q", |
| "single_transformer_blocks.10.attn.to_v", |
| "single_transformer_blocks.10.proj_mlp", |
| "single_transformer_blocks.10.proj_out", |
| "single_transformer_blocks.11.attn.to_k", |
| "single_transformer_blocks.11.attn.to_q", |
| "single_transformer_blocks.11.attn.to_v", |
| "single_transformer_blocks.11.proj_mlp", |
| "single_transformer_blocks.11.proj_out", |
| "single_transformer_blocks.12.attn.to_k", |
| "single_transformer_blocks.12.attn.to_q", |
| "single_transformer_blocks.12.attn.to_v", |
| "single_transformer_blocks.12.proj_mlp", |
| "single_transformer_blocks.12.proj_out", |
| "single_transformer_blocks.13.attn.to_k", |
| "single_transformer_blocks.13.attn.to_q", |
| "single_transformer_blocks.13.attn.to_v", |
| "single_transformer_blocks.13.proj_mlp", |
| "single_transformer_blocks.13.proj_out", |
| "single_transformer_blocks.14.attn.to_k", |
| "single_transformer_blocks.14.attn.to_q", |
| "single_transformer_blocks.14.attn.to_v", |
| "single_transformer_blocks.14.proj_mlp", |
| "single_transformer_blocks.14.proj_out", |
| "single_transformer_blocks.15.attn.to_k", |
| "single_transformer_blocks.15.attn.to_q", |
| "single_transformer_blocks.15.attn.to_v", |
| "single_transformer_blocks.15.proj_mlp", |
| "single_transformer_blocks.15.proj_out", |
| "single_transformer_blocks.16.attn.to_k", |
| "single_transformer_blocks.16.attn.to_q", |
| "single_transformer_blocks.16.attn.to_v", |
| "single_transformer_blocks.16.proj_mlp", |
| "single_transformer_blocks.16.proj_out", |
| "single_transformer_blocks.17.attn.to_k", |
| "single_transformer_blocks.17.attn.to_q", |
| "single_transformer_blocks.17.attn.to_v", |
| "single_transformer_blocks.17.proj_mlp", |
| "single_transformer_blocks.17.proj_out", |
| "single_transformer_blocks.18.attn.to_k", |
| "single_transformer_blocks.18.attn.to_q", |
| "single_transformer_blocks.18.attn.to_v", |
| "single_transformer_blocks.18.proj_mlp", |
| "single_transformer_blocks.18.proj_out", |
| "single_transformer_blocks.19.attn.to_k", |
| "single_transformer_blocks.19.attn.to_q", |
| "single_transformer_blocks.19.attn.to_v", |
| "single_transformer_blocks.19.proj_mlp", |
| "single_transformer_blocks.19.proj_out", |
| "single_transformer_blocks.2.attn.to_k", |
| "single_transformer_blocks.2.attn.to_q", |
| "single_transformer_blocks.2.attn.to_v", |
| "single_transformer_blocks.2.proj_mlp", |
| "single_transformer_blocks.2.proj_out", |
| "single_transformer_blocks.20.attn.to_k", |
| "single_transformer_blocks.20.attn.to_q", |
| "single_transformer_blocks.20.attn.to_v", |
| "single_transformer_blocks.20.proj_mlp", |
| "single_transformer_blocks.20.proj_out", |
| "single_transformer_blocks.21.attn.to_k", |
| "single_transformer_blocks.21.attn.to_q", |
| "single_transformer_blocks.21.attn.to_v", |
| "single_transformer_blocks.21.proj_mlp", |
| "single_transformer_blocks.21.proj_out", |
| "single_transformer_blocks.22.attn.to_k", |
| "single_transformer_blocks.22.attn.to_q", |
| "single_transformer_blocks.22.attn.to_v", |
| "single_transformer_blocks.22.proj_mlp", |
| "single_transformer_blocks.22.proj_out", |
| "single_transformer_blocks.23.attn.to_k", |
| "single_transformer_blocks.23.attn.to_q", |
| "single_transformer_blocks.23.attn.to_v", |
| "single_transformer_blocks.23.proj_mlp", |
| "single_transformer_blocks.23.proj_out", |
| "single_transformer_blocks.24.attn.to_k", |
| "single_transformer_blocks.24.attn.to_q", |
| "single_transformer_blocks.24.attn.to_v", |
| "single_transformer_blocks.24.proj_mlp", |
| "single_transformer_blocks.24.proj_out", |
| "single_transformer_blocks.25.attn.to_k", |
| "single_transformer_blocks.25.attn.to_q", |
| "single_transformer_blocks.25.attn.to_v", |
| "single_transformer_blocks.25.proj_mlp", |
| "single_transformer_blocks.25.proj_out", |
| "single_transformer_blocks.26.attn.to_k", |
| "single_transformer_blocks.26.attn.to_q", |
| "single_transformer_blocks.26.attn.to_v", |
| "single_transformer_blocks.26.proj_mlp", |
| "single_transformer_blocks.26.proj_out", |
| "single_transformer_blocks.27.attn.to_k", |
| "single_transformer_blocks.27.attn.to_q", |
| "single_transformer_blocks.27.attn.to_v", |
| "single_transformer_blocks.27.proj_mlp", |
| "single_transformer_blocks.27.proj_out", |
| "single_transformer_blocks.28.attn.to_k", |
| "single_transformer_blocks.28.attn.to_q", |
| "single_transformer_blocks.28.attn.to_v", |
| "single_transformer_blocks.28.proj_mlp", |
| "single_transformer_blocks.28.proj_out", |
| "single_transformer_blocks.29.attn.to_k", |
| "single_transformer_blocks.29.attn.to_q", |
| "single_transformer_blocks.29.attn.to_v", |
| "single_transformer_blocks.29.proj_mlp", |
| "single_transformer_blocks.29.proj_out", |
| "single_transformer_blocks.3.attn.to_k", |
| "single_transformer_blocks.3.attn.to_q", |
| "single_transformer_blocks.3.attn.to_v", |
| "single_transformer_blocks.3.proj_mlp", |
| "single_transformer_blocks.3.proj_out", |
| "single_transformer_blocks.30.attn.to_k", |
| "single_transformer_blocks.30.attn.to_q", |
| "single_transformer_blocks.30.attn.to_v", |
| "single_transformer_blocks.30.proj_mlp", |
| "single_transformer_blocks.30.proj_out", |
| "single_transformer_blocks.31.attn.to_k", |
| "single_transformer_blocks.31.attn.to_q", |
| "single_transformer_blocks.31.attn.to_v", |
| "single_transformer_blocks.31.proj_mlp", |
| "single_transformer_blocks.31.proj_out", |
| "single_transformer_blocks.32.attn.to_k", |
| "single_transformer_blocks.32.attn.to_q", |
| "single_transformer_blocks.32.attn.to_v", |
| "single_transformer_blocks.32.proj_mlp", |
| "single_transformer_blocks.32.proj_out", |
| "single_transformer_blocks.33.attn.to_k", |
| "single_transformer_blocks.33.attn.to_q", |
| "single_transformer_blocks.33.attn.to_v", |
| "single_transformer_blocks.33.proj_mlp", |
| "single_transformer_blocks.33.proj_out", |
| "single_transformer_blocks.34.attn.to_k", |
| "single_transformer_blocks.34.attn.to_q", |
| "single_transformer_blocks.34.attn.to_v", |
| "single_transformer_blocks.34.proj_mlp", |
| "single_transformer_blocks.34.proj_out", |
| "single_transformer_blocks.35.attn.to_k", |
| "single_transformer_blocks.35.attn.to_q", |
| "single_transformer_blocks.35.attn.to_v", |
| "single_transformer_blocks.35.proj_mlp", |
| "single_transformer_blocks.35.proj_out", |
| "single_transformer_blocks.36.attn.to_k", |
| "single_transformer_blocks.36.attn.to_q", |
| "single_transformer_blocks.36.attn.to_v", |
| "single_transformer_blocks.36.proj_mlp", |
| "single_transformer_blocks.36.proj_out", |
| "single_transformer_blocks.37.attn.to_k", |
| "single_transformer_blocks.37.attn.to_q", |
| "single_transformer_blocks.37.attn.to_v", |
| "single_transformer_blocks.37.proj_mlp", |
| "single_transformer_blocks.37.proj_out", |
| "single_transformer_blocks.4.attn.to_k", |
| "single_transformer_blocks.4.attn.to_q", |
| "single_transformer_blocks.4.attn.to_v", |
| "single_transformer_blocks.4.proj_mlp", |
| "single_transformer_blocks.4.proj_out", |
| "single_transformer_blocks.5.attn.to_k", |
| "single_transformer_blocks.5.attn.to_q", |
| "single_transformer_blocks.5.attn.to_v", |
| "single_transformer_blocks.5.proj_mlp", |
| "single_transformer_blocks.5.proj_out", |
| "single_transformer_blocks.6.attn.to_k", |
| "single_transformer_blocks.6.attn.to_q", |
| "single_transformer_blocks.6.attn.to_v", |
| "single_transformer_blocks.6.proj_mlp", |
| "single_transformer_blocks.6.proj_out", |
| "single_transformer_blocks.7.attn.to_k", |
| "single_transformer_blocks.7.attn.to_q", |
| "single_transformer_blocks.7.attn.to_v", |
| "single_transformer_blocks.7.proj_mlp", |
| "single_transformer_blocks.7.proj_out", |
| "single_transformer_blocks.8.attn.to_k", |
| "single_transformer_blocks.8.attn.to_q", |
| "single_transformer_blocks.8.attn.to_v", |
| "single_transformer_blocks.8.proj_mlp", |
| "single_transformer_blocks.8.proj_out", |
| "single_transformer_blocks.9.attn.to_k", |
| "single_transformer_blocks.9.attn.to_q", |
| "single_transformer_blocks.9.attn.to_v", |
| "single_transformer_blocks.9.proj_mlp", |
| "single_transformer_blocks.9.proj_out", |
| "transformer_blocks.0.attn.add_k_proj", |
| "transformer_blocks.0.attn.add_q_proj", |
| "transformer_blocks.0.attn.add_v_proj", |
| "transformer_blocks.0.attn.to_add_out", |
| "transformer_blocks.0.attn.to_k", |
| "transformer_blocks.0.attn.to_out.0", |
| "transformer_blocks.0.attn.to_q", |
| "transformer_blocks.0.attn.to_v", |
| "transformer_blocks.0.ff.net.0.proj", |
| "transformer_blocks.0.ff.net.2", |
| "transformer_blocks.0.ff_context.net.0.proj", |
| "transformer_blocks.0.ff_context.net.2", |
| "transformer_blocks.1.attn.add_k_proj", |
| "transformer_blocks.1.attn.add_q_proj", |
| "transformer_blocks.1.attn.add_v_proj", |
| "transformer_blocks.1.attn.to_add_out", |
| "transformer_blocks.1.attn.to_k", |
| "transformer_blocks.1.attn.to_out.0", |
| "transformer_blocks.1.attn.to_q", |
| "transformer_blocks.1.attn.to_v", |
| "transformer_blocks.1.ff.net.0.proj", |
| "transformer_blocks.1.ff.net.2", |
| "transformer_blocks.1.ff_context.net.0.proj", |
| "transformer_blocks.1.ff_context.net.2", |
| "transformer_blocks.10.attn.add_k_proj", |
| "transformer_blocks.10.attn.add_q_proj", |
| "transformer_blocks.10.attn.add_v_proj", |
| "transformer_blocks.10.attn.to_add_out", |
| "transformer_blocks.10.attn.to_k", |
| "transformer_blocks.10.attn.to_out.0", |
| "transformer_blocks.10.attn.to_q", |
| "transformer_blocks.10.attn.to_v", |
| "transformer_blocks.10.ff.net.0.proj", |
| "transformer_blocks.10.ff.net.2", |
| "transformer_blocks.10.ff_context.net.0.proj", |
| "transformer_blocks.10.ff_context.net.2", |
| "transformer_blocks.11.attn.add_k_proj", |
| "transformer_blocks.11.attn.add_q_proj", |
| "transformer_blocks.11.attn.add_v_proj", |
| "transformer_blocks.11.attn.to_add_out", |
| "transformer_blocks.11.attn.to_k", |
| "transformer_blocks.11.attn.to_out.0", |
| "transformer_blocks.11.attn.to_q", |
| "transformer_blocks.11.attn.to_v", |
| "transformer_blocks.11.ff.net.0.proj", |
| "transformer_blocks.11.ff.net.2", |
| "transformer_blocks.11.ff_context.net.0.proj", |
| "transformer_blocks.11.ff_context.net.2", |
| "transformer_blocks.12.attn.add_k_proj", |
| "transformer_blocks.12.attn.add_q_proj", |
| "transformer_blocks.12.attn.add_v_proj", |
| "transformer_blocks.12.attn.to_add_out", |
| "transformer_blocks.12.attn.to_k", |
| "transformer_blocks.12.attn.to_out.0", |
| "transformer_blocks.12.attn.to_q", |
| "transformer_blocks.12.attn.to_v", |
| "transformer_blocks.12.ff.net.0.proj", |
| "transformer_blocks.12.ff.net.2", |
| "transformer_blocks.12.ff_context.net.0.proj", |
| "transformer_blocks.12.ff_context.net.2", |
| "transformer_blocks.13.attn.add_k_proj", |
| "transformer_blocks.13.attn.add_q_proj", |
| "transformer_blocks.13.attn.add_v_proj", |
| "transformer_blocks.13.attn.to_add_out", |
| "transformer_blocks.13.attn.to_k", |
| "transformer_blocks.13.attn.to_out.0", |
| "transformer_blocks.13.attn.to_q", |
| "transformer_blocks.13.attn.to_v", |
| "transformer_blocks.13.ff.net.0.proj", |
| "transformer_blocks.13.ff.net.2", |
| "transformer_blocks.13.ff_context.net.0.proj", |
| "transformer_blocks.13.ff_context.net.2", |
| "transformer_blocks.14.attn.add_k_proj", |
| "transformer_blocks.14.attn.add_q_proj", |
| "transformer_blocks.14.attn.add_v_proj", |
| "transformer_blocks.14.attn.to_add_out", |
| "transformer_blocks.14.attn.to_k", |
| "transformer_blocks.14.attn.to_out.0", |
| "transformer_blocks.14.attn.to_q", |
| "transformer_blocks.14.attn.to_v", |
| "transformer_blocks.14.ff.net.0.proj", |
| "transformer_blocks.14.ff.net.2", |
| "transformer_blocks.14.ff_context.net.0.proj", |
| "transformer_blocks.14.ff_context.net.2", |
| "transformer_blocks.15.attn.add_k_proj", |
| "transformer_blocks.15.attn.add_q_proj", |
| "transformer_blocks.15.attn.add_v_proj", |
| "transformer_blocks.15.attn.to_add_out", |
| "transformer_blocks.15.attn.to_k", |
| "transformer_blocks.15.attn.to_out.0", |
| "transformer_blocks.15.attn.to_q", |
| "transformer_blocks.15.attn.to_v", |
| "transformer_blocks.15.ff.net.0.proj", |
| "transformer_blocks.15.ff.net.2", |
| "transformer_blocks.15.ff_context.net.0.proj", |
| "transformer_blocks.15.ff_context.net.2", |
| "transformer_blocks.16.attn.add_k_proj", |
| "transformer_blocks.16.attn.add_q_proj", |
| "transformer_blocks.16.attn.add_v_proj", |
| "transformer_blocks.16.attn.to_add_out", |
| "transformer_blocks.16.attn.to_k", |
| "transformer_blocks.16.attn.to_out.0", |
| "transformer_blocks.16.attn.to_q", |
| "transformer_blocks.16.attn.to_v", |
| "transformer_blocks.16.ff.net.0.proj", |
| "transformer_blocks.16.ff.net.2", |
| "transformer_blocks.16.ff_context.net.0.proj", |
| "transformer_blocks.16.ff_context.net.2", |
| "transformer_blocks.17.attn.add_k_proj", |
| "transformer_blocks.17.attn.add_q_proj", |
| "transformer_blocks.17.attn.add_v_proj", |
| "transformer_blocks.17.attn.to_add_out", |
| "transformer_blocks.17.attn.to_k", |
| "transformer_blocks.17.attn.to_out.0", |
| "transformer_blocks.17.attn.to_q", |
| "transformer_blocks.17.attn.to_v", |
| "transformer_blocks.17.ff.net.0.proj", |
| "transformer_blocks.17.ff.net.2", |
| "transformer_blocks.17.ff_context.net.0.proj", |
| "transformer_blocks.17.ff_context.net.2", |
| "transformer_blocks.18.attn.add_k_proj", |
| "transformer_blocks.18.attn.add_q_proj", |
| "transformer_blocks.18.attn.add_v_proj", |
| "transformer_blocks.18.attn.to_add_out", |
| "transformer_blocks.18.attn.to_k", |
| "transformer_blocks.18.attn.to_out.0", |
| "transformer_blocks.18.attn.to_q", |
| "transformer_blocks.18.attn.to_v", |
| "transformer_blocks.18.ff.net.0.proj", |
| "transformer_blocks.18.ff.net.2", |
| "transformer_blocks.18.ff_context.net.0.proj", |
| "transformer_blocks.18.ff_context.net.2", |
| "transformer_blocks.2.attn.add_k_proj", |
| "transformer_blocks.2.attn.add_q_proj", |
| "transformer_blocks.2.attn.add_v_proj", |
| "transformer_blocks.2.attn.to_add_out", |
| "transformer_blocks.2.attn.to_k", |
| "transformer_blocks.2.attn.to_out.0", |
| "transformer_blocks.2.attn.to_q", |
| "transformer_blocks.2.attn.to_v", |
| "transformer_blocks.2.ff.net.0.proj", |
| "transformer_blocks.2.ff.net.2", |
| "transformer_blocks.2.ff_context.net.0.proj", |
| "transformer_blocks.2.ff_context.net.2", |
| "transformer_blocks.3.attn.add_k_proj", |
| "transformer_blocks.3.attn.add_q_proj", |
| "transformer_blocks.3.attn.add_v_proj", |
| "transformer_blocks.3.attn.to_add_out", |
| "transformer_blocks.3.attn.to_k", |
| "transformer_blocks.3.attn.to_out.0", |
| "transformer_blocks.3.attn.to_q", |
| "transformer_blocks.3.attn.to_v", |
| "transformer_blocks.3.ff.net.0.proj", |
| "transformer_blocks.3.ff.net.2", |
| "transformer_blocks.3.ff_context.net.0.proj", |
| "transformer_blocks.3.ff_context.net.2", |
| "transformer_blocks.4.attn.add_k_proj", |
| "transformer_blocks.4.attn.add_q_proj", |
| "transformer_blocks.4.attn.add_v_proj", |
| "transformer_blocks.4.attn.to_add_out", |
| "transformer_blocks.4.attn.to_k", |
| "transformer_blocks.4.attn.to_out.0", |
| "transformer_blocks.4.attn.to_q", |
| "transformer_blocks.4.attn.to_v", |
| "transformer_blocks.4.ff.net.0.proj", |
| "transformer_blocks.4.ff.net.2", |
| "transformer_blocks.4.ff_context.net.0.proj", |
| "transformer_blocks.4.ff_context.net.2", |
| "transformer_blocks.5.attn.add_k_proj", |
| "transformer_blocks.5.attn.add_q_proj", |
| "transformer_blocks.5.attn.add_v_proj", |
| "transformer_blocks.5.attn.to_add_out", |
| "transformer_blocks.5.attn.to_k", |
| "transformer_blocks.5.attn.to_out.0", |
| "transformer_blocks.5.attn.to_q", |
| "transformer_blocks.5.attn.to_v", |
| "transformer_blocks.5.ff.net.0.proj", |
| "transformer_blocks.5.ff.net.2", |
| "transformer_blocks.5.ff_context.net.0.proj", |
| "transformer_blocks.5.ff_context.net.2", |
| "transformer_blocks.6.attn.add_k_proj", |
| "transformer_blocks.6.attn.add_q_proj", |
| "transformer_blocks.6.attn.add_v_proj", |
| "transformer_blocks.6.attn.to_add_out", |
| "transformer_blocks.6.attn.to_k", |
| "transformer_blocks.6.attn.to_out.0", |
| "transformer_blocks.6.attn.to_q", |
| "transformer_blocks.6.attn.to_v", |
| "transformer_blocks.6.ff.net.0.proj", |
| "transformer_blocks.6.ff.net.2", |
| "transformer_blocks.6.ff_context.net.0.proj", |
| "transformer_blocks.6.ff_context.net.2", |
| "transformer_blocks.7.attn.add_k_proj", |
| "transformer_blocks.7.attn.add_q_proj", |
| "transformer_blocks.7.attn.add_v_proj", |
| "transformer_blocks.7.attn.to_add_out", |
| "transformer_blocks.7.attn.to_k", |
| "transformer_blocks.7.attn.to_out.0", |
| "transformer_blocks.7.attn.to_q", |
| "transformer_blocks.7.attn.to_v", |
| "transformer_blocks.7.ff.net.0.proj", |
| "transformer_blocks.7.ff.net.2", |
| "transformer_blocks.7.ff_context.net.0.proj", |
| "transformer_blocks.7.ff_context.net.2", |
| "transformer_blocks.8.attn.add_k_proj", |
| "transformer_blocks.8.attn.add_q_proj", |
| "transformer_blocks.8.attn.add_v_proj", |
| "transformer_blocks.8.attn.to_add_out", |
| "transformer_blocks.8.attn.to_k", |
| "transformer_blocks.8.attn.to_out.0", |
| "transformer_blocks.8.attn.to_q", |
| "transformer_blocks.8.attn.to_v", |
| "transformer_blocks.8.ff.net.0.proj", |
| "transformer_blocks.8.ff.net.2", |
| "transformer_blocks.8.ff_context.net.0.proj", |
| "transformer_blocks.8.ff_context.net.2", |
| "transformer_blocks.9.attn.add_k_proj", |
| "transformer_blocks.9.attn.add_q_proj", |
| "transformer_blocks.9.attn.add_v_proj", |
| "transformer_blocks.9.attn.to_add_out", |
| "transformer_blocks.9.attn.to_k", |
| "transformer_blocks.9.attn.to_out.0", |
| "transformer_blocks.9.attn.to_q", |
| "transformer_blocks.9.attn.to_v", |
| "transformer_blocks.9.ff.net.0.proj", |
| "transformer_blocks.9.ff.net.2", |
| "transformer_blocks.9.ff_context.net.0.proj", |
| "transformer_blocks.9.ff_context.net.2" |
| ] |
| }, |
| "awq_w4a16": { |
| "precision": "int4", |
| "group_size": 64, |
| "targets": [ |
| "single_transformer_blocks.0.norm.linear", |
| "single_transformer_blocks.1.norm.linear", |
| "single_transformer_blocks.10.norm.linear", |
| "single_transformer_blocks.11.norm.linear", |
| "single_transformer_blocks.12.norm.linear", |
| "single_transformer_blocks.13.norm.linear", |
| "single_transformer_blocks.14.norm.linear", |
| "single_transformer_blocks.15.norm.linear", |
| "single_transformer_blocks.16.norm.linear", |
| "single_transformer_blocks.17.norm.linear", |
| "single_transformer_blocks.18.norm.linear", |
| "single_transformer_blocks.19.norm.linear", |
| "single_transformer_blocks.2.norm.linear", |
| "single_transformer_blocks.20.norm.linear", |
| "single_transformer_blocks.21.norm.linear", |
| "single_transformer_blocks.22.norm.linear", |
| "single_transformer_blocks.23.norm.linear", |
| "single_transformer_blocks.24.norm.linear", |
| "single_transformer_blocks.25.norm.linear", |
| "single_transformer_blocks.26.norm.linear", |
| "single_transformer_blocks.27.norm.linear", |
| "single_transformer_blocks.28.norm.linear", |
| "single_transformer_blocks.29.norm.linear", |
| "single_transformer_blocks.3.norm.linear", |
| "single_transformer_blocks.30.norm.linear", |
| "single_transformer_blocks.31.norm.linear", |
| "single_transformer_blocks.32.norm.linear", |
| "single_transformer_blocks.33.norm.linear", |
| "single_transformer_blocks.34.norm.linear", |
| "single_transformer_blocks.35.norm.linear", |
| "single_transformer_blocks.36.norm.linear", |
| "single_transformer_blocks.37.norm.linear", |
| "single_transformer_blocks.4.norm.linear", |
| "single_transformer_blocks.5.norm.linear", |
| "single_transformer_blocks.6.norm.linear", |
| "single_transformer_blocks.7.norm.linear", |
| "single_transformer_blocks.8.norm.linear", |
| "single_transformer_blocks.9.norm.linear", |
| "transformer_blocks.0.norm1.linear", |
| "transformer_blocks.0.norm1_context.linear", |
| "transformer_blocks.1.norm1.linear", |
| "transformer_blocks.1.norm1_context.linear", |
| "transformer_blocks.10.norm1.linear", |
| "transformer_blocks.10.norm1_context.linear", |
| "transformer_blocks.11.norm1.linear", |
| "transformer_blocks.11.norm1_context.linear", |
| "transformer_blocks.12.norm1.linear", |
| "transformer_blocks.12.norm1_context.linear", |
| "transformer_blocks.13.norm1.linear", |
| "transformer_blocks.13.norm1_context.linear", |
| "transformer_blocks.14.norm1.linear", |
| "transformer_blocks.14.norm1_context.linear", |
| "transformer_blocks.15.norm1.linear", |
| "transformer_blocks.15.norm1_context.linear", |
| "transformer_blocks.16.norm1.linear", |
| "transformer_blocks.16.norm1_context.linear", |
| "transformer_blocks.17.norm1.linear", |
| "transformer_blocks.17.norm1_context.linear", |
| "transformer_blocks.18.norm1.linear", |
| "transformer_blocks.18.norm1_context.linear", |
| "transformer_blocks.2.norm1.linear", |
| "transformer_blocks.2.norm1_context.linear", |
| "transformer_blocks.3.norm1.linear", |
| "transformer_blocks.3.norm1_context.linear", |
| "transformer_blocks.4.norm1.linear", |
| "transformer_blocks.4.norm1_context.linear", |
| "transformer_blocks.5.norm1.linear", |
| "transformer_blocks.5.norm1_context.linear", |
| "transformer_blocks.6.norm1.linear", |
| "transformer_blocks.6.norm1_context.linear", |
| "transformer_blocks.7.norm1.linear", |
| "transformer_blocks.7.norm1_context.linear", |
| "transformer_blocks.8.norm1.linear", |
| "transformer_blocks.8.norm1_context.linear", |
| "transformer_blocks.9.norm1.linear", |
| "transformer_blocks.9.norm1_context.linear" |
| ] |
| } |
| } |
| } |
|
|
