diff --git "a/Hyper-SDXL-8steps-CFG-lora.sdmlxpatch/manifest.json" "b/Hyper-SDXL-8steps-CFG-lora.sdmlxpatch/manifest.json" new file mode 100644--- /dev/null +++ "b/Hyper-SDXL-8steps-CFG-lora.sdmlxpatch/manifest.json" @@ -0,0 +1,12968 @@ +{ + "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 + ] + } + ] +}