generator_o.ipynb

{
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github"
      },
      "source": [
        "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/camenduru/ai-creator-archive/blob/main/generator_o.ipynb)"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "cellView": "form",
        "id": "-f1FILN0oFL4"
      },
      "outputs": [],
      "source": [
        "# https://github.com/kjsman/stable-diffusion-pytorch/issues/7#issuecomment-1426839447 modified thanks to @treeform ❤\n",
        "\n",
        "%cd /content\n",
        "!git clone -b dev https://github.com/camenduru/stable-diffusion-pytorch\n",
        "!mkdir /content/stable-diffusion-pytorch/data/ckpt\n",
        "\n",
        "%cd /content/stable-diffusion-pytorch\n",
        "!pip install piexif fold_to_ascii discord ftfy safetensors -r requirements.txt\n",
        "!apt -y install -qq aria2\n",
        "\n",
        "import torch, gc\n",
        "from safetensors.torch import save_file, load_file\n",
        "import os, gc, requests, json, piexif, torch\n",
        "from PIL.PngImagePlugin import PngInfo\n",
        "from fold_to_ascii import fold\n",
        "from stable_diffusion_pytorch import pipeline\n",
        "\n",
        "metadata = PngInfo()\n",
        "\n",
        "def closestNumber(n, m):\n",
        "    q = int(n / m)\n",
        "    n1 = m * q\n",
        "    if (n * m) > 0:\n",
        "        n2 = m * (q + 1)\n",
        "    else:\n",
        "        n2 = m * (q - 1)\n",
        "    if abs(n - n1) < abs(n - n2):\n",
        "        return n1\n",
        "    return n2\n",
        "\n",
        "root_folder = 'images'\n",
        "\n",
        "if os.path.exists(f\"{root_folder}\") == False:\n",
        "    os.mkdir(f\"{root_folder}\")\n",
        "image_folder = max([int(f) for f in os.listdir(f\"{root_folder}\")], default=0)\n",
        "if os.path.exists(f\"{root_folder}/{image_folder:04}\") == False:\n",
        "    os.mkdir(f\"{root_folder}/{image_folder:04}\")\n",
        "name = max([int(f[: f.index(\".\")]) for f in os.listdir(f\"{root_folder}/{image_folder:04}\")], default=0)\n",
        "\n",
        "def generate(discord_token, discord_channel_id, ckpt, discord_user, by, num_inference_steps, guidance_scale, sampler, width, height, prompt, negative_prompt, suffix, image_folder, name):\n",
        "    print(ckpt)\n",
        "    !rm /content/model\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M {ckpt} -d /content -o model\n",
        "\n",
        "    if 'models' in globals():  \n",
        "      del models\n",
        "    with torch.no_grad():\n",
        "      torch.cuda.empty_cache()\n",
        "    gc.collect()\n",
        "\n",
        "    old = load_file(\"/content/model\")\n",
        "    # old = torch.load(\"/content/model\", weights_only = False, map_location=torch.device('cuda'))[\"state_dict\"]\n",
        "    # old = torch.load(\"/content/model\", weights_only = False, map_location=torch.device('cuda'))\n",
        "\n",
        "    new = {}\n",
        "    new['diffusion'] = {}\n",
        "    new['encoder'] = {}\n",
        "    new['decoder'] = {}\n",
        "    new['clip'] = {}\n",
        "\n",
        "    new['diffusion']['time_embedding.linear_1.weight'] = old['model.diffusion_model.time_embed.0.weight']\n",
        "    new['diffusion']['time_embedding.linear_1.bias'] = old['model.diffusion_model.time_embed.0.bias']\n",
        "    new['diffusion']['time_embedding.linear_2.weight'] = old['model.diffusion_model.time_embed.2.weight']\n",
        "    new['diffusion']['time_embedding.linear_2.bias'] = old['model.diffusion_model.time_embed.2.bias']\n",
        "    new['diffusion']['unet.encoders.0.0.weight'] = old['model.diffusion_model.input_blocks.0.0.weight']\n",
        "    new['diffusion']['unet.encoders.0.0.bias'] = old['model.diffusion_model.input_blocks.0.0.bias']\n",
        "    new['diffusion']['unet.encoders.1.0.groupnorm_feature.weight'] = old['model.diffusion_model.input_blocks.1.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.encoders.1.0.groupnorm_feature.bias'] = old['model.diffusion_model.input_blocks.1.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.encoders.1.0.conv_feature.weight'] = old['model.diffusion_model.input_blocks.1.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.encoders.1.0.conv_feature.bias'] = old['model.diffusion_model.input_blocks.1.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.encoders.1.0.linear_time.weight'] = old['model.diffusion_model.input_blocks.1.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.encoders.1.0.linear_time.bias'] = old['model.diffusion_model.input_blocks.1.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.encoders.1.0.groupnorm_merged.weight'] = old['model.diffusion_model.input_blocks.1.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.encoders.1.0.groupnorm_merged.bias'] = old['model.diffusion_model.input_blocks.1.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.encoders.1.0.conv_merged.weight'] = old['model.diffusion_model.input_blocks.1.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.encoders.1.0.conv_merged.bias'] = old['model.diffusion_model.input_blocks.1.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.encoders.1.1.groupnorm.weight'] = old['model.diffusion_model.input_blocks.1.1.norm.weight']\n",
        "    new['diffusion']['unet.encoders.1.1.groupnorm.bias'] = old['model.diffusion_model.input_blocks.1.1.norm.bias']\n",
        "    new['diffusion']['unet.encoders.1.1.conv_input.weight'] = old['model.diffusion_model.input_blocks.1.1.proj_in.weight']\n",
        "    new['diffusion']['unet.encoders.1.1.conv_input.bias'] = old['model.diffusion_model.input_blocks.1.1.proj_in.bias']\n",
        "    new['diffusion']['unet.encoders.1.1.attention_1.out_proj.weight'] = old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn1.to_out.0.weight']\n",
        "    new['diffusion']['unet.encoders.1.1.attention_1.out_proj.bias'] = old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn1.to_out.0.bias']\n",
        "    new['diffusion']['unet.encoders.1.1.linear_geglu_1.weight'] = old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.ff.net.0.proj.weight']\n",
        "    new['diffusion']['unet.encoders.1.1.linear_geglu_1.bias'] = old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.ff.net.0.proj.bias']\n",
        "    new['diffusion']['unet.encoders.1.1.linear_geglu_2.weight'] = old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.ff.net.2.weight']\n",
        "    new['diffusion']['unet.encoders.1.1.linear_geglu_2.bias'] = old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.ff.net.2.bias']\n",
        "    new['diffusion']['unet.encoders.1.1.attention_2.q_proj.weight'] = old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn2.to_q.weight']\n",
        "    new['diffusion']['unet.encoders.1.1.attention_2.k_proj.weight'] = old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn2.to_k.weight']\n",
        "    new['diffusion']['unet.encoders.1.1.attention_2.v_proj.weight'] = old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn2.to_v.weight']\n",
        "    new['diffusion']['unet.encoders.1.1.attention_2.out_proj.weight'] = old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn2.to_out.0.weight']\n",
        "    new['diffusion']['unet.encoders.1.1.attention_2.out_proj.bias'] = old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn2.to_out.0.bias']\n",
        "    new['diffusion']['unet.encoders.1.1.layernorm_1.weight'] = old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.norm1.weight']\n",
        "    new['diffusion']['unet.encoders.1.1.layernorm_1.bias'] = old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.norm1.bias']\n",
        "    new['diffusion']['unet.encoders.1.1.layernorm_2.weight'] = old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.norm2.weight']\n",
        "    new['diffusion']['unet.encoders.1.1.layernorm_2.bias'] = old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.norm2.bias']\n",
        "    new['diffusion']['unet.encoders.1.1.layernorm_3.weight'] = old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.norm3.weight']\n",
        "    new['diffusion']['unet.encoders.1.1.layernorm_3.bias'] = old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.norm3.bias']\n",
        "    new['diffusion']['unet.encoders.1.1.conv_output.weight'] = old['model.diffusion_model.input_blocks.1.1.proj_out.weight']\n",
        "    new['diffusion']['unet.encoders.1.1.conv_output.bias'] = old['model.diffusion_model.input_blocks.1.1.proj_out.bias']\n",
        "    new['diffusion']['unet.encoders.2.0.groupnorm_feature.weight'] = old['model.diffusion_model.input_blocks.2.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.encoders.2.0.groupnorm_feature.bias'] = old['model.diffusion_model.input_blocks.2.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.encoders.2.0.conv_feature.weight'] = old['model.diffusion_model.input_blocks.2.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.encoders.2.0.conv_feature.bias'] = old['model.diffusion_model.input_blocks.2.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.encoders.2.0.linear_time.weight'] = old['model.diffusion_model.input_blocks.2.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.encoders.2.0.linear_time.bias'] = old['model.diffusion_model.input_blocks.2.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.encoders.2.0.groupnorm_merged.weight'] = old['model.diffusion_model.input_blocks.2.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.encoders.2.0.groupnorm_merged.bias'] = old['model.diffusion_model.input_blocks.2.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.encoders.2.0.conv_merged.weight'] = old['model.diffusion_model.input_blocks.2.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.encoders.2.0.conv_merged.bias'] = old['model.diffusion_model.input_blocks.2.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.encoders.2.1.groupnorm.weight'] = old['model.diffusion_model.input_blocks.2.1.norm.weight']\n",
        "    new['diffusion']['unet.encoders.2.1.groupnorm.bias'] = old['model.diffusion_model.input_blocks.2.1.norm.bias']\n",
        "    new['diffusion']['unet.encoders.2.1.conv_input.weight'] = old['model.diffusion_model.input_blocks.2.1.proj_in.weight']\n",
        "    new['diffusion']['unet.encoders.2.1.conv_input.bias'] = old['model.diffusion_model.input_blocks.2.1.proj_in.bias']\n",
        "    new['diffusion']['unet.encoders.2.1.attention_1.out_proj.weight'] = old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn1.to_out.0.weight']\n",
        "    new['diffusion']['unet.encoders.2.1.attention_1.out_proj.bias'] = old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn1.to_out.0.bias']\n",
        "    new['diffusion']['unet.encoders.2.1.linear_geglu_1.weight'] = old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.ff.net.0.proj.weight']\n",
        "    new['diffusion']['unet.encoders.2.1.linear_geglu_1.bias'] = old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.ff.net.0.proj.bias']\n",
        "    new['diffusion']['unet.encoders.2.1.linear_geglu_2.weight'] = old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.ff.net.2.weight']\n",
        "    new['diffusion']['unet.encoders.2.1.linear_geglu_2.bias'] = old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.ff.net.2.bias']\n",
        "    new['diffusion']['unet.encoders.2.1.attention_2.q_proj.weight'] = old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_q.weight']\n",
        "    new['diffusion']['unet.encoders.2.1.attention_2.k_proj.weight'] = old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_k.weight']\n",
        "    new['diffusion']['unet.encoders.2.1.attention_2.v_proj.weight'] = old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_v.weight']\n",
        "    new['diffusion']['unet.encoders.2.1.attention_2.out_proj.weight'] = old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_out.0.weight']\n",
        "    new['diffusion']['unet.encoders.2.1.attention_2.out_proj.bias'] = old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_out.0.bias']\n",
        "    new['diffusion']['unet.encoders.2.1.layernorm_1.weight'] = old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.norm1.weight']\n",
        "    new['diffusion']['unet.encoders.2.1.layernorm_1.bias'] = old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.norm1.bias']\n",
        "    new['diffusion']['unet.encoders.2.1.layernorm_2.weight'] = old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.norm2.weight']\n",
        "    new['diffusion']['unet.encoders.2.1.layernorm_2.bias'] = old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.norm2.bias']\n",
        "    new['diffusion']['unet.encoders.2.1.layernorm_3.weight'] = old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.norm3.weight']\n",
        "    new['diffusion']['unet.encoders.2.1.layernorm_3.bias'] = old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.norm3.bias']\n",
        "    new['diffusion']['unet.encoders.2.1.conv_output.weight'] = old['model.diffusion_model.input_blocks.2.1.proj_out.weight']\n",
        "    new['diffusion']['unet.encoders.2.1.conv_output.bias'] = old['model.diffusion_model.input_blocks.2.1.proj_out.bias']\n",
        "    new['diffusion']['unet.encoders.3.0.weight'] = old['model.diffusion_model.input_blocks.3.0.op.weight']\n",
        "    new['diffusion']['unet.encoders.3.0.bias'] = old['model.diffusion_model.input_blocks.3.0.op.bias']\n",
        "    new['diffusion']['unet.encoders.4.0.groupnorm_feature.weight'] = old['model.diffusion_model.input_blocks.4.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.encoders.4.0.groupnorm_feature.bias'] = old['model.diffusion_model.input_blocks.4.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.encoders.4.0.conv_feature.weight'] = old['model.diffusion_model.input_blocks.4.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.encoders.4.0.conv_feature.bias'] = old['model.diffusion_model.input_blocks.4.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.encoders.4.0.linear_time.weight'] = old['model.diffusion_model.input_blocks.4.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.encoders.4.0.linear_time.bias'] = old['model.diffusion_model.input_blocks.4.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.encoders.4.0.groupnorm_merged.weight'] = old['model.diffusion_model.input_blocks.4.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.encoders.4.0.groupnorm_merged.bias'] = old['model.diffusion_model.input_blocks.4.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.encoders.4.0.conv_merged.weight'] = old['model.diffusion_model.input_blocks.4.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.encoders.4.0.conv_merged.bias'] = old['model.diffusion_model.input_blocks.4.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.encoders.4.0.residual_layer.weight'] = old['model.diffusion_model.input_blocks.4.0.skip_connection.weight']\n",
        "    new['diffusion']['unet.encoders.4.0.residual_layer.bias'] = old['model.diffusion_model.input_blocks.4.0.skip_connection.bias']\n",
        "    new['diffusion']['unet.encoders.4.1.groupnorm.weight'] = old['model.diffusion_model.input_blocks.4.1.norm.weight']\n",
        "    new['diffusion']['unet.encoders.4.1.groupnorm.bias'] = old['model.diffusion_model.input_blocks.4.1.norm.bias']\n",
        "    new['diffusion']['unet.encoders.4.1.conv_input.weight'] = old['model.diffusion_model.input_blocks.4.1.proj_in.weight']\n",
        "    new['diffusion']['unet.encoders.4.1.conv_input.bias'] = old['model.diffusion_model.input_blocks.4.1.proj_in.bias']\n",
        "    new['diffusion']['unet.encoders.4.1.attention_1.out_proj.weight'] = old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_out.0.weight']\n",
        "    new['diffusion']['unet.encoders.4.1.attention_1.out_proj.bias'] = old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_out.0.bias']\n",
        "    new['diffusion']['unet.encoders.4.1.linear_geglu_1.weight'] = old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.ff.net.0.proj.weight']\n",
        "    new['diffusion']['unet.encoders.4.1.linear_geglu_1.bias'] = old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.ff.net.0.proj.bias']\n",
        "    new['diffusion']['unet.encoders.4.1.linear_geglu_2.weight'] = old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.ff.net.2.weight']\n",
        "    new['diffusion']['unet.encoders.4.1.linear_geglu_2.bias'] = old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.ff.net.2.bias']\n",
        "    new['diffusion']['unet.encoders.4.1.attention_2.q_proj.weight'] = old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_q.weight']\n",
        "    new['diffusion']['unet.encoders.4.1.attention_2.k_proj.weight'] = old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_k.weight']\n",
        "    new['diffusion']['unet.encoders.4.1.attention_2.v_proj.weight'] = old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_v.weight']\n",
        "    new['diffusion']['unet.encoders.4.1.attention_2.out_proj.weight'] = old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_out.0.weight']\n",
        "    new['diffusion']['unet.encoders.4.1.attention_2.out_proj.bias'] = old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_out.0.bias']\n",
        "    new['diffusion']['unet.encoders.4.1.layernorm_1.weight'] = old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.norm1.weight']\n",
        "    new['diffusion']['unet.encoders.4.1.layernorm_1.bias'] = old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.norm1.bias']\n",
        "    new['diffusion']['unet.encoders.4.1.layernorm_2.weight'] = old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.norm2.weight']\n",
        "    new['diffusion']['unet.encoders.4.1.layernorm_2.bias'] = old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.norm2.bias']\n",
        "    new['diffusion']['unet.encoders.4.1.layernorm_3.weight'] = old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.norm3.weight']\n",
        "    new['diffusion']['unet.encoders.4.1.layernorm_3.bias'] = old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.norm3.bias']\n",
        "    new['diffusion']['unet.encoders.4.1.conv_output.weight'] = old['model.diffusion_model.input_blocks.4.1.proj_out.weight']\n",
        "    new['diffusion']['unet.encoders.4.1.conv_output.bias'] = old['model.diffusion_model.input_blocks.4.1.proj_out.bias']\n",
        "    new['diffusion']['unet.encoders.5.0.groupnorm_feature.weight'] = old['model.diffusion_model.input_blocks.5.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.encoders.5.0.groupnorm_feature.bias'] = old['model.diffusion_model.input_blocks.5.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.encoders.5.0.conv_feature.weight'] = old['model.diffusion_model.input_blocks.5.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.encoders.5.0.conv_feature.bias'] = old['model.diffusion_model.input_blocks.5.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.encoders.5.0.linear_time.weight'] = old['model.diffusion_model.input_blocks.5.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.encoders.5.0.linear_time.bias'] = old['model.diffusion_model.input_blocks.5.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.encoders.5.0.groupnorm_merged.weight'] = old['model.diffusion_model.input_blocks.5.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.encoders.5.0.groupnorm_merged.bias'] = old['model.diffusion_model.input_blocks.5.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.encoders.5.0.conv_merged.weight'] = old['model.diffusion_model.input_blocks.5.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.encoders.5.0.conv_merged.bias'] = old['model.diffusion_model.input_blocks.5.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.encoders.5.1.groupnorm.weight'] = old['model.diffusion_model.input_blocks.5.1.norm.weight']\n",
        "    new['diffusion']['unet.encoders.5.1.groupnorm.bias'] = old['model.diffusion_model.input_blocks.5.1.norm.bias']\n",
        "    new['diffusion']['unet.encoders.5.1.conv_input.weight'] = old['model.diffusion_model.input_blocks.5.1.proj_in.weight']\n",
        "    new['diffusion']['unet.encoders.5.1.conv_input.bias'] = old['model.diffusion_model.input_blocks.5.1.proj_in.bias']\n",
        "    new['diffusion']['unet.encoders.5.1.attention_1.out_proj.weight'] = old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_out.0.weight']\n",
        "    new['diffusion']['unet.encoders.5.1.attention_1.out_proj.bias'] = old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_out.0.bias']\n",
        "    new['diffusion']['unet.encoders.5.1.linear_geglu_1.weight'] = old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.ff.net.0.proj.weight']\n",
        "    new['diffusion']['unet.encoders.5.1.linear_geglu_1.bias'] = old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.ff.net.0.proj.bias']\n",
        "    new['diffusion']['unet.encoders.5.1.linear_geglu_2.weight'] = old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.ff.net.2.weight']\n",
        "    new['diffusion']['unet.encoders.5.1.linear_geglu_2.bias'] = old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.ff.net.2.bias']\n",
        "    new['diffusion']['unet.encoders.5.1.attention_2.q_proj.weight'] = old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_q.weight']\n",
        "    new['diffusion']['unet.encoders.5.1.attention_2.k_proj.weight'] = old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_k.weight']\n",
        "    new['diffusion']['unet.encoders.5.1.attention_2.v_proj.weight'] = old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_v.weight']\n",
        "    new['diffusion']['unet.encoders.5.1.attention_2.out_proj.weight'] = old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_out.0.weight']\n",
        "    new['diffusion']['unet.encoders.5.1.attention_2.out_proj.bias'] = old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_out.0.bias']\n",
        "    new['diffusion']['unet.encoders.5.1.layernorm_1.weight'] = old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.norm1.weight']\n",
        "    new['diffusion']['unet.encoders.5.1.layernorm_1.bias'] = old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.norm1.bias']\n",
        "    new['diffusion']['unet.encoders.5.1.layernorm_2.weight'] = old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.norm2.weight']\n",
        "    new['diffusion']['unet.encoders.5.1.layernorm_2.bias'] = old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.norm2.bias']\n",
        "    new['diffusion']['unet.encoders.5.1.layernorm_3.weight'] = old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.norm3.weight']\n",
        "    new['diffusion']['unet.encoders.5.1.layernorm_3.bias'] = old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.norm3.bias']\n",
        "    new['diffusion']['unet.encoders.5.1.conv_output.weight'] = old['model.diffusion_model.input_blocks.5.1.proj_out.weight']\n",
        "    new['diffusion']['unet.encoders.5.1.conv_output.bias'] = old['model.diffusion_model.input_blocks.5.1.proj_out.bias']\n",
        "    new['diffusion']['unet.encoders.6.0.weight'] = old['model.diffusion_model.input_blocks.6.0.op.weight']\n",
        "    new['diffusion']['unet.encoders.6.0.bias'] = old['model.diffusion_model.input_blocks.6.0.op.bias']\n",
        "    new['diffusion']['unet.encoders.7.0.groupnorm_feature.weight'] = old['model.diffusion_model.input_blocks.7.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.encoders.7.0.groupnorm_feature.bias'] = old['model.diffusion_model.input_blocks.7.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.encoders.7.0.conv_feature.weight'] = old['model.diffusion_model.input_blocks.7.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.encoders.7.0.conv_feature.bias'] = old['model.diffusion_model.input_blocks.7.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.encoders.7.0.linear_time.weight'] = old['model.diffusion_model.input_blocks.7.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.encoders.7.0.linear_time.bias'] = old['model.diffusion_model.input_blocks.7.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.encoders.7.0.groupnorm_merged.weight'] = old['model.diffusion_model.input_blocks.7.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.encoders.7.0.groupnorm_merged.bias'] = old['model.diffusion_model.input_blocks.7.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.encoders.7.0.conv_merged.weight'] = old['model.diffusion_model.input_blocks.7.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.encoders.7.0.conv_merged.bias'] = old['model.diffusion_model.input_blocks.7.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.encoders.7.0.residual_layer.weight'] = old['model.diffusion_model.input_blocks.7.0.skip_connection.weight']\n",
        "    new['diffusion']['unet.encoders.7.0.residual_layer.bias'] = old['model.diffusion_model.input_blocks.7.0.skip_connection.bias']\n",
        "    new['diffusion']['unet.encoders.7.1.groupnorm.weight'] = old['model.diffusion_model.input_blocks.7.1.norm.weight']\n",
        "    new['diffusion']['unet.encoders.7.1.groupnorm.bias'] = old['model.diffusion_model.input_blocks.7.1.norm.bias']\n",
        "    new['diffusion']['unet.encoders.7.1.conv_input.weight'] = old['model.diffusion_model.input_blocks.7.1.proj_in.weight']\n",
        "    new['diffusion']['unet.encoders.7.1.conv_input.bias'] = old['model.diffusion_model.input_blocks.7.1.proj_in.bias']\n",
        "    new['diffusion']['unet.encoders.7.1.attention_1.out_proj.weight'] = old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_out.0.weight']\n",
        "    new['diffusion']['unet.encoders.7.1.attention_1.out_proj.bias'] = old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_out.0.bias']\n",
        "    new['diffusion']['unet.encoders.7.1.linear_geglu_1.weight'] = old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.ff.net.0.proj.weight']\n",
        "    new['diffusion']['unet.encoders.7.1.linear_geglu_1.bias'] = old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.ff.net.0.proj.bias']\n",
        "    new['diffusion']['unet.encoders.7.1.linear_geglu_2.weight'] = old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.ff.net.2.weight']\n",
        "    new['diffusion']['unet.encoders.7.1.linear_geglu_2.bias'] = old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.ff.net.2.bias']\n",
        "    new['diffusion']['unet.encoders.7.1.attention_2.q_proj.weight'] = old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_q.weight']\n",
        "    new['diffusion']['unet.encoders.7.1.attention_2.k_proj.weight'] = old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_k.weight']\n",
        "    new['diffusion']['unet.encoders.7.1.attention_2.v_proj.weight'] = old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_v.weight']\n",
        "    new['diffusion']['unet.encoders.7.1.attention_2.out_proj.weight'] = old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_out.0.weight']\n",
        "    new['diffusion']['unet.encoders.7.1.attention_2.out_proj.bias'] = old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_out.0.bias']\n",
        "    new['diffusion']['unet.encoders.7.1.layernorm_1.weight'] = old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.norm1.weight']\n",
        "    new['diffusion']['unet.encoders.7.1.layernorm_1.bias'] = old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.norm1.bias']\n",
        "    new['diffusion']['unet.encoders.7.1.layernorm_2.weight'] = old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.norm2.weight']\n",
        "    new['diffusion']['unet.encoders.7.1.layernorm_2.bias'] = old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.norm2.bias']\n",
        "    new['diffusion']['unet.encoders.7.1.layernorm_3.weight'] = old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.norm3.weight']\n",
        "    new['diffusion']['unet.encoders.7.1.layernorm_3.bias'] = old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.norm3.bias']\n",
        "    new['diffusion']['unet.encoders.7.1.conv_output.weight'] = old['model.diffusion_model.input_blocks.7.1.proj_out.weight']\n",
        "    new['diffusion']['unet.encoders.7.1.conv_output.bias'] = old['model.diffusion_model.input_blocks.7.1.proj_out.bias']\n",
        "    new['diffusion']['unet.encoders.8.0.groupnorm_feature.weight'] = old['model.diffusion_model.input_blocks.8.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.encoders.8.0.groupnorm_feature.bias'] = old['model.diffusion_model.input_blocks.8.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.encoders.8.0.conv_feature.weight'] = old['model.diffusion_model.input_blocks.8.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.encoders.8.0.conv_feature.bias'] = old['model.diffusion_model.input_blocks.8.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.encoders.8.0.linear_time.weight'] = old['model.diffusion_model.input_blocks.8.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.encoders.8.0.linear_time.bias'] = old['model.diffusion_model.input_blocks.8.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.encoders.8.0.groupnorm_merged.weight'] = old['model.diffusion_model.input_blocks.8.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.encoders.8.0.groupnorm_merged.bias'] = old['model.diffusion_model.input_blocks.8.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.encoders.8.0.conv_merged.weight'] = old['model.diffusion_model.input_blocks.8.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.encoders.8.0.conv_merged.bias'] = old['model.diffusion_model.input_blocks.8.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.encoders.8.1.groupnorm.weight'] = old['model.diffusion_model.input_blocks.8.1.norm.weight']\n",
        "    new['diffusion']['unet.encoders.8.1.groupnorm.bias'] = old['model.diffusion_model.input_blocks.8.1.norm.bias']\n",
        "    new['diffusion']['unet.encoders.8.1.conv_input.weight'] = old['model.diffusion_model.input_blocks.8.1.proj_in.weight']\n",
        "    new['diffusion']['unet.encoders.8.1.conv_input.bias'] = old['model.diffusion_model.input_blocks.8.1.proj_in.bias']\n",
        "    new['diffusion']['unet.encoders.8.1.attention_1.out_proj.weight'] = old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_out.0.weight']\n",
        "    new['diffusion']['unet.encoders.8.1.attention_1.out_proj.bias'] = old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_out.0.bias']\n",
        "    new['diffusion']['unet.encoders.8.1.linear_geglu_1.weight'] = old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.ff.net.0.proj.weight']\n",
        "    new['diffusion']['unet.encoders.8.1.linear_geglu_1.bias'] = old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.ff.net.0.proj.bias']\n",
        "    new['diffusion']['unet.encoders.8.1.linear_geglu_2.weight'] = old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.ff.net.2.weight']\n",
        "    new['diffusion']['unet.encoders.8.1.linear_geglu_2.bias'] = old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.ff.net.2.bias']\n",
        "    new['diffusion']['unet.encoders.8.1.attention_2.q_proj.weight'] = old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_q.weight']\n",
        "    new['diffusion']['unet.encoders.8.1.attention_2.k_proj.weight'] = old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_k.weight']\n",
        "    new['diffusion']['unet.encoders.8.1.attention_2.v_proj.weight'] = old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_v.weight']\n",
        "    new['diffusion']['unet.encoders.8.1.attention_2.out_proj.weight'] = old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_out.0.weight']\n",
        "    new['diffusion']['unet.encoders.8.1.attention_2.out_proj.bias'] = old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_out.0.bias']\n",
        "    new['diffusion']['unet.encoders.8.1.layernorm_1.weight'] = old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.norm1.weight']\n",
        "    new['diffusion']['unet.encoders.8.1.layernorm_1.bias'] = old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.norm1.bias']\n",
        "    new['diffusion']['unet.encoders.8.1.layernorm_2.weight'] = old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.norm2.weight']\n",
        "    new['diffusion']['unet.encoders.8.1.layernorm_2.bias'] = old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.norm2.bias']\n",
        "    new['diffusion']['unet.encoders.8.1.layernorm_3.weight'] = old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.norm3.weight']\n",
        "    new['diffusion']['unet.encoders.8.1.layernorm_3.bias'] = old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.norm3.bias']\n",
        "    new['diffusion']['unet.encoders.8.1.conv_output.weight'] = old['model.diffusion_model.input_blocks.8.1.proj_out.weight']\n",
        "    new['diffusion']['unet.encoders.8.1.conv_output.bias'] = old['model.diffusion_model.input_blocks.8.1.proj_out.bias']\n",
        "    new['diffusion']['unet.encoders.9.0.weight'] = old['model.diffusion_model.input_blocks.9.0.op.weight']\n",
        "    new['diffusion']['unet.encoders.9.0.bias'] = old['model.diffusion_model.input_blocks.9.0.op.bias']\n",
        "    new['diffusion']['unet.encoders.10.0.groupnorm_feature.weight'] = old['model.diffusion_model.input_blocks.10.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.encoders.10.0.groupnorm_feature.bias'] = old['model.diffusion_model.input_blocks.10.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.encoders.10.0.conv_feature.weight'] = old['model.diffusion_model.input_blocks.10.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.encoders.10.0.conv_feature.bias'] = old['model.diffusion_model.input_blocks.10.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.encoders.10.0.linear_time.weight'] = old['model.diffusion_model.input_blocks.10.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.encoders.10.0.linear_time.bias'] = old['model.diffusion_model.input_blocks.10.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.encoders.10.0.groupnorm_merged.weight'] = old['model.diffusion_model.input_blocks.10.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.encoders.10.0.groupnorm_merged.bias'] = old['model.diffusion_model.input_blocks.10.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.encoders.10.0.conv_merged.weight'] = old['model.diffusion_model.input_blocks.10.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.encoders.10.0.conv_merged.bias'] = old['model.diffusion_model.input_blocks.10.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.encoders.11.0.groupnorm_feature.weight'] = old['model.diffusion_model.input_blocks.11.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.encoders.11.0.groupnorm_feature.bias'] = old['model.diffusion_model.input_blocks.11.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.encoders.11.0.conv_feature.weight'] = old['model.diffusion_model.input_blocks.11.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.encoders.11.0.conv_feature.bias'] = old['model.diffusion_model.input_blocks.11.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.encoders.11.0.linear_time.weight'] = old['model.diffusion_model.input_blocks.11.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.encoders.11.0.linear_time.bias'] = old['model.diffusion_model.input_blocks.11.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.encoders.11.0.groupnorm_merged.weight'] = old['model.diffusion_model.input_blocks.11.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.encoders.11.0.groupnorm_merged.bias'] = old['model.diffusion_model.input_blocks.11.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.encoders.11.0.conv_merged.weight'] = old['model.diffusion_model.input_blocks.11.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.encoders.11.0.conv_merged.bias'] = old['model.diffusion_model.input_blocks.11.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.bottleneck.0.groupnorm_feature.weight'] = old['model.diffusion_model.middle_block.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.bottleneck.0.groupnorm_feature.bias'] = old['model.diffusion_model.middle_block.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.bottleneck.0.conv_feature.weight'] = old['model.diffusion_model.middle_block.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.bottleneck.0.conv_feature.bias'] = old['model.diffusion_model.middle_block.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.bottleneck.0.linear_time.weight'] = old['model.diffusion_model.middle_block.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.bottleneck.0.linear_time.bias'] = old['model.diffusion_model.middle_block.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.bottleneck.0.groupnorm_merged.weight'] = old['model.diffusion_model.middle_block.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.bottleneck.0.groupnorm_merged.bias'] = old['model.diffusion_model.middle_block.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.bottleneck.0.conv_merged.weight'] = old['model.diffusion_model.middle_block.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.bottleneck.0.conv_merged.bias'] = old['model.diffusion_model.middle_block.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.bottleneck.1.groupnorm.weight'] = old['model.diffusion_model.middle_block.1.norm.weight']\n",
        "    new['diffusion']['unet.bottleneck.1.groupnorm.bias'] = old['model.diffusion_model.middle_block.1.norm.bias']\n",
        "    new['diffusion']['unet.bottleneck.1.conv_input.weight'] = old['model.diffusion_model.middle_block.1.proj_in.weight']\n",
        "    new['diffusion']['unet.bottleneck.1.conv_input.bias'] = old['model.diffusion_model.middle_block.1.proj_in.bias']\n",
        "    new['diffusion']['unet.bottleneck.1.attention_1.out_proj.weight'] = old['model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_out.0.weight']\n",
        "    new['diffusion']['unet.bottleneck.1.attention_1.out_proj.bias'] = old['model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_out.0.bias']\n",
        "    new['diffusion']['unet.bottleneck.1.linear_geglu_1.weight'] = old['model.diffusion_model.middle_block.1.transformer_blocks.0.ff.net.0.proj.weight']\n",
        "    new['diffusion']['unet.bottleneck.1.linear_geglu_1.bias'] = old['model.diffusion_model.middle_block.1.transformer_blocks.0.ff.net.0.proj.bias']\n",
        "    new['diffusion']['unet.bottleneck.1.linear_geglu_2.weight'] = old['model.diffusion_model.middle_block.1.transformer_blocks.0.ff.net.2.weight']\n",
        "    new['diffusion']['unet.bottleneck.1.linear_geglu_2.bias'] = old['model.diffusion_model.middle_block.1.transformer_blocks.0.ff.net.2.bias']\n",
        "    new['diffusion']['unet.bottleneck.1.attention_2.q_proj.weight'] = old['model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_q.weight']\n",
        "    new['diffusion']['unet.bottleneck.1.attention_2.k_proj.weight'] = old['model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_k.weight']\n",
        "    new['diffusion']['unet.bottleneck.1.attention_2.v_proj.weight'] = old['model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_v.weight']\n",
        "    new['diffusion']['unet.bottleneck.1.attention_2.out_proj.weight'] = old['model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_out.0.weight']\n",
        "    new['diffusion']['unet.bottleneck.1.attention_2.out_proj.bias'] = old['model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_out.0.bias']\n",
        "    new['diffusion']['unet.bottleneck.1.layernorm_1.weight'] = old['model.diffusion_model.middle_block.1.transformer_blocks.0.norm1.weight']\n",
        "    new['diffusion']['unet.bottleneck.1.layernorm_1.bias'] = old['model.diffusion_model.middle_block.1.transformer_blocks.0.norm1.bias']\n",
        "    new['diffusion']['unet.bottleneck.1.layernorm_2.weight'] = old['model.diffusion_model.middle_block.1.transformer_blocks.0.norm2.weight']\n",
        "    new['diffusion']['unet.bottleneck.1.layernorm_2.bias'] = old['model.diffusion_model.middle_block.1.transformer_blocks.0.norm2.bias']\n",
        "    new['diffusion']['unet.bottleneck.1.layernorm_3.weight'] = old['model.diffusion_model.middle_block.1.transformer_blocks.0.norm3.weight']\n",
        "    new['diffusion']['unet.bottleneck.1.layernorm_3.bias'] = old['model.diffusion_model.middle_block.1.transformer_blocks.0.norm3.bias']\n",
        "    new['diffusion']['unet.bottleneck.1.conv_output.weight'] = old['model.diffusion_model.middle_block.1.proj_out.weight']\n",
        "    new['diffusion']['unet.bottleneck.1.conv_output.bias'] = old['model.diffusion_model.middle_block.1.proj_out.bias']\n",
        "    new['diffusion']['unet.bottleneck.2.groupnorm_feature.weight'] = old['model.diffusion_model.middle_block.2.in_layers.0.weight']\n",
        "    new['diffusion']['unet.bottleneck.2.groupnorm_feature.bias'] = old['model.diffusion_model.middle_block.2.in_layers.0.bias']\n",
        "    new['diffusion']['unet.bottleneck.2.conv_feature.weight'] = old['model.diffusion_model.middle_block.2.in_layers.2.weight']\n",
        "    new['diffusion']['unet.bottleneck.2.conv_feature.bias'] = old['model.diffusion_model.middle_block.2.in_layers.2.bias']\n",
        "    new['diffusion']['unet.bottleneck.2.linear_time.weight'] = old['model.diffusion_model.middle_block.2.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.bottleneck.2.linear_time.bias'] = old['model.diffusion_model.middle_block.2.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.bottleneck.2.groupnorm_merged.weight'] = old['model.diffusion_model.middle_block.2.out_layers.0.weight']\n",
        "    new['diffusion']['unet.bottleneck.2.groupnorm_merged.bias'] = old['model.diffusion_model.middle_block.2.out_layers.0.bias']\n",
        "    new['diffusion']['unet.bottleneck.2.conv_merged.weight'] = old['model.diffusion_model.middle_block.2.out_layers.3.weight']\n",
        "    new['diffusion']['unet.bottleneck.2.conv_merged.bias'] = old['model.diffusion_model.middle_block.2.out_layers.3.bias']\n",
        "    new['diffusion']['unet.decoders.0.0.groupnorm_feature.weight'] = old['model.diffusion_model.output_blocks.0.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.0.0.groupnorm_feature.bias'] = old['model.diffusion_model.output_blocks.0.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.0.0.conv_feature.weight'] = old['model.diffusion_model.output_blocks.0.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.decoders.0.0.conv_feature.bias'] = old['model.diffusion_model.output_blocks.0.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.decoders.0.0.linear_time.weight'] = old['model.diffusion_model.output_blocks.0.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.decoders.0.0.linear_time.bias'] = old['model.diffusion_model.output_blocks.0.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.decoders.0.0.groupnorm_merged.weight'] = old['model.diffusion_model.output_blocks.0.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.0.0.groupnorm_merged.bias'] = old['model.diffusion_model.output_blocks.0.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.0.0.conv_merged.weight'] = old['model.diffusion_model.output_blocks.0.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.decoders.0.0.conv_merged.bias'] = old['model.diffusion_model.output_blocks.0.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.decoders.0.0.residual_layer.weight'] = old['model.diffusion_model.output_blocks.0.0.skip_connection.weight']\n",
        "    new['diffusion']['unet.decoders.0.0.residual_layer.bias'] = old['model.diffusion_model.output_blocks.0.0.skip_connection.bias']\n",
        "    new['diffusion']['unet.decoders.1.0.groupnorm_feature.weight'] = old['model.diffusion_model.output_blocks.1.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.1.0.groupnorm_feature.bias'] = old['model.diffusion_model.output_blocks.1.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.1.0.conv_feature.weight'] = old['model.diffusion_model.output_blocks.1.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.decoders.1.0.conv_feature.bias'] = old['model.diffusion_model.output_blocks.1.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.decoders.1.0.linear_time.weight'] = old['model.diffusion_model.output_blocks.1.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.decoders.1.0.linear_time.bias'] = old['model.diffusion_model.output_blocks.1.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.decoders.1.0.groupnorm_merged.weight'] = old['model.diffusion_model.output_blocks.1.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.1.0.groupnorm_merged.bias'] = old['model.diffusion_model.output_blocks.1.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.1.0.conv_merged.weight'] = old['model.diffusion_model.output_blocks.1.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.decoders.1.0.conv_merged.bias'] = old['model.diffusion_model.output_blocks.1.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.decoders.1.0.residual_layer.weight'] = old['model.diffusion_model.output_blocks.1.0.skip_connection.weight']\n",
        "    new['diffusion']['unet.decoders.1.0.residual_layer.bias'] = old['model.diffusion_model.output_blocks.1.0.skip_connection.bias']\n",
        "    new['diffusion']['unet.decoders.2.0.groupnorm_feature.weight'] = old['model.diffusion_model.output_blocks.2.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.2.0.groupnorm_feature.bias'] = old['model.diffusion_model.output_blocks.2.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.2.0.conv_feature.weight'] = old['model.diffusion_model.output_blocks.2.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.decoders.2.0.conv_feature.bias'] = old['model.diffusion_model.output_blocks.2.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.decoders.2.0.linear_time.weight'] = old['model.diffusion_model.output_blocks.2.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.decoders.2.0.linear_time.bias'] = old['model.diffusion_model.output_blocks.2.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.decoders.2.0.groupnorm_merged.weight'] = old['model.diffusion_model.output_blocks.2.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.2.0.groupnorm_merged.bias'] = old['model.diffusion_model.output_blocks.2.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.2.0.conv_merged.weight'] = old['model.diffusion_model.output_blocks.2.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.decoders.2.0.conv_merged.bias'] = old['model.diffusion_model.output_blocks.2.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.decoders.2.0.residual_layer.weight'] = old['model.diffusion_model.output_blocks.2.0.skip_connection.weight']\n",
        "    new['diffusion']['unet.decoders.2.0.residual_layer.bias'] = old['model.diffusion_model.output_blocks.2.0.skip_connection.bias']\n",
        "    new['diffusion']['unet.decoders.2.1.conv.weight'] = old['model.diffusion_model.output_blocks.2.1.conv.weight']\n",
        "    new['diffusion']['unet.decoders.2.1.conv.bias'] = old['model.diffusion_model.output_blocks.2.1.conv.bias']\n",
        "    new['diffusion']['unet.decoders.3.0.groupnorm_feature.weight'] = old['model.diffusion_model.output_blocks.3.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.3.0.groupnorm_feature.bias'] = old['model.diffusion_model.output_blocks.3.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.3.0.conv_feature.weight'] = old['model.diffusion_model.output_blocks.3.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.decoders.3.0.conv_feature.bias'] = old['model.diffusion_model.output_blocks.3.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.decoders.3.0.linear_time.weight'] = old['model.diffusion_model.output_blocks.3.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.decoders.3.0.linear_time.bias'] = old['model.diffusion_model.output_blocks.3.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.decoders.3.0.groupnorm_merged.weight'] = old['model.diffusion_model.output_blocks.3.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.3.0.groupnorm_merged.bias'] = old['model.diffusion_model.output_blocks.3.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.3.0.conv_merged.weight'] = old['model.diffusion_model.output_blocks.3.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.decoders.3.0.conv_merged.bias'] = old['model.diffusion_model.output_blocks.3.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.decoders.3.0.residual_layer.weight'] = old['model.diffusion_model.output_blocks.3.0.skip_connection.weight']\n",
        "    new['diffusion']['unet.decoders.3.0.residual_layer.bias'] = old['model.diffusion_model.output_blocks.3.0.skip_connection.bias']\n",
        "    new['diffusion']['unet.decoders.3.1.groupnorm.weight'] = old['model.diffusion_model.output_blocks.3.1.norm.weight']\n",
        "    new['diffusion']['unet.decoders.3.1.groupnorm.bias'] = old['model.diffusion_model.output_blocks.3.1.norm.bias']\n",
        "    new['diffusion']['unet.decoders.3.1.conv_input.weight'] = old['model.diffusion_model.output_blocks.3.1.proj_in.weight']\n",
        "    new['diffusion']['unet.decoders.3.1.conv_input.bias'] = old['model.diffusion_model.output_blocks.3.1.proj_in.bias']\n",
        "    new['diffusion']['unet.decoders.3.1.attention_1.out_proj.weight'] = old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_out.0.weight']\n",
        "    new['diffusion']['unet.decoders.3.1.attention_1.out_proj.bias'] = old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_out.0.bias']\n",
        "    new['diffusion']['unet.decoders.3.1.linear_geglu_1.weight'] = old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.ff.net.0.proj.weight']\n",
        "    new['diffusion']['unet.decoders.3.1.linear_geglu_1.bias'] = old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.ff.net.0.proj.bias']\n",
        "    new['diffusion']['unet.decoders.3.1.linear_geglu_2.weight'] = old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.ff.net.2.weight']\n",
        "    new['diffusion']['unet.decoders.3.1.linear_geglu_2.bias'] = old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.ff.net.2.bias']\n",
        "    new['diffusion']['unet.decoders.3.1.attention_2.q_proj.weight'] = old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_q.weight']\n",
        "    new['diffusion']['unet.decoders.3.1.attention_2.k_proj.weight'] = old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_k.weight']\n",
        "    new['diffusion']['unet.decoders.3.1.attention_2.v_proj.weight'] = old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_v.weight']\n",
        "    new['diffusion']['unet.decoders.3.1.attention_2.out_proj.weight'] = old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_out.0.weight']\n",
        "    new['diffusion']['unet.decoders.3.1.attention_2.out_proj.bias'] = old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_out.0.bias']\n",
        "    new['diffusion']['unet.decoders.3.1.layernorm_1.weight'] = old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.norm1.weight']\n",
        "    new['diffusion']['unet.decoders.3.1.layernorm_1.bias'] = old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.norm1.bias']\n",
        "    new['diffusion']['unet.decoders.3.1.layernorm_2.weight'] = old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.norm2.weight']\n",
        "    new['diffusion']['unet.decoders.3.1.layernorm_2.bias'] = old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.norm2.bias']\n",
        "    new['diffusion']['unet.decoders.3.1.layernorm_3.weight'] = old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.norm3.weight']\n",
        "    new['diffusion']['unet.decoders.3.1.layernorm_3.bias'] = old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.norm3.bias']\n",
        "    new['diffusion']['unet.decoders.3.1.conv_output.weight'] = old['model.diffusion_model.output_blocks.3.1.proj_out.weight']\n",
        "    new['diffusion']['unet.decoders.3.1.conv_output.bias'] = old['model.diffusion_model.output_blocks.3.1.proj_out.bias']\n",
        "    new['diffusion']['unet.decoders.4.0.groupnorm_feature.weight'] = old['model.diffusion_model.output_blocks.4.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.4.0.groupnorm_feature.bias'] = old['model.diffusion_model.output_blocks.4.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.4.0.conv_feature.weight'] = old['model.diffusion_model.output_blocks.4.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.decoders.4.0.conv_feature.bias'] = old['model.diffusion_model.output_blocks.4.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.decoders.4.0.linear_time.weight'] = old['model.diffusion_model.output_blocks.4.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.decoders.4.0.linear_time.bias'] = old['model.diffusion_model.output_blocks.4.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.decoders.4.0.groupnorm_merged.weight'] = old['model.diffusion_model.output_blocks.4.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.4.0.groupnorm_merged.bias'] = old['model.diffusion_model.output_blocks.4.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.4.0.conv_merged.weight'] = old['model.diffusion_model.output_blocks.4.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.decoders.4.0.conv_merged.bias'] = old['model.diffusion_model.output_blocks.4.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.decoders.4.0.residual_layer.weight'] = old['model.diffusion_model.output_blocks.4.0.skip_connection.weight']\n",
        "    new['diffusion']['unet.decoders.4.0.residual_layer.bias'] = old['model.diffusion_model.output_blocks.4.0.skip_connection.bias']\n",
        "    new['diffusion']['unet.decoders.4.1.groupnorm.weight'] = old['model.diffusion_model.output_blocks.4.1.norm.weight']\n",
        "    new['diffusion']['unet.decoders.4.1.groupnorm.bias'] = old['model.diffusion_model.output_blocks.4.1.norm.bias']\n",
        "    new['diffusion']['unet.decoders.4.1.conv_input.weight'] = old['model.diffusion_model.output_blocks.4.1.proj_in.weight']\n",
        "    new['diffusion']['unet.decoders.4.1.conv_input.bias'] = old['model.diffusion_model.output_blocks.4.1.proj_in.bias']\n",
        "    new['diffusion']['unet.decoders.4.1.attention_1.out_proj.weight'] = old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_out.0.weight']\n",
        "    new['diffusion']['unet.decoders.4.1.attention_1.out_proj.bias'] = old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_out.0.bias']\n",
        "    new['diffusion']['unet.decoders.4.1.linear_geglu_1.weight'] = old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.ff.net.0.proj.weight']\n",
        "    new['diffusion']['unet.decoders.4.1.linear_geglu_1.bias'] = old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.ff.net.0.proj.bias']\n",
        "    new['diffusion']['unet.decoders.4.1.linear_geglu_2.weight'] = old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.ff.net.2.weight']\n",
        "    new['diffusion']['unet.decoders.4.1.linear_geglu_2.bias'] = old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.ff.net.2.bias']\n",
        "    new['diffusion']['unet.decoders.4.1.attention_2.q_proj.weight'] = old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_q.weight']\n",
        "    new['diffusion']['unet.decoders.4.1.attention_2.k_proj.weight'] = old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_k.weight']\n",
        "    new['diffusion']['unet.decoders.4.1.attention_2.v_proj.weight'] = old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_v.weight']\n",
        "    new['diffusion']['unet.decoders.4.1.attention_2.out_proj.weight'] = old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_out.0.weight']\n",
        "    new['diffusion']['unet.decoders.4.1.attention_2.out_proj.bias'] = old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_out.0.bias']\n",
        "    new['diffusion']['unet.decoders.4.1.layernorm_1.weight'] = old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.norm1.weight']\n",
        "    new['diffusion']['unet.decoders.4.1.layernorm_1.bias'] = old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.norm1.bias']\n",
        "    new['diffusion']['unet.decoders.4.1.layernorm_2.weight'] = old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.norm2.weight']\n",
        "    new['diffusion']['unet.decoders.4.1.layernorm_2.bias'] = old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.norm2.bias']\n",
        "    new['diffusion']['unet.decoders.4.1.layernorm_3.weight'] = old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.norm3.weight']\n",
        "    new['diffusion']['unet.decoders.4.1.layernorm_3.bias'] = old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.norm3.bias']\n",
        "    new['diffusion']['unet.decoders.4.1.conv_output.weight'] = old['model.diffusion_model.output_blocks.4.1.proj_out.weight']\n",
        "    new['diffusion']['unet.decoders.4.1.conv_output.bias'] = old['model.diffusion_model.output_blocks.4.1.proj_out.bias']\n",
        "    new['diffusion']['unet.decoders.5.0.groupnorm_feature.weight'] = old['model.diffusion_model.output_blocks.5.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.5.0.groupnorm_feature.bias'] = old['model.diffusion_model.output_blocks.5.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.5.0.conv_feature.weight'] = old['model.diffusion_model.output_blocks.5.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.decoders.5.0.conv_feature.bias'] = old['model.diffusion_model.output_blocks.5.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.decoders.5.0.linear_time.weight'] = old['model.diffusion_model.output_blocks.5.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.decoders.5.0.linear_time.bias'] = old['model.diffusion_model.output_blocks.5.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.decoders.5.0.groupnorm_merged.weight'] = old['model.diffusion_model.output_blocks.5.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.5.0.groupnorm_merged.bias'] = old['model.diffusion_model.output_blocks.5.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.5.0.conv_merged.weight'] = old['model.diffusion_model.output_blocks.5.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.decoders.5.0.conv_merged.bias'] = old['model.diffusion_model.output_blocks.5.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.decoders.5.0.residual_layer.weight'] = old['model.diffusion_model.output_blocks.5.0.skip_connection.weight']\n",
        "    new['diffusion']['unet.decoders.5.0.residual_layer.bias'] = old['model.diffusion_model.output_blocks.5.0.skip_connection.bias']\n",
        "    new['diffusion']['unet.decoders.5.1.groupnorm.weight'] = old['model.diffusion_model.output_blocks.5.1.norm.weight']\n",
        "    new['diffusion']['unet.decoders.5.1.groupnorm.bias'] = old['model.diffusion_model.output_blocks.5.1.norm.bias']\n",
        "    new['diffusion']['unet.decoders.5.1.conv_input.weight'] = old['model.diffusion_model.output_blocks.5.1.proj_in.weight']\n",
        "    new['diffusion']['unet.decoders.5.1.conv_input.bias'] = old['model.diffusion_model.output_blocks.5.1.proj_in.bias']\n",
        "    new['diffusion']['unet.decoders.5.1.attention_1.out_proj.weight'] = old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_out.0.weight']\n",
        "    new['diffusion']['unet.decoders.5.1.attention_1.out_proj.bias'] = old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_out.0.bias']\n",
        "    new['diffusion']['unet.decoders.5.1.linear_geglu_1.weight'] = old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.ff.net.0.proj.weight']\n",
        "    new['diffusion']['unet.decoders.5.1.linear_geglu_1.bias'] = old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.ff.net.0.proj.bias']\n",
        "    new['diffusion']['unet.decoders.5.1.linear_geglu_2.weight'] = old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.ff.net.2.weight']\n",
        "    new['diffusion']['unet.decoders.5.1.linear_geglu_2.bias'] = old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.ff.net.2.bias']\n",
        "    new['diffusion']['unet.decoders.5.1.attention_2.q_proj.weight'] = old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_q.weight']\n",
        "    new['diffusion']['unet.decoders.5.1.attention_2.k_proj.weight'] = old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_k.weight']\n",
        "    new['diffusion']['unet.decoders.5.1.attention_2.v_proj.weight'] = old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_v.weight']\n",
        "    new['diffusion']['unet.decoders.5.1.attention_2.out_proj.weight'] = old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_out.0.weight']\n",
        "    new['diffusion']['unet.decoders.5.1.attention_2.out_proj.bias'] = old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_out.0.bias']\n",
        "    new['diffusion']['unet.decoders.5.1.layernorm_1.weight'] = old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.norm1.weight']\n",
        "    new['diffusion']['unet.decoders.5.1.layernorm_1.bias'] = old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.norm1.bias']\n",
        "    new['diffusion']['unet.decoders.5.1.layernorm_2.weight'] = old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.norm2.weight']\n",
        "    new['diffusion']['unet.decoders.5.1.layernorm_2.bias'] = old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.norm2.bias']\n",
        "    new['diffusion']['unet.decoders.5.1.layernorm_3.weight'] = old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.norm3.weight']\n",
        "    new['diffusion']['unet.decoders.5.1.layernorm_3.bias'] = old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.norm3.bias']\n",
        "    new['diffusion']['unet.decoders.5.1.conv_output.weight'] = old['model.diffusion_model.output_blocks.5.1.proj_out.weight']\n",
        "    new['diffusion']['unet.decoders.5.1.conv_output.bias'] = old['model.diffusion_model.output_blocks.5.1.proj_out.bias']\n",
        "    new['diffusion']['unet.decoders.5.2.conv.weight'] = old['model.diffusion_model.output_blocks.5.2.conv.weight']\n",
        "    new['diffusion']['unet.decoders.5.2.conv.bias'] = old['model.diffusion_model.output_blocks.5.2.conv.bias']\n",
        "    new['diffusion']['unet.decoders.6.0.groupnorm_feature.weight'] = old['model.diffusion_model.output_blocks.6.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.6.0.groupnorm_feature.bias'] = old['model.diffusion_model.output_blocks.6.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.6.0.conv_feature.weight'] = old['model.diffusion_model.output_blocks.6.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.decoders.6.0.conv_feature.bias'] = old['model.diffusion_model.output_blocks.6.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.decoders.6.0.linear_time.weight'] = old['model.diffusion_model.output_blocks.6.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.decoders.6.0.linear_time.bias'] = old['model.diffusion_model.output_blocks.6.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.decoders.6.0.groupnorm_merged.weight'] = old['model.diffusion_model.output_blocks.6.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.6.0.groupnorm_merged.bias'] = old['model.diffusion_model.output_blocks.6.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.6.0.conv_merged.weight'] = old['model.diffusion_model.output_blocks.6.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.decoders.6.0.conv_merged.bias'] = old['model.diffusion_model.output_blocks.6.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.decoders.6.0.residual_layer.weight'] = old['model.diffusion_model.output_blocks.6.0.skip_connection.weight']\n",
        "    new['diffusion']['unet.decoders.6.0.residual_layer.bias'] = old['model.diffusion_model.output_blocks.6.0.skip_connection.bias']\n",
        "    new['diffusion']['unet.decoders.6.1.groupnorm.weight'] = old['model.diffusion_model.output_blocks.6.1.norm.weight']\n",
        "    new['diffusion']['unet.decoders.6.1.groupnorm.bias'] = old['model.diffusion_model.output_blocks.6.1.norm.bias']\n",
        "    new['diffusion']['unet.decoders.6.1.conv_input.weight'] = old['model.diffusion_model.output_blocks.6.1.proj_in.weight']\n",
        "    new['diffusion']['unet.decoders.6.1.conv_input.bias'] = old['model.diffusion_model.output_blocks.6.1.proj_in.bias']\n",
        "    new['diffusion']['unet.decoders.6.1.attention_1.out_proj.weight'] = old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn1.to_out.0.weight']\n",
        "    new['diffusion']['unet.decoders.6.1.attention_1.out_proj.bias'] = old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn1.to_out.0.bias']\n",
        "    new['diffusion']['unet.decoders.6.1.linear_geglu_1.weight'] = old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.ff.net.0.proj.weight']\n",
        "    new['diffusion']['unet.decoders.6.1.linear_geglu_1.bias'] = old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.ff.net.0.proj.bias']\n",
        "    new['diffusion']['unet.decoders.6.1.linear_geglu_2.weight'] = old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.ff.net.2.weight']\n",
        "    new['diffusion']['unet.decoders.6.1.linear_geglu_2.bias'] = old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.ff.net.2.bias']\n",
        "    new['diffusion']['unet.decoders.6.1.attention_2.q_proj.weight'] = old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn2.to_q.weight']\n",
        "    new['diffusion']['unet.decoders.6.1.attention_2.k_proj.weight'] = old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn2.to_k.weight']\n",
        "    new['diffusion']['unet.decoders.6.1.attention_2.v_proj.weight'] = old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn2.to_v.weight']\n",
        "    new['diffusion']['unet.decoders.6.1.attention_2.out_proj.weight'] = old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn2.to_out.0.weight']\n",
        "    new['diffusion']['unet.decoders.6.1.attention_2.out_proj.bias'] = old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn2.to_out.0.bias']\n",
        "    new['diffusion']['unet.decoders.6.1.layernorm_1.weight'] = old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.norm1.weight']\n",
        "    new['diffusion']['unet.decoders.6.1.layernorm_1.bias'] = old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.norm1.bias']\n",
        "    new['diffusion']['unet.decoders.6.1.layernorm_2.weight'] = old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.norm2.weight']\n",
        "    new['diffusion']['unet.decoders.6.1.layernorm_2.bias'] = old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.norm2.bias']\n",
        "    new['diffusion']['unet.decoders.6.1.layernorm_3.weight'] = old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.norm3.weight']\n",
        "    new['diffusion']['unet.decoders.6.1.layernorm_3.bias'] = old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.norm3.bias']\n",
        "    new['diffusion']['unet.decoders.6.1.conv_output.weight'] = old['model.diffusion_model.output_blocks.6.1.proj_out.weight']\n",
        "    new['diffusion']['unet.decoders.6.1.conv_output.bias'] = old['model.diffusion_model.output_blocks.6.1.proj_out.bias']\n",
        "    new['diffusion']['unet.decoders.7.0.groupnorm_feature.weight'] = old['model.diffusion_model.output_blocks.7.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.7.0.groupnorm_feature.bias'] = old['model.diffusion_model.output_blocks.7.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.7.0.conv_feature.weight'] = old['model.diffusion_model.output_blocks.7.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.decoders.7.0.conv_feature.bias'] = old['model.diffusion_model.output_blocks.7.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.decoders.7.0.linear_time.weight'] = old['model.diffusion_model.output_blocks.7.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.decoders.7.0.linear_time.bias'] = old['model.diffusion_model.output_blocks.7.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.decoders.7.0.groupnorm_merged.weight'] = old['model.diffusion_model.output_blocks.7.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.7.0.groupnorm_merged.bias'] = old['model.diffusion_model.output_blocks.7.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.7.0.conv_merged.weight'] = old['model.diffusion_model.output_blocks.7.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.decoders.7.0.conv_merged.bias'] = old['model.diffusion_model.output_blocks.7.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.decoders.7.0.residual_layer.weight'] = old['model.diffusion_model.output_blocks.7.0.skip_connection.weight']\n",
        "    new['diffusion']['unet.decoders.7.0.residual_layer.bias'] = old['model.diffusion_model.output_blocks.7.0.skip_connection.bias']\n",
        "    new['diffusion']['unet.decoders.7.1.groupnorm.weight'] = old['model.diffusion_model.output_blocks.7.1.norm.weight']\n",
        "    new['diffusion']['unet.decoders.7.1.groupnorm.bias'] = old['model.diffusion_model.output_blocks.7.1.norm.bias']\n",
        "    new['diffusion']['unet.decoders.7.1.conv_input.weight'] = old['model.diffusion_model.output_blocks.7.1.proj_in.weight']\n",
        "    new['diffusion']['unet.decoders.7.1.conv_input.bias'] = old['model.diffusion_model.output_blocks.7.1.proj_in.bias']\n",
        "    new['diffusion']['unet.decoders.7.1.attention_1.out_proj.weight'] = old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn1.to_out.0.weight']\n",
        "    new['diffusion']['unet.decoders.7.1.attention_1.out_proj.bias'] = old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn1.to_out.0.bias']\n",
        "    new['diffusion']['unet.decoders.7.1.linear_geglu_1.weight'] = old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.ff.net.0.proj.weight']\n",
        "    new['diffusion']['unet.decoders.7.1.linear_geglu_1.bias'] = old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.ff.net.0.proj.bias']\n",
        "    new['diffusion']['unet.decoders.7.1.linear_geglu_2.weight'] = old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.ff.net.2.weight']\n",
        "    new['diffusion']['unet.decoders.7.1.linear_geglu_2.bias'] = old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.ff.net.2.bias']\n",
        "    new['diffusion']['unet.decoders.7.1.attention_2.q_proj.weight'] = old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn2.to_q.weight']\n",
        "    new['diffusion']['unet.decoders.7.1.attention_2.k_proj.weight'] = old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn2.to_k.weight']\n",
        "    new['diffusion']['unet.decoders.7.1.attention_2.v_proj.weight'] = old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn2.to_v.weight']\n",
        "    new['diffusion']['unet.decoders.7.1.attention_2.out_proj.weight'] = old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn2.to_out.0.weight']\n",
        "    new['diffusion']['unet.decoders.7.1.attention_2.out_proj.bias'] = old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn2.to_out.0.bias']\n",
        "    new['diffusion']['unet.decoders.7.1.layernorm_1.weight'] = old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.norm1.weight']\n",
        "    new['diffusion']['unet.decoders.7.1.layernorm_1.bias'] = old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.norm1.bias']\n",
        "    new['diffusion']['unet.decoders.7.1.layernorm_2.weight'] = old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.norm2.weight']\n",
        "    new['diffusion']['unet.decoders.7.1.layernorm_2.bias'] = old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.norm2.bias']\n",
        "    new['diffusion']['unet.decoders.7.1.layernorm_3.weight'] = old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.norm3.weight']\n",
        "    new['diffusion']['unet.decoders.7.1.layernorm_3.bias'] = old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.norm3.bias']\n",
        "    new['diffusion']['unet.decoders.7.1.conv_output.weight'] = old['model.diffusion_model.output_blocks.7.1.proj_out.weight']\n",
        "    new['diffusion']['unet.decoders.7.1.conv_output.bias'] = old['model.diffusion_model.output_blocks.7.1.proj_out.bias']\n",
        "    new['diffusion']['unet.decoders.8.0.groupnorm_feature.weight'] = old['model.diffusion_model.output_blocks.8.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.8.0.groupnorm_feature.bias'] = old['model.diffusion_model.output_blocks.8.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.8.0.conv_feature.weight'] = old['model.diffusion_model.output_blocks.8.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.decoders.8.0.conv_feature.bias'] = old['model.diffusion_model.output_blocks.8.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.decoders.8.0.linear_time.weight'] = old['model.diffusion_model.output_blocks.8.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.decoders.8.0.linear_time.bias'] = old['model.diffusion_model.output_blocks.8.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.decoders.8.0.groupnorm_merged.weight'] = old['model.diffusion_model.output_blocks.8.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.8.0.groupnorm_merged.bias'] = old['model.diffusion_model.output_blocks.8.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.8.0.conv_merged.weight'] = old['model.diffusion_model.output_blocks.8.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.decoders.8.0.conv_merged.bias'] = old['model.diffusion_model.output_blocks.8.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.decoders.8.0.residual_layer.weight'] = old['model.diffusion_model.output_blocks.8.0.skip_connection.weight']\n",
        "    new['diffusion']['unet.decoders.8.0.residual_layer.bias'] = old['model.diffusion_model.output_blocks.8.0.skip_connection.bias']\n",
        "    new['diffusion']['unet.decoders.8.1.groupnorm.weight'] = old['model.diffusion_model.output_blocks.8.1.norm.weight']\n",
        "    new['diffusion']['unet.decoders.8.1.groupnorm.bias'] = old['model.diffusion_model.output_blocks.8.1.norm.bias']\n",
        "    new['diffusion']['unet.decoders.8.1.conv_input.weight'] = old['model.diffusion_model.output_blocks.8.1.proj_in.weight']\n",
        "    new['diffusion']['unet.decoders.8.1.conv_input.bias'] = old['model.diffusion_model.output_blocks.8.1.proj_in.bias']\n",
        "    new['diffusion']['unet.decoders.8.1.attention_1.out_proj.weight'] = old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn1.to_out.0.weight']\n",
        "    new['diffusion']['unet.decoders.8.1.attention_1.out_proj.bias'] = old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn1.to_out.0.bias']\n",
        "    new['diffusion']['unet.decoders.8.1.linear_geglu_1.weight'] = old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.ff.net.0.proj.weight']\n",
        "    new['diffusion']['unet.decoders.8.1.linear_geglu_1.bias'] = old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.ff.net.0.proj.bias']\n",
        "    new['diffusion']['unet.decoders.8.1.linear_geglu_2.weight'] = old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.ff.net.2.weight']\n",
        "    new['diffusion']['unet.decoders.8.1.linear_geglu_2.bias'] = old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.ff.net.2.bias']\n",
        "    new['diffusion']['unet.decoders.8.1.attention_2.q_proj.weight'] = old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn2.to_q.weight']\n",
        "    new['diffusion']['unet.decoders.8.1.attention_2.k_proj.weight'] = old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn2.to_k.weight']\n",
        "    new['diffusion']['unet.decoders.8.1.attention_2.v_proj.weight'] = old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn2.to_v.weight']\n",
        "    new['diffusion']['unet.decoders.8.1.attention_2.out_proj.weight'] = old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn2.to_out.0.weight']\n",
        "    new['diffusion']['unet.decoders.8.1.attention_2.out_proj.bias'] = old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn2.to_out.0.bias']\n",
        "    new['diffusion']['unet.decoders.8.1.layernorm_1.weight'] = old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.norm1.weight']\n",
        "    new['diffusion']['unet.decoders.8.1.layernorm_1.bias'] = old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.norm1.bias']\n",
        "    new['diffusion']['unet.decoders.8.1.layernorm_2.weight'] = old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.norm2.weight']\n",
        "    new['diffusion']['unet.decoders.8.1.layernorm_2.bias'] = old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.norm2.bias']\n",
        "    new['diffusion']['unet.decoders.8.1.layernorm_3.weight'] = old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.norm3.weight']\n",
        "    new['diffusion']['unet.decoders.8.1.layernorm_3.bias'] = old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.norm3.bias']\n",
        "    new['diffusion']['unet.decoders.8.1.conv_output.weight'] = old['model.diffusion_model.output_blocks.8.1.proj_out.weight']\n",
        "    new['diffusion']['unet.decoders.8.1.conv_output.bias'] = old['model.diffusion_model.output_blocks.8.1.proj_out.bias']\n",
        "    new['diffusion']['unet.decoders.8.2.conv.weight'] = old['model.diffusion_model.output_blocks.8.2.conv.weight']\n",
        "    new['diffusion']['unet.decoders.8.2.conv.bias'] = old['model.diffusion_model.output_blocks.8.2.conv.bias']\n",
        "    new['diffusion']['unet.decoders.9.0.groupnorm_feature.weight'] = old['model.diffusion_model.output_blocks.9.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.9.0.groupnorm_feature.bias'] = old['model.diffusion_model.output_blocks.9.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.9.0.conv_feature.weight'] = old['model.diffusion_model.output_blocks.9.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.decoders.9.0.conv_feature.bias'] = old['model.diffusion_model.output_blocks.9.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.decoders.9.0.linear_time.weight'] = old['model.diffusion_model.output_blocks.9.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.decoders.9.0.linear_time.bias'] = old['model.diffusion_model.output_blocks.9.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.decoders.9.0.groupnorm_merged.weight'] = old['model.diffusion_model.output_blocks.9.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.9.0.groupnorm_merged.bias'] = old['model.diffusion_model.output_blocks.9.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.9.0.conv_merged.weight'] = old['model.diffusion_model.output_blocks.9.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.decoders.9.0.conv_merged.bias'] = old['model.diffusion_model.output_blocks.9.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.decoders.9.0.residual_layer.weight'] = old['model.diffusion_model.output_blocks.9.0.skip_connection.weight']\n",
        "    new['diffusion']['unet.decoders.9.0.residual_layer.bias'] = old['model.diffusion_model.output_blocks.9.0.skip_connection.bias']\n",
        "    new['diffusion']['unet.decoders.9.1.groupnorm.weight'] = old['model.diffusion_model.output_blocks.9.1.norm.weight']\n",
        "    new['diffusion']['unet.decoders.9.1.groupnorm.bias'] = old['model.diffusion_model.output_blocks.9.1.norm.bias']\n",
        "    new['diffusion']['unet.decoders.9.1.conv_input.weight'] = old['model.diffusion_model.output_blocks.9.1.proj_in.weight']\n",
        "    new['diffusion']['unet.decoders.9.1.conv_input.bias'] = old['model.diffusion_model.output_blocks.9.1.proj_in.bias']\n",
        "    new['diffusion']['unet.decoders.9.1.attention_1.out_proj.weight'] = old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn1.to_out.0.weight']\n",
        "    new['diffusion']['unet.decoders.9.1.attention_1.out_proj.bias'] = old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn1.to_out.0.bias']\n",
        "    new['diffusion']['unet.decoders.9.1.linear_geglu_1.weight'] = old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.ff.net.0.proj.weight']\n",
        "    new['diffusion']['unet.decoders.9.1.linear_geglu_1.bias'] = old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.ff.net.0.proj.bias']\n",
        "    new['diffusion']['unet.decoders.9.1.linear_geglu_2.weight'] = old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.ff.net.2.weight']\n",
        "    new['diffusion']['unet.decoders.9.1.linear_geglu_2.bias'] = old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.ff.net.2.bias']\n",
        "    new['diffusion']['unet.decoders.9.1.attention_2.q_proj.weight'] = old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn2.to_q.weight']\n",
        "    new['diffusion']['unet.decoders.9.1.attention_2.k_proj.weight'] = old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn2.to_k.weight']\n",
        "    new['diffusion']['unet.decoders.9.1.attention_2.v_proj.weight'] = old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn2.to_v.weight']\n",
        "    new['diffusion']['unet.decoders.9.1.attention_2.out_proj.weight'] = old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn2.to_out.0.weight']\n",
        "    new['diffusion']['unet.decoders.9.1.attention_2.out_proj.bias'] = old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn2.to_out.0.bias']\n",
        "    new['diffusion']['unet.decoders.9.1.layernorm_1.weight'] = old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.norm1.weight']\n",
        "    new['diffusion']['unet.decoders.9.1.layernorm_1.bias'] = old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.norm1.bias']\n",
        "    new['diffusion']['unet.decoders.9.1.layernorm_2.weight'] = old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.norm2.weight']\n",
        "    new['diffusion']['unet.decoders.9.1.layernorm_2.bias'] = old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.norm2.bias']\n",
        "    new['diffusion']['unet.decoders.9.1.layernorm_3.weight'] = old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.norm3.weight']\n",
        "    new['diffusion']['unet.decoders.9.1.layernorm_3.bias'] = old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.norm3.bias']\n",
        "    new['diffusion']['unet.decoders.9.1.conv_output.weight'] = old['model.diffusion_model.output_blocks.9.1.proj_out.weight']\n",
        "    new['diffusion']['unet.decoders.9.1.conv_output.bias'] = old['model.diffusion_model.output_blocks.9.1.proj_out.bias']\n",
        "    new['diffusion']['unet.decoders.10.0.groupnorm_feature.weight'] = old['model.diffusion_model.output_blocks.10.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.10.0.groupnorm_feature.bias'] = old['model.diffusion_model.output_blocks.10.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.10.0.conv_feature.weight'] = old['model.diffusion_model.output_blocks.10.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.decoders.10.0.conv_feature.bias'] = old['model.diffusion_model.output_blocks.10.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.decoders.10.0.linear_time.weight'] = old['model.diffusion_model.output_blocks.10.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.decoders.10.0.linear_time.bias'] = old['model.diffusion_model.output_blocks.10.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.decoders.10.0.groupnorm_merged.weight'] = old['model.diffusion_model.output_blocks.10.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.10.0.groupnorm_merged.bias'] = old['model.diffusion_model.output_blocks.10.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.10.0.conv_merged.weight'] = old['model.diffusion_model.output_blocks.10.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.decoders.10.0.conv_merged.bias'] = old['model.diffusion_model.output_blocks.10.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.decoders.10.0.residual_layer.weight'] = old['model.diffusion_model.output_blocks.10.0.skip_connection.weight']\n",
        "    new['diffusion']['unet.decoders.10.0.residual_layer.bias'] = old['model.diffusion_model.output_blocks.10.0.skip_connection.bias']\n",
        "    new['diffusion']['unet.decoders.10.1.groupnorm.weight'] = old['model.diffusion_model.output_blocks.10.1.norm.weight']\n",
        "    new['diffusion']['unet.decoders.10.1.groupnorm.bias'] = old['model.diffusion_model.output_blocks.10.1.norm.bias']\n",
        "    new['diffusion']['unet.decoders.10.1.conv_input.weight'] = old['model.diffusion_model.output_blocks.10.1.proj_in.weight']\n",
        "    new['diffusion']['unet.decoders.10.1.conv_input.bias'] = old['model.diffusion_model.output_blocks.10.1.proj_in.bias']\n",
        "    new['diffusion']['unet.decoders.10.1.attention_1.out_proj.weight'] = old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn1.to_out.0.weight']\n",
        "    new['diffusion']['unet.decoders.10.1.attention_1.out_proj.bias'] = old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn1.to_out.0.bias']\n",
        "    new['diffusion']['unet.decoders.10.1.linear_geglu_1.weight'] = old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.ff.net.0.proj.weight']\n",
        "    new['diffusion']['unet.decoders.10.1.linear_geglu_1.bias'] = old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.ff.net.0.proj.bias']\n",
        "    new['diffusion']['unet.decoders.10.1.linear_geglu_2.weight'] = old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.ff.net.2.weight']\n",
        "    new['diffusion']['unet.decoders.10.1.linear_geglu_2.bias'] = old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.ff.net.2.bias']\n",
        "    new['diffusion']['unet.decoders.10.1.attention_2.q_proj.weight'] = old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn2.to_q.weight']\n",
        "    new['diffusion']['unet.decoders.10.1.attention_2.k_proj.weight'] = old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn2.to_k.weight']\n",
        "    new['diffusion']['unet.decoders.10.1.attention_2.v_proj.weight'] = old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn2.to_v.weight']\n",
        "    new['diffusion']['unet.decoders.10.1.attention_2.out_proj.weight'] = old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn2.to_out.0.weight']\n",
        "    new['diffusion']['unet.decoders.10.1.attention_2.out_proj.bias'] = old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn2.to_out.0.bias']\n",
        "    new['diffusion']['unet.decoders.10.1.layernorm_1.weight'] = old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.norm1.weight']\n",
        "    new['diffusion']['unet.decoders.10.1.layernorm_1.bias'] = old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.norm1.bias']\n",
        "    new['diffusion']['unet.decoders.10.1.layernorm_2.weight'] = old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.norm2.weight']\n",
        "    new['diffusion']['unet.decoders.10.1.layernorm_2.bias'] = old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.norm2.bias']\n",
        "    new['diffusion']['unet.decoders.10.1.layernorm_3.weight'] = old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.norm3.weight']\n",
        "    new['diffusion']['unet.decoders.10.1.layernorm_3.bias'] = old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.norm3.bias']\n",
        "    new['diffusion']['unet.decoders.10.1.conv_output.weight'] = old['model.diffusion_model.output_blocks.10.1.proj_out.weight']\n",
        "    new['diffusion']['unet.decoders.10.1.conv_output.bias'] = old['model.diffusion_model.output_blocks.10.1.proj_out.bias']\n",
        "    new['diffusion']['unet.decoders.11.0.groupnorm_feature.weight'] = old['model.diffusion_model.output_blocks.11.0.in_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.11.0.groupnorm_feature.bias'] = old['model.diffusion_model.output_blocks.11.0.in_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.11.0.conv_feature.weight'] = old['model.diffusion_model.output_blocks.11.0.in_layers.2.weight']\n",
        "    new['diffusion']['unet.decoders.11.0.conv_feature.bias'] = old['model.diffusion_model.output_blocks.11.0.in_layers.2.bias']\n",
        "    new['diffusion']['unet.decoders.11.0.linear_time.weight'] = old['model.diffusion_model.output_blocks.11.0.emb_layers.1.weight']\n",
        "    new['diffusion']['unet.decoders.11.0.linear_time.bias'] = old['model.diffusion_model.output_blocks.11.0.emb_layers.1.bias']\n",
        "    new['diffusion']['unet.decoders.11.0.groupnorm_merged.weight'] = old['model.diffusion_model.output_blocks.11.0.out_layers.0.weight']\n",
        "    new['diffusion']['unet.decoders.11.0.groupnorm_merged.bias'] = old['model.diffusion_model.output_blocks.11.0.out_layers.0.bias']\n",
        "    new['diffusion']['unet.decoders.11.0.conv_merged.weight'] = old['model.diffusion_model.output_blocks.11.0.out_layers.3.weight']\n",
        "    new['diffusion']['unet.decoders.11.0.conv_merged.bias'] = old['model.diffusion_model.output_blocks.11.0.out_layers.3.bias']\n",
        "    new['diffusion']['unet.decoders.11.0.residual_layer.weight'] = old['model.diffusion_model.output_blocks.11.0.skip_connection.weight']\n",
        "    new['diffusion']['unet.decoders.11.0.residual_layer.bias'] = old['model.diffusion_model.output_blocks.11.0.skip_connection.bias']\n",
        "    new['diffusion']['unet.decoders.11.1.groupnorm.weight'] = old['model.diffusion_model.output_blocks.11.1.norm.weight']\n",
        "    new['diffusion']['unet.decoders.11.1.groupnorm.bias'] = old['model.diffusion_model.output_blocks.11.1.norm.bias']\n",
        "    new['diffusion']['unet.decoders.11.1.conv_input.weight'] = old['model.diffusion_model.output_blocks.11.1.proj_in.weight']\n",
        "    new['diffusion']['unet.decoders.11.1.conv_input.bias'] = old['model.diffusion_model.output_blocks.11.1.proj_in.bias']\n",
        "    new['diffusion']['unet.decoders.11.1.attention_1.out_proj.weight'] = old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn1.to_out.0.weight']\n",
        "    new['diffusion']['unet.decoders.11.1.attention_1.out_proj.bias'] = old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn1.to_out.0.bias']\n",
        "    new['diffusion']['unet.decoders.11.1.linear_geglu_1.weight'] = old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.ff.net.0.proj.weight']\n",
        "    new['diffusion']['unet.decoders.11.1.linear_geglu_1.bias'] = old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.ff.net.0.proj.bias']\n",
        "    new['diffusion']['unet.decoders.11.1.linear_geglu_2.weight'] = old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.ff.net.2.weight']\n",
        "    new['diffusion']['unet.decoders.11.1.linear_geglu_2.bias'] = old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.ff.net.2.bias']\n",
        "    new['diffusion']['unet.decoders.11.1.attention_2.q_proj.weight'] = old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn2.to_q.weight']\n",
        "    new['diffusion']['unet.decoders.11.1.attention_2.k_proj.weight'] = old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn2.to_k.weight']\n",
        "    new['diffusion']['unet.decoders.11.1.attention_2.v_proj.weight'] = old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn2.to_v.weight']\n",
        "    new['diffusion']['unet.decoders.11.1.attention_2.out_proj.weight'] = old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn2.to_out.0.weight']\n",
        "    new['diffusion']['unet.decoders.11.1.attention_2.out_proj.bias'] = old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn2.to_out.0.bias']\n",
        "    new['diffusion']['unet.decoders.11.1.layernorm_1.weight'] = old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.norm1.weight']\n",
        "    new['diffusion']['unet.decoders.11.1.layernorm_1.bias'] = old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.norm1.bias']\n",
        "    new['diffusion']['unet.decoders.11.1.layernorm_2.weight'] = old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.norm2.weight']\n",
        "    new['diffusion']['unet.decoders.11.1.layernorm_2.bias'] = old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.norm2.bias']\n",
        "    new['diffusion']['unet.decoders.11.1.layernorm_3.weight'] = old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.norm3.weight']\n",
        "    new['diffusion']['unet.decoders.11.1.layernorm_3.bias'] = old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.norm3.bias']\n",
        "    new['diffusion']['unet.decoders.11.1.conv_output.weight'] = old['model.diffusion_model.output_blocks.11.1.proj_out.weight']\n",
        "    new['diffusion']['unet.decoders.11.1.conv_output.bias'] = old['model.diffusion_model.output_blocks.11.1.proj_out.bias']\n",
        "    new['diffusion']['final.groupnorm.weight'] = old['model.diffusion_model.out.0.weight']\n",
        "    new['diffusion']['final.groupnorm.bias'] = old['model.diffusion_model.out.0.bias']\n",
        "    new['diffusion']['final.conv.weight'] = old['model.diffusion_model.out.2.weight']\n",
        "    new['diffusion']['final.conv.bias'] = old['model.diffusion_model.out.2.bias']\n",
        "    new['encoder']['0.weight'] = old['first_stage_model.encoder.conv_in.weight']\n",
        "    new['encoder']['0.bias'] = old['first_stage_model.encoder.conv_in.bias']\n",
        "    new['encoder']['1.groupnorm_1.weight'] = old['first_stage_model.encoder.down.0.block.0.norm1.weight']\n",
        "    new['encoder']['1.groupnorm_1.bias'] = old['first_stage_model.encoder.down.0.block.0.norm1.bias']\n",
        "    new['encoder']['1.conv_1.weight'] = old['first_stage_model.encoder.down.0.block.0.conv1.weight']\n",
        "    new['encoder']['1.conv_1.bias'] = old['first_stage_model.encoder.down.0.block.0.conv1.bias']\n",
        "    new['encoder']['1.groupnorm_2.weight'] = old['first_stage_model.encoder.down.0.block.0.norm2.weight']\n",
        "    new['encoder']['1.groupnorm_2.bias'] = old['first_stage_model.encoder.down.0.block.0.norm2.bias']\n",
        "    new['encoder']['1.conv_2.weight'] = old['first_stage_model.encoder.down.0.block.0.conv2.weight']\n",
        "    new['encoder']['1.conv_2.bias'] = old['first_stage_model.encoder.down.0.block.0.conv2.bias']\n",
        "    new['encoder']['2.groupnorm_1.weight'] = old['first_stage_model.encoder.down.0.block.1.norm1.weight']\n",
        "    new['encoder']['2.groupnorm_1.bias'] = old['first_stage_model.encoder.down.0.block.1.norm1.bias']\n",
        "    new['encoder']['2.conv_1.weight'] = old['first_stage_model.encoder.down.0.block.1.conv1.weight']\n",
        "    new['encoder']['2.conv_1.bias'] = old['first_stage_model.encoder.down.0.block.1.conv1.bias']\n",
        "    new['encoder']['2.groupnorm_2.weight'] = old['first_stage_model.encoder.down.0.block.1.norm2.weight']\n",
        "    new['encoder']['2.groupnorm_2.bias'] = old['first_stage_model.encoder.down.0.block.1.norm2.bias']\n",
        "    new['encoder']['2.conv_2.weight'] = old['first_stage_model.encoder.down.0.block.1.conv2.weight']\n",
        "    new['encoder']['2.conv_2.bias'] = old['first_stage_model.encoder.down.0.block.1.conv2.bias']\n",
        "    new['encoder']['3.weight'] = old['first_stage_model.encoder.down.0.downsample.conv.weight']\n",
        "    new['encoder']['3.bias'] = old['first_stage_model.encoder.down.0.downsample.conv.bias']\n",
        "    new['encoder']['4.groupnorm_1.weight'] = old['first_stage_model.encoder.down.1.block.0.norm1.weight']\n",
        "    new['encoder']['4.groupnorm_1.bias'] = old['first_stage_model.encoder.down.1.block.0.norm1.bias']\n",
        "    new['encoder']['4.conv_1.weight'] = old['first_stage_model.encoder.down.1.block.0.conv1.weight']\n",
        "    new['encoder']['4.conv_1.bias'] = old['first_stage_model.encoder.down.1.block.0.conv1.bias']\n",
        "    new['encoder']['4.groupnorm_2.weight'] = old['first_stage_model.encoder.down.1.block.0.norm2.weight']\n",
        "    new['encoder']['4.groupnorm_2.bias'] = old['first_stage_model.encoder.down.1.block.0.norm2.bias']\n",
        "    new['encoder']['4.conv_2.weight'] = old['first_stage_model.encoder.down.1.block.0.conv2.weight']\n",
        "    new['encoder']['4.conv_2.bias'] = old['first_stage_model.encoder.down.1.block.0.conv2.bias']\n",
        "    new['encoder']['4.residual_layer.weight'] = old['first_stage_model.encoder.down.1.block.0.nin_shortcut.weight']\n",
        "    new['encoder']['4.residual_layer.bias'] = old['first_stage_model.encoder.down.1.block.0.nin_shortcut.bias']\n",
        "    new['encoder']['5.groupnorm_1.weight'] = old['first_stage_model.encoder.down.1.block.1.norm1.weight']\n",
        "    new['encoder']['5.groupnorm_1.bias'] = old['first_stage_model.encoder.down.1.block.1.norm1.bias']\n",
        "    new['encoder']['5.conv_1.weight'] = old['first_stage_model.encoder.down.1.block.1.conv1.weight']\n",
        "    new['encoder']['5.conv_1.bias'] = old['first_stage_model.encoder.down.1.block.1.conv1.bias']\n",
        "    new['encoder']['5.groupnorm_2.weight'] = old['first_stage_model.encoder.down.1.block.1.norm2.weight']\n",
        "    new['encoder']['5.groupnorm_2.bias'] = old['first_stage_model.encoder.down.1.block.1.norm2.bias']\n",
        "    new['encoder']['5.conv_2.weight'] = old['first_stage_model.encoder.down.1.block.1.conv2.weight']\n",
        "    new['encoder']['5.conv_2.bias'] = old['first_stage_model.encoder.down.1.block.1.conv2.bias']\n",
        "    new['encoder']['6.weight'] = old['first_stage_model.encoder.down.1.downsample.conv.weight']\n",
        "    new['encoder']['6.bias'] = old['first_stage_model.encoder.down.1.downsample.conv.bias']\n",
        "    new['encoder']['7.groupnorm_1.weight'] = old['first_stage_model.encoder.down.2.block.0.norm1.weight']\n",
        "    new['encoder']['7.groupnorm_1.bias'] = old['first_stage_model.encoder.down.2.block.0.norm1.bias']\n",
        "    new['encoder']['7.conv_1.weight'] = old['first_stage_model.encoder.down.2.block.0.conv1.weight']\n",
        "    new['encoder']['7.conv_1.bias'] = old['first_stage_model.encoder.down.2.block.0.conv1.bias']\n",
        "    new['encoder']['7.groupnorm_2.weight'] = old['first_stage_model.encoder.down.2.block.0.norm2.weight']\n",
        "    new['encoder']['7.groupnorm_2.bias'] = old['first_stage_model.encoder.down.2.block.0.norm2.bias']\n",
        "    new['encoder']['7.conv_2.weight'] = old['first_stage_model.encoder.down.2.block.0.conv2.weight']\n",
        "    new['encoder']['7.conv_2.bias'] = old['first_stage_model.encoder.down.2.block.0.conv2.bias']\n",
        "    new['encoder']['7.residual_layer.weight'] = old['first_stage_model.encoder.down.2.block.0.nin_shortcut.weight']\n",
        "    new['encoder']['7.residual_layer.bias'] = old['first_stage_model.encoder.down.2.block.0.nin_shortcut.bias']\n",
        "    new['encoder']['8.groupnorm_1.weight'] = old['first_stage_model.encoder.down.2.block.1.norm1.weight']\n",
        "    new['encoder']['8.groupnorm_1.bias'] = old['first_stage_model.encoder.down.2.block.1.norm1.bias']\n",
        "    new['encoder']['8.conv_1.weight'] = old['first_stage_model.encoder.down.2.block.1.conv1.weight']\n",
        "    new['encoder']['8.conv_1.bias'] = old['first_stage_model.encoder.down.2.block.1.conv1.bias']\n",
        "    new['encoder']['8.groupnorm_2.weight'] = old['first_stage_model.encoder.down.2.block.1.norm2.weight']\n",
        "    new['encoder']['8.groupnorm_2.bias'] = old['first_stage_model.encoder.down.2.block.1.norm2.bias']\n",
        "    new['encoder']['8.conv_2.weight'] = old['first_stage_model.encoder.down.2.block.1.conv2.weight']\n",
        "    new['encoder']['8.conv_2.bias'] = old['first_stage_model.encoder.down.2.block.1.conv2.bias']\n",
        "    new['encoder']['9.weight'] = old['first_stage_model.encoder.down.2.downsample.conv.weight']\n",
        "    new['encoder']['9.bias'] = old['first_stage_model.encoder.down.2.downsample.conv.bias']\n",
        "    new['encoder']['10.groupnorm_1.weight'] = old['first_stage_model.encoder.down.3.block.0.norm1.weight']\n",
        "    new['encoder']['10.groupnorm_1.bias'] = old['first_stage_model.encoder.down.3.block.0.norm1.bias']\n",
        "    new['encoder']['10.conv_1.weight'] = old['first_stage_model.encoder.down.3.block.0.conv1.weight']\n",
        "    new['encoder']['10.conv_1.bias'] = old['first_stage_model.encoder.down.3.block.0.conv1.bias']\n",
        "    new['encoder']['10.groupnorm_2.weight'] = old['first_stage_model.encoder.down.3.block.0.norm2.weight']\n",
        "    new['encoder']['10.groupnorm_2.bias'] = old['first_stage_model.encoder.down.3.block.0.norm2.bias']\n",
        "    new['encoder']['10.conv_2.weight'] = old['first_stage_model.encoder.down.3.block.0.conv2.weight']\n",
        "    new['encoder']['10.conv_2.bias'] = old['first_stage_model.encoder.down.3.block.0.conv2.bias']\n",
        "    new['encoder']['11.groupnorm_1.weight'] = old['first_stage_model.encoder.down.3.block.1.norm1.weight']\n",
        "    new['encoder']['11.groupnorm_1.bias'] = old['first_stage_model.encoder.down.3.block.1.norm1.bias']\n",
        "    new['encoder']['11.conv_1.weight'] = old['first_stage_model.encoder.down.3.block.1.conv1.weight']\n",
        "    new['encoder']['11.conv_1.bias'] = old['first_stage_model.encoder.down.3.block.1.conv1.bias']\n",
        "    new['encoder']['11.groupnorm_2.weight'] = old['first_stage_model.encoder.down.3.block.1.norm2.weight']\n",
        "    new['encoder']['11.groupnorm_2.bias'] = old['first_stage_model.encoder.down.3.block.1.norm2.bias']\n",
        "    new['encoder']['11.conv_2.weight'] = old['first_stage_model.encoder.down.3.block.1.conv2.weight']\n",
        "    new['encoder']['11.conv_2.bias'] = old['first_stage_model.encoder.down.3.block.1.conv2.bias']\n",
        "    new['encoder']['12.groupnorm_1.weight'] = old['first_stage_model.encoder.mid.block_1.norm1.weight']\n",
        "    new['encoder']['12.groupnorm_1.bias'] = old['first_stage_model.encoder.mid.block_1.norm1.bias']\n",
        "    new['encoder']['12.conv_1.weight'] = old['first_stage_model.encoder.mid.block_1.conv1.weight']\n",
        "    new['encoder']['12.conv_1.bias'] = old['first_stage_model.encoder.mid.block_1.conv1.bias']\n",
        "    new['encoder']['12.groupnorm_2.weight'] = old['first_stage_model.encoder.mid.block_1.norm2.weight']\n",
        "    new['encoder']['12.groupnorm_2.bias'] = old['first_stage_model.encoder.mid.block_1.norm2.bias']\n",
        "    new['encoder']['12.conv_2.weight'] = old['first_stage_model.encoder.mid.block_1.conv2.weight']\n",
        "    new['encoder']['12.conv_2.bias'] = old['first_stage_model.encoder.mid.block_1.conv2.bias']\n",
        "    new['encoder']['13.groupnorm.weight'] = old['first_stage_model.encoder.mid.attn_1.norm.weight']\n",
        "    new['encoder']['13.groupnorm.bias'] = old['first_stage_model.encoder.mid.attn_1.norm.bias']\n",
        "    new['encoder']['13.attention.out_proj.bias'] = old['first_stage_model.encoder.mid.attn_1.proj_out.bias']\n",
        "    new['encoder']['14.groupnorm_1.weight'] = old['first_stage_model.encoder.mid.block_2.norm1.weight']\n",
        "    new['encoder']['14.groupnorm_1.bias'] = old['first_stage_model.encoder.mid.block_2.norm1.bias']\n",
        "    new['encoder']['14.conv_1.weight'] = old['first_stage_model.encoder.mid.block_2.conv1.weight']\n",
        "    new['encoder']['14.conv_1.bias'] = old['first_stage_model.encoder.mid.block_2.conv1.bias']\n",
        "    new['encoder']['14.groupnorm_2.weight'] = old['first_stage_model.encoder.mid.block_2.norm2.weight']\n",
        "    new['encoder']['14.groupnorm_2.bias'] = old['first_stage_model.encoder.mid.block_2.norm2.bias']\n",
        "    new['encoder']['14.conv_2.weight'] = old['first_stage_model.encoder.mid.block_2.conv2.weight']\n",
        "    new['encoder']['14.conv_2.bias'] = old['first_stage_model.encoder.mid.block_2.conv2.bias']\n",
        "    new['encoder']['15.weight'] = old['first_stage_model.encoder.norm_out.weight']\n",
        "    new['encoder']['15.bias'] = old['first_stage_model.encoder.norm_out.bias']\n",
        "    new['encoder']['17.weight'] = old['first_stage_model.encoder.conv_out.weight']\n",
        "    new['encoder']['17.bias'] = old['first_stage_model.encoder.conv_out.bias']\n",
        "    new['decoder']['1.weight'] = old['first_stage_model.decoder.conv_in.weight']\n",
        "    new['decoder']['1.bias'] = old['first_stage_model.decoder.conv_in.bias']\n",
        "    new['decoder']['2.groupnorm_1.weight'] = old['first_stage_model.decoder.mid.block_1.norm1.weight']\n",
        "    new['decoder']['2.groupnorm_1.bias'] = old['first_stage_model.decoder.mid.block_1.norm1.bias']\n",
        "    new['decoder']['2.conv_1.weight'] = old['first_stage_model.decoder.mid.block_1.conv1.weight']\n",
        "    new['decoder']['2.conv_1.bias'] = old['first_stage_model.decoder.mid.block_1.conv1.bias']\n",
        "    new['decoder']['2.groupnorm_2.weight'] = old['first_stage_model.decoder.mid.block_1.norm2.weight']\n",
        "    new['decoder']['2.groupnorm_2.bias'] = old['first_stage_model.decoder.mid.block_1.norm2.bias']\n",
        "    new['decoder']['2.conv_2.weight'] = old['first_stage_model.decoder.mid.block_1.conv2.weight']\n",
        "    new['decoder']['2.conv_2.bias'] = old['first_stage_model.decoder.mid.block_1.conv2.bias']\n",
        "    new['decoder']['3.groupnorm.weight'] = old['first_stage_model.decoder.mid.attn_1.norm.weight']\n",
        "    new['decoder']['3.groupnorm.bias'] = old['first_stage_model.decoder.mid.attn_1.norm.bias']\n",
        "    new['decoder']['3.attention.out_proj.bias'] = old['first_stage_model.decoder.mid.attn_1.proj_out.bias']\n",
        "    new['decoder']['4.groupnorm_1.weight'] = old['first_stage_model.decoder.mid.block_2.norm1.weight']\n",
        "    new['decoder']['4.groupnorm_1.bias'] = old['first_stage_model.decoder.mid.block_2.norm1.bias']\n",
        "    new['decoder']['4.conv_1.weight'] = old['first_stage_model.decoder.mid.block_2.conv1.weight']\n",
        "    new['decoder']['4.conv_1.bias'] = old['first_stage_model.decoder.mid.block_2.conv1.bias']\n",
        "    new['decoder']['4.groupnorm_2.weight'] = old['first_stage_model.decoder.mid.block_2.norm2.weight']\n",
        "    new['decoder']['4.groupnorm_2.bias'] = old['first_stage_model.decoder.mid.block_2.norm2.bias']\n",
        "    new['decoder']['4.conv_2.weight'] = old['first_stage_model.decoder.mid.block_2.conv2.weight']\n",
        "    new['decoder']['4.conv_2.bias'] = old['first_stage_model.decoder.mid.block_2.conv2.bias']\n",
        "    new['decoder']['20.groupnorm_1.weight'] = old['first_stage_model.decoder.up.0.block.0.norm1.weight']\n",
        "    new['decoder']['20.groupnorm_1.bias'] = old['first_stage_model.decoder.up.0.block.0.norm1.bias']\n",
        "    new['decoder']['20.conv_1.weight'] = old['first_stage_model.decoder.up.0.block.0.conv1.weight']\n",
        "    new['decoder']['20.conv_1.bias'] = old['first_stage_model.decoder.up.0.block.0.conv1.bias']\n",
        "    new['decoder']['20.groupnorm_2.weight'] = old['first_stage_model.decoder.up.0.block.0.norm2.weight']\n",
        "    new['decoder']['20.groupnorm_2.bias'] = old['first_stage_model.decoder.up.0.block.0.norm2.bias']\n",
        "    new['decoder']['20.conv_2.weight'] = old['first_stage_model.decoder.up.0.block.0.conv2.weight']\n",
        "    new['decoder']['20.conv_2.bias'] = old['first_stage_model.decoder.up.0.block.0.conv2.bias']\n",
        "    new['decoder']['20.residual_layer.weight'] = old['first_stage_model.decoder.up.0.block.0.nin_shortcut.weight']\n",
        "    new['decoder']['20.residual_layer.bias'] = old['first_stage_model.decoder.up.0.block.0.nin_shortcut.bias']\n",
        "    new['decoder']['21.groupnorm_1.weight'] = old['first_stage_model.decoder.up.0.block.1.norm1.weight']\n",
        "    new['decoder']['21.groupnorm_1.bias'] = old['first_stage_model.decoder.up.0.block.1.norm1.bias']\n",
        "    new['decoder']['21.conv_1.weight'] = old['first_stage_model.decoder.up.0.block.1.conv1.weight']\n",
        "    new['decoder']['21.conv_1.bias'] = old['first_stage_model.decoder.up.0.block.1.conv1.bias']\n",
        "    new['decoder']['21.groupnorm_2.weight'] = old['first_stage_model.decoder.up.0.block.1.norm2.weight']\n",
        "    new['decoder']['21.groupnorm_2.bias'] = old['first_stage_model.decoder.up.0.block.1.norm2.bias']\n",
        "    new['decoder']['21.conv_2.weight'] = old['first_stage_model.decoder.up.0.block.1.conv2.weight']\n",
        "    new['decoder']['21.conv_2.bias'] = old['first_stage_model.decoder.up.0.block.1.conv2.bias']\n",
        "    new['decoder']['22.groupnorm_1.weight'] = old['first_stage_model.decoder.up.0.block.2.norm1.weight']\n",
        "    new['decoder']['22.groupnorm_1.bias'] = old['first_stage_model.decoder.up.0.block.2.norm1.bias']\n",
        "    new['decoder']['22.conv_1.weight'] = old['first_stage_model.decoder.up.0.block.2.conv1.weight']\n",
        "    new['decoder']['22.conv_1.bias'] = old['first_stage_model.decoder.up.0.block.2.conv1.bias']\n",
        "    new['decoder']['22.groupnorm_2.weight'] = old['first_stage_model.decoder.up.0.block.2.norm2.weight']\n",
        "    new['decoder']['22.groupnorm_2.bias'] = old['first_stage_model.decoder.up.0.block.2.norm2.bias']\n",
        "    new['decoder']['22.conv_2.weight'] = old['first_stage_model.decoder.up.0.block.2.conv2.weight']\n",
        "    new['decoder']['22.conv_2.bias'] = old['first_stage_model.decoder.up.0.block.2.conv2.bias']\n",
        "    new['decoder']['15.groupnorm_1.weight'] = old['first_stage_model.decoder.up.1.block.0.norm1.weight']\n",
        "    new['decoder']['15.groupnorm_1.bias'] = old['first_stage_model.decoder.up.1.block.0.norm1.bias']\n",
        "    new['decoder']['15.conv_1.weight'] = old['first_stage_model.decoder.up.1.block.0.conv1.weight']\n",
        "    new['decoder']['15.conv_1.bias'] = old['first_stage_model.decoder.up.1.block.0.conv1.bias']\n",
        "    new['decoder']['15.groupnorm_2.weight'] = old['first_stage_model.decoder.up.1.block.0.norm2.weight']\n",
        "    new['decoder']['15.groupnorm_2.bias'] = old['first_stage_model.decoder.up.1.block.0.norm2.bias']\n",
        "    new['decoder']['15.conv_2.weight'] = old['first_stage_model.decoder.up.1.block.0.conv2.weight']\n",
        "    new['decoder']['15.conv_2.bias'] = old['first_stage_model.decoder.up.1.block.0.conv2.bias']\n",
        "    new['decoder']['15.residual_layer.weight'] = old['first_stage_model.decoder.up.1.block.0.nin_shortcut.weight']\n",
        "    new['decoder']['15.residual_layer.bias'] = old['first_stage_model.decoder.up.1.block.0.nin_shortcut.bias']\n",
        "    new['decoder']['16.groupnorm_1.weight'] = old['first_stage_model.decoder.up.1.block.1.norm1.weight']\n",
        "    new['decoder']['16.groupnorm_1.bias'] = old['first_stage_model.decoder.up.1.block.1.norm1.bias']\n",
        "    new['decoder']['16.conv_1.weight'] = old['first_stage_model.decoder.up.1.block.1.conv1.weight']\n",
        "    new['decoder']['16.conv_1.bias'] = old['first_stage_model.decoder.up.1.block.1.conv1.bias']\n",
        "    new['decoder']['16.groupnorm_2.weight'] = old['first_stage_model.decoder.up.1.block.1.norm2.weight']\n",
        "    new['decoder']['16.groupnorm_2.bias'] = old['first_stage_model.decoder.up.1.block.1.norm2.bias']\n",
        "    new['decoder']['16.conv_2.weight'] = old['first_stage_model.decoder.up.1.block.1.conv2.weight']\n",
        "    new['decoder']['16.conv_2.bias'] = old['first_stage_model.decoder.up.1.block.1.conv2.bias']\n",
        "    new['decoder']['17.groupnorm_1.weight'] = old['first_stage_model.decoder.up.1.block.2.norm1.weight']\n",
        "    new['decoder']['17.groupnorm_1.bias'] = old['first_stage_model.decoder.up.1.block.2.norm1.bias']\n",
        "    new['decoder']['17.conv_1.weight'] = old['first_stage_model.decoder.up.1.block.2.conv1.weight']\n",
        "    new['decoder']['17.conv_1.bias'] = old['first_stage_model.decoder.up.1.block.2.conv1.bias']\n",
        "    new['decoder']['17.groupnorm_2.weight'] = old['first_stage_model.decoder.up.1.block.2.norm2.weight']\n",
        "    new['decoder']['17.groupnorm_2.bias'] = old['first_stage_model.decoder.up.1.block.2.norm2.bias']\n",
        "    new['decoder']['17.conv_2.weight'] = old['first_stage_model.decoder.up.1.block.2.conv2.weight']\n",
        "    new['decoder']['17.conv_2.bias'] = old['first_stage_model.decoder.up.1.block.2.conv2.bias']\n",
        "    new['decoder']['19.weight'] = old['first_stage_model.decoder.up.1.upsample.conv.weight']\n",
        "    new['decoder']['19.bias'] = old['first_stage_model.decoder.up.1.upsample.conv.bias']\n",
        "    new['decoder']['10.groupnorm_1.weight'] = old['first_stage_model.decoder.up.2.block.0.norm1.weight']\n",
        "    new['decoder']['10.groupnorm_1.bias'] = old['first_stage_model.decoder.up.2.block.0.norm1.bias']\n",
        "    new['decoder']['10.conv_1.weight'] = old['first_stage_model.decoder.up.2.block.0.conv1.weight']\n",
        "    new['decoder']['10.conv_1.bias'] = old['first_stage_model.decoder.up.2.block.0.conv1.bias']\n",
        "    new['decoder']['10.groupnorm_2.weight'] = old['first_stage_model.decoder.up.2.block.0.norm2.weight']\n",
        "    new['decoder']['10.groupnorm_2.bias'] = old['first_stage_model.decoder.up.2.block.0.norm2.bias']\n",
        "    new['decoder']['10.conv_2.weight'] = old['first_stage_model.decoder.up.2.block.0.conv2.weight']\n",
        "    new['decoder']['10.conv_2.bias'] = old['first_stage_model.decoder.up.2.block.0.conv2.bias']\n",
        "    new['decoder']['11.groupnorm_1.weight'] = old['first_stage_model.decoder.up.2.block.1.norm1.weight']\n",
        "    new['decoder']['11.groupnorm_1.bias'] = old['first_stage_model.decoder.up.2.block.1.norm1.bias']\n",
        "    new['decoder']['11.conv_1.weight'] = old['first_stage_model.decoder.up.2.block.1.conv1.weight']\n",
        "    new['decoder']['11.conv_1.bias'] = old['first_stage_model.decoder.up.2.block.1.conv1.bias']\n",
        "    new['decoder']['11.groupnorm_2.weight'] = old['first_stage_model.decoder.up.2.block.1.norm2.weight']\n",
        "    new['decoder']['11.groupnorm_2.bias'] = old['first_stage_model.decoder.up.2.block.1.norm2.bias']\n",
        "    new['decoder']['11.conv_2.weight'] = old['first_stage_model.decoder.up.2.block.1.conv2.weight']\n",
        "    new['decoder']['11.conv_2.bias'] = old['first_stage_model.decoder.up.2.block.1.conv2.bias']\n",
        "    new['decoder']['12.groupnorm_1.weight'] = old['first_stage_model.decoder.up.2.block.2.norm1.weight']\n",
        "    new['decoder']['12.groupnorm_1.bias'] = old['first_stage_model.decoder.up.2.block.2.norm1.bias']\n",
        "    new['decoder']['12.conv_1.weight'] = old['first_stage_model.decoder.up.2.block.2.conv1.weight']\n",
        "    new['decoder']['12.conv_1.bias'] = old['first_stage_model.decoder.up.2.block.2.conv1.bias']\n",
        "    new['decoder']['12.groupnorm_2.weight'] = old['first_stage_model.decoder.up.2.block.2.norm2.weight']\n",
        "    new['decoder']['12.groupnorm_2.bias'] = old['first_stage_model.decoder.up.2.block.2.norm2.bias']\n",
        "    new['decoder']['12.conv_2.weight'] = old['first_stage_model.decoder.up.2.block.2.conv2.weight']\n",
        "    new['decoder']['12.conv_2.bias'] = old['first_stage_model.decoder.up.2.block.2.conv2.bias']\n",
        "    new['decoder']['14.weight'] = old['first_stage_model.decoder.up.2.upsample.conv.weight']\n",
        "    new['decoder']['14.bias'] = old['first_stage_model.decoder.up.2.upsample.conv.bias']\n",
        "    new['decoder']['5.groupnorm_1.weight'] = old['first_stage_model.decoder.up.3.block.0.norm1.weight']\n",
        "    new['decoder']['5.groupnorm_1.bias'] = old['first_stage_model.decoder.up.3.block.0.norm1.bias']\n",
        "    new['decoder']['5.conv_1.weight'] = old['first_stage_model.decoder.up.3.block.0.conv1.weight']\n",
        "    new['decoder']['5.conv_1.bias'] = old['first_stage_model.decoder.up.3.block.0.conv1.bias']\n",
        "    new['decoder']['5.groupnorm_2.weight'] = old['first_stage_model.decoder.up.3.block.0.norm2.weight']\n",
        "    new['decoder']['5.groupnorm_2.bias'] = old['first_stage_model.decoder.up.3.block.0.norm2.bias']\n",
        "    new['decoder']['5.conv_2.weight'] = old['first_stage_model.decoder.up.3.block.0.conv2.weight']\n",
        "    new['decoder']['5.conv_2.bias'] = old['first_stage_model.decoder.up.3.block.0.conv2.bias']\n",
        "    new['decoder']['6.groupnorm_1.weight'] = old['first_stage_model.decoder.up.3.block.1.norm1.weight']\n",
        "    new['decoder']['6.groupnorm_1.bias'] = old['first_stage_model.decoder.up.3.block.1.norm1.bias']\n",
        "    new['decoder']['6.conv_1.weight'] = old['first_stage_model.decoder.up.3.block.1.conv1.weight']\n",
        "    new['decoder']['6.conv_1.bias'] = old['first_stage_model.decoder.up.3.block.1.conv1.bias']\n",
        "    new['decoder']['6.groupnorm_2.weight'] = old['first_stage_model.decoder.up.3.block.1.norm2.weight']\n",
        "    new['decoder']['6.groupnorm_2.bias'] = old['first_stage_model.decoder.up.3.block.1.norm2.bias']\n",
        "    new['decoder']['6.conv_2.weight'] = old['first_stage_model.decoder.up.3.block.1.conv2.weight']\n",
        "    new['decoder']['6.conv_2.bias'] = old['first_stage_model.decoder.up.3.block.1.conv2.bias']\n",
        "    new['decoder']['7.groupnorm_1.weight'] = old['first_stage_model.decoder.up.3.block.2.norm1.weight']\n",
        "    new['decoder']['7.groupnorm_1.bias'] = old['first_stage_model.decoder.up.3.block.2.norm1.bias']\n",
        "    new['decoder']['7.conv_1.weight'] = old['first_stage_model.decoder.up.3.block.2.conv1.weight']\n",
        "    new['decoder']['7.conv_1.bias'] = old['first_stage_model.decoder.up.3.block.2.conv1.bias']\n",
        "    new['decoder']['7.groupnorm_2.weight'] = old['first_stage_model.decoder.up.3.block.2.norm2.weight']\n",
        "    new['decoder']['7.groupnorm_2.bias'] = old['first_stage_model.decoder.up.3.block.2.norm2.bias']\n",
        "    new['decoder']['7.conv_2.weight'] = old['first_stage_model.decoder.up.3.block.2.conv2.weight']\n",
        "    new['decoder']['7.conv_2.bias'] = old['first_stage_model.decoder.up.3.block.2.conv2.bias']\n",
        "    new['decoder']['9.weight'] = old['first_stage_model.decoder.up.3.upsample.conv.weight']\n",
        "    new['decoder']['9.bias'] = old['first_stage_model.decoder.up.3.upsample.conv.bias']\n",
        "    new['decoder']['23.weight'] = old['first_stage_model.decoder.norm_out.weight']\n",
        "    new['decoder']['23.bias'] = old['first_stage_model.decoder.norm_out.bias']\n",
        "    new['decoder']['25.weight'] = old['first_stage_model.decoder.conv_out.weight']\n",
        "    new['decoder']['25.bias'] = old['first_stage_model.decoder.conv_out.bias']\n",
        "    new['encoder']['18.weight'] = old['first_stage_model.quant_conv.weight']\n",
        "    new['encoder']['18.bias'] = old['first_stage_model.quant_conv.bias']\n",
        "    new['decoder']['0.weight'] = old['first_stage_model.post_quant_conv.weight']\n",
        "    new['decoder']['0.bias'] = old['first_stage_model.post_quant_conv.bias']\n",
        "    new['clip']['embedding.token_embedding.weight'] = old['cond_stage_model.transformer.text_model.embeddings.token_embedding.weight']\n",
        "    new['clip']['embedding.position_value'] = old['cond_stage_model.transformer.text_model.embeddings.position_embedding.weight']\n",
        "    new['clip']['layers.0.attention.out_proj.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.0.self_attn.out_proj.weight']\n",
        "    new['clip']['layers.0.attention.out_proj.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.0.self_attn.out_proj.bias']\n",
        "    new['clip']['layers.0.layernorm_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.0.layer_norm1.weight']\n",
        "    new['clip']['layers.0.layernorm_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.0.layer_norm1.bias']\n",
        "    new['clip']['layers.0.linear_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.0.mlp.fc1.weight']\n",
        "    new['clip']['layers.0.linear_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.0.mlp.fc1.bias']\n",
        "    new['clip']['layers.0.linear_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.0.mlp.fc2.weight']\n",
        "    new['clip']['layers.0.linear_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.0.mlp.fc2.bias']\n",
        "    new['clip']['layers.0.layernorm_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.0.layer_norm2.weight']\n",
        "    new['clip']['layers.0.layernorm_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.0.layer_norm2.bias']\n",
        "    new['clip']['layers.1.attention.out_proj.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.1.self_attn.out_proj.weight']\n",
        "    new['clip']['layers.1.attention.out_proj.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.1.self_attn.out_proj.bias']\n",
        "    new['clip']['layers.1.layernorm_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.1.layer_norm1.weight']\n",
        "    new['clip']['layers.1.layernorm_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.1.layer_norm1.bias']\n",
        "    new['clip']['layers.1.linear_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.1.mlp.fc1.weight']\n",
        "    new['clip']['layers.1.linear_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.1.mlp.fc1.bias']\n",
        "    new['clip']['layers.1.linear_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.1.mlp.fc2.weight']\n",
        "    new['clip']['layers.1.linear_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.1.mlp.fc2.bias']\n",
        "    new['clip']['layers.1.layernorm_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.1.layer_norm2.weight']\n",
        "    new['clip']['layers.1.layernorm_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.1.layer_norm2.bias']\n",
        "    new['clip']['layers.2.attention.out_proj.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.2.self_attn.out_proj.weight']\n",
        "    new['clip']['layers.2.attention.out_proj.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.2.self_attn.out_proj.bias']\n",
        "    new['clip']['layers.2.layernorm_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.2.layer_norm1.weight']\n",
        "    new['clip']['layers.2.layernorm_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.2.layer_norm1.bias']\n",
        "    new['clip']['layers.2.linear_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.2.mlp.fc1.weight']\n",
        "    new['clip']['layers.2.linear_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.2.mlp.fc1.bias']\n",
        "    new['clip']['layers.2.linear_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.2.mlp.fc2.weight']\n",
        "    new['clip']['layers.2.linear_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.2.mlp.fc2.bias']\n",
        "    new['clip']['layers.2.layernorm_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.2.layer_norm2.weight']\n",
        "    new['clip']['layers.2.layernorm_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.2.layer_norm2.bias']\n",
        "    new['clip']['layers.3.attention.out_proj.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.3.self_attn.out_proj.weight']\n",
        "    new['clip']['layers.3.attention.out_proj.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.3.self_attn.out_proj.bias']\n",
        "    new['clip']['layers.3.layernorm_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.3.layer_norm1.weight']\n",
        "    new['clip']['layers.3.layernorm_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.3.layer_norm1.bias']\n",
        "    new['clip']['layers.3.linear_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.3.mlp.fc1.weight']\n",
        "    new['clip']['layers.3.linear_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.3.mlp.fc1.bias']\n",
        "    new['clip']['layers.3.linear_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.3.mlp.fc2.weight']\n",
        "    new['clip']['layers.3.linear_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.3.mlp.fc2.bias']\n",
        "    new['clip']['layers.3.layernorm_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.3.layer_norm2.weight']\n",
        "    new['clip']['layers.3.layernorm_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.3.layer_norm2.bias']\n",
        "    new['clip']['layers.4.attention.out_proj.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.4.self_attn.out_proj.weight']\n",
        "    new['clip']['layers.4.attention.out_proj.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.4.self_attn.out_proj.bias']\n",
        "    new['clip']['layers.4.layernorm_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.4.layer_norm1.weight']\n",
        "    new['clip']['layers.4.layernorm_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.4.layer_norm1.bias']\n",
        "    new['clip']['layers.4.linear_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.4.mlp.fc1.weight']\n",
        "    new['clip']['layers.4.linear_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.4.mlp.fc1.bias']\n",
        "    new['clip']['layers.4.linear_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.4.mlp.fc2.weight']\n",
        "    new['clip']['layers.4.linear_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.4.mlp.fc2.bias']\n",
        "    new['clip']['layers.4.layernorm_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.4.layer_norm2.weight']\n",
        "    new['clip']['layers.4.layernorm_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.4.layer_norm2.bias']\n",
        "    new['clip']['layers.5.attention.out_proj.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.5.self_attn.out_proj.weight']\n",
        "    new['clip']['layers.5.attention.out_proj.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.5.self_attn.out_proj.bias']\n",
        "    new['clip']['layers.5.layernorm_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.5.layer_norm1.weight']\n",
        "    new['clip']['layers.5.layernorm_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.5.layer_norm1.bias']\n",
        "    new['clip']['layers.5.linear_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.5.mlp.fc1.weight']\n",
        "    new['clip']['layers.5.linear_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.5.mlp.fc1.bias']\n",
        "    new['clip']['layers.5.linear_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.5.mlp.fc2.weight']\n",
        "    new['clip']['layers.5.linear_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.5.mlp.fc2.bias']\n",
        "    new['clip']['layers.5.layernorm_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.5.layer_norm2.weight']\n",
        "    new['clip']['layers.5.layernorm_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.5.layer_norm2.bias']\n",
        "    new['clip']['layers.6.attention.out_proj.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.6.self_attn.out_proj.weight']\n",
        "    new['clip']['layers.6.attention.out_proj.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.6.self_attn.out_proj.bias']\n",
        "    new['clip']['layers.6.layernorm_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.6.layer_norm1.weight']\n",
        "    new['clip']['layers.6.layernorm_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.6.layer_norm1.bias']\n",
        "    new['clip']['layers.6.linear_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.6.mlp.fc1.weight']\n",
        "    new['clip']['layers.6.linear_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.6.mlp.fc1.bias']\n",
        "    new['clip']['layers.6.linear_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.6.mlp.fc2.weight']\n",
        "    new['clip']['layers.6.linear_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.6.mlp.fc2.bias']\n",
        "    new['clip']['layers.6.layernorm_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.6.layer_norm2.weight']\n",
        "    new['clip']['layers.6.layernorm_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.6.layer_norm2.bias']\n",
        "    new['clip']['layers.7.attention.out_proj.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.7.self_attn.out_proj.weight']\n",
        "    new['clip']['layers.7.attention.out_proj.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.7.self_attn.out_proj.bias']\n",
        "    new['clip']['layers.7.layernorm_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.7.layer_norm1.weight']\n",
        "    new['clip']['layers.7.layernorm_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.7.layer_norm1.bias']\n",
        "    new['clip']['layers.7.linear_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.7.mlp.fc1.weight']\n",
        "    new['clip']['layers.7.linear_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.7.mlp.fc1.bias']\n",
        "    new['clip']['layers.7.linear_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.7.mlp.fc2.weight']\n",
        "    new['clip']['layers.7.linear_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.7.mlp.fc2.bias']\n",
        "    new['clip']['layers.7.layernorm_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.7.layer_norm2.weight']\n",
        "    new['clip']['layers.7.layernorm_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.7.layer_norm2.bias']\n",
        "    new['clip']['layers.8.attention.out_proj.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.8.self_attn.out_proj.weight']\n",
        "    new['clip']['layers.8.attention.out_proj.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.8.self_attn.out_proj.bias']\n",
        "    new['clip']['layers.8.layernorm_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.8.layer_norm1.weight']\n",
        "    new['clip']['layers.8.layernorm_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.8.layer_norm1.bias']\n",
        "    new['clip']['layers.8.linear_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.8.mlp.fc1.weight']\n",
        "    new['clip']['layers.8.linear_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.8.mlp.fc1.bias']\n",
        "    new['clip']['layers.8.linear_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.8.mlp.fc2.weight']\n",
        "    new['clip']['layers.8.linear_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.8.mlp.fc2.bias']\n",
        "    new['clip']['layers.8.layernorm_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.8.layer_norm2.weight']\n",
        "    new['clip']['layers.8.layernorm_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.8.layer_norm2.bias']\n",
        "    new['clip']['layers.9.attention.out_proj.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.9.self_attn.out_proj.weight']\n",
        "    new['clip']['layers.9.attention.out_proj.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.9.self_attn.out_proj.bias']\n",
        "    new['clip']['layers.9.layernorm_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.9.layer_norm1.weight']\n",
        "    new['clip']['layers.9.layernorm_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.9.layer_norm1.bias']\n",
        "    new['clip']['layers.9.linear_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.9.mlp.fc1.weight']\n",
        "    new['clip']['layers.9.linear_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.9.mlp.fc1.bias']\n",
        "    new['clip']['layers.9.linear_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.9.mlp.fc2.weight']\n",
        "    new['clip']['layers.9.linear_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.9.mlp.fc2.bias']\n",
        "    new['clip']['layers.9.layernorm_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.9.layer_norm2.weight']\n",
        "    new['clip']['layers.9.layernorm_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.9.layer_norm2.bias']\n",
        "    new['clip']['layers.10.attention.out_proj.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.10.self_attn.out_proj.weight']\n",
        "    new['clip']['layers.10.attention.out_proj.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.10.self_attn.out_proj.bias']\n",
        "    new['clip']['layers.10.layernorm_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.10.layer_norm1.weight']\n",
        "    new['clip']['layers.10.layernorm_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.10.layer_norm1.bias']\n",
        "    new['clip']['layers.10.linear_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.10.mlp.fc1.weight']\n",
        "    new['clip']['layers.10.linear_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.10.mlp.fc1.bias']\n",
        "    new['clip']['layers.10.linear_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.10.mlp.fc2.weight']\n",
        "    new['clip']['layers.10.linear_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.10.mlp.fc2.bias']\n",
        "    new['clip']['layers.10.layernorm_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.10.layer_norm2.weight']\n",
        "    new['clip']['layers.10.layernorm_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.10.layer_norm2.bias']\n",
        "    new['clip']['layers.11.attention.out_proj.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.11.self_attn.out_proj.weight']\n",
        "    new['clip']['layers.11.attention.out_proj.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.11.self_attn.out_proj.bias']\n",
        "    new['clip']['layers.11.layernorm_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.11.layer_norm1.weight']\n",
        "    new['clip']['layers.11.layernorm_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.11.layer_norm1.bias']\n",
        "    new['clip']['layers.11.linear_1.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.11.mlp.fc1.weight']\n",
        "    new['clip']['layers.11.linear_1.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.11.mlp.fc1.bias']\n",
        "    new['clip']['layers.11.linear_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.11.mlp.fc2.weight']\n",
        "    new['clip']['layers.11.linear_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.11.mlp.fc2.bias']\n",
        "    new['clip']['layers.11.layernorm_2.weight'] = old['cond_stage_model.transformer.text_model.encoder.layers.11.layer_norm2.weight']\n",
        "    new['clip']['layers.11.layernorm_2.bias'] = old['cond_stage_model.transformer.text_model.encoder.layers.11.layer_norm2.bias']\n",
        "    new['clip']['layernorm.weight'] = old['cond_stage_model.transformer.text_model.final_layer_norm.weight']\n",
        "    new['clip']['layernorm.bias'] = old['cond_stage_model.transformer.text_model.final_layer_norm.bias']\n",
        "    new['diffusion']['unet.encoders.1.1.attention_1.in_proj.weight'] = torch.cat((old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn1.to_q.weight'], old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn1.to_k.weight'], old['model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn1.to_v.weight']), 0)\n",
        "    new['diffusion']['unet.encoders.2.1.attention_1.in_proj.weight'] = torch.cat((old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn1.to_q.weight'], old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn1.to_k.weight'], old['model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn1.to_v.weight']), 0)\n",
        "    new['diffusion']['unet.encoders.4.1.attention_1.in_proj.weight'] = torch.cat((old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_q.weight'], old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_k.weight'], old['model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_v.weight']), 0)\n",
        "    new['diffusion']['unet.encoders.5.1.attention_1.in_proj.weight'] = torch.cat((old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_q.weight'], old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_k.weight'], old['model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_v.weight']), 0)\n",
        "    new['diffusion']['unet.encoders.7.1.attention_1.in_proj.weight'] = torch.cat((old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_q.weight'], old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_k.weight'], old['model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_v.weight']), 0)\n",
        "    new['diffusion']['unet.encoders.8.1.attention_1.in_proj.weight'] = torch.cat((old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_q.weight'], old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_k.weight'], old['model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_v.weight']), 0)\n",
        "    new['diffusion']['unet.bottleneck.1.attention_1.in_proj.weight'] = torch.cat((old['model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_q.weight'], old['model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_k.weight'], old['model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_v.weight']), 0)\n",
        "    new['diffusion']['unet.decoders.3.1.attention_1.in_proj.weight'] = torch.cat((old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_q.weight'], old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_k.weight'], old['model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_v.weight']), 0)\n",
        "    new['diffusion']['unet.decoders.4.1.attention_1.in_proj.weight'] = torch.cat((old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_q.weight'], old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_k.weight'], old['model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_v.weight']), 0)\n",
        "    new['diffusion']['unet.decoders.5.1.attention_1.in_proj.weight'] = torch.cat((old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_q.weight'], old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_k.weight'], old['model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_v.weight']), 0)\n",
        "    new['diffusion']['unet.decoders.6.1.attention_1.in_proj.weight'] = torch.cat((old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn1.to_q.weight'], old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn1.to_k.weight'], old['model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn1.to_v.weight']), 0)\n",
        "    new['diffusion']['unet.decoders.7.1.attention_1.in_proj.weight'] = torch.cat((old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn1.to_q.weight'], old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn1.to_k.weight'], old['model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn1.to_v.weight']), 0)\n",
        "    new['diffusion']['unet.decoders.8.1.attention_1.in_proj.weight'] = torch.cat((old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn1.to_q.weight'], old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn1.to_k.weight'], old['model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn1.to_v.weight']), 0)\n",
        "    new['diffusion']['unet.decoders.9.1.attention_1.in_proj.weight'] = torch.cat((old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn1.to_q.weight'], old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn1.to_k.weight'], old['model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn1.to_v.weight']), 0)\n",
        "    new['diffusion']['unet.decoders.10.1.attention_1.in_proj.weight'] = torch.cat((old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn1.to_q.weight'], old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn1.to_k.weight'], old['model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn1.to_v.weight']), 0)\n",
        "    new['diffusion']['unet.decoders.11.1.attention_1.in_proj.weight'] = torch.cat((old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn1.to_q.weight'], old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn1.to_k.weight'], old['model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn1.to_v.weight']), 0)\n",
        "    new['encoder']['13.attention.in_proj.weight'] = torch.cat((old['first_stage_model.encoder.mid.attn_1.q.weight'], old['first_stage_model.encoder.mid.attn_1.k.weight'], old['first_stage_model.encoder.mid.attn_1.v.weight']), 0).reshape((1536, 512))\n",
        "    new['encoder']['13.attention.in_proj.bias'] = torch.cat((old['first_stage_model.encoder.mid.attn_1.q.bias'], old['first_stage_model.encoder.mid.attn_1.k.bias'], old['first_stage_model.encoder.mid.attn_1.v.bias']), 0)\n",
        "    new['encoder']['13.attention.out_proj.weight'] = old['first_stage_model.encoder.mid.attn_1.proj_out.weight'].reshape((512, 512))\n",
        "    new['decoder']['3.attention.in_proj.weight'] = torch.cat((old['first_stage_model.decoder.mid.attn_1.q.weight'], old['first_stage_model.decoder.mid.attn_1.k.weight'], old['first_stage_model.decoder.mid.attn_1.v.weight']), 0).reshape((1536, 512))\n",
        "    new['decoder']['3.attention.in_proj.bias'] = torch.cat((old['first_stage_model.decoder.mid.attn_1.q.bias'], old['first_stage_model.decoder.mid.attn_1.k.bias'], old['first_stage_model.decoder.mid.attn_1.v.bias']), 0)\n",
        "    new['decoder']['3.attention.out_proj.weight'] = old['first_stage_model.decoder.mid.attn_1.proj_out.weight'].reshape((512, 512))\n",
        "    new['clip']['layers.0.attention.in_proj.weight'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.0.self_attn.q_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.0.self_attn.k_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.0.self_attn.v_proj.weight']), 0)\n",
        "    new['clip']['layers.0.attention.in_proj.bias'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.0.self_attn.q_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.0.self_attn.k_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.0.self_attn.v_proj.bias']), 0)\n",
        "    new['clip']['layers.1.attention.in_proj.weight'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.1.self_attn.q_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.1.self_attn.k_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.1.self_attn.v_proj.weight']), 0)\n",
        "    new['clip']['layers.1.attention.in_proj.bias'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.1.self_attn.q_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.1.self_attn.k_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.1.self_attn.v_proj.bias']), 0)\n",
        "    new['clip']['layers.2.attention.in_proj.weight'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.2.self_attn.q_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.2.self_attn.k_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.2.self_attn.v_proj.weight']), 0)\n",
        "    new['clip']['layers.2.attention.in_proj.bias'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.2.self_attn.q_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.2.self_attn.k_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.2.self_attn.v_proj.bias']), 0)\n",
        "    new['clip']['layers.3.attention.in_proj.weight'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.3.self_attn.q_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.3.self_attn.k_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.3.self_attn.v_proj.weight']), 0)\n",
        "    new['clip']['layers.3.attention.in_proj.bias'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.3.self_attn.q_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.3.self_attn.k_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.3.self_attn.v_proj.bias']), 0)\n",
        "    new['clip']['layers.4.attention.in_proj.weight'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.4.self_attn.q_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.4.self_attn.k_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.4.self_attn.v_proj.weight']), 0)\n",
        "    new['clip']['layers.4.attention.in_proj.bias'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.4.self_attn.q_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.4.self_attn.k_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.4.self_attn.v_proj.bias']), 0)\n",
        "    new['clip']['layers.5.attention.in_proj.weight'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.5.self_attn.q_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.5.self_attn.k_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.5.self_attn.v_proj.weight']), 0)\n",
        "    new['clip']['layers.5.attention.in_proj.bias'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.5.self_attn.q_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.5.self_attn.k_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.5.self_attn.v_proj.bias']), 0)\n",
        "    new['clip']['layers.6.attention.in_proj.weight'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.6.self_attn.q_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.6.self_attn.k_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.6.self_attn.v_proj.weight']), 0)\n",
        "    new['clip']['layers.6.attention.in_proj.bias'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.6.self_attn.q_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.6.self_attn.k_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.6.self_attn.v_proj.bias']), 0)\n",
        "    new['clip']['layers.7.attention.in_proj.weight'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.7.self_attn.q_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.7.self_attn.k_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.7.self_attn.v_proj.weight']), 0)\n",
        "    new['clip']['layers.7.attention.in_proj.bias'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.7.self_attn.q_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.7.self_attn.k_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.7.self_attn.v_proj.bias']), 0)\n",
        "    new['clip']['layers.8.attention.in_proj.weight'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.8.self_attn.q_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.8.self_attn.k_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.8.self_attn.v_proj.weight']), 0)\n",
        "    new['clip']['layers.8.attention.in_proj.bias'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.8.self_attn.q_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.8.self_attn.k_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.8.self_attn.v_proj.bias']), 0)\n",
        "    new['clip']['layers.9.attention.in_proj.weight'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.9.self_attn.q_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.9.self_attn.k_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.9.self_attn.v_proj.weight']), 0)\n",
        "    new['clip']['layers.9.attention.in_proj.bias'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.9.self_attn.q_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.9.self_attn.k_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.9.self_attn.v_proj.bias']), 0)\n",
        "    new['clip']['layers.10.attention.in_proj.weight'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.10.self_attn.q_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.10.self_attn.k_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.10.self_attn.v_proj.weight']), 0)\n",
        "    new['clip']['layers.10.attention.in_proj.bias'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.10.self_attn.q_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.10.self_attn.k_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.10.self_attn.v_proj.bias']), 0)\n",
        "    new['clip']['layers.11.attention.in_proj.weight'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.11.self_attn.q_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.11.self_attn.k_proj.weight'], old['cond_stage_model.transformer.text_model.encoder.layers.11.self_attn.v_proj.weight']), 0)\n",
        "    new['clip']['layers.11.attention.in_proj.bias'] = torch.cat((old['cond_stage_model.transformer.text_model.encoder.layers.11.self_attn.q_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.11.self_attn.k_proj.bias'], old['cond_stage_model.transformer.text_model.encoder.layers.11.self_attn.v_proj.bias']), 0)\n",
        "\n",
        "    torch.save(new['diffusion'], \"/content/stable-diffusion-pytorch/data/ckpt/diffusion.pt\")\n",
        "    torch.save(new['encoder'], \"/content/stable-diffusion-pytorch/data/ckpt/encoder.pt\")\n",
        "    torch.save(new['decoder'], \"/content/stable-diffusion-pytorch/data/ckpt/decoder.pt\")\n",
        "    torch.save(new['clip'], \"/content/stable-diffusion-pytorch/data/ckpt/clip.pt\")\n",
        "\n",
        "    del old, new\n",
        "    with torch.no_grad():\n",
        "      torch.cuda.empty_cache()\n",
        "    gc.collect()\n",
        "\n",
        "    %cd /content/stable-diffusion-pytorch\n",
        "    from stable_diffusion_pytorch import model_loader\n",
        "    models = model_loader.preload_models('cpu')\n",
        "\n",
        "    width = closestNumber(width, 8)\n",
        "    height = closestNumber(height, 8)\n",
        "    metadata.add_text(\"Prompt\", f\"{prompt}\")\n",
        "    metadata.add_text(\"by\", f\"{by}\")\n",
        "    gc.collect()\n",
        "    real_seed = torch.cuda.initial_seed()\n",
        "\n",
        "    # https://github.com/kjsman/stable-diffusion-pytorch/blob/main/demo.ipynb modified thanks to @kjsman ❤\n",
        "    image = pipeline.generate(prompts=[prompt], uncond_prompts=[negative_prompt],\n",
        "                    input_images=None, strength=0.8,\n",
        "                    do_cfg=True, cfg_scale=guidance_scale,\n",
        "                    height=height, width=width, sampler=\"k_lms\",\n",
        "                    n_inference_steps=num_inference_steps, seed=None,\n",
        "                    models=models, device='cuda', idle_device='cpu')[0]\n",
        "\n",
        "    if(suffix == 'png'):\n",
        "      image.save(f\"{root_folder}/{image_folder:04}/{name:04}.{suffix}\", pnginfo=metadata)\n",
        "    else:\n",
        "      zeroth_ifd = {piexif.ImageIFD.ImageDescription: f\"{fold(prompt)}\", piexif.ImageIFD.Make: f\"{fold(by)}\", piexif.ImageIFD.Model: f\"{ckpt}\", piexif.ImageIFD.Copyright: f\"Attribution 4.0 International (CC BY 4.0)\"}\n",
        "      exif_dict = {\"0th\": zeroth_ifd}\n",
        "      exif_bytes = piexif.dump(exif_dict)\n",
        "      image.save(f\"{root_folder}/{image_folder:04}/{name:04}.{suffix}\", \"JPEG\", quality=70, exif=exif_bytes)\n",
        "    files = {f\"{image_folder:04}_{name:04}.{suffix}\": open(f\"{root_folder}/{image_folder:04}/{name:04}.{suffix}\", \"rb\").read()}\n",
        "    payload = {\"content\": f\"{prompt}\\nSteps: {num_inference_steps}, Sampler: {sampler}, CFG scale: {guidance_scale}, Seed: {real_seed}, Size: {width}x{height}, Model: {ckpt} - {discord_user}\"}\n",
        "    requests.post(f\"https://discord.com/api/v9/channels/{discord_channel_id}/messages\", data=payload, headers={\"authorization\": f\"Bot {discord_token}\"}, files=files)\n",
        "    os.remove(f\"{root_folder}/{image_folder:04}/{name:04}.{suffix}\")"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "ut2HKwJ7FlqR"
      },
      "outputs": [],
      "source": [
        "with open(\"template.json\", \"r\") as file:\n",
        "    prompts = file.readlines()\n",
        "for prompt in prompts:\n",
        "    d = json.loads(prompt)\n",
        "    name += 1\n",
        "    generate(d[\"discord_token\"], d[\"discord_channel_id\"], d[\"ckpt\"], \"camenduru\", d[\"by\"], d[\"num_inference_steps\"], d[\"guidance_scale\"], d[\"sampler\"], d[\"width\"], d[\"height\"], d[\"prompt\"], d[\"negative_prompt\"], d[\"suffix\"], image_folder, name)"
      ]
    }
  ],
  "metadata": {
    "accelerator": "GPU",
    "colab": {
      "provenance": []
    },
    "gpuClass": "standard",
    "kernelspec": {
      "display_name": "Python 3",
      "language": "python",
      "name": "python3"
    },
    "language_info": {
      "codemirror_mode": {
        "name": "ipython",
        "version": 3
      },
      "file_extension": ".py",
      "mimetype": "text/x-python",
      "name": "python",
      "nbconvert_exporter": "python",
      "pygments_lexer": "ipython3",
      "version": "3.10.8"
    },
    "orig_nbformat": 4
  },
  "nbformat": 4,
  "nbformat_minor": 0
}