Upload folder using huggingface_hub

flux_transformer/config.json

@@ -0,0 +1,27 @@
+{
+  "_class_name": "Flux2Transformer2DModel",
+  "_diffusers_version": "0.37.0",
+  "_name_or_path": "/inspire/hdd/global_user/nieming-CUCXS125005/jiangjunzhe/hf_models/hub/models--black-forest-labs--FLUX.2-klein-base-9B/snapshots/32773329fbe7e81a90ef971740e8ba4b0364ecf3",
+  "attention_head_dim": 128,
+  "axes_dims_rope": [
+    32,
+    32,
+    32,
+    32
+  ],
+  "eps": 1e-06,
+  "guidance_embeds": false,
+  "in_channels": 128,
+  "joint_attention_dim": 12288,
+  "mask_channels": 256,
+  "mask_layers": 8,
+  "mask_single_layers": 0,
+  "mlp_ratio": 3.0,
+  "num_attention_heads": 32,
+  "num_layers": 8,
+  "num_single_layers": 24,
+  "out_channels": null,
+  "patch_size": 1,
+  "rope_theta": 2000,
+  "timestep_guidance_channels": 256
+}

flux_transformer/diffusion_pytorch_model-00001-of-00004.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6ceb2fd0dd5343a56ef903e2596a255a52e1a89b2721928a710d9e7fd98490c9
+size 9905929104

flux_transformer/diffusion_pytorch_model-00002-of-00004.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:196fae3747e2d112c0c1b85c02ce460bd661ac9e4ac1dcbd90990a3a660dd0a8
+size 9999240512

flux_transformer/diffusion_pytorch_model-00003-of-00004.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:772ca9531c0adcb6422e5ff14b38e217cc3bd3d5583ef15ec0a3f195f0184165
+size 9965689280

flux_transformer/diffusion_pytorch_model-00004-of-00004.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:87a0755a03a9d17b373c28f873e3d31612ce603a3f8012a9a1a8ea6d177b5061
+size 2786204360

flux_transformer/diffusion_pytorch_model.safetensors.index.json

@@ -0,0 +1,385 @@
+{
+  "metadata": {
+    "total_size": 32657018880
+  },
+  "weight_map": {
+    "context_embedder.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "controlnet_blocks.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "controlnet_blocks.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "controlnet_blocks.1.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "controlnet_blocks.1.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "controlnet_blocks.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "controlnet_blocks.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "controlnet_blocks.3.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "controlnet_blocks.3.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "controlnet_blocks.4.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "controlnet_blocks.4.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "controlnet_blocks.5.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "controlnet_blocks.5.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "controlnet_blocks.6.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "controlnet_blocks.6.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "controlnet_blocks.7.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "controlnet_blocks.7.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "double_stream_modulation_img.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "double_stream_modulation_txt.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.0.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.0.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.0.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.0.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.0.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.0.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.0.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.0.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.0.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.0.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.0.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.0.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.0.ff.linear_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.0.ff.linear_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.0.ff_context.linear_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.0.ff_context.linear_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.1.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.1.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.1.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.1.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.1.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.1.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.1.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.1.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.1.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.1.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.1.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.1.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "mask_blocks.1.ff.linear_in.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.1.ff.linear_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.1.ff_context.linear_in.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.1.ff_context.linear_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.2.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.2.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.2.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.2.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.2.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.2.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.2.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.2.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.2.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.2.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.2.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.2.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.2.ff.linear_in.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.2.ff.linear_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.2.ff_context.linear_in.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.2.ff_context.linear_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.3.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.3.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.3.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.3.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.3.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.3.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.3.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.3.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.3.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.3.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.3.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.3.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.3.ff.linear_in.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.3.ff.linear_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.3.ff_context.linear_in.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.3.ff_context.linear_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.4.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.4.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.4.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.4.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.4.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.4.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.4.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.4.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.4.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.4.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.4.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.4.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.4.ff.linear_in.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.4.ff.linear_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.4.ff_context.linear_in.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.4.ff_context.linear_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.5.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.5.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.5.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.5.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.5.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.5.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.5.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.5.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.5.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.5.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.5.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.5.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.5.ff.linear_in.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.5.ff.linear_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.5.ff_context.linear_in.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.5.ff_context.linear_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.6.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.6.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.6.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.6.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.6.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.6.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.6.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.6.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.6.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.6.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.6.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.6.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.6.ff.linear_in.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.6.ff.linear_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.6.ff_context.linear_in.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.6.ff_context.linear_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.7.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "mask_blocks.7.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "mask_blocks.7.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "mask_blocks.7.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "mask_blocks.7.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "mask_blocks.7.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "mask_blocks.7.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "mask_blocks.7.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "mask_blocks.7.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "mask_blocks.7.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "mask_blocks.7.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "mask_blocks.7.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "mask_blocks.7.ff.linear_in.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "mask_blocks.7.ff.linear_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "mask_blocks.7.ff_context.linear_in.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "mask_blocks.7.ff_context.linear_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "mask_embedder.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "norm_out.linear.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "proj_out.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_stream_modulation.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "single_transformer_blocks.0.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.0.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.0.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.0.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.1.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.1.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.1.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.1.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.10.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.10.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.10.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.10.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.11.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.11.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.11.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.11.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.12.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.12.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.12.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.12.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.13.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.13.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.13.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.13.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.14.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.14.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.14.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.14.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.15.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.15.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.15.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.15.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.16.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.16.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.16.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.16.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.17.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.17.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.17.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.17.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.18.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.18.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.18.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.18.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.19.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.19.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.19.attn.to_out.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.19.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.2.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.2.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.2.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.2.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.20.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.20.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.20.attn.to_out.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.20.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.21.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.21.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.21.attn.to_out.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.21.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.22.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.22.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.22.attn.to_out.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.22.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.23.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.23.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.23.attn.to_out.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.23.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "single_transformer_blocks.3.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.3.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.3.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.3.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.4.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.4.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.4.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.4.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.5.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.5.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.5.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.5.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.6.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.6.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.6.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.6.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.7.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.7.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.7.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.7.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.8.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.8.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.8.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.8.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.9.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.9.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.9.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "single_transformer_blocks.9.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "time_guidance_embed.timestep_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "time_guidance_embed.timestep_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.ff.linear_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.ff.linear_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.ff_context.linear_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.ff_context.linear_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.ff.linear_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.ff.linear_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.ff_context.linear_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.ff_context.linear_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.ff.linear_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.ff.linear_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.ff_context.linear_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.ff_context.linear_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.ff.linear_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.ff.linear_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.ff_context.linear_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.ff_context.linear_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.ff.linear_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.ff.linear_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.ff_context.linear_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.ff_context.linear_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.ff.linear_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.ff.linear_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.ff_context.linear_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.ff_context.linear_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.ff.linear_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.ff.linear_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.ff_context.linear_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.ff_context.linear_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.ff.linear_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.ff.linear_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.ff_context.linear_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.ff_context.linear_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "x_embedder.weight": "diffusion_pytorch_model-00001-of-00004.safetensors"
+  }
+}