{ "base_model_family": "sdxl", "components": { "factors": "patch.safetensors" }, "created_at_unix": 1779345841, "factor_layout": { "conv2d": { "lora_down": "[rank, kernel_h, kernel_w, in]", "lora_up": "[out, rank]" }, "linear": { "lora_down": "[rank, in]", "lora_up": "[out, rank]" }, "scale": "runtime_strength * alpha / rank" }, "format": "sdmlx-acceleration-patch-v1", "module_count": 858, "modules": [ { "alpha": 8.0, "down_shape": [ 64, 3, 3, 320 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.0.downsample.weight", "up_shape": [ 320, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 320 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.0.resnets.0.conv1.weight", "up_shape": [ 320, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 320 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.0.resnets.0.conv2.weight", "up_shape": [ 320, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.0.resnets.0.time_emb_proj.weight", "up_shape": [ 320, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 320 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.0.resnets.1.conv1.weight", "up_shape": [ 320, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 320 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.0.resnets.1.conv2.weight", "up_shape": [ 320, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.0.resnets.1.time_emb_proj.weight", "up_shape": [ 320, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.proj_in.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.proj_out.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.0.attn1.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.0.attn1.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.0.attn1.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.0.attn1.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.0.attn2.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.0.attn2.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.0.attn2.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.0.attn2.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.0.linear1.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.0.linear2.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2560 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.0.linear3.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.1.attn1.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.1.attn1.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.1.attn1.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.1.attn1.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.1.attn2.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.1.attn2.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.1.attn2.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.1.attn2.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.1.linear1.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.1.linear2.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2560 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.0.transformer_blocks.1.linear3.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.proj_in.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.proj_out.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.0.attn1.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.0.attn1.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.0.attn1.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.0.attn1.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.0.attn2.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.0.attn2.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.0.attn2.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.0.attn2.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.0.linear1.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.0.linear2.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2560 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.0.linear3.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.1.attn1.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.1.attn1.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.1.attn1.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.1.attn1.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.1.attn2.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.1.attn2.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.1.attn2.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.1.attn2.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.1.linear1.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.1.linear2.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2560 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.attentions.1.transformer_blocks.1.linear3.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.downsample.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 320 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.resnets.0.conv1.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.resnets.0.conv2.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 320 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.resnets.0.conv_shortcut.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.resnets.0.time_emb_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.resnets.1.conv1.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.resnets.1.conv2.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.1.resnets.1.time_emb_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.proj_in.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.proj_out.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.0.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.0.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.0.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.0.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.0.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.0.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.0.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.0.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.0.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.0.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.0.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.1.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.1.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.1.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.1.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.1.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.1.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.1.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.1.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.1.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.1.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.1.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.2.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.2.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.2.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.2.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.2.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.2.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.2.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.2.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.2.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.2.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.2.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.3.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.3.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.3.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.3.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.3.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.3.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.3.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.3.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.3.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.3.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.3.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.4.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.4.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.4.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.4.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.4.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.4.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.4.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.4.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.4.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.4.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.4.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.5.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.5.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.5.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.5.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.5.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.5.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.5.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.5.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.5.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.5.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.5.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.6.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.6.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.6.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.6.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.6.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.6.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.6.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.6.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.6.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.6.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.6.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.7.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.7.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.7.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.7.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.7.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.7.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.7.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.7.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.7.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.7.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.7.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.8.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.8.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.8.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.8.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.8.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.8.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.8.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.8.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.8.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.8.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.8.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.9.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.9.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.9.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.9.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.9.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.9.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.9.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.9.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.9.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.9.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.0.transformer_blocks.9.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.proj_in.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.proj_out.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.0.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.0.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.0.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.0.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.0.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.0.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.0.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.0.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.0.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.0.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.0.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.1.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.1.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.1.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.1.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.1.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.1.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.1.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.1.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.1.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.1.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.1.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.2.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.2.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.2.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.2.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.2.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.2.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.2.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.2.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.2.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.2.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.2.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.3.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.3.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.3.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.3.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.3.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.3.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.3.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.3.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.3.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.3.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.3.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.4.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.4.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.4.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.4.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.4.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.4.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.4.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.4.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.4.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.4.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.4.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.5.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.5.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.5.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.5.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.5.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.5.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.5.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.5.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.5.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.5.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.5.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.6.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.6.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.6.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.6.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.6.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.6.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.6.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.6.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.6.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.6.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.6.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.7.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.7.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.7.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.7.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.7.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.7.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.7.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.7.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.7.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.7.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.7.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.8.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.8.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.8.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.8.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.8.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.8.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.8.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.8.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.8.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.8.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.8.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.9.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.9.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.9.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.9.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.9.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.9.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.9.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.9.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.9.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.9.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.attentions.1.transformer_blocks.9.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.resnets.0.conv1.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.resnets.0.conv2.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.resnets.0.conv_shortcut.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.resnets.0.time_emb_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.resnets.1.conv1.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.resnets.1.conv2.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "down_blocks.2.resnets.1.time_emb_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.proj_in.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.proj_out.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.0.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.0.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.0.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.0.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.0.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.0.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.0.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.0.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.0.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.0.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.0.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.1.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.1.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.1.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.1.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.1.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.1.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.1.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.1.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.1.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.1.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.1.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.2.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.2.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.2.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.2.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.2.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.2.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.2.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.2.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.2.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.2.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.2.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.3.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.3.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.3.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.3.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.3.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.3.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.3.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.3.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.3.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.3.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.3.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.4.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.4.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.4.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.4.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.4.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.4.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.4.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.4.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.4.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.4.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.4.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.5.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.5.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.5.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.5.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.5.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.5.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.5.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.5.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.5.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.5.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.5.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.6.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.6.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.6.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.6.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.6.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.6.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.6.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.6.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.6.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.6.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.6.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.7.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.7.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.7.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.7.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.7.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.7.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.7.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.7.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.7.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.7.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.7.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.8.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.8.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.8.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.8.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.8.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.8.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.8.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.8.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.8.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.8.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.8.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.9.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.9.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.9.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.9.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.9.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.9.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.9.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.9.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.9.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.9.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.1.transformer_blocks.9.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.0.conv1.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.0.conv2.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.0.time_emb_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.2.conv1.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.2.conv2.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "mid_blocks.2.time_emb_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.proj_in.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.proj_out.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.0.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.0.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.0.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.0.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.0.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.0.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.0.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.0.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.0.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.0.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.0.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.1.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.1.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.1.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.1.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.1.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.1.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.1.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.1.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.1.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.1.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.1.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.2.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.2.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.2.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.2.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.2.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.2.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.2.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.2.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.2.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.2.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.2.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.3.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.3.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.3.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.3.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.3.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.3.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.3.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.3.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.3.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.3.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.3.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.4.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.4.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.4.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.4.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.4.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.4.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.4.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.4.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.4.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.4.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.4.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.5.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.5.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.5.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.5.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.5.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.5.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.5.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.5.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.5.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.5.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.5.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.6.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.6.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.6.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.6.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.6.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.6.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.6.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.6.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.6.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.6.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.6.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.7.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.7.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.7.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.7.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.7.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.7.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.7.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.7.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.7.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.7.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.7.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.8.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.8.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.8.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.8.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.8.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.8.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.8.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.8.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.8.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.8.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.8.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.9.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.9.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.9.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.9.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.9.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.9.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.9.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.9.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.9.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.9.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.0.transformer_blocks.9.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.proj_in.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.proj_out.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.0.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.0.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.0.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.0.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.0.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.0.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.0.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.0.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.0.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.0.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.0.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.1.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.1.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.1.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.1.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.1.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.1.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.1.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.1.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.1.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.1.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.1.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.2.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.2.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.2.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.2.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.2.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.2.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.2.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.2.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.2.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.2.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.2.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.3.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.3.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.3.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.3.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.3.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.3.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.3.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.3.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.3.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.3.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.3.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.4.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.4.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.4.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.4.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.4.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.4.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.4.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.4.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.4.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.4.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.4.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.5.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.5.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.5.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.5.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.5.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.5.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.5.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.5.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.5.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.5.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.5.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.6.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.6.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.6.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.6.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.6.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.6.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.6.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.6.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.6.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.6.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.6.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.7.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.7.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.7.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.7.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.7.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.7.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.7.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.7.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.7.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.7.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.7.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.8.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.8.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.8.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.8.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.8.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.8.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.8.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.8.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.8.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.8.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.8.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.9.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.9.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.9.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.9.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.9.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.9.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.9.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.9.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.9.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.9.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.1.transformer_blocks.9.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.proj_in.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.proj_out.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.0.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.0.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.0.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.0.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.0.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.0.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.0.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.0.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.0.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.0.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.0.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.1.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.1.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.1.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.1.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.1.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.1.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.1.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.1.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.1.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.1.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.1.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.2.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.2.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.2.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.2.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.2.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.2.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.2.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.2.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.2.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.2.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.2.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.3.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.3.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.3.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.3.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.3.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.3.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.3.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.3.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.3.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.3.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.3.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.4.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.4.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.4.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.4.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.4.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.4.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.4.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.4.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.4.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.4.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.4.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.5.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.5.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.5.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.5.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.5.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.5.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.5.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.5.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.5.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.5.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.5.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.6.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.6.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.6.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.6.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.6.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.6.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.6.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.6.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.6.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.6.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.6.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.7.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.7.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.7.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.7.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.7.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.7.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.7.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.7.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.7.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.7.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.7.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.8.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.8.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.8.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.8.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.8.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.8.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.8.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.8.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.8.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.8.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.8.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.9.attn1.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.9.attn1.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.9.attn1.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.9.attn1.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.9.attn2.key_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.9.attn2.out_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.9.attn2.query_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.9.attn2.value_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.9.linear1.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.9.linear2.weight", "up_shape": [ 5120, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 5120 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.attentions.2.transformer_blocks.9.linear3.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 2560 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.resnets.0.conv1.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.resnets.0.conv2.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2560 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.resnets.0.conv_shortcut.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.resnets.0.time_emb_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 2560 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.resnets.1.conv1.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.resnets.1.conv2.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2560 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.resnets.1.conv_shortcut.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.resnets.1.time_emb_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 1920 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.resnets.2.conv1.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.resnets.2.conv2.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1920 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.resnets.2.conv_shortcut.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.resnets.2.time_emb_proj.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.0.upsample.weight", "up_shape": [ 1280, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.proj_in.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.proj_out.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.0.attn1.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.0.attn1.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.0.attn1.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.0.attn1.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.0.attn2.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.0.attn2.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.0.attn2.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.0.attn2.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.0.linear1.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.0.linear2.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2560 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.0.linear3.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.1.attn1.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.1.attn1.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.1.attn1.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.1.attn1.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.1.attn2.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.1.attn2.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.1.attn2.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.1.attn2.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.1.linear1.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.1.linear2.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2560 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.0.transformer_blocks.1.linear3.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.proj_in.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.proj_out.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.0.attn1.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.0.attn1.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.0.attn1.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.0.attn1.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.0.attn2.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.0.attn2.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.0.attn2.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.0.attn2.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.0.linear1.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.0.linear2.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2560 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.0.linear3.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.1.attn1.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.1.attn1.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.1.attn1.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.1.attn1.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.1.attn2.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.1.attn2.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.1.attn2.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.1.attn2.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.1.linear1.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.1.linear2.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2560 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.1.transformer_blocks.1.linear3.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.proj_in.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.proj_out.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.0.attn1.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.0.attn1.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.0.attn1.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.0.attn1.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.0.attn2.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.0.attn2.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.0.attn2.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.0.attn2.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.0.linear1.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.0.linear2.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2560 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.0.linear3.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.1.attn1.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.1.attn1.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.1.attn1.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.1.attn1.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.1.attn2.key_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.1.attn2.out_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.1.attn2.query_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2048 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.1.attn2.value_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.1.linear1.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.1.linear2.weight", "up_shape": [ 2560, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 2560 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.attentions.2.transformer_blocks.1.linear3.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 1920 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.resnets.0.conv1.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.resnets.0.conv2.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1920 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.resnets.0.conv_shortcut.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.resnets.0.time_emb_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.resnets.1.conv1.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.resnets.1.conv2.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.resnets.1.conv_shortcut.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.resnets.1.time_emb_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 960 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.resnets.2.conv1.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.resnets.2.conv2.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 960 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.resnets.2.conv_shortcut.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.resnets.2.time_emb_proj.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.1.upsample.weight", "up_shape": [ 640, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 960 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.2.resnets.0.conv1.weight", "up_shape": [ 320, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 320 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.2.resnets.0.conv2.weight", "up_shape": [ 320, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 960 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.2.resnets.0.conv_shortcut.weight", "up_shape": [ 320, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.2.resnets.0.time_emb_proj.weight", "up_shape": [ 320, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.2.resnets.1.conv1.weight", "up_shape": [ 320, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 320 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.2.resnets.1.conv2.weight", "up_shape": [ 320, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.2.resnets.1.conv_shortcut.weight", "up_shape": [ 320, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.2.resnets.1.time_emb_proj.weight", "up_shape": [ 320, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.2.resnets.2.conv1.weight", "up_shape": [ 320, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 3, 3, 320 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.2.resnets.2.conv2.weight", "up_shape": [ 320, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 640 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.2.resnets.2.conv_shortcut.weight", "up_shape": [ 320, 64 ] }, { "alpha": 8.0, "down_shape": [ 64, 1280 ], "rank": 64, "source_prefix": null, "split_index": null, "target": "up_blocks.2.resnets.2.time_emb_proj.weight", "up_shape": [ 320, 64 ] } ] }