{
  "model_version": "stabilityai/stable-diffusion-2-1-base",
  "single_layer": {
    "1": {
      "time_embedding.linear_1": [
        60.6,
        60.5,
        60.4,
        60.4
      ],
      "time_embedding.linear_2": [
        60.0,
        60.2,
        59.9,
        59.9
      ],
      "down_blocks.0.attentions.0.proj_in": [
        73.0,
        74.4,
        74.5,
        72.7
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        79.6,
        79.6,
        76.6,
        78.0
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        80.8,
        80.7,
        78.5,
        79.9
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        73.9,
        75.9,
        74.8,
        73.5
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        76.2,
        76.0,
        76.6,
        74.2
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        83.2,
        81.8,
        84.0,
        83.6
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        77.3,
        76.8,
        77.5,
        75.6
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        83.9,
        83.6,
        84.4,
        83.8
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        84.2,
        84.2,
        83.2,
        82.8
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        76.1,
        76.2,
        76.2,
        71.5
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        74.4,
        77.8,
        77.4,
        74.8
      ],
      "down_blocks.0.attentions.0.proj_out": [
        74.9,
        75.6,
        75.5,
        74.3
      ],
      "down_blocks.0.attentions.1.proj_in": [
        73.5,
        73.5,
        74.3,
        72.9
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        77.0,
        75.1,
        76.8,
        75.4
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        76.8,
        76.3,
        77.8,
        74.9
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        76.3,
        75.1,
        77.1,
        75.6
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        75.8,
        76.3,
        75.8,
        74.4
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        80.2,
        79.6,
        78.4,
        73.4
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        79.3,
        78.2,
        76.9,
        75.1
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        82.6,
        81.9,
        77.4,
        74.9
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        82.8,
        82.4,
        77.2,
        79.8
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        72.6,
        73.9,
        73.1,
        73.0
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        76.6,
        76.8,
        76.1,
        75.6
      ],
      "down_blocks.0.attentions.1.proj_out": [
        75.2,
        75.5,
        76.0,
        72.5
      ],
      "down_blocks.0.resnets.0.conv1": [
        74.5,
        75.7,
        75.5,
        74.6
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        73.5,
        73.5,
        73.7,
        70.4
      ],
      "down_blocks.0.resnets.0.conv2": [
        70.5,
        70.3,
        70.3,
        69.5
      ],
      "down_blocks.0.resnets.1.conv1": [
        73.2,
        73.2,
        73.3,
        72.4
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        76.3,
        76.6,
        77.0,
        75.9
      ],
      "down_blocks.0.resnets.1.conv2": [
        75.7,
        73.9,
        75.7,
        74.7
      ],
      "down_blocks.0.downsamplers.0.conv": [
        70.5,
        73.0,
        72.6,
        71.2
      ],
      "down_blocks.1.attentions.0.proj_in": [
        75.0,
        76.2,
        76.6,
        75.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        81.5,
        81.2,
        77.2,
        78.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        82.2,
        82.0,
        82.7,
        78.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        77.0,
        78.5,
        75.2,
        71.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        75.7,
        75.2,
        74.9,
        73.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        78.1,
        80.2,
        80.7,
        75.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        70.3,
        74.3,
        73.6,
        73.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        83.2,
        82.8,
        76.9,
        78.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        83.9,
        84.1,
        86.5,
        74.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        74.0,
        76.3,
        76.4,
        73.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        77.5,
        78.7,
        78.6,
        75.4
      ],
      "down_blocks.1.attentions.0.proj_out": [
        75.6,
        77.3,
        76.7,
        75.7
      ],
      "down_blocks.1.attentions.1.proj_in": [
        77.7,
        80.3,
        70.4,
        71.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        82.5,
        82.1,
        78.1,
        75.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        83.0,
        82.0,
        77.4,
        75.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        80.3,
        80.8,
        80.1,
        77.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        80.3,
        80.2,
        77.1,
        76.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        82.8,
        82.9,
        76.2,
        74.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        75.9,
        76.2,
        72.1,
        75.3
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        85.7,
        86.4,
        84.7,
        84.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        87.1,
        87.2,
        87.1,
        79.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        77.6,
        80.4,
        79.8,
        76.3
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        77.8,
        83.4,
        83.7,
        81.5
      ],
      "down_blocks.1.attentions.1.proj_out": [
        80.8,
        82.8,
        82.7,
        79.9
      ],
      "down_blocks.1.resnets.0.conv1": [
        80.0,
        81.1,
        76.5,
        77.0
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        78.5,
        79.3,
        79.6,
        76.7
      ],
      "down_blocks.1.resnets.0.conv2": [
        76.9,
        79.7,
        81.1,
        77.6
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        72.9,
        72.5,
        74.4,
        72.7
      ],
      "down_blocks.1.resnets.1.conv1": [
        81.2,
        82.1,
        77.1,
        77.3
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        82.0,
        82.5,
        83.8,
        78.2
      ],
      "down_blocks.1.resnets.1.conv2": [
        81.4,
        83.1,
        78.8,
        76.2
      ],
      "down_blocks.1.downsamplers.0.conv": [
        78.5,
        78.9,
        81.8,
        76.3
      ],
      "down_blocks.2.attentions.0.proj_in": [
        84.3,
        84.7,
        78.8,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        86.5,
        87.2,
        80.1,
        73.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        87.0,
        87.3,
        76.5,
        73.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        80.6,
        84.2,
        83.1,
        77.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        82.1,
        82.9,
        81.6,
        77.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        80.1,
        80.8,
        82.2,
        75.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        75.1,
        76.5,
        78.0,
        75.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        83.3,
        83.9,
        78.0,
        74.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        85.6,
        85.4,
        86.4,
        74.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        80.8,
        81.8,
        82.0,
        75.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        83.4,
        84.2,
        81.4,
        80.7
      ],
      "down_blocks.2.attentions.0.proj_out": [
        84.0,
        83.4,
        82.2,
        73.2
      ],
      "down_blocks.2.attentions.1.proj_in": [
        83.1,
        84.6,
        80.3,
        78.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        87.9,
        87.7,
        79.0,
        86.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        87.8,
        87.9,
        77.0,
        86.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        81.5,
        81.9,
        79.0,
        83.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        83.9,
        84.1,
        78.3,
        78.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        81.1,
        79.9,
        82.0,
        76.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        79.8,
        79.8,
        76.9,
        80.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        84.5,
        83.7,
        84.6,
        73.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        87.9,
        88.1,
        87.4,
        86.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        81.4,
        82.3,
        76.5,
        77.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        84.9,
        84.8,
        79.4,
        84.8
      ],
      "down_blocks.2.attentions.1.proj_out": [
        83.1,
        83.1,
        79.0,
        78.4
      ],
      "down_blocks.2.resnets.0.conv1": [
        83.0,
        81.7,
        84.1,
        77.8
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        84.8,
        83.7,
        86.5,
        78.1
      ],
      "down_blocks.2.resnets.0.conv2": [
        77.8,
        76.8,
        84.8,
        78.1
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        78.9,
        81.1,
        82.0,
        75.7
      ],
      "down_blocks.2.resnets.1.conv1": [
        85.8,
        87.1,
        82.0,
        78.3
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        85.9,
        85.9,
        87.1,
        74.1
      ],
      "down_blocks.2.resnets.1.conv2": [
        84.6,
        83.8,
        86.5,
        83.3
      ],
      "down_blocks.2.downsamplers.0.conv": [
        86.5,
        86.7,
        87.5,
        78.9
      ],
      "down_blocks.3.resnets.0.conv1": [
        87.2,
        87.4,
        87.5,
        83.4
      ],
      "down_blocks.3.resnets.0.time_emb_proj": [
        88.3,
        88.4,
        88.5,
        88.2
      ],
      "down_blocks.3.resnets.0.conv2": [
        86.6,
        86.5,
        86.7,
        86.0
      ],
      "down_blocks.3.resnets.1.conv1": [
        88.1,
        88.1,
        87.5,
        86.9
      ],
      "down_blocks.3.resnets.1.time_emb_proj": [
        88.6,
        89.0,
        88.5,
        88.5
      ],
      "down_blocks.3.resnets.1.conv2": [
        87.5,
        88.0,
        87.8,
        84.8
      ],
      "up_blocks.0.resnets.0.conv1": [
        86.5,
        86.6,
        87.4,
        87.2
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        88.6,
        88.4,
        88.9,
        88.5
      ],
      "up_blocks.0.resnets.0.conv2": [
        84.8,
        86.4,
        86.2,
        85.8
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        88.0,
        88.3,
        88.2,
        83.7
      ],
      "up_blocks.0.resnets.1.conv1": [
        86.1,
        86.5,
        87.4,
        85.5
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        88.6,
        88.7,
        88.6,
        88.8
      ],
      "up_blocks.0.resnets.1.conv2": [
        84.5,
        85.4,
        85.4,
        85.1
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        87.7,
        88.3,
        87.7,
        82.3
      ],
      "up_blocks.0.resnets.2.conv1": [
        85.3,
        86.2,
        85.8,
        85.4
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        88.3,
        88.4,
        88.4,
        88.3
      ],
      "up_blocks.0.resnets.2.conv2": [
        81.3,
        83.1,
        83.6,
        80.2
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        87.5,
        88.1,
        88.2,
        87.6
      ],
      "up_blocks.0.upsamplers.0.conv": [
        81.4,
        81.9,
        80.0,
        79.8
      ],
      "up_blocks.1.attentions.0.proj_in": [
        83.0,
        82.6,
        79.5,
        77.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        85.6,
        86.7,
        81.4,
        81.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        85.8,
        86.9,
        78.1,
        82.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        83.0,
        84.2,
        77.0,
        77.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        83.6,
        84.5,
        81.2,
        77.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        83.3,
        82.5,
        85.0,
        83.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        81.5,
        81.4,
        83.8,
        79.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        85.2,
        85.6,
        86.7,
        81.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        85.4,
        84.9,
        87.4,
        87.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        82.0,
        83.3,
        82.7,
        79.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        84.0,
        83.9,
        79.6,
        77.3
      ],
      "up_blocks.1.attentions.0.proj_out": [
        83.6,
        84.1,
        82.2,
        83.0
      ],
      "up_blocks.1.attentions.1.proj_in": [
        83.2,
        82.5,
        84.6,
        73.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        87.3,
        87.3,
        82.3,
        78.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        87.4,
        87.7,
        80.2,
        79.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        84.1,
        84.3,
        81.7,
        77.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        84.6,
        84.1,
        80.6,
        77.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        82.1,
        82.2,
        83.6,
        80.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        77.9,
        80.0,
        80.7,
        79.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        83.7,
        84.4,
        84.8,
        82.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        87.4,
        88.0,
        88.4,
        86.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        82.9,
        82.9,
        84.2,
        81.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        84.7,
        84.8,
        85.9,
        79.9
      ],
      "up_blocks.1.attentions.1.proj_out": [
        84.1,
        84.6,
        85.2,
        83.5
      ],
      "up_blocks.1.attentions.2.proj_in": [
        80.7,
        81.3,
        82.2,
        75.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        85.8,
        86.0,
        86.9,
        80.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        86.5,
        86.6,
        87.2,
        84.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        83.6,
        84.2,
        85.3,
        83.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        83.1,
        83.1,
        84.0,
        81.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        81.2,
        81.3,
        82.5,
        81.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        78.3,
        78.7,
        80.5,
        79.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        84.2,
        83.6,
        84.6,
        82.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        85.5,
        85.6,
        86.1,
        85.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        82.7,
        82.4,
        83.5,
        79.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        84.5,
        84.6,
        85.2,
        82.0
      ],
      "up_blocks.1.attentions.2.proj_out": [
        80.6,
        80.7,
        81.7,
        77.3
      ],
      "up_blocks.1.resnets.0.conv1": [
        83.4,
        84.1,
        84.7,
        81.3
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        84.2,
        84.4,
        85.5,
        84.8
      ],
      "up_blocks.1.resnets.0.conv2": [
        82.9,
        83.3,
        84.6,
        78.3
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        82.9,
        79.6,
        84.0,
        77.6
      ],
      "up_blocks.1.resnets.1.conv1": [
        85.0,
        85.1,
        81.9,
        79.4
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        86.2,
        86.6,
        86.9,
        85.7
      ],
      "up_blocks.1.resnets.1.conv2": [
        85.1,
        85.4,
        81.5,
        77.4
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        82.4,
        82.2,
        83.1,
        77.9
      ],
      "up_blocks.1.resnets.2.conv1": [
        83.4,
        83.1,
        84.2,
        81.0
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        86.3,
        86.3,
        86.9,
        86.5
      ],
      "up_blocks.1.resnets.2.conv2": [
        84.0,
        85.0,
        85.1,
        80.8
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        81.1,
        82.4,
        81.6,
        79.0
      ],
      "up_blocks.1.upsamplers.0.conv": [
        81.1,
        82.2,
        81.8,
        79.3
      ],
      "up_blocks.2.attentions.0.proj_in": [
        78.0,
        78.4,
        76.0,
        68.1
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        84.7,
        84.4,
        85.0,
        82.8
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        84.6,
        84.0,
        85.0,
        83.0
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        81.5,
        81.2,
        82.0,
        80.4
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        81.4,
        81.0,
        83.0,
        80.5
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        80.7,
        80.5,
        82.3,
        81.2
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        78.1,
        77.7,
        78.0,
        78.0
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        85.2,
        85.3,
        85.4,
        84.2
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        83.8,
        84.1,
        84.4,
        83.0
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        81.0,
        80.9,
        81.7,
        78.3
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        82.8,
        83.0,
        83.6,
        79.5
      ],
      "up_blocks.2.attentions.0.proj_out": [
        75.9,
        76.1,
        76.6,
        69.0
      ],
      "up_blocks.2.attentions.1.proj_in": [
        77.0,
        77.0,
        77.7,
        71.8
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        81.5,
        81.4,
        82.3,
        79.1
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        81.1,
        81.3,
        81.8,
        79.0
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        77.4,
        77.3,
        77.6,
        76.2
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        77.8,
        78.1,
        78.3,
        76.8
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        80.7,
        80.7,
        81.1,
        79.9
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        74.3,
        74.7,
        76.7,
        75.9
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        83.5,
        84.9,
        84.5,
        83.6
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        87.8,
        87.8,
        88.3,
        88.0
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        76.7,
        75.7,
        78.6,
        77.1
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        78.4,
        78.0,
        79.9,
        78.6
      ],
      "up_blocks.2.attentions.1.proj_out": [
        77.4,
        76.9,
        77.9,
        72.2
      ],
      "up_blocks.2.attentions.2.proj_in": [
        72.6,
        74.1,
        73.3,
        71.9
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q": [
        80.1,
        80.3,
        80.8,
        79.3
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k": [
        80.5,
        80.6,
        81.2,
        79.5
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v": [
        75.0,
        75.7,
        75.9,
        74.1
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        74.2,
        75.4,
        74.8,
        73.8
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q": [
        78.1,
        78.6,
        79.0,
        78.4
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k": [
        74.9,
        75.5,
        76.7,
        76.7
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v": [
        79.4,
        79.5,
        80.6,
        80.2
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        79.0,
        80.7,
        80.3,
        79.8
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        75.2,
        75.8,
        76.6,
        74.2
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2": [
        77.8,
        78.3,
        78.3,
        76.4
      ],
      "up_blocks.2.attentions.2.proj_out": [
        74.4,
        75.3,
        75.3,
        69.6
      ],
      "up_blocks.2.resnets.0.conv1": [
        82.5,
        83.0,
        83.9,
        79.3
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        86.1,
        86.2,
        86.9,
        85.2
      ],
      "up_blocks.2.resnets.0.conv2": [
        81.0,
        81.7,
        81.9,
        77.4
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        70.7,
        71.0,
        71.5,
        61.3
      ],
      "up_blocks.2.resnets.1.conv1": [
        81.4,
        82.1,
        82.4,
        78.4
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        81.0,
        81.2,
        81.1,
        79.8
      ],
      "up_blocks.2.resnets.1.conv2": [
        80.4,
        81.2,
        82.0,
        79.0
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        74.7,
        75.3,
        75.5,
        69.3
      ],
      "up_blocks.2.resnets.2.conv1": [
        79.4,
        79.9,
        80.2,
        76.2
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        75.6,
        74.7,
        76.1,
        74.3
      ],
      "up_blocks.2.resnets.2.conv2": [
        77.9,
        77.6,
        78.7,
        77.4
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        72.6,
        73.1,
        72.9,
        69.4
      ],
      "up_blocks.2.upsamplers.0.conv": [
        72.9,
        74.1,
        73.5,
        72.0
      ],
      "up_blocks.3.attentions.0.proj_in": [
        71.7,
        72.6,
        72.1,
        70.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q": [
        73.8,
        74.5,
        74.6,
        73.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k": [
        75.1,
        75.5,
        75.9,
        74.9
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v": [
        74.4,
        74.1,
        75.3,
        74.8
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        72.0,
        72.7,
        73.2,
        71.8
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q": [
        80.1,
        79.1,
        80.9,
        80.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k": [
        74.1,
        74.6,
        72.7,
        72.9
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.8,
        81.4,
        82.2,
        81.3
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        77.4,
        77.0,
        78.0,
        77.0
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        74.1,
        74.7,
        74.4,
        73.3
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2": [
        75.8,
        76.2,
        76.1,
        75.6
      ],
      "up_blocks.3.attentions.0.proj_out": [
        74.1,
        74.2,
        74.4,
        73.3
      ],
      "up_blocks.3.attentions.1.proj_in": [
        67.9,
        68.0,
        68.4,
        67.5
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q": [
        73.0,
        73.2,
        74.0,
        74.1
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k": [
        71.2,
        71.3,
        72.3,
        72.4
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v": [
        69.5,
        69.5,
        70.5,
        70.4
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        65.2,
        64.9,
        66.0,
        65.4
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q": [
        76.7,
        76.1,
        76.2,
        76.7
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k": [
        75.9,
        74.7,
        75.7,
        76.0
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v": [
        81.8,
        81.4,
        82.1,
        82.0
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        80.1,
        79.3,
        80.2,
        80.1
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        69.5,
        69.5,
        70.2,
        69.5
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2": [
        72.0,
        71.9,
        72.5,
        71.9
      ],
      "up_blocks.3.attentions.1.proj_out": [
        70.7,
        70.9,
        71.3,
        70.6
      ],
      "up_blocks.3.attentions.2.proj_in": [
        63.3,
        63.6,
        63.4,
        63.3
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q": [
        74.5,
        74.5,
        75.3,
        74.4
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k": [
        72.9,
        72.9,
        73.3,
        72.6
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v": [
        68.2,
        67.7,
        68.9,
        68.2
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        69.5,
        69.2,
        69.4,
        69.1
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q": [
        74.2,
        74.1,
        75.8,
        75.2
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k": [
        71.2,
        73.6,
        70.2,
        70.6
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v": [
        75.0,
        75.5,
        75.4,
        74.4
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        75.4,
        75.5,
        75.9,
        74.8
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        65.9,
        65.6,
        65.8,
        65.8
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2": [
        70.5,
        70.4,
        70.7,
        70.5
      ],
      "up_blocks.3.attentions.2.proj_out": [
        69.2,
        68.9,
        69.3,
        69.2
      ],
      "up_blocks.3.resnets.0.conv1": [
        71.4,
        71.9,
        72.5,
        71.7
      ],
      "up_blocks.3.resnets.0.time_emb_proj": [
        79.4,
        80.0,
        79.8,
        79.6
      ],
      "up_blocks.3.resnets.0.conv2": [
        72.3,
        72.7,
        73.2,
        72.3
      ],
      "up_blocks.3.resnets.0.conv_shortcut": [
        60.6,
        60.9,
        60.7,
        53.1
      ],
      "up_blocks.3.resnets.1.conv1": [
        68.0,
        68.2,
        68.1,
        66.9
      ],
      "up_blocks.3.resnets.1.time_emb_proj": [
        71.5,
        71.9,
        71.8,
        71.5
      ],
      "up_blocks.3.resnets.1.conv2": [
        65.9,
        66.0,
        66.1,
        65.6
      ],
      "up_blocks.3.resnets.1.conv_shortcut": [
        65.3,
        66.0,
        65.6,
        65.2
      ],
      "up_blocks.3.resnets.2.conv1": [
        66.4,
        66.8,
        66.8,
        66.3
      ],
      "up_blocks.3.resnets.2.time_emb_proj": [
        63.5,
        63.6,
        63.4,
        63.4
      ],
      "up_blocks.3.resnets.2.conv2": [
        62.8,
        62.8,
        62.7,
        62.6
      ],
      "up_blocks.3.resnets.2.conv_shortcut": [
        59.6,
        59.9,
        60.0,
        59.9
      ],
      "mid_block.attentions.0.proj_in": [
        87.9,
        88.2,
        88.4,
        87.3
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.5,
        88.5,
        88.8,
        88.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.4,
        88.7,
        88.8,
        88.5
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        87.8,
        88.4,
        88.3,
        88.4
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        88.0,
        88.3,
        88.3,
        86.6
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.2,
        88.0,
        88.5,
        87.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        86.9,
        87.1,
        87.2,
        85.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.3,
        88.0,
        88.1,
        88.4
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.8,
        88.6,
        88.9,
        88.6
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        87.1,
        87.0,
        87.5,
        86.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        87.6,
        87.7,
        87.8,
        87.9
      ],
      "mid_block.attentions.0.proj_out": [
        88.4,
        88.4,
        88.2,
        86.4
      ],
      "mid_block.resnets.0.conv1": [
        87.9,
        88.1,
        87.6,
        87.0
      ],
      "mid_block.resnets.0.time_emb_proj": [
        88.6,
        88.4,
        88.7,
        87.7
      ],
      "mid_block.resnets.0.conv2": [
        87.7,
        87.8,
        88.1,
        81.7
      ],
      "mid_block.resnets.1.conv1": [
        87.9,
        87.9,
        88.3,
        80.7
      ],
      "mid_block.resnets.1.time_emb_proj": [
        88.7,
        88.6,
        88.7,
        88.9
      ],
      "mid_block.resnets.1.conv2": [
        88.0,
        88.1,
        87.3,
        79.7
      ]
    },
    "2": {
      "time_embedding.linear_1": [
        72.6,
        72.8,
        72.8,
        72.6
      ],
      "time_embedding.linear_2": [
        80.7,
        80.1,
        80.6,
        80.2
      ],
      "down_blocks.0.attentions.0.proj_in": [
        79.0,
        78.7,
        75.9,
        76.1
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        84.7,
        84.6,
        77.4,
        79.0
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        83.3,
        83.4,
        83.8,
        73.6
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        79.3,
        80.0,
        79.8,
        76.3
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        79.7,
        80.0,
        77.0,
        79.6
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        85.0,
        84.7,
        85.7,
        84.8
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        75.6,
        76.7,
        78.1,
        77.0
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        85.2,
        85.1,
        85.0,
        76.1
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        85.9,
        85.6,
        84.9,
        84.1
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        79.1,
        79.2,
        79.4,
        77.1
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        81.0,
        80.9,
        80.6,
        80.5
      ],
      "down_blocks.0.attentions.0.proj_out": [
        79.1,
        79.3,
        79.2,
        76.1
      ],
      "down_blocks.0.attentions.1.proj_in": [
        77.2,
        77.1,
        77.5,
        72.1
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        82.2,
        82.2,
        82.9,
        81.6
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        80.8,
        81.4,
        82.2,
        80.6
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        80.7,
        80.7,
        76.5,
        75.5
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        78.9,
        79.2,
        74.5,
        76.7
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        84.6,
        84.4,
        83.7,
        78.3
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        78.3,
        77.3,
        78.0,
        75.6
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        83.6,
        82.5,
        83.6,
        74.0
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        83.7,
        83.9,
        84.5,
        83.7
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        78.2,
        78.4,
        76.4,
        76.1
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        80.2,
        80.3,
        79.8,
        78.5
      ],
      "down_blocks.0.attentions.1.proj_out": [
        79.2,
        79.3,
        79.8,
        73.5
      ],
      "down_blocks.0.resnets.0.conv1": [
        80.0,
        80.1,
        80.2,
        79.3
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        81.7,
        81.9,
        79.2,
        80.0
      ],
      "down_blocks.0.resnets.0.conv2": [
        75.7,
        75.9,
        76.4,
        75.1
      ],
      "down_blocks.0.resnets.1.conv1": [
        77.7,
        78.3,
        78.6,
        76.0
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        77.9,
        81.1,
        76.7,
        73.5
      ],
      "down_blocks.0.resnets.1.conv2": [
        79.1,
        79.5,
        79.8,
        73.7
      ],
      "down_blocks.0.downsamplers.0.conv": [
        76.8,
        76.8,
        77.4,
        76.0
      ],
      "down_blocks.1.attentions.0.proj_in": [
        80.7,
        81.8,
        82.2,
        80.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        85.5,
        85.4,
        85.7,
        75.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        85.8,
        85.5,
        85.8,
        73.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        82.4,
        83.0,
        81.7,
        81.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        80.4,
        79.4,
        78.5,
        78.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        85.3,
        85.4,
        86.0,
        73.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        73.3,
        75.5,
        73.6,
        73.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        84.4,
        84.9,
        85.2,
        80.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        86.1,
        86.5,
        88.1,
        84.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        81.0,
        81.2,
        75.9,
        78.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        81.4,
        81.9,
        82.3,
        75.5
      ],
      "down_blocks.1.attentions.0.proj_out": [
        77.0,
        80.4,
        82.6,
        77.1
      ],
      "down_blocks.1.attentions.1.proj_in": [
        82.1,
        82.4,
        77.3,
        77.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        86.6,
        85.6,
        78.6,
        76.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        86.8,
        85.9,
        77.0,
        73.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        82.4,
        83.1,
        77.9,
        73.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        82.2,
        83.0,
        83.5,
        78.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        87.2,
        87.3,
        87.2,
        80.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        75.3,
        79.6,
        76.6,
        77.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        87.5,
        87.4,
        77.1,
        85.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.1,
        88.0,
        88.1,
        87.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        80.4,
        83.4,
        79.1,
        77.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        83.9,
        85.6,
        78.1,
        83.9
      ],
      "down_blocks.1.attentions.1.proj_out": [
        83.9,
        84.6,
        85.2,
        78.2
      ],
      "down_blocks.1.resnets.0.conv1": [
        82.9,
        83.5,
        84.6,
        82.9
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        78.9,
        79.7,
        79.9,
        75.0
      ],
      "down_blocks.1.resnets.0.conv2": [
        80.9,
        83.0,
        84.3,
        78.3
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        75.0,
        73.4,
        74.7,
        74.9
      ],
      "down_blocks.1.resnets.1.conv1": [
        84.5,
        85.0,
        85.4,
        78.1
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        83.7,
        84.1,
        85.0,
        78.4
      ],
      "down_blocks.1.resnets.1.conv2": [
        85.0,
        85.5,
        77.8,
        77.6
      ],
      "down_blocks.1.downsamplers.0.conv": [
        83.0,
        82.4,
        83.8,
        78.0
      ],
      "down_blocks.2.attentions.0.proj_in": [
        86.6,
        86.4,
        79.5,
        78.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        87.9,
        88.0,
        77.4,
        76.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.2,
        88.3,
        87.5,
        77.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        86.5,
        85.6,
        86.8,
        75.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        83.6,
        85.1,
        85.7,
        76.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        84.9,
        84.9,
        86.0,
        73.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        75.7,
        77.9,
        79.1,
        75.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        85.6,
        86.0,
        85.5,
        85.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        87.3,
        87.0,
        85.6,
        79.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        84.5,
        84.8,
        77.7,
        73.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        86.2,
        86.2,
        86.6,
        85.2
      ],
      "down_blocks.2.attentions.0.proj_out": [
        85.8,
        86.9,
        78.0,
        72.8
      ],
      "down_blocks.2.attentions.1.proj_in": [
        85.6,
        86.6,
        78.0,
        78.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.2,
        88.3,
        87.5,
        87.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        88.5,
        88.3,
        78.7,
        87.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        85.7,
        86.4,
        86.6,
        85.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        87.0,
        86.3,
        86.9,
        85.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        86.1,
        85.3,
        77.1,
        78.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        80.6,
        79.2,
        76.4,
        76.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        86.8,
        86.6,
        87.4,
        73.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.3,
        88.3,
        88.7,
        85.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        85.1,
        85.5,
        87.0,
        78.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        87.3,
        87.3,
        87.3,
        74.4
      ],
      "down_blocks.2.attentions.1.proj_out": [
        86.0,
        86.4,
        78.0,
        78.3
      ],
      "down_blocks.2.resnets.0.conv1": [
        84.4,
        83.9,
        84.4,
        78.1
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        86.8,
        86.2,
        87.7,
        74.5
      ],
      "down_blocks.2.resnets.0.conv2": [
        86.3,
        86.5,
        87.4,
        78.4
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        82.4,
        82.5,
        82.1,
        78.0
      ],
      "down_blocks.2.resnets.1.conv1": [
        87.6,
        87.7,
        87.6,
        85.5
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        86.9,
        87.1,
        87.5,
        74.1
      ],
      "down_blocks.2.resnets.1.conv2": [
        86.5,
        87.1,
        87.4,
        85.7
      ],
      "down_blocks.2.downsamplers.0.conv": [
        87.6,
        87.6,
        87.6,
        81.9
      ],
      "down_blocks.3.resnets.0.conv1": [
        87.5,
        87.9,
        87.8,
        83.3
      ],
      "down_blocks.3.resnets.0.time_emb_proj": [
        88.8,
        88.8,
        88.7,
        89.0
      ],
      "down_blocks.3.resnets.0.conv2": [
        87.8,
        88.2,
        88.3,
        87.8
      ],
      "down_blocks.3.resnets.1.conv1": [
        88.6,
        88.8,
        87.9,
        88.4
      ],
      "down_blocks.3.resnets.1.time_emb_proj": [
        88.9,
        88.8,
        88.9,
        88.7
      ],
      "down_blocks.3.resnets.1.conv2": [
        88.3,
        88.6,
        88.2,
        87.9
      ],
      "up_blocks.0.resnets.0.conv1": [
        87.1,
        87.8,
        88.0,
        87.9
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        88.7,
        88.9,
        88.9,
        88.9
      ],
      "up_blocks.0.resnets.0.conv2": [
        87.3,
        88.2,
        87.9,
        87.0
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        88.5,
        88.7,
        88.5,
        84.8
      ],
      "up_blocks.0.resnets.1.conv1": [
        88.1,
        88.2,
        88.2,
        88.2
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        89.0,
        88.7,
        88.7,
        88.9
      ],
      "up_blocks.0.resnets.1.conv2": [
        88.4,
        88.3,
        87.8,
        88.0
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        88.3,
        88.6,
        88.3,
        85.4
      ],
      "up_blocks.0.resnets.2.conv1": [
        87.1,
        87.4,
        86.7,
        77.8
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        88.9,
        88.9,
        88.8,
        88.8
      ],
      "up_blocks.0.resnets.2.conv2": [
        82.8,
        84.5,
        84.6,
        78.3
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        87.8,
        88.4,
        87.9,
        86.6
      ],
      "up_blocks.0.upsamplers.0.conv": [
        83.1,
        83.5,
        78.5,
        84.6
      ],
      "up_blocks.1.attentions.0.proj_in": [
        83.4,
        84.4,
        82.0,
        79.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        87.1,
        87.7,
        87.1,
        86.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        87.8,
        87.9,
        86.3,
        87.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        86.5,
        86.4,
        86.8,
        81.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        85.5,
        86.5,
        80.3,
        78.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        86.8,
        86.4,
        87.1,
        86.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.7,
        81.3,
        78.0,
        73.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        87.6,
        87.5,
        88.1,
        86.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        87.9,
        87.8,
        88.6,
        87.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        86.2,
        85.5,
        81.0,
        73.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        86.8,
        85.9,
        80.9,
        85.4
      ],
      "up_blocks.1.attentions.0.proj_out": [
        86.3,
        86.4,
        86.3,
        85.9
      ],
      "up_blocks.1.attentions.1.proj_in": [
        84.9,
        84.6,
        85.0,
        78.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.2,
        87.8,
        78.9,
        78.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        88.4,
        88.4,
        81.7,
        79.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        86.3,
        86.3,
        85.0,
        80.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        86.5,
        85.9,
        85.4,
        81.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        86.3,
        86.1,
        86.8,
        85.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        79.5,
        80.4,
        81.8,
        79.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        87.1,
        87.3,
        87.6,
        86.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.4,
        88.7,
        88.5,
        88.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        86.0,
        86.0,
        86.2,
        84.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        86.9,
        86.9,
        87.1,
        83.3
      ],
      "up_blocks.1.attentions.1.proj_out": [
        86.0,
        86.4,
        87.0,
        85.4
      ],
      "up_blocks.1.attentions.2.proj_in": [
        82.2,
        83.1,
        83.7,
        76.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        86.4,
        87.0,
        87.3,
        82.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        88.2,
        87.9,
        88.2,
        87.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        85.8,
        85.9,
        86.9,
        85.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        84.0,
        83.4,
        85.4,
        82.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        85.1,
        85.6,
        86.0,
        85.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        79.4,
        79.8,
        80.9,
        79.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        86.7,
        87.0,
        87.1,
        86.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        85.2,
        85.5,
        86.2,
        85.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        85.9,
        86.2,
        86.2,
        83.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        86.4,
        86.8,
        86.9,
        85.5
      ],
      "up_blocks.1.attentions.2.proj_out": [
        83.7,
        84.0,
        85.1,
        81.7
      ],
      "up_blocks.1.resnets.0.conv1": [
        86.7,
        86.6,
        84.0,
        78.3
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        85.7,
        85.9,
        86.6,
        86.5
      ],
      "up_blocks.1.resnets.0.conv2": [
        85.8,
        85.8,
        86.0,
        80.0
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        85.6,
        84.0,
        78.2,
        84.8
      ],
      "up_blocks.1.resnets.1.conv1": [
        86.6,
        86.4,
        80.6,
        77.2
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        87.3,
        87.7,
        87.8,
        87.0
      ],
      "up_blocks.1.resnets.1.conv2": [
        86.8,
        86.9,
        78.2,
        78.1
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        84.2,
        84.4,
        84.7,
        79.9
      ],
      "up_blocks.1.resnets.2.conv1": [
        84.2,
        84.6,
        85.6,
        82.7
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        87.3,
        87.3,
        87.6,
        87.3
      ],
      "up_blocks.1.resnets.2.conv2": [
        87.0,
        87.0,
        86.8,
        83.7
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        83.7,
        84.2,
        83.7,
        82.9
      ],
      "up_blocks.1.upsamplers.0.conv": [
        83.9,
        84.3,
        84.1,
        82.3
      ],
      "up_blocks.2.attentions.0.proj_in": [
        81.5,
        81.2,
        82.7,
        78.6
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        87.1,
        87.4,
        87.5,
        85.6
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        87.0,
        87.0,
        87.6,
        86.1
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        85.3,
        85.5,
        85.8,
        83.8
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        82.1,
        82.4,
        82.8,
        81.2
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        86.1,
        85.9,
        86.6,
        85.8
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        79.1,
        79.9,
        81.1,
        80.5
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        87.0,
        86.7,
        87.0,
        86.3
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        85.5,
        85.1,
        85.8,
        84.9
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        84.4,
        84.6,
        85.0,
        81.8
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        85.5,
        85.6,
        85.8,
        83.0
      ],
      "up_blocks.2.attentions.0.proj_out": [
        81.1,
        82.2,
        82.2,
        75.4
      ],
      "up_blocks.2.attentions.1.proj_in": [
        78.5,
        77.9,
        78.6,
        76.1
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        84.8,
        84.2,
        85.4,
        82.3
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        84.4,
        84.7,
        84.7,
        82.8
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        80.3,
        81.4,
        80.7,
        79.0
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        79.5,
        79.4,
        80.2,
        78.1
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        85.2,
        84.1,
        84.9,
        83.3
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        75.9,
        77.1,
        77.5,
        76.6
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        87.1,
        87.1,
        87.3,
        87.1
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.3,
        88.3,
        88.4,
        88.3
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        81.7,
        80.3,
        82.6,
        80.4
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        80.8,
        80.7,
        82.3,
        80.8
      ],
      "up_blocks.2.attentions.1.proj_out": [
        81.3,
        79.9,
        81.3,
        77.7
      ],
      "up_blocks.2.attentions.2.proj_in": [
        76.7,
        73.7,
        77.4,
        74.9
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q": [
        83.5,
        83.7,
        84.3,
        82.8
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k": [
        84.5,
        84.7,
        84.9,
        83.3
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v": [
        78.0,
        78.7,
        78.5,
        77.1
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        79.0,
        79.7,
        79.5,
        78.5
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q": [
        79.9,
        81.2,
        80.8,
        80.8
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k": [
        74.8,
        76.6,
        76.9,
        77.0
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v": [
        81.6,
        79.8,
        82.3,
        82.6
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        80.7,
        82.5,
        81.3,
        80.7
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        79.1,
        79.5,
        80.3,
        78.5
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2": [
        81.3,
        81.9,
        82.1,
        80.6
      ],
      "up_blocks.2.attentions.2.proj_out": [
        79.4,
        79.4,
        79.8,
        74.8
      ],
      "up_blocks.2.resnets.0.conv1": [
        85.4,
        86.1,
        86.4,
        83.8
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        86.7,
        86.8,
        87.4,
        86.0
      ],
      "up_blocks.2.resnets.0.conv2": [
        84.9,
        85.5,
        85.8,
        81.4
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        78.3,
        79.3,
        79.3,
        73.9
      ],
      "up_blocks.2.resnets.1.conv1": [
        83.6,
        84.3,
        84.4,
        81.3
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        82.3,
        82.7,
        82.9,
        81.2
      ],
      "up_blocks.2.resnets.1.conv2": [
        82.5,
        84.1,
        83.9,
        81.9
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        79.7,
        80.2,
        80.3,
        74.9
      ],
      "up_blocks.2.resnets.2.conv1": [
        82.6,
        82.7,
        83.2,
        79.9
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        77.9,
        77.1,
        78.6,
        76.6
      ],
      "up_blocks.2.resnets.2.conv2": [
        81.5,
        82.0,
        82.1,
        81.0
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        78.0,
        78.2,
        78.6,
        73.9
      ],
      "up_blocks.2.upsamplers.0.conv": [
        77.7,
        77.9,
        78.0,
        76.2
      ],
      "up_blocks.3.attentions.0.proj_in": [
        75.7,
        76.3,
        76.1,
        75.1
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q": [
        79.5,
        80.2,
        80.2,
        79.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k": [
        79.6,
        79.9,
        80.2,
        79.2
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v": [
        78.4,
        78.7,
        79.2,
        78.6
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        77.0,
        77.4,
        77.8,
        76.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q": [
        84.7,
        84.6,
        84.8,
        84.6
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k": [
        78.7,
        79.0,
        77.9,
        77.4
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v": [
        84.1,
        84.3,
        84.6,
        84.2
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        79.8,
        79.8,
        80.2,
        79.1
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        77.9,
        78.2,
        78.3,
        77.4
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2": [
        79.6,
        80.1,
        80.1,
        79.4
      ],
      "up_blocks.3.attentions.0.proj_out": [
        78.4,
        78.4,
        78.5,
        77.7
      ],
      "up_blocks.3.attentions.1.proj_in": [
        72.9,
        72.8,
        72.9,
        72.4
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q": [
        79.9,
        79.9,
        80.5,
        80.5
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k": [
        78.4,
        78.7,
        79.9,
        79.7
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v": [
        75.7,
        75.6,
        76.4,
        76.0
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        74.0,
        73.8,
        74.8,
        74.3
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q": [
        83.7,
        83.4,
        84.7,
        84.4
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k": [
        79.3,
        76.9,
        78.9,
        78.8
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v": [
        83.1,
        82.8,
        83.7,
        83.8
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        81.6,
        81.1,
        81.3,
        81.3
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        74.7,
        74.9,
        75.6,
        74.9
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2": [
        77.1,
        77.4,
        77.6,
        76.7
      ],
      "up_blocks.3.attentions.1.proj_out": [
        75.5,
        75.7,
        76.0,
        75.3
      ],
      "up_blocks.3.attentions.2.proj_in": [
        67.7,
        67.6,
        68.0,
        67.4
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q": [
        78.8,
        78.9,
        79.6,
        78.8
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k": [
        77.2,
        77.4,
        77.9,
        77.0
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v": [
        74.6,
        74.1,
        75.2,
        74.6
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        73.3,
        72.8,
        73.7,
        73.5
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q": [
        80.3,
        79.7,
        80.7,
        80.0
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k": [
        75.5,
        76.2,
        76.0,
        75.9
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v": [
        77.9,
        78.0,
        78.7,
        78.6
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        77.5,
        76.9,
        77.1,
        76.5
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        72.3,
        72.3,
        72.7,
        72.3
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2": [
        76.2,
        75.9,
        76.0,
        75.8
      ],
      "up_blocks.3.attentions.2.proj_out": [
        75.3,
        75.1,
        75.4,
        75.1
      ],
      "up_blocks.3.resnets.0.conv1": [
        76.4,
        76.4,
        77.1,
        76.1
      ],
      "up_blocks.3.resnets.0.time_emb_proj": [
        83.5,
        84.0,
        83.6,
        83.6
      ],
      "up_blocks.3.resnets.0.conv2": [
        76.9,
        77.6,
        77.7,
        76.9
      ],
      "up_blocks.3.resnets.0.conv_shortcut": [
        72.2,
        72.3,
        72.7,
        68.6
      ],
      "up_blocks.3.resnets.1.conv1": [
        73.5,
        73.6,
        73.5,
        72.1
      ],
      "up_blocks.3.resnets.1.time_emb_proj": [
        78.9,
        79.3,
        79.1,
        78.9
      ],
      "up_blocks.3.resnets.1.conv2": [
        72.0,
        71.9,
        72.0,
        71.5
      ],
      "up_blocks.3.resnets.1.conv_shortcut": [
        69.5,
        70.3,
        69.4,
        68.6
      ],
      "up_blocks.3.resnets.2.conv1": [
        73.7,
        73.5,
        74.4,
        73.6
      ],
      "up_blocks.3.resnets.2.time_emb_proj": [
        73.1,
        73.3,
        73.0,
        73.0
      ],
      "up_blocks.3.resnets.2.conv2": [
        69.7,
        69.5,
        70.0,
        69.7
      ],
      "up_blocks.3.resnets.2.conv_shortcut": [
        66.1,
        66.4,
        66.4,
        66.1
      ],
      "mid_block.attentions.0.proj_in": [
        88.5,
        88.6,
        88.5,
        88.7
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.8,
        88.9,
        88.9,
        88.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.7,
        89.0,
        88.7,
        88.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.6,
        88.9,
        88.7,
        89.0
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        88.9,
        88.7,
        88.9,
        88.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.6,
        89.2,
        89.0,
        88.7
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        87.2,
        87.8,
        87.9,
        87.6
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.5,
        88.5,
        88.5,
        88.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        89.0,
        89.0,
        88.8,
        88.7
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        88.2,
        88.3,
        88.2,
        87.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        88.6,
        88.7,
        88.8,
        88.9
      ],
      "mid_block.attentions.0.proj_out": [
        88.8,
        88.7,
        89.0,
        88.3
      ],
      "mid_block.resnets.0.conv1": [
        88.5,
        88.6,
        88.6,
        88.5
      ],
      "mid_block.resnets.0.time_emb_proj": [
        88.8,
        88.7,
        88.9,
        89.0
      ],
      "mid_block.resnets.0.conv2": [
        88.3,
        88.5,
        88.3,
        88.8
      ],
      "mid_block.resnets.1.conv1": [
        88.6,
        88.5,
        89.0,
        84.6
      ],
      "mid_block.resnets.1.time_emb_proj": [
        88.8,
        88.9,
        88.6,
        88.8
      ],
      "mid_block.resnets.1.conv2": [
        88.8,
        88.3,
        88.7,
        87.7
      ]
    },
    "4": {
      "time_embedding.linear_1": [
        87.3,
        87.5,
        87.3,
        86.4
      ],
      "time_embedding.linear_2": [
        82.3,
        82.5,
        82.4,
        81.9
      ],
      "down_blocks.0.attentions.0.proj_in": [
        83.7,
        83.6,
        84.1,
        73.4
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        87.7,
        87.8,
        87.6,
        87.1
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        87.3,
        87.5,
        87.6,
        86.6
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        85.7,
        85.9,
        85.8,
        85.4
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        85.4,
        85.5,
        77.0,
        85.0
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        87.8,
        88.1,
        88.3,
        88.4
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.6,
        83.5,
        83.1,
        82.3
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        87.4,
        87.7,
        87.2,
        86.7
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        87.9,
        87.9,
        87.6,
        86.9
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        84.9,
        85.1,
        77.9,
        82.9
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        86.0,
        85.8,
        86.1,
        78.3
      ],
      "down_blocks.0.attentions.0.proj_out": [
        84.8,
        85.0,
        84.9,
        74.0
      ],
      "down_blocks.0.attentions.1.proj_in": [
        83.3,
        83.3,
        82.5,
        82.6
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        87.3,
        87.2,
        84.4,
        87.2
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        86.6,
        86.5,
        86.8,
        80.8
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        85.8,
        86.3,
        86.2,
        73.9
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        84.6,
        84.5,
        77.3,
        73.8
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        87.2,
        87.6,
        86.8,
        86.9
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        85.6,
        85.1,
        84.7,
        78.3
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        85.7,
        85.2,
        86.6,
        78.5
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        87.1,
        87.0,
        87.0,
        87.1
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        84.6,
        85.0,
        78.3,
        81.7
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        85.5,
        85.5,
        85.7,
        78.4
      ],
      "down_blocks.0.attentions.1.proj_out": [
        85.6,
        85.6,
        84.7,
        74.6
      ],
      "down_blocks.0.resnets.0.conv1": [
        86.1,
        86.1,
        84.9,
        85.6
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        85.0,
        85.9,
        86.0,
        74.6
      ],
      "down_blocks.0.resnets.0.conv2": [
        83.2,
        83.1,
        77.8,
        73.7
      ],
      "down_blocks.0.resnets.1.conv1": [
        84.5,
        84.9,
        84.6,
        75.4
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        87.3,
        87.1,
        87.1,
        86.9
      ],
      "down_blocks.0.resnets.1.conv2": [
        84.5,
        84.7,
        84.9,
        83.5
      ],
      "down_blocks.0.downsamplers.0.conv": [
        82.6,
        82.8,
        83.0,
        81.9
      ],
      "down_blocks.1.attentions.0.proj_in": [
        86.5,
        85.6,
        86.5,
        78.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.2,
        88.0,
        87.8,
        87.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.2,
        88.2,
        88.1,
        83.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        86.5,
        86.1,
        80.2,
        73.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        86.3,
        86.4,
        77.8,
        78.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        87.9,
        87.8,
        88.1,
        87.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        78.8,
        79.3,
        80.3,
        70.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        87.7,
        87.9,
        87.8,
        86.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.6,
        88.3,
        88.3,
        88.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        86.2,
        86.3,
        86.2,
        84.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        86.4,
        86.0,
        86.5,
        83.0
      ],
      "down_blocks.1.attentions.0.proj_out": [
        86.6,
        85.9,
        86.6,
        79.1
      ],
      "down_blocks.1.attentions.1.proj_in": [
        86.5,
        87.2,
        77.1,
        75.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.6,
        88.3,
        86.5,
        86.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        88.9,
        88.3,
        88.3,
        88.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        87.3,
        87.2,
        87.9,
        85.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        86.1,
        86.9,
        86.9,
        86.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        89.0,
        88.4,
        88.1,
        88.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        81.8,
        82.8,
        76.7,
        77.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        88.3,
        88.6,
        87.3,
        83.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.7,
        88.5,
        88.8,
        87.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        87.3,
        87.9,
        87.6,
        86.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        87.1,
        87.7,
        87.5,
        86.7
      ],
      "down_blocks.1.attentions.1.proj_out": [
        87.6,
        88.0,
        87.9,
        76.1
      ],
      "down_blocks.1.resnets.0.conv1": [
        87.4,
        87.7,
        87.6,
        86.2
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        87.4,
        87.0,
        87.4,
        85.8
      ],
      "down_blocks.1.resnets.0.conv2": [
        87.5,
        87.2,
        80.2,
        73.6
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        82.7,
        83.2,
        83.7,
        77.5
      ],
      "down_blocks.1.resnets.1.conv1": [
        87.0,
        87.0,
        86.6,
        86.6
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        87.5,
        87.9,
        87.5,
        78.0
      ],
      "down_blocks.1.resnets.1.conv2": [
        88.3,
        88.0,
        88.1,
        74.4
      ],
      "down_blocks.1.downsamplers.0.conv": [
        84.5,
        84.8,
        84.2,
        78.4
      ],
      "down_blocks.2.attentions.0.proj_in": [
        87.8,
        87.7,
        88.2,
        86.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        89.0,
        88.8,
        88.8,
        88.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.8,
        88.9,
        88.7,
        88.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.4,
        88.2,
        87.7,
        88.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        86.3,
        87.3,
        86.4,
        87.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.2,
        88.5,
        88.2,
        83.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        80.7,
        81.0,
        83.3,
        79.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.6,
        88.2,
        87.0,
        87.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.5,
        88.6,
        88.3,
        87.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        88.3,
        87.8,
        87.3,
        73.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        88.3,
        88.3,
        82.5,
        77.6
      ],
      "down_blocks.2.attentions.0.proj_out": [
        88.3,
        87.9,
        87.5,
        87.5
      ],
      "down_blocks.2.attentions.1.proj_in": [
        87.6,
        87.6,
        87.8,
        86.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.9,
        88.6,
        88.1,
        88.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        89.1,
        89.2,
        88.9,
        88.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        88.1,
        87.9,
        88.2,
        75.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        87.5,
        86.9,
        88.4,
        76.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.6,
        88.5,
        87.9,
        79.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        77.1,
        81.4,
        79.5,
        74.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        89.0,
        88.6,
        87.7,
        87.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.9,
        88.8,
        88.5,
        88.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        88.4,
        88.1,
        85.3,
        86.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        88.6,
        88.7,
        88.4,
        88.1
      ],
      "down_blocks.2.attentions.1.proj_out": [
        88.3,
        88.3,
        88.3,
        87.4
      ],
      "down_blocks.2.resnets.0.conv1": [
        88.0,
        87.7,
        88.0,
        73.9
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        88.3,
        88.5,
        88.2,
        87.8
      ],
      "down_blocks.2.resnets.0.conv2": [
        88.5,
        88.3,
        87.9,
        73.7
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        85.8,
        86.0,
        83.7,
        85.3
      ],
      "down_blocks.2.resnets.1.conv1": [
        88.8,
        88.6,
        88.9,
        86.3
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        88.6,
        88.7,
        88.8,
        88.3
      ],
      "down_blocks.2.resnets.1.conv2": [
        88.5,
        88.7,
        88.3,
        87.7
      ],
      "down_blocks.2.downsamplers.0.conv": [
        88.8,
        88.8,
        88.9,
        87.6
      ],
      "down_blocks.3.resnets.0.conv1": [
        88.9,
        88.9,
        88.5,
        87.5
      ],
      "down_blocks.3.resnets.0.time_emb_proj": [
        89.3,
        88.6,
        89.2,
        88.7
      ],
      "down_blocks.3.resnets.0.conv2": [
        88.9,
        88.7,
        88.5,
        88.8
      ],
      "down_blocks.3.resnets.1.conv1": [
        88.7,
        89.1,
        88.8,
        88.5
      ],
      "down_blocks.3.resnets.1.time_emb_proj": [
        89.0,
        89.0,
        88.6,
        87.8
      ],
      "down_blocks.3.resnets.1.conv2": [
        89.1,
        89.0,
        88.8,
        88.4
      ],
      "up_blocks.0.resnets.0.conv1": [
        88.8,
        88.9,
        88.7,
        88.7
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        89.1,
        89.0,
        88.6,
        88.5
      ],
      "up_blocks.0.resnets.0.conv2": [
        89.2,
        89.0,
        89.1,
        88.8
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        89.2,
        88.8,
        88.7,
        88.5
      ],
      "up_blocks.0.resnets.1.conv1": [
        88.9,
        88.8,
        88.5,
        88.2
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        88.9,
        89.1,
        89.0,
        88.6
      ],
      "up_blocks.0.resnets.1.conv2": [
        89.1,
        88.8,
        88.7,
        88.7
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        88.9,
        88.9,
        88.8,
        88.4
      ],
      "up_blocks.0.resnets.2.conv1": [
        88.5,
        88.2,
        87.3,
        84.3
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        88.9,
        89.0,
        88.8,
        88.7
      ],
      "up_blocks.0.resnets.2.conv2": [
        88.6,
        88.6,
        88.5,
        87.9
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        88.8,
        88.6,
        88.3,
        88.7
      ],
      "up_blocks.0.upsamplers.0.conv": [
        87.6,
        87.9,
        88.0,
        85.5
      ],
      "up_blocks.1.attentions.0.proj_in": [
        86.0,
        87.2,
        87.4,
        86.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.3,
        88.4,
        88.6,
        87.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        89.1,
        88.8,
        88.7,
        88.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.5,
        88.4,
        88.4,
        87.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        88.1,
        88.1,
        86.6,
        81.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.3,
        88.9,
        88.8,
        88.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        84.0,
        82.8,
        86.1,
        85.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.7,
        88.8,
        88.6,
        88.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.6,
        88.8,
        88.6,
        88.3
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        87.5,
        87.8,
        78.7,
        78.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        88.6,
        88.4,
        88.6,
        85.7
      ],
      "up_blocks.1.attentions.0.proj_out": [
        87.8,
        88.3,
        88.0,
        88.0
      ],
      "up_blocks.1.attentions.1.proj_in": [
        87.6,
        87.4,
        87.8,
        84.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.9,
        88.3,
        87.5,
        86.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        88.9,
        89.3,
        88.6,
        88.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        88.6,
        88.0,
        88.3,
        88.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        87.7,
        87.4,
        87.8,
        86.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.3,
        88.2,
        88.4,
        88.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        82.4,
        82.8,
        83.8,
        81.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        88.7,
        88.2,
        88.5,
        88.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.9,
        88.9,
        88.6,
        88.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        88.1,
        87.6,
        87.2,
        85.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        87.8,
        88.3,
        88.4,
        87.9
      ],
      "up_blocks.1.attentions.1.proj_out": [
        88.2,
        88.3,
        88.4,
        88.1
      ],
      "up_blocks.1.attentions.2.proj_in": [
        86.9,
        87.4,
        87.4,
        85.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        87.6,
        87.9,
        87.5,
        86.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        88.8,
        88.6,
        88.5,
        88.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        88.3,
        88.4,
        88.0,
        88.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        87.7,
        87.5,
        87.8,
        87.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        87.9,
        87.9,
        88.1,
        87.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        81.4,
        82.0,
        83.1,
        81.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        88.5,
        88.1,
        88.2,
        88.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        88.2,
        87.9,
        88.0,
        88.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        88.1,
        88.1,
        88.4,
        87.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        88.3,
        88.2,
        88.4,
        88.1
      ],
      "up_blocks.1.attentions.2.proj_out": [
        87.5,
        87.6,
        87.9,
        86.4
      ],
      "up_blocks.1.resnets.0.conv1": [
        88.7,
        88.3,
        86.3,
        80.1
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        89.0,
        88.6,
        88.6,
        88.1
      ],
      "up_blocks.1.resnets.0.conv2": [
        88.2,
        88.2,
        78.6,
        73.5
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        88.1,
        87.6,
        86.7,
        84.1
      ],
      "up_blocks.1.resnets.1.conv1": [
        88.0,
        88.2,
        88.5,
        87.7
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        88.3,
        88.3,
        88.6,
        88.1
      ],
      "up_blocks.1.resnets.1.conv2": [
        88.4,
        88.5,
        88.4,
        88.1
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        86.5,
        86.2,
        86.3,
        84.3
      ],
      "up_blocks.1.resnets.2.conv1": [
        88.5,
        88.3,
        88.4,
        88.3
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        88.4,
        88.6,
        88.6,
        88.2
      ],
      "up_blocks.1.resnets.2.conv2": [
        88.4,
        88.3,
        88.3,
        87.9
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        85.9,
        86.3,
        86.2,
        85.4
      ],
      "up_blocks.1.upsamplers.0.conv": [
        87.0,
        86.9,
        86.8,
        86.1
      ],
      "up_blocks.2.attentions.0.proj_in": [
        86.2,
        86.7,
        86.7,
        85.1
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.5,
        88.5,
        88.2,
        87.8
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.6,
        88.4,
        88.2,
        88.5
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.2,
        87.9,
        88.2,
        87.7
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        86.8,
        87.1,
        86.6,
        85.5
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        87.8,
        88.3,
        88.6,
        87.6
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.8,
        83.2,
        84.2,
        83.4
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.6,
        88.2,
        88.6,
        87.9
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        87.3,
        87.0,
        87.5,
        86.4
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        87.8,
        87.7,
        87.8,
        87.2
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        88.2,
        87.8,
        88.2,
        86.6
      ],
      "up_blocks.2.attentions.0.proj_out": [
        86.5,
        86.2,
        86.7,
        84.8
      ],
      "up_blocks.2.attentions.1.proj_in": [
        84.9,
        85.0,
        85.1,
        83.0
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.0,
        87.4,
        88.0,
        87.1
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        88.2,
        88.2,
        88.1,
        87.8
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        86.8,
        86.3,
        87.0,
        86.1
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        83.3,
        82.9,
        84.4,
        82.3
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.0,
        87.6,
        88.2,
        88.4
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        82.2,
        81.6,
        82.9,
        82.7
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        88.4,
        88.1,
        88.3,
        88.2
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.9,
        88.6,
        88.8,
        88.8
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        85.8,
        86.0,
        86.0,
        84.7
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        86.4,
        86.6,
        87.1,
        86.2
      ],
      "up_blocks.2.attentions.1.proj_out": [
        86.2,
        86.0,
        86.7,
        85.0
      ],
      "up_blocks.2.attentions.2.proj_in": [
        82.2,
        82.4,
        83.3,
        82.4
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q": [
        87.8,
        87.8,
        87.9,
        87.2
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k": [
        87.9,
        87.9,
        88.0,
        87.3
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v": [
        85.1,
        85.9,
        85.5,
        84.2
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        80.9,
        82.2,
        82.6,
        81.9
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q": [
        86.4,
        86.8,
        86.7,
        86.9
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k": [
        79.1,
        80.7,
        80.9,
        81.1
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v": [
        85.2,
        86.8,
        86.2,
        86.4
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        85.5,
        85.5,
        85.9,
        85.8
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        83.7,
        82.8,
        84.1,
        82.9
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2": [
        86.4,
        86.5,
        86.6,
        85.4
      ],
      "up_blocks.2.attentions.2.proj_out": [
        84.8,
        85.1,
        85.1,
        82.8
      ],
      "up_blocks.2.resnets.0.conv1": [
        88.0,
        87.9,
        88.0,
        87.1
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        88.9,
        88.5,
        88.6,
        88.3
      ],
      "up_blocks.2.resnets.0.conv2": [
        87.6,
        87.8,
        87.9,
        86.7
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        85.7,
        86.2,
        86.2,
        84.6
      ],
      "up_blocks.2.resnets.1.conv1": [
        87.4,
        87.3,
        87.5,
        85.9
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        88.0,
        88.0,
        88.1,
        87.6
      ],
      "up_blocks.2.resnets.1.conv2": [
        87.5,
        87.5,
        87.2,
        86.7
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        83.9,
        84.4,
        83.9,
        82.4
      ],
      "up_blocks.2.resnets.2.conv1": [
        86.8,
        86.6,
        86.7,
        85.8
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        87.6,
        87.4,
        87.8,
        87.0
      ],
      "up_blocks.2.resnets.2.conv2": [
        86.9,
        86.0,
        86.8,
        86.0
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        82.7,
        82.9,
        83.3,
        79.6
      ],
      "up_blocks.2.upsamplers.0.conv": [
        83.9,
        84.6,
        84.2,
        82.2
      ],
      "up_blocks.3.attentions.0.proj_in": [
        82.6,
        83.0,
        82.9,
        82.2
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q": [
        86.2,
        86.1,
        86.4,
        85.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k": [
        86.0,
        86.0,
        86.0,
        85.6
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v": [
        85.1,
        85.5,
        85.3,
        84.9
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        83.5,
        83.8,
        83.9,
        83.0
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.0,
        88.2,
        88.3,
        88.2
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k": [
        84.9,
        83.9,
        84.3,
        83.8
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v": [
        86.6,
        86.1,
        87.1,
        86.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        86.5,
        86.2,
        86.7,
        86.3
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        84.4,
        84.7,
        84.6,
        83.9
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2": [
        84.5,
        85.0,
        85.2,
        84.5
      ],
      "up_blocks.3.attentions.0.proj_out": [
        84.2,
        84.5,
        84.8,
        84.1
      ],
      "up_blocks.3.attentions.1.proj_in": [
        80.1,
        80.4,
        80.6,
        79.8
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q": [
        86.8,
        86.2,
        86.6,
        86.2
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k": [
        86.3,
        86.0,
        86.5,
        86.3
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v": [
        83.0,
        82.9,
        83.4,
        83.2
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        81.5,
        81.7,
        82.0,
        81.3
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q": [
        87.7,
        87.7,
        88.0,
        87.7
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k": [
        85.2,
        85.1,
        84.8,
        85.3
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v": [
        87.1,
        87.2,
        87.6,
        87.4
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        85.7,
        85.8,
        85.9,
        85.7
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        81.8,
        81.9,
        82.1,
        81.7
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2": [
        83.7,
        83.9,
        84.1,
        83.6
      ],
      "up_blocks.3.attentions.1.proj_out": [
        83.1,
        83.1,
        83.3,
        83.0
      ],
      "up_blocks.3.attentions.2.proj_in": [
        78.9,
        78.7,
        79.3,
        78.9
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q": [
        84.7,
        84.5,
        85.2,
        84.8
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k": [
        84.0,
        84.0,
        84.5,
        84.1
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v": [
        82.7,
        82.8,
        83.5,
        83.2
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        82.1,
        81.9,
        82.6,
        82.2
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q": [
        87.0,
        86.9,
        87.1,
        86.7
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k": [
        84.1,
        84.2,
        84.5,
        84.7
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v": [
        83.8,
        84.0,
        84.4,
        84.4
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        82.7,
        81.9,
        82.3,
        81.8
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        80.9,
        80.8,
        81.4,
        81.0
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2": [
        84.6,
        84.7,
        84.6,
        84.6
      ],
      "up_blocks.3.attentions.2.proj_out": [
        81.6,
        81.5,
        82.1,
        81.8
      ],
      "up_blocks.3.resnets.0.conv1": [
        83.5,
        83.5,
        83.8,
        82.9
      ],
      "up_blocks.3.resnets.0.time_emb_proj": [
        87.6,
        87.7,
        87.4,
        86.9
      ],
      "up_blocks.3.resnets.0.conv2": [
        83.5,
        83.7,
        84.0,
        83.5
      ],
      "up_blocks.3.resnets.0.conv_shortcut": [
        80.8,
        81.0,
        81.0,
        77.9
      ],
      "up_blocks.3.resnets.1.conv1": [
        81.8,
        81.8,
        81.8,
        81.3
      ],
      "up_blocks.3.resnets.1.time_emb_proj": [
        84.5,
        84.7,
        85.1,
        84.8
      ],
      "up_blocks.3.resnets.1.conv2": [
        81.7,
        81.7,
        81.6,
        81.3
      ],
      "up_blocks.3.resnets.1.conv_shortcut": [
        79.1,
        79.4,
        79.7,
        79.2
      ],
      "up_blocks.3.resnets.2.conv1": [
        81.5,
        81.5,
        81.9,
        81.3
      ],
      "up_blocks.3.resnets.2.time_emb_proj": [
        82.4,
        82.3,
        82.3,
        82.3
      ],
      "up_blocks.3.resnets.2.conv2": [
        80.3,
        80.6,
        80.9,
        80.3
      ],
      "up_blocks.3.resnets.2.conv_shortcut": [
        80.1,
        80.3,
        80.4,
        80.1
      ],
      "mid_block.attentions.0.proj_in": [
        88.9,
        88.6,
        88.8,
        87.7
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        89.0,
        89.0,
        89.1,
        88.6
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.8,
        88.7,
        88.9,
        88.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.8,
        89.2,
        88.9,
        88.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        89.1,
        89.0,
        88.7,
        88.6
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        89.1,
        89.0,
        88.9,
        89.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        89.1,
        88.9,
        89.1,
        88.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        89.1,
        89.1,
        88.5,
        88.0
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.8,
        88.8,
        88.7,
        88.7
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        88.8,
        89.0,
        89.0,
        88.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        89.2,
        89.1,
        88.9,
        88.4
      ],
      "mid_block.attentions.0.proj_out": [
        89.1,
        88.9,
        88.9,
        88.6
      ],
      "mid_block.resnets.0.conv1": [
        89.0,
        89.0,
        88.7,
        88.8
      ],
      "mid_block.resnets.0.time_emb_proj": [
        88.8,
        89.1,
        88.6,
        88.3
      ],
      "mid_block.resnets.0.conv2": [
        89.0,
        88.8,
        88.7,
        89.3
      ],
      "mid_block.resnets.1.conv1": [
        89.0,
        88.7,
        88.6,
        88.6
      ],
      "mid_block.resnets.1.time_emb_proj": [
        89.1,
        89.2,
        88.8,
        88.9
      ],
      "mid_block.resnets.1.conv2": [
        88.8,
        89.0,
        88.7,
        88.8
      ]
    },
    "6": {
      "time_embedding.linear_1": [
        88.5,
        88.8,
        88.5,
        88.6
      ],
      "time_embedding.linear_2": [
        87.7,
        87.9,
        87.8,
        87.0
      ],
      "down_blocks.0.attentions.0.proj_in": [
        87.6,
        87.7,
        87.9,
        86.6
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.5,
        88.4,
        88.5,
        87.7
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.4,
        88.7,
        88.3,
        88.3
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        87.7,
        88.0,
        87.6,
        86.9
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        87.3,
        87.6,
        86.2,
        86.4
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.4,
        88.5,
        88.5,
        88.1
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        88.0,
        88.2,
        88.0,
        87.6
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.2,
        88.4,
        87.8,
        88.2
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.4,
        88.5,
        88.7,
        88.2
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        87.4,
        87.9,
        85.9,
        88.0
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        87.5,
        88.1,
        87.9,
        87.9
      ],
      "down_blocks.0.attentions.0.proj_out": [
        87.7,
        87.7,
        87.7,
        87.2
      ],
      "down_blocks.0.attentions.1.proj_in": [
        87.6,
        87.5,
        87.4,
        85.9
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.6,
        88.5,
        87.5,
        87.6
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        88.2,
        88.3,
        87.1,
        88.4
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        88.0,
        88.1,
        87.4,
        87.6
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        87.7,
        87.9,
        85.7,
        86.0
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.9,
        88.7,
        88.7,
        88.3
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        88.3,
        88.3,
        88.4,
        88.0
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        88.1,
        88.3,
        88.3,
        88.2
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.6,
        88.6,
        88.5,
        88.3
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        87.9,
        87.7,
        87.8,
        86.0
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        87.9,
        88.0,
        87.7,
        87.6
      ],
      "down_blocks.0.attentions.1.proj_out": [
        87.4,
        88.0,
        86.8,
        79.7
      ],
      "down_blocks.0.resnets.0.conv1": [
        87.9,
        88.3,
        87.5,
        83.7
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        87.9,
        88.2,
        88.2,
        87.7
      ],
      "down_blocks.0.resnets.0.conv2": [
        87.3,
        87.4,
        86.2,
        86.5
      ],
      "down_blocks.0.resnets.1.conv1": [
        87.6,
        88.0,
        86.3,
        86.4
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        88.2,
        88.5,
        88.3,
        87.5
      ],
      "down_blocks.0.resnets.1.conv2": [
        87.7,
        87.7,
        88.0,
        86.8
      ],
      "down_blocks.0.downsamplers.0.conv": [
        87.9,
        88.1,
        88.1,
        85.9
      ],
      "down_blocks.1.attentions.0.proj_in": [
        87.9,
        88.3,
        87.7,
        84.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.6,
        88.5,
        88.3,
        87.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.7,
        88.9,
        88.9,
        88.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.3,
        88.7,
        86.4,
        88.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        88.0,
        88.0,
        86.8,
        85.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.9,
        88.9,
        89.0,
        88.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        88.6,
        88.4,
        88.3,
        88.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.4,
        88.8,
        88.4,
        86.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        89.0,
        88.8,
        88.7,
        88.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        88.0,
        88.0,
        88.4,
        87.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        88.2,
        88.5,
        87.3,
        87.1
      ],
      "down_blocks.1.attentions.0.proj_out": [
        87.3,
        88.2,
        87.6,
        87.2
      ],
      "down_blocks.1.attentions.1.proj_in": [
        88.4,
        88.7,
        88.5,
        87.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.9,
        88.9,
        88.7,
        88.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        88.7,
        89.1,
        88.5,
        88.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        88.6,
        88.3,
        88.3,
        87.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        88.3,
        88.5,
        88.4,
        88.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.8,
        89.0,
        88.6,
        88.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        88.7,
        89.2,
        88.6,
        87.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        88.7,
        89.0,
        89.0,
        88.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.9,
        89.1,
        88.7,
        88.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        88.0,
        88.5,
        88.6,
        75.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        88.9,
        88.5,
        87.3,
        79.9
      ],
      "down_blocks.1.attentions.1.proj_out": [
        88.4,
        88.7,
        88.1,
        86.8
      ],
      "down_blocks.1.resnets.0.conv1": [
        88.5,
        88.5,
        87.8,
        88.3
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        88.5,
        88.3,
        88.2,
        87.8
      ],
      "down_blocks.1.resnets.0.conv2": [
        88.3,
        88.7,
        88.4,
        88.1
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        87.8,
        88.1,
        87.4,
        77.6
      ],
      "down_blocks.1.resnets.1.conv1": [
        88.2,
        88.6,
        88.2,
        87.6
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        88.6,
        88.7,
        88.5,
        87.7
      ],
      "down_blocks.1.resnets.1.conv2": [
        88.6,
        88.6,
        88.6,
        88.0
      ],
      "down_blocks.1.downsamplers.0.conv": [
        88.8,
        88.4,
        87.9,
        86.3
      ],
      "down_blocks.2.attentions.0.proj_in": [
        88.6,
        89.0,
        88.6,
        87.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.9,
        89.0,
        89.0,
        87.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.9,
        88.9,
        89.1,
        88.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.8,
        88.8,
        88.9,
        88.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        89.0,
        88.8,
        88.7,
        86.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        89.0,
        88.9,
        88.6,
        88.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        88.7,
        88.5,
        88.3,
        88.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.9,
        88.8,
        88.7,
        88.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        89.0,
        88.9,
        88.7,
        88.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        88.7,
        88.5,
        87.4,
        87.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        88.9,
        88.6,
        88.7,
        88.1
      ],
      "down_blocks.2.attentions.0.proj_out": [
        88.6,
        88.6,
        88.8,
        88.7
      ],
      "down_blocks.2.attentions.1.proj_in": [
        88.7,
        88.8,
        88.6,
        88.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.9,
        89.1,
        88.9,
        88.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        88.8,
        89.1,
        89.0,
        88.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        88.9,
        88.9,
        89.0,
        88.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        88.9,
        89.1,
        88.6,
        88.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        89.0,
        89.0,
        88.9,
        88.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        88.8,
        88.8,
        87.9,
        87.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        88.8,
        88.8,
        88.7,
        89.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        89.0,
        89.0,
        88.6,
        88.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        88.5,
        89.0,
        86.9,
        88.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        89.0,
        89.0,
        88.6,
        88.4
      ],
      "down_blocks.2.attentions.1.proj_out": [
        88.9,
        88.8,
        88.2,
        88.0
      ],
      "down_blocks.2.resnets.0.conv1": [
        88.7,
        89.1,
        88.0,
        87.9
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        89.0,
        89.3,
        89.1,
        88.8
      ],
      "down_blocks.2.resnets.0.conv2": [
        88.6,
        88.8,
        88.9,
        82.7
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        88.2,
        88.9,
        88.4,
        85.3
      ],
      "down_blocks.2.resnets.1.conv1": [
        89.1,
        88.9,
        88.8,
        88.1
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        89.0,
        89.1,
        88.9,
        89.1
      ],
      "down_blocks.2.resnets.1.conv2": [
        89.0,
        88.8,
        88.8,
        89.1
      ],
      "down_blocks.2.downsamplers.0.conv": [
        88.9,
        88.9,
        88.7,
        88.5
      ],
      "down_blocks.3.resnets.0.conv1": [
        89.1,
        89.3,
        88.7,
        89.0
      ],
      "down_blocks.3.resnets.0.time_emb_proj": [
        89.1,
        88.9,
        88.7,
        89.0
      ],
      "down_blocks.3.resnets.0.conv2": [
        89.2,
        89.0,
        88.7,
        88.7
      ],
      "down_blocks.3.resnets.1.conv1": [
        89.0,
        89.0,
        89.2,
        88.6
      ],
      "down_blocks.3.resnets.1.time_emb_proj": [
        89.1,
        88.9,
        88.7,
        88.0
      ],
      "down_blocks.3.resnets.1.conv2": [
        89.2,
        89.0,
        88.7,
        89.2
      ],
      "up_blocks.0.resnets.0.conv1": [
        89.0,
        88.8,
        88.9,
        89.1
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        88.8,
        89.0,
        88.9,
        89.0
      ],
      "up_blocks.0.resnets.0.conv2": [
        88.7,
        88.9,
        89.3,
        88.8
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        89.0,
        89.3,
        89.0,
        89.2
      ],
      "up_blocks.0.resnets.1.conv1": [
        89.1,
        89.0,
        88.9,
        89.0
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        89.2,
        89.0,
        89.1,
        88.7
      ],
      "up_blocks.0.resnets.1.conv2": [
        88.9,
        89.0,
        88.9,
        89.1
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        89.1,
        89.0,
        88.8,
        88.9
      ],
      "up_blocks.0.resnets.2.conv1": [
        89.0,
        89.1,
        89.1,
        89.1
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        89.2,
        89.1,
        89.0,
        88.8
      ],
      "up_blocks.0.resnets.2.conv2": [
        88.7,
        89.1,
        88.9,
        88.7
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        89.0,
        89.0,
        89.0,
        88.8
      ],
      "up_blocks.0.upsamplers.0.conv": [
        88.8,
        89.0,
        88.8,
        88.5
      ],
      "up_blocks.1.attentions.0.proj_in": [
        88.7,
        88.6,
        88.5,
        87.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        89.1,
        88.9,
        89.0,
        89.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        89.0,
        88.9,
        88.9,
        88.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.6,
        88.9,
        88.9,
        88.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        88.6,
        88.8,
        88.5,
        88.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        89.1,
        88.9,
        89.0,
        89.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        88.4,
        88.5,
        88.7,
        88.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.8,
        89.4,
        88.8,
        88.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        89.1,
        89.1,
        88.9,
        88.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        88.5,
        89.0,
        88.7,
        88.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        88.7,
        89.0,
        88.7,
        88.8
      ],
      "up_blocks.1.attentions.0.proj_out": [
        88.9,
        88.8,
        88.6,
        88.5
      ],
      "up_blocks.1.attentions.1.proj_in": [
        88.7,
        88.4,
        88.4,
        87.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        89.0,
        89.4,
        88.9,
        89.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        89.2,
        89.0,
        88.7,
        88.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        88.6,
        88.8,
        89.0,
        88.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        88.6,
        88.8,
        88.9,
        88.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.7,
        88.9,
        88.9,
        89.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        88.8,
        88.7,
        88.8,
        88.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        89.0,
        88.9,
        88.8,
        88.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.7,
        89.1,
        89.1,
        89.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        88.9,
        88.7,
        88.6,
        87.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        88.9,
        88.8,
        88.9,
        88.5
      ],
      "up_blocks.1.attentions.1.proj_out": [
        88.8,
        88.6,
        88.8,
        88.5
      ],
      "up_blocks.1.attentions.2.proj_in": [
        88.4,
        88.5,
        88.5,
        88.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        88.5,
        88.8,
        88.6,
        88.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        88.9,
        89.1,
        88.9,
        88.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        88.6,
        88.9,
        88.8,
        88.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        88.4,
        88.6,
        88.6,
        88.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        88.6,
        88.7,
        88.8,
        88.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        88.3,
        88.7,
        88.4,
        88.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        88.7,
        88.8,
        88.6,
        88.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        88.8,
        89.2,
        88.9,
        88.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        88.5,
        88.6,
        89.0,
        88.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        88.8,
        88.7,
        88.7,
        88.7
      ],
      "up_blocks.1.attentions.2.proj_out": [
        88.2,
        88.8,
        88.8,
        88.7
      ],
      "up_blocks.1.resnets.0.conv1": [
        88.8,
        89.1,
        87.0,
        88.2
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        88.9,
        89.2,
        88.6,
        88.8
      ],
      "up_blocks.1.resnets.0.conv2": [
        88.9,
        88.8,
        88.7,
        88.6
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        88.8,
        88.7,
        88.7,
        88.5
      ],
      "up_blocks.1.resnets.1.conv1": [
        88.9,
        88.6,
        88.8,
        89.0
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        89.0,
        88.9,
        88.8,
        88.9
      ],
      "up_blocks.1.resnets.1.conv2": [
        88.9,
        89.2,
        88.8,
        88.2
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        88.4,
        88.7,
        88.6,
        88.8
      ],
      "up_blocks.1.resnets.2.conv1": [
        88.9,
        89.1,
        88.7,
        88.7
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        89.0,
        88.8,
        89.1,
        88.9
      ],
      "up_blocks.1.resnets.2.conv2": [
        88.9,
        88.8,
        88.8,
        88.9
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        88.8,
        88.4,
        88.6,
        88.5
      ],
      "up_blocks.1.upsamplers.0.conv": [
        88.3,
        88.7,
        88.7,
        88.3
      ],
      "up_blocks.2.attentions.0.proj_in": [
        88.2,
        88.3,
        88.4,
        88.2
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.6,
        88.6,
        88.6,
        88.8
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.6,
        89.2,
        88.7,
        88.9
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.2,
        88.7,
        88.6,
        87.6
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        88.3,
        88.1,
        88.6,
        88.3
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        89.1,
        89.3,
        89.2,
        88.6
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        88.9,
        89.0,
        88.9,
        88.8
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.8,
        89.0,
        88.8,
        88.4
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.6,
        88.7,
        88.4,
        88.6
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        88.8,
        88.9,
        88.5,
        88.6
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        88.6,
        88.5,
        88.7,
        88.8
      ],
      "up_blocks.2.attentions.0.proj_out": [
        88.4,
        88.3,
        88.4,
        88.3
      ],
      "up_blocks.2.attentions.1.proj_in": [
        87.9,
        88.1,
        88.1,
        88.0
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.4,
        88.7,
        88.6,
        88.5
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        88.4,
        88.5,
        88.7,
        88.5
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        88.1,
        88.3,
        88.6,
        88.1
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        88.1,
        88.3,
        88.2,
        87.9
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.8,
        88.7,
        88.9,
        88.6
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        88.8,
        89.0,
        88.9,
        88.5
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        88.4,
        88.6,
        88.7,
        88.6
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        89.0,
        89.1,
        88.9,
        88.4
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        88.1,
        88.5,
        88.4,
        88.0
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        88.4,
        88.4,
        88.3,
        88.4
      ],
      "up_blocks.2.attentions.1.proj_out": [
        87.9,
        88.0,
        88.1,
        87.9
      ],
      "up_blocks.2.attentions.2.proj_in": [
        87.8,
        87.9,
        87.9,
        87.1
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q": [
        88.7,
        88.7,
        88.6,
        88.0
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k": [
        88.4,
        88.5,
        88.4,
        88.3
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v": [
        88.3,
        88.1,
        88.1,
        87.4
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        87.0,
        87.7,
        87.5,
        87.0
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q": [
        89.0,
        88.5,
        88.7,
        88.7
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k": [
        88.0,
        88.1,
        88.3,
        87.8
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v": [
        88.1,
        88.3,
        88.5,
        88.3
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        88.6,
        88.4,
        88.3,
        88.6
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        86.8,
        87.3,
        87.6,
        86.6
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2": [
        88.0,
        88.6,
        88.3,
        87.9
      ],
      "up_blocks.2.attentions.2.proj_out": [
        87.8,
        87.6,
        87.9,
        87.5
      ],
      "up_blocks.2.resnets.0.conv1": [
        88.4,
        88.6,
        89.0,
        88.1
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        89.0,
        88.9,
        88.9,
        88.9
      ],
      "up_blocks.2.resnets.0.conv2": [
        88.4,
        88.6,
        88.7,
        88.0
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        88.1,
        88.2,
        88.2,
        87.7
      ],
      "up_blocks.2.resnets.1.conv1": [
        88.3,
        88.2,
        88.6,
        88.3
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        88.5,
        88.4,
        88.6,
        88.8
      ],
      "up_blocks.2.resnets.1.conv2": [
        88.4,
        88.3,
        88.6,
        88.5
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        88.0,
        88.1,
        88.1,
        87.5
      ],
      "up_blocks.2.resnets.2.conv1": [
        88.4,
        88.3,
        88.3,
        87.3
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        88.3,
        88.5,
        88.2,
        88.1
      ],
      "up_blocks.2.resnets.2.conv2": [
        88.2,
        88.1,
        88.2,
        88.3
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        87.6,
        87.8,
        87.9,
        87.0
      ],
      "up_blocks.2.upsamplers.0.conv": [
        88.1,
        87.8,
        88.2,
        87.6
      ],
      "up_blocks.3.attentions.0.proj_in": [
        86.9,
        87.4,
        87.4,
        86.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.2,
        88.4,
        88.2,
        88.2
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.2,
        87.7,
        87.9,
        88.0
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v": [
        87.6,
        87.8,
        87.5,
        87.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        87.5,
        87.4,
        87.7,
        87.2
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.7,
        89.0,
        88.5,
        88.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k": [
        88.2,
        88.4,
        88.3,
        87.9
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.4,
        88.6,
        88.6,
        88.5
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.5,
        88.2,
        88.2,
        88.5
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        87.9,
        87.9,
        87.8,
        87.3
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2": [
        87.9,
        88.1,
        88.0,
        87.9
      ],
      "up_blocks.3.attentions.0.proj_out": [
        87.3,
        87.9,
        87.9,
        87.5
      ],
      "up_blocks.3.attentions.1.proj_in": [
        86.0,
        86.5,
        86.4,
        86.0
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.2,
        88.4,
        88.4,
        88.2
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k": [
        88.5,
        88.7,
        88.3,
        88.5
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v": [
        88.2,
        87.9,
        88.1,
        87.8
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        87.2,
        87.2,
        87.2,
        87.3
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.7,
        88.7,
        88.7,
        88.9
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k": [
        88.1,
        88.6,
        88.8,
        88.2
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v": [
        88.3,
        88.2,
        88.4,
        88.6
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.5,
        88.6,
        88.6,
        88.4
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        86.9,
        86.7,
        87.3,
        86.6
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2": [
        87.7,
        87.9,
        87.9,
        87.7
      ],
      "up_blocks.3.attentions.1.proj_out": [
        87.1,
        87.5,
        87.4,
        87.4
      ],
      "up_blocks.3.attentions.2.proj_in": [
        85.4,
        85.1,
        85.9,
        85.8
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q": [
        87.9,
        88.2,
        88.1,
        87.8
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k": [
        87.7,
        87.9,
        87.5,
        87.7
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v": [
        87.0,
        86.8,
        87.3,
        86.9
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        86.6,
        86.6,
        87.1,
        86.8
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q": [
        88.7,
        88.7,
        88.7,
        88.4
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k": [
        88.2,
        87.9,
        88.6,
        88.1
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v": [
        87.7,
        87.9,
        87.9,
        87.5
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        87.6,
        87.5,
        87.5,
        87.1
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        85.8,
        85.8,
        86.1,
        85.9
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2": [
        88.0,
        88.5,
        88.0,
        87.6
      ],
      "up_blocks.3.attentions.2.proj_out": [
        87.5,
        87.4,
        87.4,
        87.4
      ],
      "up_blocks.3.resnets.0.conv1": [
        87.7,
        87.7,
        87.8,
        87.7
      ],
      "up_blocks.3.resnets.0.time_emb_proj": [
        88.5,
        88.7,
        88.4,
        88.6
      ],
      "up_blocks.3.resnets.0.conv2": [
        87.4,
        87.4,
        87.8,
        87.0
      ],
      "up_blocks.3.resnets.0.conv_shortcut": [
        86.5,
        86.5,
        86.5,
        85.0
      ],
      "up_blocks.3.resnets.1.conv1": [
        86.5,
        87.0,
        86.9,
        86.4
      ],
      "up_blocks.3.resnets.1.time_emb_proj": [
        88.1,
        87.9,
        88.3,
        88.2
      ],
      "up_blocks.3.resnets.1.conv2": [
        86.9,
        86.7,
        87.2,
        86.6
      ],
      "up_blocks.3.resnets.1.conv_shortcut": [
        85.9,
        85.6,
        86.2,
        86.0
      ],
      "up_blocks.3.resnets.2.conv1": [
        87.2,
        87.3,
        87.1,
        86.7
      ],
      "up_blocks.3.resnets.2.time_emb_proj": [
        88.0,
        88.3,
        88.3,
        88.3
      ],
      "up_blocks.3.resnets.2.conv2": [
        87.0,
        86.7,
        87.1,
        86.6
      ],
      "up_blocks.3.resnets.2.conv_shortcut": [
        85.9,
        86.1,
        85.7,
        85.8
      ],
      "mid_block.attentions.0.proj_in": [
        88.9,
        89.2,
        88.8,
        89.0
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        89.1,
        89.2,
        88.6,
        89.0
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.9,
        89.0,
        89.1,
        88.6
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.8,
        89.1,
        88.9,
        88.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        89.1,
        89.2,
        88.9,
        88.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.9,
        89.2,
        88.9,
        88.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        89.1,
        89.3,
        89.0,
        89.0
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        89.0,
        89.4,
        89.0,
        89.0
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.9,
        89.2,
        89.0,
        88.6
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        88.8,
        89.2,
        88.8,
        89.0
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        88.9,
        89.0,
        89.0,
        89.1
      ],
      "mid_block.attentions.0.proj_out": [
        88.9,
        89.1,
        89.2,
        89.1
      ],
      "mid_block.resnets.0.conv1": [
        89.1,
        89.3,
        89.1,
        89.0
      ],
      "mid_block.resnets.0.time_emb_proj": [
        89.0,
        89.0,
        89.0,
        89.1
      ],
      "mid_block.resnets.0.conv2": [
        88.8,
        89.0,
        88.7,
        88.9
      ],
      "mid_block.resnets.1.conv1": [
        88.9,
        89.1,
        88.9,
        88.9
      ],
      "mid_block.resnets.1.time_emb_proj": [
        89.0,
        89.3,
        89.0,
        88.8
      ],
      "mid_block.resnets.1.conv2": [
        89.0,
        88.9,
        89.0,
        88.9
      ]
    },
    "8": {
      "time_embedding.linear_1": [
        89.1,
        88.8,
        88.9,
        88.5
      ],
      "time_embedding.linear_2": [
        88.8,
        88.6,
        88.8,
        87.6
      ],
      "down_blocks.0.attentions.0.proj_in": [
        88.3,
        88.3,
        88.2,
        88.0
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.7,
        88.9,
        88.6,
        88.5
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.8,
        88.7,
        88.3,
        88.7
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.6,
        88.6,
        88.5,
        88.3
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        88.5,
        88.5,
        88.6,
        88.1
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.9,
        88.8,
        88.4,
        88.3
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        88.8,
        88.4,
        88.4,
        88.2
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.8,
        88.6,
        88.8,
        87.9
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.8,
        88.9,
        88.6,
        88.2
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        88.4,
        88.4,
        88.4,
        88.5
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        88.4,
        88.2,
        88.5,
        87.9
      ],
      "down_blocks.0.attentions.0.proj_out": [
        88.4,
        87.9,
        88.4,
        88.5
      ],
      "down_blocks.0.attentions.1.proj_in": [
        88.2,
        88.3,
        87.5,
        87.9
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.8,
        89.0,
        88.7,
        88.3
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        89.0,
        88.7,
        88.7,
        88.8
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        88.8,
        88.5,
        88.4,
        88.2
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        88.8,
        88.7,
        88.1,
        88.4
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.8,
        89.2,
        88.8,
        88.1
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        88.9,
        88.6,
        88.6,
        87.7
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        89.1,
        88.9,
        88.8,
        88.4
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        89.1,
        88.8,
        89.2,
        88.9
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        88.6,
        88.4,
        88.5,
        87.0
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        88.8,
        88.7,
        88.5,
        88.5
      ],
      "down_blocks.0.attentions.1.proj_out": [
        88.9,
        88.3,
        88.2,
        87.8
      ],
      "down_blocks.0.resnets.0.conv1": [
        88.6,
        88.6,
        88.5,
        88.3
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        88.7,
        88.7,
        88.5,
        88.1
      ],
      "down_blocks.0.resnets.0.conv2": [
        88.4,
        88.5,
        88.1,
        87.8
      ],
      "down_blocks.0.resnets.1.conv1": [
        88.7,
        88.3,
        88.2,
        86.9
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        88.4,
        88.7,
        88.7,
        88.8
      ],
      "down_blocks.0.resnets.1.conv2": [
        88.5,
        88.3,
        87.9,
        88.7
      ],
      "down_blocks.0.downsamplers.0.conv": [
        88.8,
        88.4,
        88.5,
        86.1
      ],
      "down_blocks.1.attentions.0.proj_in": [
        89.1,
        88.8,
        88.8,
        87.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        89.0,
        88.9,
        89.0,
        88.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        89.0,
        89.1,
        89.0,
        88.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        89.1,
        88.4,
        88.6,
        88.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        88.7,
        88.5,
        88.0,
        88.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        89.2,
        88.8,
        89.1,
        88.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        89.0,
        89.1,
        89.0,
        88.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        89.0,
        89.0,
        89.1,
        88.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        89.1,
        89.0,
        88.8,
        88.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        88.8,
        88.4,
        88.6,
        87.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        88.8,
        88.3,
        88.8,
        88.5
      ],
      "down_blocks.1.attentions.0.proj_out": [
        88.9,
        88.8,
        89.0,
        87.1
      ],
      "down_blocks.1.attentions.1.proj_in": [
        88.9,
        88.6,
        89.2,
        87.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        89.3,
        89.0,
        88.6,
        88.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        89.3,
        89.1,
        88.9,
        89.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        89.2,
        88.8,
        88.7,
        88.3
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        88.9,
        88.8,
        88.6,
        88.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        89.5,
        88.9,
        88.9,
        88.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        89.2,
        88.9,
        88.9,
        88.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        89.0,
        88.9,
        88.9,
        87.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        89.4,
        89.0,
        88.7,
        88.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        89.1,
        88.6,
        88.7,
        87.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        89.1,
        89.2,
        89.0,
        88.8
      ],
      "down_blocks.1.attentions.1.proj_out": [
        89.0,
        88.7,
        89.0,
        88.3
      ],
      "down_blocks.1.resnets.0.conv1": [
        89.1,
        88.7,
        88.9,
        88.4
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        89.0,
        88.9,
        88.8,
        88.3
      ],
      "down_blocks.1.resnets.0.conv2": [
        89.0,
        89.0,
        88.8,
        88.4
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        88.9,
        88.4,
        88.6,
        86.8
      ],
      "down_blocks.1.resnets.1.conv1": [
        88.9,
        88.9,
        88.6,
        88.2
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        88.8,
        89.1,
        88.5,
        88.3
      ],
      "down_blocks.1.resnets.1.conv2": [
        88.9,
        88.8,
        88.4,
        88.7
      ],
      "down_blocks.1.downsamplers.0.conv": [
        89.0,
        88.7,
        88.7,
        88.5
      ],
      "down_blocks.2.attentions.0.proj_in": [
        88.9,
        89.0,
        88.7,
        88.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        89.2,
        88.9,
        89.0,
        88.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        89.0,
        89.1,
        88.4,
        88.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        89.1,
        88.8,
        89.4,
        88.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        89.2,
        88.9,
        88.8,
        88.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        89.1,
        88.8,
        89.1,
        88.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        89.1,
        88.9,
        88.6,
        89.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        89.3,
        89.0,
        88.8,
        88.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        89.1,
        88.8,
        89.0,
        89.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        89.1,
        89.1,
        89.0,
        88.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        89.2,
        89.1,
        89.1,
        88.7
      ],
      "down_blocks.2.attentions.0.proj_out": [
        89.1,
        89.0,
        88.8,
        88.6
      ],
      "down_blocks.2.attentions.1.proj_in": [
        88.9,
        89.1,
        88.8,
        87.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        89.0,
        88.7,
        89.0,
        88.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        89.2,
        89.1,
        88.7,
        87.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        89.2,
        88.9,
        88.8,
        88.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        89.2,
        89.0,
        88.6,
        88.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        89.2,
        88.9,
        88.7,
        88.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        89.1,
        89.0,
        89.0,
        88.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        89.3,
        89.0,
        88.9,
        89.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        89.3,
        89.2,
        88.6,
        88.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        89.2,
        89.0,
        89.0,
        89.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        88.9,
        89.1,
        89.0,
        88.8
      ],
      "down_blocks.2.attentions.1.proj_out": [
        89.1,
        88.6,
        88.9,
        89.0
      ],
      "down_blocks.2.resnets.0.conv1": [
        89.1,
        89.3,
        88.8,
        88.2
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        88.9,
        89.0,
        88.8,
        88.0
      ],
      "down_blocks.2.resnets.0.conv2": [
        89.3,
        89.0,
        88.7,
        88.5
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        89.1,
        88.8,
        88.7,
        89.1
      ],
      "down_blocks.2.resnets.1.conv1": [
        89.2,
        89.1,
        88.6,
        88.9
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        89.1,
        89.0,
        89.0,
        89.2
      ],
      "down_blocks.2.resnets.1.conv2": [
        89.2,
        89.3,
        88.9,
        88.4
      ],
      "down_blocks.2.downsamplers.0.conv": [
        88.9,
        88.8,
        88.9,
        88.1
      ],
      "down_blocks.3.resnets.0.conv1": [
        89.3,
        89.2,
        88.8,
        89.0
      ],
      "down_blocks.3.resnets.0.time_emb_proj": [
        89.0,
        88.8,
        89.1,
        87.9
      ],
      "down_blocks.3.resnets.0.conv2": [
        88.9,
        89.1,
        89.2,
        88.6
      ],
      "down_blocks.3.resnets.1.conv1": [
        89.4,
        89.1,
        88.9,
        88.8
      ],
      "down_blocks.3.resnets.1.time_emb_proj": [
        89.1,
        89.0,
        89.1,
        88.9
      ],
      "down_blocks.3.resnets.1.conv2": [
        89.1,
        88.9,
        89.0,
        88.8
      ],
      "up_blocks.0.resnets.0.conv1": [
        89.3,
        89.2,
        89.0,
        88.4
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        89.0,
        88.8,
        88.7,
        87.9
      ],
      "up_blocks.0.resnets.0.conv2": [
        89.3,
        88.9,
        89.0,
        88.8
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        89.2,
        89.0,
        88.9,
        89.1
      ],
      "up_blocks.0.resnets.1.conv1": [
        89.3,
        88.9,
        88.8,
        89.2
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        89.1,
        88.9,
        89.0,
        88.6
      ],
      "up_blocks.0.resnets.1.conv2": [
        89.3,
        88.9,
        88.8,
        88.5
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        89.2,
        88.9,
        89.1,
        89.0
      ],
      "up_blocks.0.resnets.2.conv1": [
        89.3,
        89.0,
        88.7,
        88.7
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        89.0,
        89.2,
        88.9,
        89.1
      ],
      "up_blocks.0.resnets.2.conv2": [
        89.1,
        89.0,
        89.1,
        88.9
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        89.2,
        89.3,
        88.9,
        88.3
      ],
      "up_blocks.0.upsamplers.0.conv": [
        89.1,
        88.9,
        88.8,
        89.0
      ],
      "up_blocks.1.attentions.0.proj_in": [
        89.0,
        88.8,
        88.8,
        88.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        89.0,
        89.0,
        88.8,
        87.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        89.1,
        89.0,
        88.8,
        88.3
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        89.3,
        88.8,
        88.7,
        88.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        88.9,
        89.1,
        88.7,
        88.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        89.2,
        89.0,
        88.8,
        87.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        89.1,
        89.2,
        88.6,
        89.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        89.0,
        88.8,
        88.9,
        88.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.9,
        89.1,
        88.8,
        88.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        89.4,
        89.2,
        88.8,
        88.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        88.9,
        88.9,
        89.1,
        88.1
      ],
      "up_blocks.1.attentions.0.proj_out": [
        88.8,
        89.1,
        88.7,
        89.1
      ],
      "up_blocks.1.attentions.1.proj_in": [
        88.8,
        89.2,
        88.8,
        88.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        89.2,
        88.8,
        88.8,
        88.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        89.3,
        89.1,
        88.9,
        88.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        88.8,
        89.0,
        88.9,
        87.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        89.3,
        88.9,
        88.7,
        88.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        89.2,
        89.0,
        89.2,
        89.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        89.0,
        89.0,
        89.1,
        88.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        88.9,
        88.8,
        89.3,
        88.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        89.0,
        88.9,
        88.5,
        89.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        89.2,
        88.9,
        88.6,
        88.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        88.9,
        89.1,
        88.7,
        88.7
      ],
      "up_blocks.1.attentions.1.proj_out": [
        89.2,
        88.9,
        88.9,
        88.7
      ],
      "up_blocks.1.attentions.2.proj_in": [
        89.1,
        89.0,
        88.6,
        88.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        89.1,
        89.2,
        89.1,
        89.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        89.1,
        88.9,
        89.2,
        89.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        89.2,
        88.8,
        88.9,
        89.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        89.0,
        88.9,
        89.1,
        87.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        89.0,
        89.0,
        88.7,
        88.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        89.0,
        89.2,
        88.9,
        88.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        89.1,
        88.9,
        89.3,
        88.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        89.1,
        88.9,
        88.7,
        88.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        88.9,
        89.2,
        89.3,
        88.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        88.8,
        88.9,
        89.1,
        88.9
      ],
      "up_blocks.1.attentions.2.proj_out": [
        89.2,
        88.7,
        88.9,
        88.5
      ],
      "up_blocks.1.resnets.0.conv1": [
        89.1,
        88.8,
        88.7,
        88.6
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        89.0,
        89.1,
        89.1,
        88.6
      ],
      "up_blocks.1.resnets.0.conv2": [
        89.2,
        89.1,
        88.7,
        87.9
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        89.2,
        88.9,
        89.0,
        88.9
      ],
      "up_blocks.1.resnets.1.conv1": [
        88.9,
        88.9,
        88.9,
        89.0
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        89.1,
        88.8,
        88.8,
        88.2
      ],
      "up_blocks.1.resnets.1.conv2": [
        89.0,
        89.2,
        89.0,
        88.4
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        89.4,
        89.1,
        89.3,
        88.1
      ],
      "up_blocks.1.resnets.2.conv1": [
        89.5,
        88.8,
        88.8,
        88.5
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        89.0,
        89.1,
        89.0,
        88.9
      ],
      "up_blocks.1.resnets.2.conv2": [
        89.3,
        88.8,
        88.6,
        88.0
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        88.9,
        88.9,
        88.6,
        87.9
      ],
      "up_blocks.1.upsamplers.0.conv": [
        89.1,
        88.8,
        88.8,
        89.0
      ],
      "up_blocks.2.attentions.0.proj_in": [
        88.8,
        88.7,
        89.0,
        88.3
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        89.2,
        89.1,
        88.9,
        88.3
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        89.2,
        89.0,
        88.9,
        88.5
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.9,
        88.9,
        88.8,
        88.3
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        89.2,
        88.9,
        88.9,
        88.6
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        89.2,
        88.9,
        88.6,
        88.9
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        89.3,
        89.0,
        89.2,
        89.2
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        89.1,
        88.7,
        88.9,
        89.1
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        89.1,
        89.0,
        88.8,
        88.4
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        89.1,
        88.9,
        88.8,
        88.1
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        89.2,
        88.7,
        88.9,
        89.0
      ],
      "up_blocks.2.attentions.0.proj_out": [
        89.1,
        88.8,
        88.6,
        88.9
      ],
      "up_blocks.2.attentions.1.proj_in": [
        88.6,
        88.6,
        88.8,
        87.7
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        89.0,
        88.8,
        88.8,
        88.7
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        88.9,
        88.9,
        89.0,
        88.4
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        89.0,
        88.5,
        88.7,
        89.1
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        88.6,
        88.6,
        88.5,
        88.6
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        89.2,
        89.2,
        88.8,
        88.8
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        89.2,
        89.1,
        89.2,
        88.9
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        89.1,
        88.9,
        89.0,
        88.8
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        89.2,
        88.8,
        89.3,
        88.4
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        88.6,
        88.5,
        89.0,
        88.8
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        88.9,
        88.6,
        88.7,
        88.4
      ],
      "up_blocks.2.attentions.1.proj_out": [
        89.0,
        88.5,
        88.7,
        88.4
      ],
      "up_blocks.2.attentions.2.proj_in": [
        88.6,
        88.1,
        88.3,
        88.3
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q": [
        88.8,
        88.6,
        88.8,
        88.6
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k": [
        89.0,
        88.9,
        88.9,
        88.5
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v": [
        88.8,
        88.3,
        88.8,
        88.2
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        88.4,
        88.5,
        88.5,
        88.7
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q": [
        89.0,
        89.1,
        88.9,
        89.2
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k": [
        88.9,
        88.7,
        88.9,
        88.7
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v": [
        89.2,
        88.9,
        88.6,
        88.8
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        88.8,
        89.0,
        88.7,
        88.8
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        88.7,
        88.4,
        88.7,
        88.6
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2": [
        88.9,
        88.4,
        88.7,
        88.7
      ],
      "up_blocks.2.attentions.2.proj_out": [
        88.7,
        88.6,
        88.7,
        88.0
      ],
      "up_blocks.2.resnets.0.conv1": [
        89.0,
        88.8,
        89.1,
        89.1
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        89.2,
        89.2,
        88.9,
        88.6
      ],
      "up_blocks.2.resnets.0.conv2": [
        89.1,
        89.0,
        88.7,
        88.3
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        88.8,
        88.5,
        88.6,
        88.3
      ],
      "up_blocks.2.resnets.1.conv1": [
        89.0,
        88.7,
        89.0,
        88.5
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        89.2,
        89.0,
        89.3,
        88.8
      ],
      "up_blocks.2.resnets.1.conv2": [
        89.2,
        88.9,
        88.8,
        87.9
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        88.5,
        89.0,
        88.4,
        88.6
      ],
      "up_blocks.2.resnets.2.conv1": [
        89.0,
        88.9,
        88.9,
        88.7
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        88.9,
        88.7,
        88.6,
        88.8
      ],
      "up_blocks.2.resnets.2.conv2": [
        88.8,
        88.3,
        88.8,
        89.1
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        88.7,
        88.6,
        88.5,
        88.6
      ],
      "up_blocks.2.upsamplers.0.conv": [
        88.7,
        88.7,
        88.5,
        88.1
      ],
      "up_blocks.3.attentions.0.proj_in": [
        88.3,
        88.3,
        88.4,
        88.0
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.6,
        88.8,
        88.7,
        88.6
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.8,
        88.6,
        88.9,
        88.2
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.6,
        88.6,
        88.7,
        88.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        88.7,
        88.4,
        88.7,
        88.8
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q": [
        89.4,
        88.9,
        88.8,
        88.8
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k": [
        88.8,
        88.5,
        88.8,
        87.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v": [
        89.1,
        88.7,
        88.9,
        88.6
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.8,
        88.8,
        88.8,
        87.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        88.6,
        88.4,
        88.6,
        88.5
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2": [
        88.6,
        88.6,
        88.5,
        88.4
      ],
      "up_blocks.3.attentions.0.proj_out": [
        88.4,
        88.3,
        88.5,
        88.1
      ],
      "up_blocks.3.attentions.1.proj_in": [
        88.1,
        88.0,
        88.3,
        88.2
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.9,
        88.8,
        88.5,
        88.7
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k": [
        89.0,
        88.5,
        88.9,
        88.2
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v": [
        88.6,
        88.4,
        88.7,
        88.3
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        88.6,
        88.3,
        88.6,
        88.6
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q": [
        89.0,
        89.0,
        88.9,
        89.0
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k": [
        88.9,
        89.0,
        88.7,
        88.3
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v": [
        89.3,
        88.8,
        89.2,
        88.7
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.9,
        88.7,
        88.9,
        88.3
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        88.8,
        88.5,
        88.3,
        88.3
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2": [
        88.3,
        88.3,
        88.5,
        88.1
      ],
      "up_blocks.3.attentions.1.proj_out": [
        88.5,
        88.4,
        88.7,
        88.6
      ],
      "up_blocks.3.attentions.2.proj_in": [
        88.3,
        88.3,
        88.4,
        88.2
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q": [
        88.7,
        88.9,
        88.5,
        88.8
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k": [
        89.0,
        88.7,
        88.8,
        88.2
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v": [
        88.8,
        88.8,
        88.5,
        88.1
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        88.4,
        88.6,
        88.5,
        88.0
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q": [
        89.0,
        88.6,
        89.0,
        88.8
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k": [
        89.2,
        89.0,
        89.0,
        88.9
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v": [
        89.2,
        88.5,
        88.9,
        87.9
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        88.8,
        89.0,
        88.8,
        88.8
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        88.7,
        88.3,
        88.8,
        88.3
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2": [
        88.7,
        88.7,
        88.8,
        88.4
      ],
      "up_blocks.3.attentions.2.proj_out": [
        88.8,
        88.8,
        89.0,
        89.0
      ],
      "up_blocks.3.resnets.0.conv1": [
        88.6,
        88.3,
        88.6,
        87.9
      ],
      "up_blocks.3.resnets.0.time_emb_proj": [
        88.8,
        89.0,
        88.8,
        88.9
      ],
      "up_blocks.3.resnets.0.conv2": [
        88.6,
        88.5,
        88.7,
        88.7
      ],
      "up_blocks.3.resnets.0.conv_shortcut": [
        88.5,
        88.2,
        88.4,
        87.8
      ],
      "up_blocks.3.resnets.1.conv1": [
        88.2,
        88.3,
        88.6,
        88.6
      ],
      "up_blocks.3.resnets.1.time_emb_proj": [
        88.5,
        88.6,
        88.8,
        88.5
      ],
      "up_blocks.3.resnets.1.conv2": [
        88.6,
        88.2,
        88.6,
        87.9
      ],
      "up_blocks.3.resnets.1.conv_shortcut": [
        88.0,
        87.9,
        88.1,
        88.3
      ],
      "up_blocks.3.resnets.2.conv1": [
        88.8,
        88.2,
        88.5,
        88.7
      ],
      "up_blocks.3.resnets.2.time_emb_proj": [
        88.7,
        88.2,
        88.7,
        88.7
      ],
      "up_blocks.3.resnets.2.conv2": [
        88.6,
        88.5,
        88.7,
        88.8
      ],
      "up_blocks.3.resnets.2.conv_shortcut": [
        88.4,
        88.3,
        88.6,
        87.3
      ],
      "mid_block.attentions.0.proj_in": [
        89.2,
        89.1,
        89.2,
        88.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        89.4,
        89.0,
        89.1,
        87.6
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        89.3,
        89.2,
        89.1,
        88.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        89.0,
        89.0,
        89.0,
        88.6
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        89.0,
        89.1,
        88.8,
        88.5
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        89.1,
        88.8,
        88.9,
        88.4
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        89.1,
        89.0,
        88.8,
        88.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        89.2,
        88.7,
        88.8,
        89.0
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        89.2,
        89.1,
        88.9,
        89.0
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        88.9,
        89.0,
        88.9,
        89.0
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        89.4,
        88.9,
        89.0,
        89.3
      ],
      "mid_block.attentions.0.proj_out": [
        89.0,
        88.7,
        89.0,
        89.0
      ],
      "mid_block.resnets.0.conv1": [
        89.2,
        88.8,
        88.9,
        88.8
      ],
      "mid_block.resnets.0.time_emb_proj": [
        89.0,
        89.2,
        89.0,
        89.0
      ],
      "mid_block.resnets.0.conv2": [
        89.2,
        89.0,
        89.0,
        89.2
      ],
      "mid_block.resnets.1.conv1": [
        89.1,
        88.8,
        88.9,
        88.9
      ],
      "mid_block.resnets.1.time_emb_proj": [
        89.2,
        88.9,
        88.7,
        88.7
      ],
      "mid_block.resnets.1.conv2": [
        89.1,
        88.9,
        88.7,
        87.8
      ]
    }
  },
  "cumulative": {
    "1": {
      "metadata": {
        "nbits": 1,
        "out_ngroups": 1,
        "in_ngroups": 1,
        "cumulative": true,
        "candidates": [
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "mid_block.resnets.1.time_emb_proj",
          "up_blocks.0.resnets.1.time_emb_proj",
          "down_blocks.3.resnets.1.time_emb_proj",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.resnets.0.time_emb_proj",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.resnets.2.time_emb_proj",
          "down_blocks.3.resnets.0.time_emb_proj",
          "mid_block.resnets.0.time_emb_proj",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "mid_block.attentions.0.proj_in",
          "up_blocks.0.resnets.2.conv_shortcut",
          "mid_block.attentions.0.proj_out",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.2",
          "down_blocks.3.resnets.1.conv1",
          "mid_block.resnets.0.conv1",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.resnets.0.conv_shortcut",
          "down_blocks.3.resnets.1.conv2",
          "up_blocks.0.resnets.0.conv1",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.resnets.1.conv_shortcut",
          "up_blocks.1.resnets.2.time_emb_proj",
          "down_blocks.3.resnets.0.conv2",
          "down_blocks.3.resnets.0.conv1",
          "up_blocks.0.resnets.1.conv1",
          "up_blocks.1.resnets.1.time_emb_proj",
          "mid_block.resnets.0.conv2",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k",
          "mid_block.resnets.1.conv1",
          "up_blocks.2.resnets.0.time_emb_proj",
          "up_blocks.0.resnets.0.conv2",
          "mid_block.resnets.1.conv2",
          "up_blocks.0.resnets.2.conv1",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.resnets.1.conv2",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.downsamplers.0.conv",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.resnets.0.time_emb_proj",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.resnets.1.conv2",
          "up_blocks.1.attentions.1.proj_out",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.resnets.2.conv2",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.1.resnets.0.conv1",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.resnets.1.conv1",
          "down_blocks.2.resnets.0.time_emb_proj",
          "down_blocks.2.resnets.1.time_emb_proj",
          "up_blocks.1.attentions.0.proj_out",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.resnets.2.conv1",
          "up_blocks.1.resnets.1.conv1",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.1.resnets.1.conv2",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.resnets.0.conv2",
          "down_blocks.2.attentions.0.proj_in",
          "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.resnets.0.conv1",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.resnets.2.conv2",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.resnets.0.conv1",
          "down_blocks.1.resnets.1.time_emb_proj",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q",
          "down_blocks.1.attentions.1.proj_out",
          "down_blocks.2.attentions.1.proj_in",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.resnets.1.conv_shortcut",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.upsamplers.0.conv",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.2.resnets.1.conv1",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.resnets.0.conv_shortcut",
          "up_blocks.1.resnets.2.conv_shortcut",
          "down_blocks.2.attentions.1.proj_out",
          "up_blocks.1.attentions.1.proj_in",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.upsamplers.0.conv",
          "up_blocks.2.resnets.1.time_emb_proj",
          "down_blocks.2.attentions.0.proj_out",
          "up_blocks.1.attentions.0.proj_in",
          "up_blocks.2.resnets.1.conv2",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.2.resnets.0.conv2",
          "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.attentions.2.proj_out",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.attentions.2.proj_in",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.1.resnets.1.conv2",
          "up_blocks.3.resnets.0.time_emb_proj",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.1.resnets.1.conv1",
          "down_blocks.2.resnets.0.conv_shortcut",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.resnets.0.conv2",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.resnets.2.conv1",
          "down_blocks.1.downsamplers.0.conv",
          "down_blocks.1.resnets.0.conv2",
          "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.1.resnets.0.conv1",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.resnets.0.time_emb_proj",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.resnets.2.conv2",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.0.resnets.1.time_emb_proj",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q",
          "down_blocks.1.attentions.0.proj_out",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.2.attentions.1.proj_out",
          "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k",
          "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.2.attentions.1.proj_in",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.attentions.0.proj_in",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v",
          "up_blocks.2.resnets.2.time_emb_proj",
          "up_blocks.2.attentions.0.proj_in",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.0.attentions.0.proj_out",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v",
          "down_blocks.0.resnets.0.conv1",
          "down_blocks.1.attentions.1.proj_in",
          "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.0.resnets.1.conv2",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q",
          "down_blocks.0.attentions.1.proj_out",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.2.attentions.0.proj_out",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.attentions.0.proj_out",
          "up_blocks.2.resnets.1.conv_shortcut",
          "down_blocks.0.attentions.0.proj_in",
          "up_blocks.2.attentions.2.proj_out",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.0.attentions.1.proj_in",
          "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.resnets.0.conv_shortcut",
          "up_blocks.2.upsamplers.0.conv",
          "down_blocks.0.resnets.1.conv1",
          "up_blocks.2.attentions.2.proj_in",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.0.resnets.0.time_emb_proj",
          "up_blocks.3.resnets.0.conv2",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.2.resnets.2.conv_shortcut",
          "up_blocks.3.resnets.0.conv1",
          "down_blocks.0.downsamplers.0.conv",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.3.attentions.0.proj_in",
          "up_blocks.3.resnets.1.time_emb_proj",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k",
          "up_blocks.3.attentions.1.proj_out",
          "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2",
          "down_blocks.0.resnets.0.conv2",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.attentions.2.proj_out",
          "up_blocks.2.resnets.0.conv_shortcut",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v",
          "up_blocks.3.attentions.1.proj_in",
          "up_blocks.3.resnets.1.conv1",
          "up_blocks.3.resnets.2.conv1",
          "up_blocks.3.resnets.1.conv2",
          "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.resnets.1.conv_shortcut",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.resnets.2.time_emb_proj",
          "up_blocks.3.attentions.2.proj_in",
          "up_blocks.3.resnets.2.conv2",
          "time_embedding.linear_1",
          "time_embedding.linear_2",
          "up_blocks.3.resnets.2.conv_shortcut",
          "up_blocks.3.resnets.0.conv_shortcut"
        ],
        "sizes": [
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1310720,
          1638400,
          409600,
          1638400,
          3276800,
          1638400,
          1638400,
          6553600,
          14745600,
          14745600,
          1638400,
          1638400,
          13107200,
          3276800,
          14745600,
          29491200,
          1310720,
          3276800,
          1638400,
          14745600,
          14745600,
          29491200,
          1638400,
          14745600,
          1638400,
          1638400,
          14745600,
          819200,
          14745600,
          14745600,
          29491200,
          1638400,
          655360,
          1638400,
          409600,
          14745600,
          655360,
          14745600,
          1638400,
          1638400,
          1638400,
          1310720,
          14745600,
          1638400,
          409600,
          409600,
          655360,
          1638400,
          6553600,
          327680,
          1638400,
          6553600,
          409600,
          1638400,
          1310720,
          1310720,
          14745600,
          1638400,
          1638400,
          102400,
          6553600,
          29491200,
          1638400,
          14745600,
          1638400,
          1638400,
          1638400,
          102400,
          1638400,
          1638400,
          22118400,
          29491200,
          13107200,
          6553600,
          14745600,
          409600,
          14745600,
          1638400,
          1638400,
          1638400,
          11059200,
          13107200,
          14745600,
          1638400,
          1638400,
          327680,
          13107200,
          1638400,
          327680,
          7372800,
          819200,
          1638400,
          1310720,
          1638400,
          1638400,
          409600,
          1638400,
          1310720,
          409600,
          1638400,
          1638400,
          3276800,
          409600,
          409600,
          6553600,
          409600,
          1638400,
          14745600,
          1638400,
          1638400,
          7372800,
          409600,
          3276800,
          2457600,
          1638400,
          1638400,
          409600,
          14745600,
          819200,
          1638400,
          1638400,
          3686400,
          409600,
          102400,
          3686400,
          3276800,
          409600,
          655360,
          1638400,
          102400,
          409600,
          13107200,
          1638400,
          102400,
          1310720,
          1638400,
          409600,
          1638400,
          655360,
          102400,
          3686400,
          409600,
          1638400,
          1310720,
          409600,
          409600,
          13107200,
          409600,
          3686400,
          819200,
          409600,
          14745600,
          1310720,
          1310720,
          327680,
          409600,
          5529600,
          3686400,
          3686400,
          1638400,
          1843200,
          409600,
          409600,
          3276800,
          819200,
          409600,
          102400,
          655360,
          102400,
          3686400,
          409600,
          1638400,
          1638400,
          327680,
          102400,
          409600,
          3276800,
          327680,
          102400,
          409600,
          102400,
          409600,
          1310720,
          409600,
          409600,
          409600,
          102400,
          102400,
          655360,
          409600,
          409600,
          102400,
          409600,
          409600,
          327680,
          102400,
          3276800,
          655360,
          102400,
          102400,
          409600,
          819200,
          409600,
          3276800,
          102400,
          327680,
          921600,
          409600,
          819200,
          921600,
          655360,
          409600,
          102400,
          102400,
          102400,
          102400,
          409600,
          102400,
          409600,
          102400,
          819200,
          102400,
          819200,
          102400,
          409600,
          327680,
          102400,
          102400,
          819200,
          204800,
          3686400,
          921600,
          409600,
          102400,
          655360,
          409600,
          921600,
          102400,
          409600,
          614400,
          2764800,
          921600,
          102400,
          102400,
          409600,
          327680,
          102400,
          409600,
          921600,
          102400,
          819200,
          102400,
          102400,
          1228800,
          102400,
          102400,
          1843200,
          1843200,
          921600,
          819200,
          204800,
          102400,
          409600,
          102400,
          921600,
          409600,
          1638400,
          204800,
          307200
        ]
      },
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.7,
        88.7,
        88.5,
        88.5
      ],
      "mid_block.resnets.1.time_emb_proj": [
        88.5,
        88.4,
        88.7,
        88.0
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        88.5,
        88.3,
        88.4,
        88.2
      ],
      "down_blocks.3.resnets.1.time_emb_proj": [
        88.3,
        87.9,
        88.3,
        87.3
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.1,
        87.9,
        88.4,
        88.1
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        88.1,
        88.3,
        88.4,
        88.3
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        87.9,
        88.0,
        88.3,
        88.0
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        87.8,
        87.9,
        87.8,
        88.0
      ],
      "down_blocks.3.resnets.0.time_emb_proj": [
        87.4,
        87.9,
        88.1,
        87.5
      ],
      "mid_block.resnets.0.time_emb_proj": [
        87.4,
        87.6,
        88.1,
        86.7
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        87.4,
        87.7,
        88.0,
        87.6
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        87.3,
        87.6,
        88.0,
        87.3
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        87.6,
        87.4,
        87.8,
        87.3
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        87.2,
        86.9,
        87.6,
        86.7
      ],
      "mid_block.attentions.0.proj_in": [
        86.8,
        87.2,
        87.4,
        85.9
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        86.3,
        86.7,
        87.2,
        85.5
      ],
      "mid_block.attentions.0.proj_out": [
        86.1,
        86.8,
        87.1,
        85.5
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        86.3,
        86.7,
        87.0,
        82.7
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        86.1,
        86.5,
        86.8,
        85.2
      ],
      "down_blocks.3.resnets.1.conv1": [
        86.6,
        86.8,
        86.7,
        75.5
      ],
      "mid_block.resnets.0.conv1": [
        86.4,
        86.9,
        87.1,
        78.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        86.3,
        86.3,
        86.8,
        78.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        86.3,
        86.2,
        86.8,
        77.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        86.2,
        86.2,
        86.9,
        77.9
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        86.0,
        86.3,
        87.0,
        76.2
      ],
      "down_blocks.3.resnets.1.conv2": [
        86.2,
        86.3,
        86.8,
        76.9
      ],
      "up_blocks.0.resnets.0.conv1": [
        85.3,
        85.6,
        86.3,
        73.6
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        85.3,
        85.6,
        86.4,
        73.6
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        85.2,
        85.4,
        86.4,
        73.5
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        84.9,
        84.6,
        86.1,
        73.7
      ],
      "down_blocks.3.resnets.0.conv2": [
        84.1,
        84.5,
        84.7,
        73.5
      ],
      "down_blocks.3.resnets.0.conv1": [
        83.9,
        84.2,
        79.1,
        73.5
      ],
      "up_blocks.0.resnets.1.conv1": [
        83.6,
        83.8,
        85.1,
        83.0
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        82.6,
        82.8,
        84.3,
        81.9
      ],
      "mid_block.resnets.0.conv2": [
        82.3,
        82.8,
        84.0,
        81.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        83.3,
        83.5,
        84.3,
        82.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        83.3,
        83.2,
        84.3,
        81.4
      ],
      "mid_block.resnets.1.conv1": [
        83.3,
        83.2,
        84.2,
        81.5
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        82.8,
        82.5,
        83.8,
        80.5
      ],
      "up_blocks.0.resnets.0.conv2": [
        80.4,
        82.2,
        82.7,
        76.8
      ],
      "mid_block.resnets.1.conv2": [
        80.7,
        82.0,
        82.5,
        80.0
      ],
      "up_blocks.0.resnets.2.conv1": [
        80.7,
        81.7,
        76.4,
        79.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        80.2,
        80.9,
        76.1,
        78.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        79.8,
        81.2,
        75.1,
        77.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        80.2,
        81.2,
        76.3,
        77.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        80.4,
        81.0,
        76.8,
        78.2
      ],
      "up_blocks.0.resnets.1.conv2": [
        79.0,
        79.7,
        78.4,
        78.4
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        78.7,
        79.7,
        78.2,
        78.1
      ],
      "down_blocks.2.downsamplers.0.conv": [
        78.7,
        79.8,
        77.9,
        77.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        78.7,
        79.6,
        76.3,
        76.8
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        77.7,
        78.3,
        75.8,
        76.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        77.8,
        78.3,
        76.8,
        75.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        78.2,
        78.9,
        74.2,
        76.1
      ],
      "down_blocks.2.resnets.1.conv2": [
        77.2,
        77.9,
        75.5,
        76.0
      ],
      "up_blocks.1.attentions.1.proj_out": [
        76.7,
        78.4,
        75.7,
        75.9
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        76.6,
        78.5,
        75.0,
        75.7
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        76.5,
        78.4,
        73.8,
        75.6
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        76.7,
        78.5,
        73.9,
        75.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        77.1,
        78.3,
        75.0,
        75.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        77.1,
        78.6,
        75.1,
        75.7
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        77.0,
        78.4,
        78.2,
        75.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        77.4,
        78.0,
        68.6,
        70.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        77.6,
        78.2,
        69.6,
        70.5
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        77.9,
        78.3,
        68.7,
        70.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        78.0,
        77.9,
        73.9,
        74.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        78.1,
        78.0,
        74.6,
        74.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        78.1,
        78.2,
        74.9,
        74.8
      ],
      "up_blocks.1.resnets.2.conv2": [
        77.6,
        77.6,
        74.9,
        74.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        78.1,
        77.1,
        72.9,
        74.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        77.5,
        76.4,
        77.4,
        73.5
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        77.2,
        76.3,
        76.9,
        73.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        76.6,
        76.0,
        76.6,
        73.0
      ],
      "up_blocks.1.resnets.0.conv1": [
        72.8,
        75.5,
        77.1,
        73.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        76.2,
        76.6,
        76.6,
        72.8
      ],
      "down_blocks.2.resnets.1.conv1": [
        76.5,
        76.2,
        76.4,
        73.4
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        75.5,
        75.8,
        76.3,
        73.9
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        73.8,
        75.9,
        74.1,
        74.2
      ],
      "up_blocks.1.attentions.0.proj_out": [
        74.1,
        75.1,
        76.4,
        72.8
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        74.0,
        74.9,
        76.5,
        72.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        73.9,
        75.0,
        76.6,
        74.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        73.0,
        74.1,
        75.3,
        74.2
      ],
      "up_blocks.1.resnets.2.conv1": [
        72.2,
        74.9,
        75.2,
        74.4
      ],
      "up_blocks.1.resnets.1.conv1": [
        71.6,
        74.5,
        74.8,
        74.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        76.5,
        76.2,
        77.4,
        75.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        76.6,
        75.5,
        77.1,
        75.2
      ],
      "up_blocks.1.resnets.1.conv2": [
        76.2,
        75.4,
        77.1,
        75.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        76.1,
        75.4,
        77.1,
        75.2
      ],
      "up_blocks.1.resnets.0.conv2": [
        76.0,
        75.6,
        76.7,
        75.0
      ],
      "down_blocks.2.attentions.0.proj_in": [
        76.2,
        76.1,
        76.8,
        74.9
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        76.6,
        76.5,
        77.0,
        75.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        75.7,
        75.6,
        76.5,
        74.8
      ],
      "up_blocks.2.resnets.0.conv1": [
        75.8,
        75.4,
        76.3,
        74.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        76.0,
        75.6,
        76.2,
        74.7
      ],
      "up_blocks.0.resnets.2.conv2": [
        76.3,
        75.7,
        76.6,
        74.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        76.3,
        75.6,
        76.5,
        74.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        76.2,
        75.5,
        76.4,
        74.7
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v": [
        75.5,
        75.0,
        75.6,
        74.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        75.8,
        75.0,
        75.8,
        74.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        76.1,
        75.4,
        75.8,
        74.5
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v": [
        75.9,
        75.0,
        75.5,
        74.3
      ],
      "down_blocks.2.resnets.0.conv1": [
        75.8,
        74.8,
        75.3,
        74.4
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        75.3,
        74.3,
        75.0,
        74.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        74.8,
        73.8,
        74.5,
        73.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        74.8,
        73.4,
        74.6,
        73.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        74.6,
        73.6,
        74.5,
        73.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        74.3,
        72.9,
        74.5,
        73.6
      ],
      "down_blocks.1.attentions.1.proj_out": [
        74.9,
        73.6,
        75.2,
        73.9
      ],
      "down_blocks.2.attentions.1.proj_in": [
        74.7,
        73.2,
        75.4,
        73.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        74.7,
        73.3,
        75.4,
        73.9
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        74.2,
        72.8,
        74.7,
        73.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        74.1,
        72.8,
        74.7,
        73.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        74.1,
        73.1,
        74.9,
        73.2
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        73.9,
        73.2,
        74.7,
        73.0
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        73.2,
        72.4,
        74.7,
        72.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        72.8,
        71.7,
        74.0,
        72.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        72.8,
        71.6,
        73.9,
        72.4
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        72.6,
        71.8,
        74.0,
        72.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        72.6,
        71.9,
        74.0,
        72.4
      ],
      "up_blocks.1.upsamplers.0.conv": [
        73.4,
        72.3,
        74.3,
        72.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        73.4,
        72.3,
        74.2,
        72.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        73.3,
        72.1,
        74.2,
        72.9
      ],
      "up_blocks.2.resnets.1.conv1": [
        73.2,
        71.9,
        74.2,
        73.0
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        73.2,
        72.1,
        74.3,
        73.1
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        73.1,
        72.7,
        73.9,
        72.9
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        73.3,
        72.6,
        74.0,
        72.8
      ],
      "down_blocks.2.attentions.1.proj_out": [
        73.3,
        72.4,
        74.0,
        72.8
      ],
      "up_blocks.1.attentions.1.proj_in": [
        73.5,
        72.5,
        74.2,
        72.8
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        73.5,
        72.9,
        74.2,
        72.7
      ],
      "up_blocks.0.upsamplers.0.conv": [
        73.3,
        72.6,
        73.9,
        72.5
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        71.7,
        71.5,
        72.3,
        70.9
      ],
      "down_blocks.2.attentions.0.proj_out": [
        71.6,
        71.5,
        72.2,
        70.9
      ],
      "up_blocks.1.attentions.0.proj_in": [
        71.8,
        71.7,
        72.4,
        71.0
      ],
      "up_blocks.2.resnets.1.conv2": [
        71.2,
        71.7,
        72.3,
        71.1
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        71.4,
        71.7,
        72.5,
        71.3
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        71.1,
        71.4,
        72.1,
        71.1
      ],
      "up_blocks.2.resnets.0.conv2": [
        71.3,
        71.6,
        71.8,
        70.9
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        69.8,
        70.9,
        71.1,
        70.4
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k": [
        69.9,
        71.0,
        70.9,
        70.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        69.9,
        70.8,
        70.8,
        70.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        69.6,
        70.7,
        70.7,
        70.0
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q": [
        69.6,
        70.6,
        70.6,
        70.0
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q": [
        69.1,
        70.0,
        70.2,
        69.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        69.1,
        70.0,
        70.2,
        69.6
      ],
      "up_blocks.1.attentions.2.proj_out": [
        68.8,
        70.2,
        69.9,
        69.2
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        68.4,
        69.8,
        69.6,
        69.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        68.4,
        69.9,
        69.6,
        69.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        68.5,
        69.9,
        69.6,
        69.0
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        68.1,
        69.7,
        69.6,
        69.0
      ],
      "up_blocks.1.attentions.2.proj_in": [
        68.1,
        69.3,
        69.9,
        69.2
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v": [
        68.2,
        69.3,
        70.1,
        69.4
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        68.3,
        69.6,
        70.4,
        69.5
      ],
      "down_blocks.1.resnets.1.conv2": [
        68.2,
        69.4,
        70.2,
        69.6
      ],
      "up_blocks.3.resnets.0.time_emb_proj": [
        68.2,
        69.4,
        70.1,
        69.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        68.2,
        69.4,
        69.9,
        69.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        67.7,
        68.3,
        70.1,
        69.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        67.6,
        68.3,
        69.8,
        69.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        67.7,
        68.2,
        69.7,
        69.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        67.7,
        68.3,
        69.8,
        69.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        67.7,
        68.2,
        69.7,
        69.2
      ],
      "down_blocks.1.resnets.1.conv1": [
        67.7,
        67.9,
        69.5,
        68.9
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        67.6,
        67.8,
        69.2,
        68.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        67.6,
        67.8,
        69.3,
        68.7
      ],
      "down_blocks.2.resnets.0.conv2": [
        67.7,
        67.7,
        69.1,
        68.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        67.5,
        67.9,
        69.2,
        68.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        67.4,
        68.5,
        69.6,
        69.1
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        67.3,
        68.5,
        69.6,
        69.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        67.3,
        68.5,
        69.5,
        69.0
      ],
      "up_blocks.2.resnets.2.conv1": [
        67.3,
        68.3,
        69.4,
        69.1
      ],
      "down_blocks.1.downsamplers.0.conv": [
        67.1,
        68.1,
        69.5,
        69.2
      ],
      "down_blocks.1.resnets.0.conv2": [
        67.2,
        67.4,
        69.5,
        69.2
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        68.5,
        68.6,
        70.3,
        69.8
      ],
      "down_blocks.1.resnets.0.conv1": [
        68.7,
        68.9,
        70.4,
        69.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        68.4,
        69.0,
        70.3,
        69.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        68.6,
        69.1,
        70.1,
        69.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        68.7,
        69.1,
        70.0,
        69.7
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        68.3,
        67.7,
        69.4,
        68.9
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q": [
        68.0,
        67.4,
        69.4,
        68.9
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        67.8,
        67.3,
        69.2,
        68.5
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        67.8,
        67.7,
        68.7,
        68.2
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        67.8,
        67.7,
        68.7,
        68.2
      ],
      "up_blocks.2.resnets.2.conv2": [
        67.1,
        66.8,
        68.0,
        67.7
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        66.4,
        65.9,
        68.3,
        67.5
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2": [
        66.2,
        65.5,
        67.4,
        66.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        67.2,
        65.4,
        67.7,
        67.1
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        67.1,
        65.4,
        67.7,
        67.1
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        66.8,
        65.0,
        67.3,
        66.6
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        66.2,
        64.4,
        66.5,
        65.8
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        66.1,
        64.1,
        66.0,
        65.2
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        66.1,
        64.0,
        66.0,
        65.2
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        66.4,
        64.6,
        66.4,
        65.5
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        65.4,
        63.9,
        65.3,
        64.5
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q": [
        65.2,
        63.8,
        65.2,
        64.5
      ],
      "down_blocks.1.attentions.0.proj_out": [
        65.4,
        64.1,
        65.4,
        64.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        65.3,
        64.0,
        65.5,
        64.7
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        65.0,
        63.6,
        65.0,
        64.2
      ],
      "up_blocks.2.attentions.1.proj_out": [
        65.0,
        63.9,
        65.1,
        64.4
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        64.8,
        63.8,
        64.9,
        64.3
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        64.8,
        63.8,
        64.8,
        64.2
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        64.6,
        63.7,
        64.3,
        63.8
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k": [
        64.4,
        63.6,
        64.3,
        63.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2": [
        64.1,
        63.5,
        64.3,
        63.7
      ],
      "up_blocks.2.attentions.1.proj_in": [
        64.1,
        63.6,
        64.3,
        63.7
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        64.0,
        63.6,
        64.1,
        63.5
      ],
      "down_blocks.1.attentions.0.proj_in": [
        63.7,
        63.4,
        64.1,
        63.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        63.5,
        63.4,
        63.7,
        63.1
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k": [
        63.5,
        63.3,
        63.8,
        63.1
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        63.7,
        63.3,
        64.4,
        63.6
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        62.8,
        62.4,
        62.9,
        62.5
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        63.0,
        62.5,
        63.0,
        62.5
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        63.1,
        62.6,
        62.8,
        62.4
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k": [
        62.9,
        62.1,
        62.4,
        62.0
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v": [
        62.1,
        60.9,
        61.4,
        61.3
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        61.3,
        60.3,
        60.9,
        60.8
      ],
      "up_blocks.2.attentions.0.proj_in": [
        61.1,
        60.7,
        61.3,
        61.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        61.1,
        60.5,
        61.2,
        60.9
      ],
      "down_blocks.0.attentions.0.proj_out": [
        61.1,
        60.4,
        61.2,
        61.0
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v": [
        61.3,
        60.5,
        61.3,
        61.0
      ],
      "down_blocks.0.resnets.0.conv1": [
        61.0,
        60.3,
        61.1,
        60.9
      ],
      "down_blocks.1.attentions.1.proj_in": [
        61.0,
        60.4,
        61.2,
        60.9
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        60.8,
        60.2,
        60.9,
        60.7
      ],
      "down_blocks.0.resnets.1.conv2": [
        60.3,
        60.0,
        60.8,
        60.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        60.4,
        60.1,
        60.8,
        60.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        60.2,
        59.9,
        60.7,
        60.5
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q": [
        60.2,
        59.9,
        60.8,
        60.6
      ],
      "down_blocks.0.attentions.1.proj_out": [
        60.3,
        60.3,
        61.0,
        60.7
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q": [
        59.7,
        59.9,
        60.4,
        60.1
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v": [
        60.0,
        59.8,
        60.3,
        60.1
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        59.3,
        59.1,
        60.0,
        59.9
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        59.3,
        59.1,
        60.1,
        59.9
      ],
      "up_blocks.2.attentions.0.proj_out": [
        60.1,
        59.3,
        60.2,
        60.2
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q": [
        59.8,
        59.1,
        59.7,
        59.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        59.6,
        58.9,
        59.6,
        59.6
      ],
      "up_blocks.3.attentions.0.proj_out": [
        59.7,
        59.5,
        60.2,
        60.1
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        59.4,
        58.4,
        59.9,
        59.8
      ],
      "down_blocks.0.attentions.0.proj_in": [
        59.2,
        58.4,
        59.8,
        59.8
      ],
      "up_blocks.2.attentions.2.proj_out": [
        59.5,
        59.1,
        60.0,
        60.2
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k": [
        59.4,
        59.0,
        60.1,
        60.2
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q": [
        59.0,
        58.8,
        59.6,
        59.8
      ],
      "down_blocks.0.attentions.1.proj_in": [
        59.1,
        58.9,
        59.6,
        59.7
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        58.8,
        58.5,
        59.4,
        59.5
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        58.7,
        58.2,
        59.4,
        59.5
      ],
      "up_blocks.2.upsamplers.0.conv": [
        58.8,
        59.3,
        60.1,
        60.1
      ],
      "down_blocks.0.resnets.1.conv1": [
        58.6,
        59.2,
        60.0,
        59.9
      ],
      "up_blocks.2.attentions.2.proj_in": [
        57.9,
        59.1,
        59.4,
        59.3
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k": [
        57.8,
        58.8,
        59.2,
        59.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        58.0,
        58.8,
        59.4,
        59.3
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        57.9,
        58.2,
        58.7,
        58.8
      ],
      "up_blocks.3.resnets.0.conv2": [
        57.9,
        58.1,
        58.4,
        58.5
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        57.9,
        58.0,
        58.4,
        58.6
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2": [
        57.6,
        57.5,
        58.2,
        58.2
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        58.0,
        58.5,
        58.7,
        58.6
      ],
      "up_blocks.3.resnets.0.conv1": [
        57.9,
        57.9,
        58.3,
        58.3
      ],
      "down_blocks.0.downsamplers.0.conv": [
        58.1,
        58.0,
        58.7,
        58.5
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k": [
        58.2,
        58.2,
        58.7,
        58.5
      ],
      "up_blocks.3.attentions.0.proj_in": [
        58.3,
        58.4,
        58.9,
        58.6
      ],
      "up_blocks.3.resnets.1.time_emb_proj": [
        57.9,
        57.9,
        58.4,
        58.0
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k": [
        57.8,
        57.8,
        58.3,
        57.9
      ],
      "up_blocks.3.attentions.1.proj_out": [
        57.6,
        57.8,
        57.9,
        57.6
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2": [
        56.5,
        56.9,
        56.8,
        56.5
      ],
      "down_blocks.0.resnets.0.conv2": [
        56.2,
        56.9,
        56.8,
        56.6
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v": [
        55.6,
        56.3,
        56.4,
        56.1
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        55.4,
        55.9,
        56.2,
        56.0
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        54.9,
        55.7,
        55.8,
        55.5
      ],
      "up_blocks.3.attentions.2.proj_out": [
        54.2,
        55.0,
        54.9,
        54.7
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        54.6,
        54.7,
        53.6,
        53.7
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v": [
        54.4,
        54.4,
        53.3,
        53.5
      ],
      "up_blocks.3.attentions.1.proj_in": [
        54.2,
        53.9,
        53.2,
        53.4
      ],
      "up_blocks.3.resnets.1.conv1": [
        53.8,
        53.0,
        52.6,
        52.8
      ],
      "up_blocks.3.resnets.2.conv1": [
        53.7,
        53.5,
        53.4,
        53.5
      ],
      "up_blocks.3.resnets.1.conv2": [
        53.7,
        54.0,
        53.8,
        54.0
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        52.4,
        52.4,
        52.5,
        52.6
      ],
      "up_blocks.3.resnets.1.conv_shortcut": [
        52.4,
        52.1,
        52.5,
        52.5
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        52.2,
        52.1,
        52.2,
        52.3
      ],
      "up_blocks.3.resnets.2.time_emb_proj": [
        51.1,
        51.3,
        51.2,
        51.3
      ],
      "up_blocks.3.attentions.2.proj_in": [
        51.1,
        51.4,
        51.4,
        51.5
      ],
      "up_blocks.3.resnets.2.conv2": [
        50.9,
        51.2,
        51.3,
        51.3
      ],
      "time_embedding.linear_1": [
        51.0,
        51.2,
        51.2,
        51.1
      ],
      "time_embedding.linear_2": [
        51.0,
        51.2,
        51.1,
        51.1
      ],
      "up_blocks.3.resnets.2.conv_shortcut": [
        54.8,
        54.8,
        54.5,
        54.5
      ],
      "up_blocks.3.resnets.0.conv_shortcut": [
        54.3,
        53.5,
        55.8,
        55.3
      ]
    },
    "2": {
      "metadata": {
        "nbits": 2,
        "out_ngroups": 1,
        "in_ngroups": 1,
        "cumulative": true,
        "candidates": [
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.resnets.2.time_emb_proj",
          "up_blocks.0.resnets.0.time_emb_proj",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "mid_block.resnets.0.time_emb_proj",
          "down_blocks.3.resnets.0.time_emb_proj",
          "down_blocks.3.resnets.1.time_emb_proj",
          "up_blocks.0.resnets.1.time_emb_proj",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_v",
          "mid_block.resnets.1.time_emb_proj",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.2",
          "mid_block.attentions.0.proj_out",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_q",
          "mid_block.attentions.0.proj_in",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_v",
          "mid_block.resnets.0.conv1",
          "mid_block.resnets.0.conv2",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.3.resnets.1.conv1",
          "mid_block.resnets.1.conv2",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.3.resnets.1.conv2",
          "up_blocks.0.resnets.1.conv1",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.resnets.1.conv2",
          "down_blocks.3.resnets.0.conv2",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.resnets.0.conv1",
          "up_blocks.0.resnets.2.conv_shortcut",
          "mid_block.resnets.1.conv1",
          "up_blocks.0.resnets.1.conv_shortcut",
          "up_blocks.0.resnets.0.conv_shortcut",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.resnets.0.conv2",
          "up_blocks.1.resnets.1.time_emb_proj",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.resnets.2.time_emb_proj",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.resnets.1.conv1",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.2.resnets.0.time_emb_proj",
          "down_blocks.2.resnets.1.conv2",
          "down_blocks.3.resnets.0.conv1",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.0.proj_out",
          "up_blocks.1.attentions.1.proj_out",
          "up_blocks.1.resnets.0.time_emb_proj",
          "down_blocks.2.downsamplers.0.conv",
          "up_blocks.1.resnets.2.conv2",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.resnets.0.conv1",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.resnets.2.conv1",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.resnets.0.conv2",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.resnets.0.conv2",
          "up_blocks.2.resnets.0.conv2",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.resnets.2.conv1",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.2.resnets.1.time_emb_proj",
          "up_blocks.1.resnets.0.conv1",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.2.resnets.0.time_emb_proj",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.3.resnets.0.time_emb_proj",
          "up_blocks.1.upsamplers.0.conv",
          "up_blocks.1.attentions.2.proj_out",
          "up_blocks.1.resnets.2.conv_shortcut",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.1.resnets.0.conv1",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.2.resnets.1.conv1",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.resnets.1.conv_shortcut",
          "down_blocks.1.resnets.1.conv1",
          "up_blocks.1.attentions.1.proj_in",
          "up_blocks.1.resnets.0.conv_shortcut",
          "up_blocks.2.resnets.1.conv2",
          "down_blocks.1.attentions.1.proj_out",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.1.resnets.1.time_emb_proj",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.proj_in",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.2.resnets.0.conv1",
          "up_blocks.1.resnets.1.conv1",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.resnets.2.conv2",
          "up_blocks.1.resnets.1.conv2",
          "up_blocks.0.upsamplers.0.conv",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.resnets.1.time_emb_proj",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.0.proj_in",
          "down_blocks.2.attentions.1.proj_out",
          "down_blocks.2.attentions.1.proj_in",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.2.resnets.2.conv1",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.downsamplers.0.conv",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.resnets.2.conv2",
          "down_blocks.1.resnets.0.conv2",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v",
          "down_blocks.1.resnets.1.conv2",
          "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.2.proj_in",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.resnets.0.conv_shortcut",
          "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.attentions.0.proj_in",
          "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.2.attentions.0.proj_in",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.attentions.0.proj_out",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2",
          "down_blocks.0.resnets.0.time_emb_proj",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q",
          "time_embedding.linear_2",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.2.attentions.0.proj_out",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.2.attentions.1.proj_out",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k",
          "down_blocks.0.resnets.0.conv1",
          "down_blocks.1.attentions.1.proj_in",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
          "down_blocks.1.attentions.0.proj_out",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.resnets.1.time_emb_proj",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.2.resnets.1.conv_shortcut",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.0.attentions.0.proj_out",
          "down_blocks.1.resnets.0.time_emb_proj",
          "up_blocks.2.attentions.2.proj_out",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.3.attentions.0.proj_out",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v",
          "down_blocks.0.resnets.1.conv2",
          "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.0.attentions.1.proj_out",
          "up_blocks.2.attentions.1.proj_in",
          "up_blocks.2.resnets.0.conv_shortcut",
          "down_blocks.0.resnets.1.conv1",
          "up_blocks.2.resnets.2.time_emb_proj",
          "up_blocks.2.upsamplers.0.conv",
          "down_blocks.0.attentions.0.proj_in",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.0.resnets.1.time_emb_proj",
          "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.resnets.0.conv2",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.2.resnets.2.conv_shortcut",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.0.downsamplers.0.conv",
          "up_blocks.3.resnets.0.conv1",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k",
          "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2",
          "down_blocks.0.attentions.1.proj_in",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k",
          "up_blocks.3.attentions.0.proj_in",
          "down_blocks.0.resnets.0.conv2",
          "up_blocks.2.attentions.2.proj_in",
          "up_blocks.3.attentions.1.proj_out",
          "up_blocks.3.attentions.2.proj_out",
          "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v",
          "down_blocks.1.resnets.0.conv_shortcut",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.3.resnets.2.conv1",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.resnets.1.conv1",
          "up_blocks.3.resnets.2.time_emb_proj",
          "up_blocks.3.attentions.1.proj_in",
          "time_embedding.linear_1",
          "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.resnets.1.conv2",
          "up_blocks.3.resnets.0.conv_shortcut",
          "up_blocks.3.resnets.2.conv2",
          "up_blocks.3.resnets.1.conv_shortcut",
          "up_blocks.3.attentions.2.proj_in",
          "up_blocks.3.resnets.2.conv_shortcut"
        ],
        "sizes": [
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1310720,
          14745600,
          14745600,
          1638400,
          14745600,
          14745600,
          409600,
          14745600,
          29491200,
          13107200,
          14745600,
          14745600,
          1638400,
          1638400,
          409600,
          1638400,
          29491200,
          3276800,
          14745600,
          3276800,
          3276800,
          1310720,
          1638400,
          14745600,
          1638400,
          1310720,
          1638400,
          1638400,
          1638400,
          655360,
          14745600,
          1310720,
          409600,
          409600,
          1310720,
          655360,
          819200,
          14745600,
          14745600,
          1638400,
          1638400,
          409600,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          14745600,
          14745600,
          409600,
          6553600,
          6553600,
          1638400,
          1638400,
          1638400,
          1310720,
          1638400,
          13107200,
          1638400,
          409600,
          1638400,
          1638400,
          11059200,
          1638400,
          13107200,
          409600,
          102400,
          409600,
          102400,
          1638400,
          1638400,
          29491200,
          6553600,
          1638400,
          102400,
          14745600,
          1638400,
          14745600,
          3686400,
          655360,
          1638400,
          409600,
          409600,
          327680,
          22118400,
          409600,
          409600,
          6553600,
          102400,
          13107200,
          102400,
          3276800,
          1638400,
          29491200,
          1638400,
          1638400,
          655360,
          409600,
          14745600,
          1638400,
          2457600,
          409600,
          1310720,
          1638400,
          1843200,
          1638400,
          7372800,
          327680,
          3276800,
          3686400,
          1638400,
          3276800,
          3686400,
          409600,
          409600,
          1638400,
          327680,
          819200,
          1638400,
          102400,
          1638400,
          409600,
          1638400,
          7372800,
          29491200,
          409600,
          1638400,
          14745600,
          14745600,
          14745600,
          1638400,
          819200,
          102400,
          1638400,
          1638400,
          1638400,
          409600,
          409600,
          5529600,
          409600,
          409600,
          3686400,
          1638400,
          3686400,
          3686400,
          13107200,
          655360,
          3686400,
          1638400,
          102400,
          1638400,
          102400,
          409600,
          102400,
          819200,
          3276800,
          409600,
          1638400,
          102400,
          409600,
          327680,
          1638400,
          409600,
          409600,
          409600,
          409600,
          1638400,
          409600,
          1638400,
          409600,
          102400,
          1310720,
          3276800,
          102400,
          655360,
          13107200,
          409600,
          1310720,
          921600,
          409600,
          102400,
          409600,
          102400,
          102400,
          409600,
          3276800,
          409600,
          409600,
          409600,
          409600,
          102400,
          409600,
          102400,
          409600,
          3276800,
          102400,
          1310720,
          102400,
          819200,
          102400,
          819200,
          327680,
          102400,
          819200,
          409600,
          102400,
          327680,
          327680,
          102400,
          1310720,
          409600,
          921600,
          819200,
          102400,
          409600,
          1228800,
          921600,
          819200,
          3686400,
          102400,
          102400,
          102400,
          327680,
          409600,
          819200,
          921600,
          655360,
          102400,
          409600,
          614400,
          1310720,
          102400,
          327680,
          655360,
          921600,
          2764800,
          655360,
          409600,
          102400,
          102400,
          327680,
          102400,
          921600,
          409600,
          102400,
          102400,
          819200,
          102400,
          204800,
          102400,
          655360,
          1843200,
          102400,
          1843200,
          409600,
          102400,
          409600,
          819200,
          921600,
          307200,
          921600,
          204800,
          102400,
          204800
        ]
      },
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        89.1,
        88.7,
        88.9,
        88.5
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        89.1,
        88.6,
        88.7,
        88.9
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        89.0,
        88.8,
        89.0,
        88.4
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        88.7,
        88.7,
        88.8,
        88.5
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        88.6,
        88.4,
        88.4,
        88.4
      ],
      "mid_block.resnets.0.time_emb_proj": [
        88.3,
        88.5,
        88.8,
        88.1
      ],
      "down_blocks.3.resnets.0.time_emb_proj": [
        88.4,
        88.3,
        88.5,
        88.2
      ],
      "down_blocks.3.resnets.1.time_emb_proj": [
        88.5,
        88.5,
        88.8,
        88.1
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        88.3,
        88.4,
        88.4,
        87.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.2,
        88.3,
        88.1,
        88.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.1,
        88.2,
        88.4,
        87.5
      ],
      "mid_block.resnets.1.time_emb_proj": [
        88.1,
        88.2,
        88.3,
        87.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        88.1,
        88.2,
        88.2,
        87.9
      ],
      "mid_block.attentions.0.proj_out": [
        87.7,
        88.1,
        88.0,
        87.4
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        87.7,
        87.9,
        88.1,
        87.0
      ],
      "mid_block.attentions.0.proj_in": [
        87.9,
        88.1,
        88.2,
        86.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        87.8,
        88.2,
        88.0,
        86.3
      ],
      "mid_block.resnets.0.conv1": [
        88.0,
        88.2,
        88.3,
        83.7
      ],
      "mid_block.resnets.0.conv2": [
        87.6,
        88.1,
        88.0,
        82.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        87.7,
        88.0,
        87.8,
        82.7
      ],
      "down_blocks.3.resnets.1.conv1": [
        87.4,
        87.7,
        87.9,
        80.3
      ],
      "mid_block.resnets.1.conv2": [
        87.4,
        87.6,
        87.8,
        79.7
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        87.1,
        87.5,
        87.5,
        78.7
      ],
      "down_blocks.3.resnets.1.conv2": [
        87.0,
        87.0,
        87.4,
        77.8
      ],
      "up_blocks.0.resnets.1.conv1": [
        86.8,
        87.6,
        87.5,
        76.5
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        86.6,
        87.3,
        87.5,
        77.0
      ],
      "up_blocks.0.resnets.1.conv2": [
        86.4,
        87.0,
        87.2,
        85.7
      ],
      "down_blocks.3.resnets.0.conv2": [
        86.5,
        86.8,
        87.3,
        86.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        86.6,
        87.1,
        87.3,
        85.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        86.2,
        86.6,
        87.3,
        85.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        85.7,
        86.1,
        87.2,
        85.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        85.5,
        85.9,
        87.0,
        85.0
      ],
      "up_blocks.0.resnets.0.conv1": [
        85.7,
        85.8,
        87.1,
        85.0
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        85.6,
        85.8,
        87.0,
        81.8
      ],
      "mid_block.resnets.1.conv1": [
        85.7,
        85.9,
        86.9,
        83.9
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        85.5,
        85.8,
        87.0,
        84.8
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        85.8,
        86.0,
        86.9,
        84.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        85.6,
        86.2,
        87.0,
        85.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        85.6,
        86.2,
        87.0,
        81.5
      ],
      "up_blocks.0.resnets.0.conv2": [
        85.4,
        85.9,
        86.8,
        83.9
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        84.8,
        85.4,
        86.6,
        83.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        85.5,
        85.5,
        86.7,
        81.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        85.1,
        85.5,
        78.3,
        83.7
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        84.6,
        84.8,
        78.6,
        83.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        83.9,
        84.5,
        77.3,
        83.4
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        84.0,
        84.2,
        77.4,
        83.2
      ],
      "down_blocks.2.resnets.1.conv1": [
        84.2,
        84.0,
        80.6,
        82.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        84.0,
        84.0,
        80.5,
        82.2
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        83.9,
        83.8,
        80.4,
        82.4
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        83.3,
        83.4,
        80.3,
        81.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        83.4,
        83.5,
        80.3,
        81.7
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        83.3,
        83.2,
        80.2,
        81.7
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        82.8,
        82.9,
        80.0,
        81.1
      ],
      "down_blocks.2.resnets.1.conv2": [
        82.6,
        82.9,
        77.2,
        77.7
      ],
      "down_blocks.3.resnets.0.conv1": [
        82.7,
        82.9,
        76.3,
        77.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.6,
        82.6,
        78.0,
        77.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        82.7,
        82.5,
        77.2,
        77.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        82.3,
        82.3,
        78.5,
        77.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        82.2,
        82.2,
        78.4,
        77.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        81.6,
        82.2,
        78.3,
        77.6
      ],
      "up_blocks.1.attentions.0.proj_out": [
        81.4,
        82.0,
        78.2,
        77.5
      ],
      "up_blocks.1.attentions.1.proj_out": [
        81.5,
        82.0,
        78.9,
        77.5
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        81.1,
        81.7,
        79.3,
        77.5
      ],
      "down_blocks.2.downsamplers.0.conv": [
        81.0,
        81.6,
        82.9,
        77.4
      ],
      "up_blocks.1.resnets.2.conv2": [
        81.0,
        81.6,
        82.8,
        77.5
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        80.9,
        81.5,
        82.6,
        77.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        81.1,
        81.5,
        82.0,
        77.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        80.8,
        81.2,
        81.7,
        77.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        76.7,
        81.0,
        76.1,
        77.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        76.5,
        80.9,
        76.5,
        77.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        75.9,
        80.9,
        75.8,
        77.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        79.7,
        80.7,
        75.4,
        77.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        80.3,
        80.7,
        75.2,
        77.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        79.8,
        80.3,
        77.6,
        76.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        78.7,
        79.5,
        76.8,
        76.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        78.8,
        79.7,
        79.8,
        76.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        78.9,
        79.6,
        80.7,
        76.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        77.8,
        78.7,
        80.1,
        76.2
      ],
      "up_blocks.2.resnets.0.conv1": [
        77.3,
        78.3,
        79.9,
        76.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        77.5,
        78.6,
        80.1,
        76.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        77.9,
        78.9,
        80.2,
        76.2
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        77.6,
        78.4,
        79.8,
        76.0
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        76.8,
        78.4,
        79.7,
        76.1
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        76.3,
        77.7,
        79.4,
        76.0
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        76.7,
        77.5,
        79.1,
        75.9
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        76.9,
        77.9,
        79.1,
        75.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        76.9,
        78.0,
        79.1,
        75.8
      ],
      "up_blocks.0.resnets.2.conv1": [
        77.2,
        77.8,
        79.0,
        75.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        77.3,
        77.7,
        79.0,
        75.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        76.9,
        77.4,
        78.0,
        75.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q": [
        76.7,
        77.2,
        78.1,
        75.6
      ],
      "down_blocks.2.resnets.0.conv2": [
        76.9,
        77.2,
        78.7,
        76.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        77.1,
        77.3,
        77.1,
        75.8
      ],
      "up_blocks.1.resnets.0.conv2": [
        77.2,
        77.4,
        78.8,
        75.3
      ],
      "up_blocks.2.resnets.0.conv2": [
        77.3,
        77.4,
        78.8,
        75.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        77.2,
        77.4,
        78.9,
        75.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        77.3,
        77.4,
        78.9,
        74.5
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        77.3,
        77.4,
        78.5,
        74.3
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k": [
        77.2,
        77.4,
        78.6,
        74.4
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v": [
        77.1,
        77.2,
        78.5,
        74.4
      ],
      "up_blocks.1.resnets.2.conv1": [
        77.1,
        76.9,
        77.9,
        74.4
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        77.1,
        76.9,
        78.0,
        74.4
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        77.0,
        76.9,
        77.9,
        74.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        77.1,
        77.1,
        77.9,
        74.7
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q": [
        76.7,
        76.8,
        77.6,
        74.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        76.4,
        76.4,
        77.6,
        73.7
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        76.4,
        76.2,
        77.4,
        73.9
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        76.4,
        76.5,
        77.5,
        73.9
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        76.1,
        76.3,
        77.3,
        73.9
      ],
      "up_blocks.1.resnets.0.conv1": [
        76.3,
        76.2,
        76.9,
        73.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        75.6,
        75.2,
        76.9,
        73.8
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        75.7,
        75.2,
        76.8,
        73.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        75.8,
        75.5,
        76.9,
        73.9
      ],
      "up_blocks.3.resnets.0.time_emb_proj": [
        75.5,
        75.2,
        76.2,
        73.6
      ],
      "up_blocks.1.upsamplers.0.conv": [
        75.3,
        75.2,
        76.3,
        74.0
      ],
      "up_blocks.1.attentions.2.proj_out": [
        75.2,
        75.1,
        75.9,
        73.9
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        75.2,
        75.2,
        76.1,
        74.1
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q": [
        75.2,
        75.1,
        76.0,
        74.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        75.2,
        75.1,
        75.8,
        74.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        75.2,
        75.0,
        75.5,
        73.8
      ],
      "down_blocks.1.resnets.0.conv1": [
        74.7,
        74.8,
        74.9,
        73.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        74.7,
        74.8,
        75.1,
        73.6
      ],
      "up_blocks.2.resnets.1.conv1": [
        74.7,
        74.6,
        74.8,
        73.5
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v": [
        74.6,
        74.5,
        74.8,
        73.3
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        74.6,
        74.5,
        74.4,
        73.2
      ],
      "down_blocks.1.resnets.1.conv1": [
        74.1,
        74.6,
        74.3,
        73.0
      ],
      "up_blocks.1.attentions.1.proj_in": [
        73.6,
        74.2,
        73.8,
        72.6
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        73.8,
        74.5,
        74.1,
        72.6
      ],
      "up_blocks.2.resnets.1.conv2": [
        73.6,
        74.2,
        74.0,
        72.5
      ],
      "down_blocks.1.attentions.1.proj_out": [
        73.4,
        73.6,
        73.3,
        72.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        73.1,
        73.4,
        72.9,
        72.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        73.0,
        73.4,
        73.0,
        72.5
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        73.0,
        73.4,
        73.0,
        72.5
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        72.7,
        72.0,
        72.4,
        72.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        73.1,
        72.7,
        73.0,
        72.2
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        73.1,
        72.8,
        73.3,
        72.3
      ],
      "down_blocks.2.attentions.0.proj_in": [
        73.6,
        73.4,
        73.1,
        72.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        73.4,
        73.2,
        72.9,
        72.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        73.7,
        73.6,
        73.0,
        72.3
      ],
      "down_blocks.2.resnets.0.conv1": [
        73.3,
        73.5,
        73.2,
        72.3
      ],
      "up_blocks.1.resnets.1.conv1": [
        73.4,
        73.3,
        73.6,
        72.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        73.3,
        73.1,
        73.7,
        72.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        73.3,
        73.1,
        73.8,
        72.5
      ],
      "up_blocks.0.resnets.2.conv2": [
        72.8,
        73.3,
        73.1,
        72.1
      ],
      "up_blocks.1.resnets.1.conv2": [
        72.5,
        73.1,
        73.2,
        71.7
      ],
      "up_blocks.0.upsamplers.0.conv": [
        72.8,
        73.3,
        73.3,
        71.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        72.2,
        73.2,
        73.1,
        71.6
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        71.5,
        72.5,
        72.4,
        70.7
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        71.2,
        72.2,
        72.2,
        70.9
      ],
      "up_blocks.1.attentions.0.proj_in": [
        71.3,
        72.3,
        72.4,
        71.2
      ],
      "down_blocks.2.attentions.1.proj_out": [
        71.4,
        72.3,
        72.3,
        71.1
      ],
      "down_blocks.2.attentions.1.proj_in": [
        71.6,
        72.4,
        72.3,
        71.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        71.7,
        72.2,
        72.5,
        71.4
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        71.6,
        72.0,
        72.7,
        71.5
      ],
      "up_blocks.2.resnets.2.conv1": [
        71.8,
        72.0,
        72.8,
        71.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        71.9,
        72.2,
        72.9,
        71.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        71.9,
        72.2,
        72.9,
        71.4
      ],
      "down_blocks.1.downsamplers.0.conv": [
        71.8,
        72.1,
        72.9,
        71.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        71.5,
        71.9,
        72.4,
        70.9
      ],
      "up_blocks.2.resnets.2.conv2": [
        70.9,
        71.6,
        72.2,
        70.8
      ],
      "down_blocks.1.resnets.0.conv2": [
        70.7,
        71.5,
        72.2,
        70.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        70.6,
        71.5,
        72.2,
        70.5
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v": [
        70.6,
        71.4,
        72.0,
        70.3
      ],
      "down_blocks.1.resnets.1.conv2": [
        70.2,
        71.3,
        72.0,
        70.4
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2": [
        70.1,
        71.2,
        72.0,
        70.3
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        70.0,
        71.2,
        71.8,
        70.3
      ],
      "up_blocks.1.attentions.2.proj_in": [
        69.5,
        70.5,
        71.2,
        69.8
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        69.5,
        70.5,
        71.1,
        69.7
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        69.3,
        70.4,
        71.1,
        69.7
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        69.1,
        70.1,
        70.6,
        69.3
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        68.7,
        70.3,
        69.8,
        68.5
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        68.8,
        70.2,
        69.9,
        68.5
      ],
      "down_blocks.1.attentions.0.proj_in": [
        68.8,
        69.9,
        69.5,
        68.4
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        68.8,
        69.5,
        69.4,
        68.1
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        69.0,
        69.6,
        69.5,
        68.2
      ],
      "up_blocks.2.attentions.0.proj_in": [
        68.9,
        69.5,
        69.3,
        67.9
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        68.9,
        69.5,
        69.4,
        68.0
      ],
      "down_blocks.2.attentions.0.proj_out": [
        69.0,
        69.6,
        69.7,
        68.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        69.0,
        69.4,
        69.4,
        68.5
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        68.5,
        69.4,
        69.4,
        68.5
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        68.4,
        69.1,
        69.4,
        68.4
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q": [
        68.4,
        68.9,
        69.3,
        68.3
      ],
      "time_embedding.linear_2": [
        67.6,
        67.9,
        68.3,
        67.5
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        67.7,
        68.7,
        68.7,
        67.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        67.5,
        68.0,
        68.6,
        67.8
      ],
      "up_blocks.2.attentions.0.proj_out": [
        68.1,
        68.1,
        68.8,
        68.1
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q": [
        67.8,
        67.8,
        68.6,
        67.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        67.1,
        67.3,
        68.0,
        67.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        66.9,
        66.8,
        67.7,
        66.9
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q": [
        66.7,
        66.5,
        67.5,
        66.6
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        66.3,
        66.2,
        67.4,
        66.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        66.4,
        66.2,
        67.6,
        66.9
      ],
      "up_blocks.2.attentions.1.proj_out": [
        66.2,
        66.0,
        67.5,
        66.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        66.9,
        64.7,
        66.2,
        65.9
      ],
      "down_blocks.0.resnets.0.conv1": [
        66.4,
        64.4,
        66.0,
        65.7
      ],
      "down_blocks.1.attentions.1.proj_in": [
        66.9,
        64.5,
        65.7,
        65.4
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q": [
        66.7,
        64.3,
        65.5,
        65.4
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2": [
        66.5,
        64.2,
        65.4,
        65.2
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k": [
        66.3,
        64.1,
        65.2,
        65.1
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        65.8,
        63.6,
        64.8,
        64.9
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        65.9,
        63.7,
        65.2,
        64.9
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        65.2,
        63.5,
        64.8,
        64.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        65.1,
        63.4,
        64.6,
        64.6
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        65.6,
        63.1,
        64.4,
        64.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        65.2,
        63.0,
        64.4,
        64.8
      ],
      "down_blocks.1.attentions.0.proj_out": [
        65.6,
        63.0,
        64.3,
        64.4
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k": [
        65.2,
        62.7,
        64.0,
        64.3
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        64.9,
        62.3,
        63.9,
        64.3
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        65.1,
        62.3,
        63.9,
        64.4
      ],
      "up_blocks.3.resnets.1.time_emb_proj": [
        64.4,
        61.7,
        63.2,
        63.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        64.1,
        61.6,
        63.1,
        63.5
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q": [
        63.8,
        61.5,
        62.9,
        63.3
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        63.7,
        61.8,
        62.7,
        63.3
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        63.4,
        61.8,
        62.5,
        62.9
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        62.8,
        60.4,
        62.2,
        62.2
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v": [
        62.7,
        60.7,
        61.9,
        62.0
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        62.7,
        60.6,
        61.7,
        61.8
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k": [
        62.8,
        60.8,
        61.7,
        61.8
      ],
      "down_blocks.0.attentions.0.proj_out": [
        62.9,
        60.8,
        62.0,
        62.1
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        62.8,
        60.2,
        61.6,
        62.2
      ],
      "up_blocks.2.attentions.2.proj_out": [
        62.6,
        60.6,
        61.5,
        62.2
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        62.6,
        60.0,
        61.0,
        62.2
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v": [
        62.6,
        60.2,
        61.0,
        62.1
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k": [
        62.6,
        60.4,
        61.0,
        62.1
      ],
      "up_blocks.3.attentions.0.proj_out": [
        62.6,
        60.7,
        61.0,
        62.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        61.1,
        61.2,
        60.3,
        60.8
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v": [
        61.2,
        61.0,
        60.4,
        61.1
      ],
      "down_blocks.0.resnets.1.conv2": [
        61.4,
        61.3,
        60.5,
        61.2
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        61.5,
        61.4,
        60.7,
        61.3
      ],
      "down_blocks.0.attentions.1.proj_out": [
        62.0,
        61.8,
        60.8,
        61.3
      ],
      "up_blocks.2.attentions.1.proj_in": [
        61.6,
        61.4,
        60.7,
        61.3
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        61.2,
        61.7,
        62.0,
        62.5
      ],
      "down_blocks.0.resnets.1.conv1": [
        61.0,
        61.9,
        61.9,
        62.5
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        60.3,
        61.2,
        61.0,
        62.0
      ],
      "up_blocks.2.upsamplers.0.conv": [
        60.8,
        61.2,
        61.5,
        62.4
      ],
      "down_blocks.0.attentions.0.proj_in": [
        60.9,
        61.8,
        61.6,
        62.4
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k": [
        61.0,
        61.7,
        61.5,
        62.3
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        60.9,
        61.5,
        61.4,
        62.1
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        61.0,
        61.8,
        61.4,
        62.1
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        60.7,
        61.6,
        61.2,
        62.0
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        60.1,
        61.2,
        60.8,
        61.4
      ],
      "up_blocks.3.resnets.0.conv2": [
        60.3,
        61.0,
        60.5,
        61.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        60.4,
        61.3,
        60.7,
        61.1
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        59.8,
        61.0,
        60.2,
        60.6
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2": [
        59.6,
        60.8,
        60.1,
        60.5
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        59.1,
        60.3,
        59.6,
        60.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        58.4,
        60.0,
        59.6,
        59.9
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        58.3,
        59.8,
        59.5,
        59.7
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        58.3,
        59.7,
        59.4,
        59.7
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        58.2,
        59.7,
        59.5,
        59.7
      ],
      "down_blocks.0.downsamplers.0.conv": [
        56.9,
        58.7,
        59.2,
        59.3
      ],
      "up_blocks.3.resnets.0.conv1": [
        56.1,
        58.2,
        58.5,
        58.6
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k": [
        55.6,
        57.8,
        58.4,
        58.5
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2": [
        55.2,
        57.4,
        57.9,
        58.0
      ],
      "down_blocks.0.attentions.1.proj_in": [
        55.6,
        57.4,
        58.4,
        58.4
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v": [
        55.4,
        57.1,
        57.8,
        58.0
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k": [
        55.4,
        57.1,
        57.7,
        57.9
      ],
      "up_blocks.3.attentions.0.proj_in": [
        55.3,
        56.9,
        57.0,
        57.2
      ],
      "down_blocks.0.resnets.0.conv2": [
        54.5,
        56.9,
        56.5,
        56.9
      ],
      "up_blocks.2.attentions.2.proj_in": [
        54.6,
        56.6,
        56.8,
        56.8
      ],
      "up_blocks.3.attentions.1.proj_out": [
        54.1,
        56.1,
        56.3,
        56.4
      ],
      "up_blocks.3.attentions.2.proj_out": [
        53.8,
        55.7,
        55.9,
        56.0
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        53.0,
        55.1,
        55.3,
        55.5
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v": [
        52.8,
        54.9,
        55.2,
        55.3
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        52.9,
        54.6,
        54.1,
        54.1
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        52.4,
        54.2,
        53.9,
        54.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        52.6,
        54.4,
        53.5,
        53.7
      ],
      "up_blocks.3.resnets.2.conv1": [
        53.0,
        54.8,
        54.0,
        54.3
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        52.7,
        54.5,
        53.9,
        54.6
      ],
      "up_blocks.3.resnets.1.conv1": [
        52.2,
        54.2,
        53.5,
        54.4
      ],
      "up_blocks.3.resnets.2.time_emb_proj": [
        51.7,
        53.6,
        52.8,
        53.6
      ],
      "up_blocks.3.attentions.1.proj_in": [
        51.4,
        53.4,
        52.7,
        53.4
      ],
      "time_embedding.linear_1": [
        51.0,
        52.9,
        52.3,
        52.9
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        50.7,
        52.5,
        52.0,
        52.6
      ],
      "up_blocks.3.resnets.1.conv2": [
        50.7,
        52.7,
        52.1,
        52.7
      ],
      "up_blocks.3.resnets.0.conv_shortcut": [
        50.3,
        52.2,
        51.9,
        52.3
      ],
      "up_blocks.3.resnets.2.conv2": [
        50.6,
        52.4,
        52.1,
        52.6
      ],
      "up_blocks.3.resnets.1.conv_shortcut": [
        50.1,
        52.0,
        51.9,
        52.5
      ],
      "up_blocks.3.attentions.2.proj_in": [
        49.6,
        51.6,
        51.5,
        52.2
      ],
      "up_blocks.3.resnets.2.conv_shortcut": [
        49.4,
        51.5,
        51.6,
        52.3
      ]
    },
    "4": {
      "metadata": {
        "nbits": 4,
        "out_ngroups": 1,
        "in_ngroups": 1,
        "cumulative": true,
        "candidates": [
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.resnets.0.conv2",
          "mid_block.resnets.1.time_emb_proj",
          "mid_block.resnets.0.conv2",
          "down_blocks.3.resnets.0.time_emb_proj",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.resnets.1.time_emb_proj",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "mid_block.attentions.0.proj_out",
          "mid_block.resnets.0.conv1",
          "up_blocks.0.resnets.2.time_emb_proj",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_k",
          "mid_block.resnets.1.conv2",
          "down_blocks.3.resnets.1.conv2",
          "up_blocks.0.resnets.1.conv2",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.resnets.0.time_emb_proj",
          "up_blocks.0.resnets.0.conv_shortcut",
          "down_blocks.3.resnets.1.conv1",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.resnets.0.conv1",
          "up_blocks.0.resnets.1.conv_shortcut",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.3.resnets.0.conv2",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "mid_block.resnets.1.conv1",
          "mid_block.resnets.0.time_emb_proj",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.resnets.1.time_emb_proj",
          "down_blocks.3.resnets.1.time_emb_proj",
          "up_blocks.0.resnets.1.conv1",
          "up_blocks.0.resnets.2.conv_shortcut",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.2.resnets.0.time_emb_proj",
          "up_blocks.1.resnets.0.time_emb_proj",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.downsamplers.0.conv",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "mid_block.attentions.0.proj_in",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.3.resnets.0.conv1",
          "up_blocks.1.resnets.2.time_emb_proj",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.resnets.2.conv2",
          "up_blocks.1.resnets.2.conv1",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.resnets.1.conv2",
          "up_blocks.1.resnets.1.time_emb_proj",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.resnets.1.conv2",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.1.proj_out",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.resnets.2.conv2",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.2.resnets.0.time_emb_proj",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.resnets.1.conv1",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.1.resnets.1.conv1",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.attentions.1.proj_out",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.0.proj_out",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.2.resnets.1.time_emb_proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "down_blocks.2.attentions.0.proj_out",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.resnets.0.conv1",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q",
          "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.0.proj_in",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.2.resnets.0.conv2",
          "down_blocks.2.attentions.1.proj_in",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q",
          "up_blocks.2.resnets.2.time_emb_proj",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.resnets.0.time_emb_proj",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.2.proj_out",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.0.upsamplers.0.conv",
          "down_blocks.1.resnets.0.conv1",
          "up_blocks.2.resnets.1.conv2",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "time_embedding.linear_1",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.0.resnets.1.time_emb_proj",
          "up_blocks.0.resnets.2.conv1",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.2.resnets.1.conv1",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q",
          "down_blocks.1.resnets.0.time_emb_proj",
          "up_blocks.1.attentions.2.proj_in",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.0.proj_in",
          "down_blocks.1.resnets.1.conv1",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.1.proj_in",
          "up_blocks.1.upsamplers.0.conv",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.resnets.0.conv_shortcut",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.2.resnets.2.conv1",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.2.resnets.2.conv2",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2",
          "up_blocks.2.attentions.0.proj_in",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.2.attentions.0.proj_out",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.2.attentions.1.proj_out",
          "up_blocks.1.resnets.2.conv_shortcut",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.resnets.0.conv1",
          "up_blocks.1.resnets.1.conv_shortcut",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.0.resnets.0.conv1",
          "up_blocks.2.resnets.0.conv_shortcut",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "down_blocks.1.resnets.1.time_emb_proj",
          "down_blocks.2.resnets.0.conv_shortcut",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.attentions.1.proj_out",
          "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q",
          "up_blocks.3.resnets.1.time_emb_proj",
          "down_blocks.1.resnets.1.conv2",
          "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2",
          "down_blocks.2.resnets.0.conv2",
          "down_blocks.1.attentions.0.proj_out",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.2.attentions.1.proj_in",
          "up_blocks.2.attentions.2.proj_out",
          "down_blocks.0.resnets.1.conv2",
          "down_blocks.2.resnets.0.conv1",
          "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.attentions.0.proj_out",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k",
          "down_blocks.1.attentions.0.proj_in",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v",
          "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.2.upsamplers.0.conv",
          "up_blocks.3.resnets.0.conv2",
          "up_blocks.2.resnets.1.conv_shortcut",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.resnets.0.conv1",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.3.attentions.1.proj_out",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v",
          "down_blocks.1.downsamplers.0.conv",
          "down_blocks.0.attentions.1.proj_in",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.0.resnets.0.time_emb_proj",
          "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.attentions.0.proj_in",
          "down_blocks.0.attentions.1.proj_out",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.2.attentions.2.proj_in",
          "down_blocks.0.downsamplers.0.conv",
          "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.0.resnets.1.conv1",
          "up_blocks.3.resnets.2.time_emb_proj",
          "time_embedding.linear_2",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.0.attentions.0.proj_out",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.resnets.0.conv2",
          "up_blocks.1.resnets.0.conv2",
          "up_blocks.2.resnets.2.conv_shortcut",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.resnets.0.conv_shortcut",
          "up_blocks.3.attentions.2.proj_out",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.3.resnets.1.conv1",
          "down_blocks.1.attentions.1.proj_in",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.resnets.1.conv2",
          "up_blocks.3.resnets.2.conv1",
          "down_blocks.0.attentions.0.proj_in",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.resnets.2.conv2",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k",
          "up_blocks.3.attentions.1.proj_in",
          "up_blocks.3.resnets.2.conv_shortcut",
          "up_blocks.3.resnets.0.conv_shortcut",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.0.resnets.0.conv2",
          "up_blocks.3.resnets.1.conv_shortcut",
          "up_blocks.3.attentions.2.proj_in",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k"
        ],
        "sizes": [
          1638400,
          1638400,
          14745600,
          1638400,
          14745600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          6553600,
          1638400,
          1638400,
          14745600,
          1638400,
          1638400,
          1310720,
          14745600,
          14745600,
          14745600,
          1638400,
          1638400,
          3276800,
          14745600,
          409600,
          1638400,
          29491200,
          3276800,
          1638400,
          14745600,
          1638400,
          14745600,
          1638400,
          1638400,
          1310720,
          1638400,
          1638400,
          1310720,
          1638400,
          1638400,
          1638400,
          29491200,
          3276800,
          1638400,
          819200,
          1638400,
          1638400,
          14745600,
          409600,
          1638400,
          409600,
          1310720,
          6553600,
          14745600,
          1638400,
          409600,
          409600,
          14745600,
          22118400,
          1638400,
          14745600,
          1638400,
          655360,
          1310720,
          14745600,
          1638400,
          1310720,
          1638400,
          409600,
          1638400,
          1638400,
          6553600,
          409600,
          655360,
          1638400,
          1638400,
          14745600,
          1638400,
          1638400,
          102400,
          102400,
          14745600,
          13107200,
          6553600,
          29491200,
          409600,
          1638400,
          409600,
          1638400,
          409600,
          1638400,
          409600,
          819200,
          1638400,
          1310720,
          409600,
          6553600,
          1638400,
          409600,
          102400,
          11059200,
          409600,
          1638400,
          1638400,
          409600,
          3276800,
          409600,
          102400,
          1638400,
          102400,
          1638400,
          3686400,
          1638400,
          1638400,
          819200,
          409600,
          1638400,
          409600,
          3276800,
          655360,
          1638400,
          327680,
          102400,
          327680,
          1638400,
          14745600,
          1843200,
          3686400,
          13107200,
          409600,
          102400,
          1638400,
          13107200,
          409600,
          29491200,
          102400,
          409600,
          409600,
          7372800,
          655360,
          102400,
          819200,
          1638400,
          409600,
          1638400,
          3686400,
          1638400,
          1638400,
          14745600,
          409600,
          3276800,
          327680,
          1638400,
          409600,
          102400,
          409600,
          409600,
          5529600,
          102400,
          3686400,
          102400,
          102400,
          1638400,
          409600,
          1638400,
          655360,
          102400,
          409600,
          1638400,
          409600,
          2457600,
          3276800,
          102400,
          29491200,
          3276800,
          102400,
          102400,
          921600,
          1228800,
          409600,
          3276800,
          1638400,
          819200,
          819200,
          102400,
          102400,
          409600,
          327680,
          1638400,
          1638400,
          409600,
          409600,
          102400,
          409600,
          3686400,
          409600,
          14745600,
          409600,
          1310720,
          409600,
          409600,
          921600,
          7372800,
          819200,
          102400,
          327680,
          409600,
          13107200,
          327680,
          6553600,
          102400,
          327680,
          409600,
          327680,
          409600,
          409600,
          3686400,
          921600,
          819200,
          102400,
          2764800,
          327680,
          655360,
          3276800,
          102400,
          409600,
          102400,
          102400,
          102400,
          13107200,
          102400,
          3686400,
          102400,
          327680,
          409600,
          819200,
          102400,
          102400,
          1310720,
          409600,
          921600,
          819200,
          655360,
          921600,
          409600,
          1638400,
          102400,
          102400,
          102400,
          409600,
          3686400,
          14745600,
          614400,
          1310720,
          409600,
          819200,
          204800,
          102400,
          409600,
          1843200,
          409600,
          102400,
          921600,
          1843200,
          102400,
          1310720,
          819200,
          921600,
          655360,
          102400,
          204800,
          307200,
          102400,
          655360,
          921600,
          204800,
          102400,
          1310720,
          655360
        ]
      },
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        89.0,
        89.0,
        89.0,
        88.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        88.8,
        88.9,
        88.8,
        88.3
      ],
      "up_blocks.0.resnets.0.conv2": [
        88.7,
        88.9,
        88.7,
        88.2
      ],
      "mid_block.resnets.1.time_emb_proj": [
        89.0,
        88.9,
        88.7,
        88.0
      ],
      "mid_block.resnets.0.conv2": [
        88.5,
        88.8,
        88.7,
        88.5
      ],
      "down_blocks.3.resnets.0.time_emb_proj": [
        88.8,
        88.7,
        88.6,
        87.5
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.6,
        88.8,
        88.8,
        87.7
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.5,
        89.0,
        88.9,
        88.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        88.6,
        88.6,
        88.8,
        87.6
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        88.5,
        88.7,
        88.2,
        87.7
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        88.9,
        88.7,
        87.9,
        86.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        88.9,
        88.6,
        88.8,
        86.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.8,
        88.5,
        87.6,
        86.7
      ],
      "mid_block.attentions.0.proj_out": [
        88.6,
        88.3,
        87.7,
        86.9
      ],
      "mid_block.resnets.0.conv1": [
        88.7,
        88.6,
        87.5,
        86.6
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        88.5,
        88.3,
        88.0,
        86.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        88.7,
        88.4,
        87.8,
        86.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        88.7,
        88.6,
        87.8,
        86.5
      ],
      "mid_block.resnets.1.conv2": [
        88.6,
        88.6,
        87.8,
        86.4
      ],
      "down_blocks.3.resnets.1.conv2": [
        88.6,
        88.6,
        87.4,
        86.4
      ],
      "up_blocks.0.resnets.1.conv2": [
        88.6,
        88.4,
        88.1,
        87.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.5,
        88.2,
        88.3,
        86.3
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        88.5,
        88.5,
        88.2,
        86.1
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        88.3,
        88.3,
        87.4,
        86.1
      ],
      "down_blocks.3.resnets.1.conv1": [
        88.5,
        88.4,
        87.8,
        86.0
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.2,
        88.4,
        87.4,
        86.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        87.7,
        88.3,
        87.3,
        86.3
      ],
      "up_blocks.0.resnets.0.conv1": [
        88.2,
        88.3,
        87.4,
        85.8
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        87.9,
        88.3,
        88.2,
        86.0
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        87.8,
        88.0,
        88.2,
        85.9
      ],
      "down_blocks.3.resnets.0.conv2": [
        88.0,
        88.1,
        88.2,
        86.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.1,
        88.1,
        88.3,
        86.2
      ],
      "mid_block.resnets.1.conv1": [
        87.8,
        88.2,
        87.8,
        86.0
      ],
      "mid_block.resnets.0.time_emb_proj": [
        88.1,
        88.5,
        88.1,
        85.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.1,
        88.3,
        87.5,
        85.4
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        87.8,
        88.2,
        87.4,
        85.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        88.1,
        88.3,
        87.3,
        85.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.5,
        88.1,
        87.6,
        85.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        87.8,
        88.1,
        88.3,
        86.0
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.1,
        88.2,
        88.2,
        85.9
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        87.9,
        88.1,
        87.9,
        86.4
      ],
      "down_blocks.3.resnets.1.time_emb_proj": [
        87.8,
        88.1,
        88.2,
        86.3
      ],
      "up_blocks.0.resnets.1.conv1": [
        87.7,
        88.2,
        88.0,
        86.3
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        87.7,
        87.9,
        88.0,
        86.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        87.9,
        88.0,
        88.1,
        86.3
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        87.8,
        87.8,
        87.8,
        86.1
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        87.5,
        87.9,
        87.9,
        86.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        87.5,
        87.6,
        87.3,
        85.6
      ],
      "down_blocks.2.downsamplers.0.conv": [
        87.5,
        87.7,
        87.2,
        85.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        87.5,
        87.5,
        86.9,
        85.0
      ],
      "mid_block.attentions.0.proj_in": [
        87.6,
        87.4,
        86.7,
        85.3
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        87.3,
        87.2,
        87.2,
        85.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        87.2,
        86.6,
        88.1,
        85.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        87.1,
        87.2,
        87.3,
        85.1
      ],
      "down_blocks.3.resnets.0.conv1": [
        87.0,
        87.1,
        87.7,
        85.3
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        87.0,
        87.3,
        87.3,
        85.2
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        86.7,
        86.9,
        87.2,
        85.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        86.2,
        86.4,
        87.1,
        85.5
      ],
      "up_blocks.0.resnets.2.conv2": [
        86.2,
        86.5,
        87.6,
        85.3
      ],
      "up_blocks.1.resnets.2.conv1": [
        86.3,
        86.2,
        87.2,
        84.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        86.5,
        86.4,
        86.9,
        85.0
      ],
      "up_blocks.1.resnets.1.conv2": [
        86.5,
        86.4,
        86.7,
        84.8
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        86.4,
        86.0,
        86.7,
        84.8
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        86.5,
        86.0,
        86.7,
        84.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        86.4,
        86.1,
        87.0,
        84.7
      ],
      "down_blocks.2.resnets.1.conv2": [
        86.4,
        86.2,
        86.2,
        84.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        86.4,
        86.1,
        86.5,
        84.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        86.3,
        85.8,
        81.9,
        84.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        86.1,
        85.7,
        81.1,
        84.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        85.8,
        85.7,
        84.0,
        84.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        85.9,
        85.8,
        84.8,
        83.7
      ],
      "up_blocks.1.attentions.1.proj_out": [
        85.8,
        85.9,
        84.9,
        83.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        85.8,
        85.7,
        85.0,
        83.5
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        85.6,
        85.6,
        83.8,
        84.1
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        85.6,
        85.6,
        84.3,
        82.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        85.4,
        85.7,
        84.4,
        83.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        85.4,
        85.0,
        84.2,
        83.8
      ],
      "up_blocks.1.resnets.2.conv2": [
        85.1,
        85.4,
        83.8,
        83.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        84.7,
        85.1,
        81.8,
        83.3
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        84.6,
        84.7,
        78.7,
        82.6
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q": [
        84.8,
        85.2,
        79.6,
        81.1
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        84.7,
        85.0,
        79.1,
        81.0
      ],
      "down_blocks.2.resnets.1.conv1": [
        84.7,
        85.1,
        79.7,
        83.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        84.8,
        84.5,
        79.6,
        83.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        84.7,
        84.4,
        79.9,
        83.5
      ],
      "up_blocks.1.resnets.1.conv1": [
        84.9,
        84.6,
        78.9,
        83.5
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        84.8,
        84.6,
        78.3,
        83.6
      ],
      "down_blocks.2.attentions.1.proj_out": [
        84.5,
        84.6,
        79.9,
        83.8
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        84.5,
        84.8,
        80.4,
        83.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        84.4,
        84.5,
        81.2,
        83.6
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        84.4,
        84.8,
        80.4,
        83.6
      ],
      "up_blocks.1.attentions.0.proj_out": [
        84.4,
        84.6,
        81.5,
        83.6
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        84.4,
        84.4,
        81.3,
        83.4
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        84.4,
        84.5,
        81.2,
        83.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        84.2,
        84.7,
        81.2,
        83.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        84.1,
        84.3,
        81.6,
        83.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        84.1,
        84.3,
        81.8,
        83.3
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        84.1,
        84.6,
        81.7,
        83.2
      ],
      "down_blocks.2.attentions.0.proj_out": [
        83.9,
        84.4,
        79.2,
        83.2
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k": [
        83.7,
        84.0,
        79.1,
        82.8
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q": [
        83.7,
        84.0,
        78.7,
        82.9
      ],
      "up_blocks.2.resnets.0.conv1": [
        83.7,
        83.9,
        79.3,
        82.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        83.8,
        83.6,
        82.6,
        82.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        83.8,
        83.7,
        80.9,
        82.5
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        83.7,
        83.5,
        80.7,
        82.3
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q": [
        83.6,
        83.4,
        80.4,
        82.1
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        83.4,
        83.2,
        81.1,
        82.4
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        83.4,
        83.2,
        80.8,
        82.4
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        83.3,
        83.2,
        81.1,
        82.2
      ],
      "down_blocks.2.attentions.0.proj_in": [
        83.0,
        83.2,
        80.6,
        81.8
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        83.1,
        83.2,
        83.7,
        80.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        82.8,
        82.9,
        83.4,
        79.8
      ],
      "up_blocks.2.resnets.0.conv2": [
        82.5,
        82.7,
        83.6,
        80.1
      ],
      "down_blocks.2.attentions.1.proj_in": [
        82.2,
        82.7,
        83.6,
        78.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        82.1,
        82.4,
        83.6,
        78.9
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        82.2,
        82.6,
        83.7,
        80.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        82.3,
        82.4,
        83.6,
        80.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        82.1,
        82.5,
        83.8,
        81.2
      ],
      "up_blocks.3.resnets.0.time_emb_proj": [
        82.1,
        82.3,
        83.8,
        81.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        82.0,
        82.3,
        83.6,
        80.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        82.0,
        82.4,
        83.6,
        81.0
      ],
      "up_blocks.1.attentions.2.proj_out": [
        81.8,
        82.2,
        83.6,
        80.9
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v": [
        81.7,
        82.2,
        83.3,
        80.8
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        81.6,
        82.1,
        80.9,
        78.3
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.7,
        82.1,
        81.9,
        78.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        81.7,
        82.1,
        78.5,
        78.7
      ],
      "up_blocks.0.upsamplers.0.conv": [
        81.8,
        82.1,
        76.6,
        78.3
      ],
      "down_blocks.1.resnets.0.conv1": [
        81.7,
        82.1,
        81.7,
        78.2
      ],
      "up_blocks.2.resnets.1.conv2": [
        81.7,
        82.1,
        81.2,
        78.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        81.7,
        82.0,
        82.5,
        78.1
      ],
      "time_embedding.linear_1": [
        81.5,
        81.7,
        82.6,
        78.1
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        81.3,
        81.7,
        82.6,
        78.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        81.3,
        81.7,
        82.6,
        78.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        81.3,
        81.8,
        80.2,
        78.0
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        81.2,
        81.7,
        77.5,
        77.9
      ],
      "up_blocks.0.resnets.2.conv1": [
        81.2,
        81.7,
        77.1,
        77.9
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        81.1,
        81.6,
        77.7,
        77.9
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        80.9,
        81.4,
        77.4,
        77.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        80.8,
        81.3,
        82.1,
        77.9
      ],
      "up_blocks.2.resnets.1.conv1": [
        80.8,
        81.4,
        82.0,
        77.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        80.8,
        81.4,
        82.1,
        77.9
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q": [
        80.7,
        81.3,
        81.9,
        77.8
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        80.5,
        81.1,
        81.6,
        77.7
      ],
      "up_blocks.1.attentions.2.proj_in": [
        80.1,
        81.0,
        81.5,
        77.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        80.1,
        81.0,
        81.4,
        77.5
      ],
      "up_blocks.1.attentions.0.proj_in": [
        79.9,
        81.0,
        80.7,
        77.6
      ],
      "down_blocks.1.resnets.1.conv1": [
        79.8,
        80.8,
        81.6,
        77.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        79.8,
        80.7,
        81.5,
        77.5
      ],
      "up_blocks.1.attentions.1.proj_in": [
        79.6,
        80.5,
        81.5,
        77.5
      ],
      "up_blocks.1.upsamplers.0.conv": [
        79.2,
        80.2,
        81.3,
        77.5
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q": [
        79.4,
        80.3,
        81.2,
        77.5
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        79.2,
        80.2,
        81.0,
        77.5
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v": [
        79.2,
        80.1,
        81.1,
        77.5
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        79.0,
        79.8,
        80.8,
        77.6
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        78.9,
        80.0,
        80.7,
        77.5
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        78.9,
        79.9,
        77.5,
        77.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        78.4,
        79.7,
        76.8,
        77.5
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        78.3,
        79.4,
        76.6,
        77.2
      ],
      "up_blocks.2.resnets.2.conv1": [
        78.3,
        79.3,
        76.6,
        77.2
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q": [
        78.3,
        79.3,
        76.4,
        77.0
      ],
      "up_blocks.2.resnets.2.conv2": [
        78.2,
        79.1,
        76.4,
        76.9
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        78.2,
        79.2,
        76.3,
        76.8
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k": [
        78.2,
        79.1,
        76.2,
        76.8
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2": [
        78.0,
        79.0,
        76.1,
        76.5
      ],
      "up_blocks.2.attentions.0.proj_in": [
        77.8,
        78.7,
        76.2,
        76.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        77.8,
        78.8,
        75.9,
        76.5
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v": [
        77.8,
        78.7,
        76.1,
        76.6
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q": [
        77.8,
        78.6,
        76.0,
        76.5
      ],
      "up_blocks.2.attentions.0.proj_out": [
        77.8,
        78.6,
        75.7,
        76.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        77.6,
        78.5,
        75.5,
        76.5
      ],
      "up_blocks.2.attentions.1.proj_out": [
        77.5,
        78.4,
        75.4,
        76.3
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        77.4,
        78.2,
        75.6,
        76.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        77.3,
        78.1,
        77.1,
        76.3
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k": [
        77.2,
        77.9,
        77.1,
        76.1
      ],
      "up_blocks.1.resnets.0.conv1": [
        77.2,
        77.9,
        77.1,
        76.1
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        77.2,
        77.8,
        77.0,
        76.1
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        77.1,
        77.7,
        77.1,
        76.2
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        77.0,
        77.6,
        76.9,
        76.1
      ],
      "down_blocks.0.resnets.0.conv1": [
        76.9,
        77.4,
        74.8,
        76.1
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        76.6,
        77.3,
        75.4,
        76.0
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        76.6,
        77.1,
        74.7,
        75.9
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        76.4,
        76.8,
        75.3,
        76.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        76.5,
        76.9,
        77.0,
        76.0
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        76.5,
        77.0,
        76.5,
        76.0
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        76.6,
        77.1,
        75.7,
        75.9
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v": [
        76.5,
        76.9,
        75.6,
        75.7
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        76.5,
        76.6,
        75.5,
        75.8
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v": [
        76.7,
        76.7,
        75.5,
        75.5
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k": [
        76.6,
        76.8,
        75.4,
        75.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        76.6,
        76.7,
        75.4,
        75.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        76.5,
        76.6,
        75.6,
        75.5
      ],
      "down_blocks.1.attentions.1.proj_out": [
        76.7,
        76.7,
        74.4,
        75.5
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2": [
        76.5,
        76.4,
        74.3,
        75.4
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q": [
        76.2,
        76.1,
        74.1,
        75.3
      ],
      "up_blocks.3.resnets.1.time_emb_proj": [
        75.9,
        75.6,
        74.0,
        75.1
      ],
      "down_blocks.1.resnets.1.conv2": [
        75.9,
        75.5,
        73.9,
        75.1
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2": [
        75.7,
        75.5,
        73.9,
        75.1
      ],
      "down_blocks.2.resnets.0.conv2": [
        75.8,
        75.6,
        75.1,
        75.0
      ],
      "down_blocks.1.attentions.0.proj_out": [
        75.7,
        75.5,
        75.7,
        74.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        75.3,
        75.1,
        76.6,
        74.8
      ],
      "up_blocks.2.attentions.1.proj_in": [
        75.1,
        74.5,
        76.2,
        74.5
      ],
      "up_blocks.2.attentions.2.proj_out": [
        75.0,
        74.5,
        76.1,
        74.3
      ],
      "down_blocks.0.resnets.1.conv2": [
        74.8,
        74.4,
        75.8,
        74.2
      ],
      "down_blocks.2.resnets.0.conv1": [
        74.8,
        74.5,
        74.5,
        74.2
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        74.7,
        74.3,
        74.4,
        74.0
      ],
      "up_blocks.3.attentions.0.proj_out": [
        74.6,
        74.2,
        74.3,
        74.0
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k": [
        74.5,
        74.1,
        74.2,
        74.0
      ],
      "down_blocks.1.attentions.0.proj_in": [
        74.4,
        74.3,
        73.3,
        74.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        74.4,
        74.3,
        73.8,
        74.0
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k": [
        74.4,
        74.4,
        73.5,
        73.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        74.4,
        74.5,
        74.9,
        73.8
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k": [
        74.1,
        74.2,
        74.7,
        73.6
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v": [
        74.0,
        74.1,
        74.5,
        73.4
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        73.9,
        73.9,
        74.8,
        73.1
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        73.8,
        73.8,
        74.5,
        72.8
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2": [
        73.7,
        73.7,
        74.3,
        72.7
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        73.7,
        73.6,
        74.5,
        72.9
      ],
      "up_blocks.2.upsamplers.0.conv": [
        73.6,
        73.6,
        74.3,
        72.8
      ],
      "up_blocks.3.resnets.0.conv2": [
        73.6,
        73.5,
        74.2,
        72.7
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        73.4,
        73.5,
        73.8,
        72.4
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        73.0,
        73.1,
        73.5,
        72.0
      ],
      "up_blocks.3.resnets.0.conv1": [
        72.9,
        73.0,
        73.3,
        71.9
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        72.7,
        72.7,
        73.3,
        71.6
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        72.7,
        72.9,
        73.0,
        71.6
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        72.6,
        72.8,
        72.8,
        71.3
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        72.3,
        72.7,
        72.0,
        71.7
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        72.1,
        72.7,
        72.0,
        71.8
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v": [
        72.0,
        72.4,
        71.6,
        71.3
      ],
      "up_blocks.3.attentions.1.proj_out": [
        71.9,
        72.2,
        71.5,
        71.2
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        71.8,
        72.1,
        71.0,
        71.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        71.8,
        72.1,
        71.7,
        71.0
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v": [
        71.8,
        72.2,
        71.7,
        71.1
      ],
      "down_blocks.1.downsamplers.0.conv": [
        71.7,
        72.0,
        71.3,
        70.4
      ],
      "down_blocks.0.attentions.1.proj_in": [
        71.2,
        70.3,
        71.3,
        70.3
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        70.5,
        70.0,
        68.8,
        70.5
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        70.5,
        71.2,
        69.6,
        70.4
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        70.6,
        71.6,
        70.7,
        70.7
      ],
      "up_blocks.3.attentions.0.proj_in": [
        70.5,
        71.4,
        70.4,
        70.5
      ],
      "down_blocks.0.attentions.1.proj_out": [
        70.3,
        71.3,
        71.1,
        70.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        69.4,
        70.2,
        69.4,
        70.5
      ],
      "up_blocks.2.attentions.2.proj_in": [
        69.3,
        70.0,
        69.4,
        70.1
      ],
      "down_blocks.0.downsamplers.0.conv": [
        69.4,
        70.1,
        67.7,
        69.6
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        69.3,
        70.0,
        69.7,
        69.5
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        68.3,
        68.9,
        69.5,
        69.5
      ],
      "down_blocks.0.resnets.1.conv1": [
        68.5,
        69.1,
        70.5,
        69.0
      ],
      "up_blocks.3.resnets.2.time_emb_proj": [
        68.2,
        68.7,
        70.1,
        68.6
      ],
      "time_embedding.linear_2": [
        67.2,
        67.8,
        68.6,
        67.6
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        67.1,
        67.6,
        68.5,
        67.5
      ],
      "down_blocks.0.attentions.0.proj_out": [
        67.0,
        67.5,
        68.1,
        67.7
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        67.0,
        67.3,
        68.1,
        67.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        67.1,
        67.4,
        67.8,
        67.5
      ],
      "down_blocks.1.resnets.0.conv2": [
        67.0,
        67.6,
        66.9,
        67.1
      ],
      "up_blocks.1.resnets.0.conv2": [
        67.0,
        67.5,
        66.8,
        67.7
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        66.6,
        67.2,
        66.3,
        67.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        67.3,
        66.9,
        67.2,
        66.6
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        66.8,
        66.5,
        66.9,
        66.1
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        66.7,
        66.4,
        66.8,
        66.1
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        66.0,
        65.9,
        67.2,
        65.9
      ],
      "up_blocks.3.attentions.2.proj_out": [
        65.9,
        65.8,
        67.1,
        65.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        65.9,
        65.9,
        67.0,
        65.6
      ],
      "up_blocks.3.resnets.1.conv1": [
        65.9,
        65.8,
        66.8,
        65.5
      ],
      "down_blocks.1.attentions.1.proj_in": [
        65.3,
        65.9,
        66.6,
        65.4
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        65.2,
        65.8,
        66.5,
        65.3
      ],
      "up_blocks.3.resnets.1.conv2": [
        65.0,
        65.6,
        66.4,
        65.1
      ],
      "up_blocks.3.resnets.2.conv1": [
        65.0,
        65.6,
        66.4,
        65.2
      ],
      "down_blocks.0.attentions.0.proj_in": [
        65.3,
        65.7,
        66.8,
        65.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        65.8,
        66.3,
        66.3,
        65.8
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        65.7,
        66.1,
        66.2,
        65.6
      ],
      "up_blocks.3.resnets.2.conv2": [
        65.4,
        65.9,
        66.0,
        65.4
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k": [
        64.6,
        65.2,
        65.3,
        64.6
      ],
      "up_blocks.3.attentions.1.proj_in": [
        64.7,
        65.4,
        65.4,
        64.7
      ],
      "up_blocks.3.resnets.2.conv_shortcut": [
        64.9,
        65.6,
        65.7,
        64.9
      ],
      "up_blocks.3.resnets.0.conv_shortcut": [
        64.8,
        65.6,
        65.1,
        64.8
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        64.8,
        65.4,
        65.2,
        65.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        64.3,
        64.2,
        65.9,
        65.0
      ],
      "down_blocks.0.resnets.0.conv2": [
        64.7,
        64.0,
        65.7,
        64.5
      ],
      "up_blocks.3.resnets.1.conv_shortcut": [
        64.7,
        63.9,
        65.5,
        64.4
      ],
      "up_blocks.3.attentions.2.proj_in": [
        64.2,
        63.6,
        65.1,
        64.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        63.1,
        63.5,
        65.9,
        64.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        63.3,
        62.8,
        65.7,
        64.4
      ]
    },
    "6": {
      "metadata": {
        "nbits": 6,
        "out_ngroups": 1,
        "in_ngroups": 1,
        "cumulative": true,
        "candidates": [
          "up_blocks.0.resnets.0.conv_shortcut",
          "mid_block.resnets.0.conv1",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.resnets.2.conv1",
          "mid_block.attentions.0.proj_out",
          "down_blocks.2.resnets.0.time_emb_proj",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.resnets.1.time_emb_proj",
          "up_blocks.0.resnets.2.time_emb_proj",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "mid_block.resnets.0.time_emb_proj",
          "mid_block.resnets.1.time_emb_proj",
          "down_blocks.3.resnets.0.conv1",
          "down_blocks.3.resnets.1.conv2",
          "up_blocks.0.resnets.1.conv1",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.0.resnets.1.time_emb_proj",
          "mid_block.attentions.0.proj_in",
          "up_blocks.0.resnets.1.conv2",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.resnets.0.conv1",
          "up_blocks.0.resnets.2.conv_shortcut",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.3.resnets.1.conv1",
          "up_blocks.0.resnets.1.conv_shortcut",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.resnets.2.time_emb_proj",
          "mid_block.resnets.1.conv1",
          "mid_block.resnets.1.conv2",
          "down_blocks.2.resnets.1.conv2",
          "up_blocks.0.resnets.0.time_emb_proj",
          "up_blocks.0.resnets.0.conv2",
          "up_blocks.2.resnets.0.time_emb_proj",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.3.resnets.0.time_emb_proj",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.resnets.1.time_emb_proj",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.3.resnets.0.conv2",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.resnets.0.time_emb_proj",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.resnets.2.conv2",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.resnets.2.conv1",
          "up_blocks.1.resnets.2.conv2",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "mid_block.resnets.0.conv2",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.resnets.1.conv1",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.upsamplers.0.conv",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.1.resnets.1.conv2",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.2.downsamplers.0.conv",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.resnets.0.conv2",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.2.resnets.1.conv1",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.0.proj_out",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.attentions.0.proj_out",
          "down_blocks.3.resnets.1.time_emb_proj",
          "up_blocks.1.attentions.1.proj_out",
          "up_blocks.1.resnets.0.conv_shortcut",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.attentions.1.proj_in",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.resnets.1.conv_shortcut",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.2.proj_out",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q",
          "time_embedding.linear_1",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.2.resnets.1.time_emb_proj",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.resnets.2.conv_shortcut",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.3.resnets.0.time_emb_proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.attentions.0.proj_in",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.2.resnets.0.conv1",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k",
          "up_blocks.1.upsamplers.0.conv",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.attentions.1.proj_out",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.1.resnets.1.conv2",
          "up_blocks.1.attentions.2.proj_in",
          "up_blocks.2.resnets.1.conv2",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.resnets.0.conv1",
          "up_blocks.2.resnets.0.conv2",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.1.resnets.0.conv2",
          "down_blocks.1.resnets.1.time_emb_proj",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.2.attentions.0.proj_out",
          "up_blocks.2.resnets.1.conv1",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.attentions.1.proj_in",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.0.proj_in",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.1.resnets.0.conv1",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.2.resnets.2.time_emb_proj",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.resnets.0.conv1",
          "up_blocks.2.attentions.0.proj_in",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.3.resnets.2.time_emb_proj",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k",
          "down_blocks.1.resnets.0.time_emb_proj",
          "up_blocks.2.resnets.2.conv2",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2",
          "down_blocks.1.attentions.1.proj_in",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.1.resnets.1.conv1",
          "down_blocks.0.resnets.1.time_emb_proj",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.resnets.1.time_emb_proj",
          "up_blocks.2.resnets.2.conv1",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k",
          "up_blocks.2.resnets.0.conv_shortcut",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.2.attentions.1.proj_in",
          "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2",
          "down_blocks.1.attentions.1.proj_out",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.0.resnets.0.time_emb_proj",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.2.attentions.1.proj_out",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v",
          "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.2.resnets.1.conv_shortcut",
          "up_blocks.2.upsamplers.0.conv",
          "down_blocks.1.downsamplers.0.conv",
          "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v",
          "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.resnets.0.conv1",
          "down_blocks.2.resnets.0.conv_shortcut",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k",
          "up_blocks.2.attentions.2.proj_out",
          "up_blocks.2.attentions.2.proj_in",
          "up_blocks.3.attentions.0.proj_out",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v",
          "time_embedding.linear_2",
          "down_blocks.0.attentions.0.proj_out",
          "down_blocks.1.attentions.0.proj_out",
          "up_blocks.2.resnets.2.conv_shortcut",
          "down_blocks.0.resnets.1.conv2",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.0.downsamplers.0.conv",
          "down_blocks.0.attentions.0.proj_in",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.3.attentions.2.proj_out",
          "up_blocks.3.resnets.0.conv2",
          "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.attentions.1.proj_out",
          "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.2.resnets.0.conv2",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.attentions.0.proj_in",
          "up_blocks.3.attentions.0.proj_in",
          "down_blocks.0.attentions.1.proj_in",
          "up_blocks.3.resnets.2.conv1",
          "down_blocks.0.resnets.1.conv1",
          "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.0.resnets.0.conv1",
          "down_blocks.0.resnets.0.conv2",
          "up_blocks.3.resnets.1.conv2",
          "up_blocks.3.resnets.2.conv2",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.resnets.1.conv1",
          "up_blocks.3.attentions.1.proj_in",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.3.resnets.0.conv_shortcut",
          "up_blocks.3.resnets.1.conv_shortcut",
          "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.resnets.2.conv_shortcut",
          "up_blocks.3.attentions.2.proj_in",
          "down_blocks.0.attentions.1.proj_out",
          "down_blocks.1.resnets.0.conv_shortcut",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj"
        ],
        "sizes": [
          3276800,
          14745600,
          1310720,
          1310720,
          1638400,
          29491200,
          1638400,
          1638400,
          409600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          14745600,
          14745600,
          29491200,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          14745600,
          1638400,
          1638400,
          29491200,
          3276800,
          13107200,
          14745600,
          3276800,
          1638400,
          1638400,
          14745600,
          14745600,
          14745600,
          1638400,
          14745600,
          819200,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1310720,
          1638400,
          655360,
          1638400,
          14745600,
          1638400,
          1638400,
          1638400,
          409600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          14745600,
          409600,
          1638400,
          1310720,
          22118400,
          14745600,
          409600,
          409600,
          14745600,
          1310720,
          29491200,
          655360,
          655360,
          409600,
          1638400,
          6553600,
          1638400,
          14745600,
          6553600,
          14745600,
          1310720,
          409600,
          1638400,
          6553600,
          14745600,
          1638400,
          1638400,
          1638400,
          1310720,
          14745600,
          655360,
          409600,
          102400,
          13107200,
          14745600,
          6553600,
          409600,
          102400,
          1638400,
          1638400,
          1638400,
          3276800,
          1638400,
          1638400,
          1638400,
          1638400,
          3276800,
          102400,
          409600,
          1638400,
          1638400,
          1310720,
          1638400,
          409600,
          3276800,
          409600,
          1638400,
          102400,
          409600,
          409600,
          409600,
          655360,
          1310720,
          819200,
          6553600,
          1638400,
          13107200,
          2457600,
          655360,
          409600,
          409600,
          1638400,
          1638400,
          409600,
          11059200,
          327680,
          102400,
          102400,
          1638400,
          1310720,
          14745600,
          409600,
          102400,
          1638400,
          409600,
          13107200,
          102400,
          3686400,
          1638400,
          3686400,
          102400,
          655360,
          7372800,
          3686400,
          327680,
          409600,
          102400,
          3686400,
          819200,
          1310720,
          1638400,
          327680,
          409600,
          7372800,
          102400,
          1638400,
          409600,
          409600,
          1638400,
          1638400,
          655360,
          102400,
          1843200,
          409600,
          409600,
          819200,
          102400,
          29491200,
          409600,
          327680,
          409600,
          13107200,
          3276800,
          102400,
          409600,
          327680,
          327680,
          327680,
          819200,
          3686400,
          409600,
          1638400,
          409600,
          1310720,
          327680,
          3686400,
          409600,
          409600,
          409600,
          5529600,
          102400,
          655360,
          1228800,
          655360,
          409600,
          409600,
          409600,
          102400,
          102400,
          102400,
          409600,
          13107200,
          409600,
          409600,
          409600,
          102400,
          327680,
          3276800,
          819200,
          3686400,
          3686400,
          409600,
          409600,
          409600,
          409600,
          102400,
          1638400,
          327680,
          819200,
          2764800,
          819200,
          102400,
          409600,
          409600,
          102400,
          102400,
          1638400,
          102400,
          409600,
          614400,
          921600,
          102400,
          921600,
          102400,
          102400,
          102400,
          102400,
          921600,
          819200,
          102400,
          819200,
          409600,
          14745600,
          102400,
          409600,
          102400,
          102400,
          1843200,
          921600,
          3276800,
          409600,
          102400,
          102400,
          819200,
          921600,
          921600,
          921600,
          921600,
          102400,
          102400,
          1843200,
          102400,
          1638400,
          307200,
          204800,
          819200,
          204800,
          102400,
          102400,
          204800,
          3276800
        ]
      },
      "up_blocks.0.resnets.0.conv_shortcut": [
        89.4,
        88.9,
        88.8,
        88.9
      ],
      "mid_block.resnets.0.conv1": [
        89.1,
        89.0,
        89.1,
        88.7
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        89.0,
        88.8,
        88.7,
        88.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        89.2,
        89.0,
        88.9,
        88.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.9,
        88.5,
        88.9,
        88.8
      ],
      "up_blocks.0.resnets.2.conv1": [
        89.2,
        88.9,
        88.6,
        88.9
      ],
      "mid_block.attentions.0.proj_out": [
        89.2,
        88.7,
        88.8,
        88.4
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        89.0,
        88.7,
        89.1,
        87.4
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.9,
        88.9,
        89.1,
        87.6
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        89.0,
        89.0,
        88.7,
        86.7
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        88.9,
        89.1,
        89.1,
        87.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        89.0,
        88.9,
        88.9,
        87.1
      ],
      "mid_block.resnets.0.time_emb_proj": [
        89.3,
        88.9,
        88.9,
        86.4
      ],
      "mid_block.resnets.1.time_emb_proj": [
        89.1,
        88.9,
        88.9,
        86.6
      ],
      "down_blocks.3.resnets.0.conv1": [
        89.1,
        88.8,
        88.8,
        86.8
      ],
      "down_blocks.3.resnets.1.conv2": [
        89.2,
        88.7,
        88.7,
        86.9
      ],
      "up_blocks.0.resnets.1.conv1": [
        89.0,
        88.7,
        88.6,
        85.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        89.0,
        88.6,
        88.6,
        87.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.9,
        88.9,
        88.7,
        87.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        89.0,
        88.7,
        88.6,
        88.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.9,
        88.9,
        89.0,
        86.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        89.1,
        89.0,
        88.6,
        87.9
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        88.8,
        88.8,
        88.8,
        87.7
      ],
      "mid_block.attentions.0.proj_in": [
        89.0,
        88.7,
        88.7,
        87.2
      ],
      "up_blocks.0.resnets.1.conv2": [
        88.8,
        88.8,
        88.9,
        86.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.8,
        88.7,
        88.7,
        86.6
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        89.0,
        88.6,
        88.7,
        86.5
      ],
      "up_blocks.0.resnets.0.conv1": [
        88.8,
        89.0,
        88.8,
        86.0
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        88.9,
        88.8,
        88.7,
        87.5
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        89.0,
        88.8,
        88.6,
        87.3
      ],
      "down_blocks.3.resnets.1.conv1": [
        89.0,
        88.7,
        88.8,
        85.8
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        88.9,
        88.8,
        88.5,
        86.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        88.8,
        88.7,
        88.7,
        87.4
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        88.6,
        88.5,
        89.1,
        86.8
      ],
      "mid_block.resnets.1.conv1": [
        89.0,
        88.7,
        88.6,
        87.3
      ],
      "mid_block.resnets.1.conv2": [
        88.9,
        88.7,
        88.7,
        87.7
      ],
      "down_blocks.2.resnets.1.conv2": [
        88.7,
        88.8,
        88.5,
        87.9
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        88.8,
        88.6,
        88.4,
        86.7
      ],
      "up_blocks.0.resnets.0.conv2": [
        88.7,
        88.7,
        88.7,
        87.4
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        88.8,
        88.6,
        88.7,
        87.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.7,
        88.5,
        88.2,
        86.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.8,
        88.5,
        88.4,
        87.1
      ],
      "down_blocks.3.resnets.0.time_emb_proj": [
        88.8,
        88.7,
        88.6,
        87.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.6,
        88.7,
        88.4,
        88.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.8,
        88.8,
        88.2,
        86.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        89.0,
        88.8,
        88.4,
        86.7
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        89.0,
        88.6,
        88.4,
        87.2
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        88.5,
        88.3,
        88.1,
        87.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.8,
        88.4,
        88.4,
        87.3
      ],
      "down_blocks.3.resnets.0.conv2": [
        88.7,
        88.5,
        88.1,
        87.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        88.6,
        88.5,
        88.1,
        87.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.4,
        88.7,
        88.3,
        86.8
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        88.5,
        88.4,
        88.0,
        86.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.5,
        88.3,
        88.0,
        87.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.4,
        88.2,
        88.3,
        87.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        88.3,
        88.2,
        88.1,
        87.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.2,
        88.1,
        88.2,
        85.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        88.4,
        88.3,
        88.0,
        86.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.3,
        88.4,
        88.0,
        86.7
      ],
      "up_blocks.0.resnets.2.conv2": [
        88.3,
        88.3,
        88.0,
        86.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.4,
        88.2,
        88.0,
        86.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.3,
        88.1,
        88.0,
        85.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        88.4,
        88.3,
        88.0,
        86.7
      ],
      "up_blocks.1.resnets.2.conv1": [
        88.1,
        88.2,
        87.8,
        86.9
      ],
      "up_blocks.1.resnets.2.conv2": [
        88.4,
        88.1,
        87.8,
        86.7
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.4,
        88.3,
        87.9,
        86.5
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.4,
        88.3,
        88.1,
        87.4
      ],
      "mid_block.resnets.0.conv2": [
        88.7,
        88.2,
        87.8,
        86.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        88.7,
        88.1,
        88.1,
        87.2
      ],
      "up_blocks.1.resnets.1.conv1": [
        88.4,
        88.0,
        88.0,
        85.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        88.5,
        88.1,
        87.9,
        84.8
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        88.3,
        88.3,
        87.9,
        84.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.2,
        87.9,
        87.9,
        84.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.2,
        87.7,
        87.7,
        85.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        88.3,
        88.0,
        87.9,
        86.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.1,
        87.8,
        87.6,
        78.5
      ],
      "up_blocks.0.upsamplers.0.conv": [
        87.9,
        88.1,
        87.8,
        81.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        88.0,
        87.9,
        87.7,
        82.3
      ],
      "up_blocks.1.resnets.1.conv2": [
        88.1,
        88.3,
        87.9,
        83.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.3,
        87.9,
        87.5,
        82.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.1,
        88.0,
        87.7,
        85.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.2,
        87.9,
        88.0,
        84.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        88.3,
        87.7,
        87.8,
        86.1
      ],
      "down_blocks.2.downsamplers.0.conv": [
        87.9,
        87.8,
        87.8,
        85.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        88.1,
        88.0,
        87.8,
        86.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        87.8,
        87.5,
        87.8,
        86.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        87.8,
        87.9,
        87.7,
        86.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        87.7,
        87.7,
        87.7,
        85.9
      ],
      "up_blocks.1.resnets.0.conv2": [
        87.9,
        87.6,
        87.5,
        87.0
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.1,
        87.8,
        87.6,
        86.7
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.1,
        87.9,
        87.7,
        86.5
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q": [
        87.9,
        87.5,
        87.5,
        86.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        87.7,
        87.6,
        87.5,
        86.7
      ],
      "down_blocks.2.resnets.1.conv1": [
        87.7,
        87.8,
        87.7,
        86.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        87.7,
        87.7,
        87.6,
        86.2
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q": [
        87.7,
        87.6,
        87.9,
        86.9
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q": [
        87.8,
        87.5,
        87.8,
        85.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        87.8,
        87.5,
        87.6,
        87.7
      ],
      "up_blocks.1.attentions.0.proj_out": [
        87.9,
        87.8,
        87.2,
        86.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        87.8,
        87.5,
        87.5,
        86.8
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        87.7,
        87.7,
        87.5,
        87.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        87.6,
        87.5,
        87.6,
        87.2
      ],
      "down_blocks.2.attentions.0.proj_out": [
        87.7,
        87.6,
        87.6,
        87.4
      ],
      "down_blocks.3.resnets.1.time_emb_proj": [
        87.7,
        87.6,
        87.3,
        87.2
      ],
      "up_blocks.1.attentions.1.proj_out": [
        87.8,
        87.5,
        87.4,
        87.3
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        87.7,
        87.4,
        87.6,
        87.3
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        87.5,
        87.5,
        87.2,
        87.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        87.6,
        87.5,
        87.3,
        87.6
      ],
      "down_blocks.2.attentions.1.proj_in": [
        87.5,
        87.1,
        87.3,
        86.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        87.3,
        87.0,
        87.4,
        86.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        87.2,
        86.9,
        87.3,
        86.7
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        87.4,
        86.9,
        87.3,
        86.6
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        87.5,
        87.1,
        87.8,
        86.9
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        87.6,
        87.1,
        87.3,
        87.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        87.5,
        87.0,
        87.3,
        84.5
      ],
      "up_blocks.1.attentions.2.proj_out": [
        87.4,
        87.0,
        87.3,
        85.0
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q": [
        87.1,
        86.7,
        87.2,
        85.1
      ],
      "time_embedding.linear_1": [
        87.1,
        86.7,
        87.3,
        83.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        87.2,
        86.8,
        87.3,
        85.9
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        87.1,
        86.7,
        87.2,
        85.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        87.0,
        86.5,
        87.1,
        86.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        86.8,
        86.8,
        86.9,
        87.0
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        87.1,
        86.7,
        87.1,
        86.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        87.0,
        86.5,
        87.0,
        87.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        87.1,
        86.5,
        87.0,
        87.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        86.6,
        86.5,
        87.1,
        87.1
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        86.7,
        86.4,
        87.0,
        87.0
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        86.7,
        86.5,
        87.1,
        86.9
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        86.6,
        86.5,
        87.0,
        86.8
      ],
      "up_blocks.3.resnets.0.time_emb_proj": [
        86.5,
        86.5,
        87.0,
        86.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        86.6,
        86.3,
        86.9,
        86.8
      ],
      "down_blocks.2.attentions.0.proj_in": [
        86.6,
        86.3,
        86.9,
        86.4
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        86.3,
        86.2,
        86.7,
        86.3
      ],
      "up_blocks.2.resnets.0.conv1": [
        86.4,
        86.5,
        86.8,
        86.2
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v": [
        86.3,
        86.4,
        86.8,
        86.4
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        86.5,
        86.2,
        86.9,
        86.5
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        86.5,
        86.2,
        86.9,
        86.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        86.6,
        85.9,
        86.9,
        86.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        86.7,
        86.2,
        86.7,
        86.3
      ],
      "up_blocks.1.upsamplers.0.conv": [
        86.4,
        86.1,
        86.6,
        86.0
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q": [
        86.3,
        86.2,
        86.6,
        85.8
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k": [
        86.2,
        86.0,
        86.8,
        85.8
      ],
      "down_blocks.2.attentions.1.proj_out": [
        86.2,
        86.0,
        86.6,
        85.7
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        86.3,
        85.8,
        86.5,
        85.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        86.2,
        85.8,
        86.5,
        85.4
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        86.1,
        85.7,
        86.3,
        85.3
      ],
      "down_blocks.1.resnets.1.conv2": [
        86.4,
        85.6,
        86.3,
        85.6
      ],
      "up_blocks.1.attentions.2.proj_in": [
        86.3,
        85.7,
        86.5,
        85.5
      ],
      "up_blocks.2.resnets.1.conv2": [
        86.2,
        85.7,
        86.4,
        85.5
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        86.2,
        85.4,
        86.4,
        85.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        85.9,
        85.1,
        86.4,
        85.6
      ],
      "down_blocks.2.resnets.0.conv1": [
        86.0,
        85.2,
        86.3,
        85.2
      ],
      "up_blocks.2.resnets.0.conv2": [
        85.9,
        85.2,
        86.3,
        84.9
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k": [
        85.9,
        85.0,
        86.1,
        84.9
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k": [
        85.8,
        85.3,
        86.1,
        84.8
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        85.6,
        85.4,
        86.0,
        84.9
      ],
      "down_blocks.1.resnets.0.conv2": [
        85.8,
        85.4,
        85.9,
        84.7
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        85.7,
        85.6,
        86.1,
        84.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        85.7,
        85.4,
        85.9,
        84.7
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        85.6,
        85.4,
        85.8,
        84.4
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v": [
        85.6,
        85.1,
        85.8,
        84.5
      ],
      "up_blocks.2.attentions.0.proj_out": [
        85.6,
        85.4,
        85.7,
        84.5
      ],
      "up_blocks.2.resnets.1.conv1": [
        85.6,
        85.3,
        85.8,
        84.4
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        85.4,
        85.4,
        85.4,
        84.4
      ],
      "up_blocks.1.attentions.1.proj_in": [
        85.5,
        85.4,
        85.5,
        84.4
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        85.5,
        85.5,
        85.7,
        84.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        85.5,
        85.5,
        85.5,
        84.3
      ],
      "up_blocks.1.attentions.0.proj_in": [
        85.2,
        85.3,
        85.5,
        84.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        85.3,
        85.4,
        85.5,
        84.3
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v": [
        85.3,
        85.5,
        85.3,
        84.3
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q": [
        85.1,
        85.4,
        85.4,
        84.2
      ],
      "down_blocks.1.resnets.0.conv1": [
        85.1,
        85.3,
        85.5,
        84.2
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        85.2,
        85.4,
        85.4,
        84.2
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        85.0,
        85.0,
        85.5,
        84.2
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        85.2,
        85.4,
        85.5,
        84.2
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        85.1,
        85.2,
        85.6,
        84.1
      ],
      "up_blocks.1.resnets.0.conv1": [
        85.2,
        85.3,
        85.5,
        84.1
      ],
      "up_blocks.2.attentions.0.proj_in": [
        85.3,
        85.1,
        85.5,
        84.0
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        85.1,
        85.2,
        85.6,
        84.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        85.2,
        85.0,
        85.6,
        83.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        85.3,
        85.0,
        85.2,
        84.1
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        85.0,
        85.0,
        85.0,
        84.0
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q": [
        85.1,
        84.7,
        85.1,
        84.0
      ],
      "up_blocks.3.resnets.2.time_emb_proj": [
        84.9,
        84.9,
        84.9,
        83.9
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        84.8,
        84.8,
        84.6,
        83.6
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k": [
        84.6,
        84.8,
        84.7,
        83.6
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k": [
        84.5,
        85.1,
        84.5,
        83.7
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        84.6,
        84.9,
        85.0,
        83.4
      ],
      "up_blocks.2.resnets.2.conv2": [
        84.5,
        84.8,
        85.0,
        83.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        84.5,
        84.9,
        85.1,
        81.3
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2": [
        84.4,
        84.8,
        85.0,
        80.7
      ],
      "down_blocks.1.attentions.1.proj_in": [
        84.5,
        84.7,
        85.2,
        78.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        84.5,
        84.7,
        84.8,
        78.5
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        84.3,
        84.7,
        84.8,
        78.6
      ],
      "down_blocks.1.resnets.1.conv1": [
        84.4,
        84.7,
        85.1,
        78.4
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        84.3,
        84.7,
        85.0,
        78.4
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        84.3,
        84.7,
        85.1,
        78.4
      ],
      "up_blocks.3.resnets.1.time_emb_proj": [
        84.4,
        84.6,
        84.9,
        78.4
      ],
      "up_blocks.2.resnets.2.conv1": [
        84.4,
        84.7,
        85.1,
        78.3
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        84.3,
        84.6,
        84.7,
        78.3
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k": [
        84.4,
        84.3,
        84.7,
        78.4
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        84.4,
        84.3,
        84.6,
        78.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        84.4,
        84.3,
        85.0,
        78.9
      ],
      "up_blocks.2.attentions.1.proj_in": [
        84.4,
        84.2,
        84.7,
        79.1
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2": [
        84.2,
        84.2,
        84.6,
        79.2
      ],
      "down_blocks.1.attentions.1.proj_out": [
        84.2,
        84.2,
        84.5,
        78.3
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v": [
        84.1,
        84.0,
        84.4,
        78.3
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q": [
        84.0,
        84.0,
        84.3,
        78.3
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        83.9,
        84.0,
        83.8,
        78.3
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        84.0,
        84.0,
        83.8,
        78.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        84.0,
        84.0,
        78.5,
        78.4
      ],
      "up_blocks.2.attentions.1.proj_out": [
        83.8,
        83.8,
        78.4,
        78.3
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v": [
        83.4,
        83.5,
        78.4,
        78.3
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2": [
        83.5,
        83.5,
        78.4,
        78.2
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k": [
        83.5,
        83.4,
        78.3,
        78.2
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        83.5,
        83.4,
        78.6,
        78.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        83.4,
        83.6,
        79.0,
        78.3
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        83.3,
        83.3,
        79.1,
        78.3
      ],
      "up_blocks.2.upsamplers.0.conv": [
        83.2,
        83.1,
        78.5,
        78.2
      ],
      "down_blocks.1.downsamplers.0.conv": [
        83.2,
        83.1,
        83.4,
        78.2
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        83.1,
        83.2,
        82.8,
        78.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        82.9,
        83.0,
        77.4,
        78.2
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        83.0,
        83.0,
        80.6,
        78.3
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2": [
        82.9,
        82.9,
        79.3,
        78.2
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        83.0,
        82.8,
        77.8,
        78.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        83.0,
        82.8,
        76.9,
        78.2
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v": [
        82.8,
        82.8,
        76.9,
        78.2
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        82.8,
        82.9,
        76.9,
        78.2
      ],
      "up_blocks.3.resnets.0.conv1": [
        82.8,
        82.7,
        76.9,
        78.2
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        82.9,
        82.7,
        77.1,
        78.2
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k": [
        82.7,
        82.6,
        77.1,
        78.2
      ],
      "up_blocks.2.attentions.2.proj_out": [
        82.7,
        82.6,
        77.1,
        78.2
      ],
      "up_blocks.2.attentions.2.proj_in": [
        82.6,
        82.3,
        77.1,
        78.1
      ],
      "up_blocks.3.attentions.0.proj_out": [
        82.6,
        82.3,
        77.0,
        78.1
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v": [
        82.4,
        82.3,
        77.0,
        78.0
      ],
      "time_embedding.linear_2": [
        82.3,
        82.0,
        77.1,
        78.0
      ],
      "down_blocks.0.attentions.0.proj_out": [
        82.2,
        81.9,
        76.9,
        78.0
      ],
      "down_blocks.1.attentions.0.proj_out": [
        82.2,
        81.8,
        77.9,
        78.0
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        82.2,
        81.8,
        77.5,
        78.0
      ],
      "down_blocks.0.resnets.1.conv2": [
        82.0,
        81.9,
        77.9,
        78.0
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        82.1,
        81.9,
        77.5,
        78.0
      ],
      "down_blocks.0.downsamplers.0.conv": [
        81.9,
        81.7,
        78.3,
        77.9
      ],
      "down_blocks.0.attentions.0.proj_in": [
        81.9,
        81.7,
        78.1,
        77.9
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        82.0,
        81.6,
        77.9,
        77.9
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        81.9,
        81.5,
        77.9,
        77.9
      ],
      "up_blocks.3.attentions.2.proj_out": [
        81.8,
        81.4,
        77.9,
        77.9
      ],
      "up_blocks.3.resnets.0.conv2": [
        81.7,
        81.5,
        77.6,
        77.9
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        81.7,
        81.3,
        79.5,
        77.8
      ],
      "up_blocks.3.attentions.1.proj_out": [
        81.6,
        81.3,
        79.3,
        77.8
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        81.7,
        81.2,
        76.8,
        77.8
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        81.4,
        81.0,
        76.8,
        77.8
      ],
      "down_blocks.2.resnets.0.conv2": [
        81.4,
        81.0,
        76.9,
        77.8
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        81.2,
        80.8,
        76.8,
        77.7
      ],
      "down_blocks.1.attentions.0.proj_in": [
        81.3,
        80.9,
        77.7,
        77.6
      ],
      "up_blocks.3.attentions.0.proj_in": [
        81.1,
        80.9,
        77.6,
        77.6
      ],
      "down_blocks.0.attentions.1.proj_in": [
        81.1,
        81.0,
        81.7,
        77.6
      ],
      "up_blocks.3.resnets.2.conv1": [
        81.0,
        80.9,
        81.5,
        77.6
      ],
      "down_blocks.0.resnets.1.conv1": [
        81.0,
        80.9,
        81.5,
        77.6
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        80.6,
        80.3,
        81.2,
        77.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        80.6,
        80.2,
        81.1,
        77.5
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v": [
        80.2,
        80.0,
        80.8,
        77.3
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        80.3,
        80.1,
        79.1,
        77.4
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        80.1,
        79.8,
        79.0,
        77.4
      ],
      "down_blocks.0.resnets.0.conv1": [
        80.1,
        79.9,
        80.6,
        77.4
      ],
      "down_blocks.0.resnets.0.conv2": [
        80.2,
        80.0,
        77.3,
        77.3
      ],
      "up_blocks.3.resnets.1.conv2": [
        80.0,
        79.9,
        77.3,
        77.3
      ],
      "up_blocks.3.resnets.2.conv2": [
        79.8,
        79.7,
        77.1,
        77.2
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        79.8,
        79.6,
        75.9,
        77.3
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        79.9,
        79.7,
        76.0,
        77.2
      ],
      "up_blocks.3.resnets.1.conv1": [
        79.7,
        79.7,
        75.9,
        77.1
      ],
      "up_blocks.3.attentions.1.proj_in": [
        79.7,
        79.5,
        75.9,
        77.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        79.5,
        79.5,
        75.7,
        77.2
      ],
      "up_blocks.3.resnets.0.conv_shortcut": [
        79.5,
        79.3,
        75.6,
        77.1
      ],
      "up_blocks.3.resnets.1.conv_shortcut": [
        79.3,
        79.1,
        75.5,
        76.9
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        79.1,
        79.0,
        75.5,
        76.8
      ],
      "up_blocks.3.resnets.2.conv_shortcut": [
        79.1,
        78.9,
        75.5,
        76.9
      ],
      "up_blocks.3.attentions.2.proj_in": [
        79.1,
        79.0,
        75.5,
        76.8
      ],
      "down_blocks.0.attentions.1.proj_out": [
        79.1,
        79.1,
        77.1,
        76.8
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        79.1,
        79.0,
        78.6,
        76.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        79.2,
        79.0,
        78.6,
        76.7
      ]
    },
    "8": {
      "metadata": {
        "nbits": 8,
        "out_ngroups": 1,
        "in_ngroups": 1,
        "cumulative": true,
        "candidates": [
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k",
          "mid_block.attentions.0.proj_in",
          "mid_block.resnets.0.conv2",
          "down_blocks.2.resnets.1.time_emb_proj",
          "down_blocks.3.resnets.0.conv1",
          "up_blocks.2.resnets.1.time_emb_proj",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.resnets.0.conv_shortcut",
          "up_blocks.0.resnets.1.conv_shortcut",
          "up_blocks.0.resnets.2.time_emb_proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.3.resnets.1.conv1",
          "up_blocks.0.resnets.1.conv1",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q",
          "mid_block.resnets.0.time_emb_proj",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.3.resnets.1.time_emb_proj",
          "up_blocks.0.resnets.2.conv2",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.resnets.0.conv2",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.1.resnets.0.conv_shortcut",
          "up_blocks.1.resnets.2.time_emb_proj",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.resnets.0.conv1",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.resnets.0.conv1",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.resnets.1.conv_shortcut",
          "up_blocks.2.resnets.0.time_emb_proj",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.3.resnets.1.conv2",
          "up_blocks.0.upsamplers.0.conv",
          "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.2.resnets.1.conv1",
          "down_blocks.2.resnets.1.conv2",
          "down_blocks.3.resnets.0.conv2",
          "up_blocks.1.resnets.0.time_emb_proj",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k",
          "up_blocks.1.resnets.1.conv1",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.resnets.2.conv1",
          "up_blocks.0.resnets.2.conv_shortcut",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.1.proj_out",
          "up_blocks.1.upsamplers.0.conv",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_v",
          "mid_block.attentions.0.proj_out",
          "mid_block.resnets.0.conv1",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.2.resnets.0.conv_shortcut",
          "up_blocks.1.attentions.0.proj_out",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2",
          "mid_block.resnets.1.conv1",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.attentions.1.proj_out",
          "up_blocks.0.resnets.1.time_emb_proj",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.resnets.1.conv2",
          "up_blocks.1.resnets.2.conv1",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.3.attentions.2.proj_out",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.2.attentions.0.proj_out",
          "down_blocks.2.resnets.0.conv2",
          "up_blocks.0.resnets.1.conv2",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.attentions.1.proj_in",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.2.resnets.2.conv1",
          "up_blocks.3.resnets.0.time_emb_proj",
          "mid_block.resnets.1.time_emb_proj",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.0.proj_in",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.resnets.0.conv1",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.2.attentions.0.proj_out",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.2.proj_out",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "time_embedding.linear_1",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v",
          "down_blocks.1.resnets.0.conv2",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k",
          "up_blocks.2.resnets.1.conv1",
          "up_blocks.1.resnets.0.conv1",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.1.resnets.0.conv1",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.2.resnets.0.conv2",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.2.resnets.2.time_emb_proj",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
          "down_blocks.1.attentions.1.proj_out",
          "down_blocks.1.resnets.0.time_emb_proj",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.2.proj_in",
          "up_blocks.2.resnets.2.conv2",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.attentions.0.proj_in",
          "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q",
          "down_blocks.1.downsamplers.0.conv",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.resnets.0.conv2",
          "up_blocks.1.resnets.1.time_emb_proj",
          "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.2.resnets.1.conv2",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.1.resnets.1.conv2",
          "up_blocks.2.attentions.0.proj_in",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.3.resnets.0.time_emb_proj",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.downsamplers.0.conv",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.resnets.1.time_emb_proj",
          "down_blocks.2.resnets.0.time_emb_proj",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.resnets.2.conv2",
          "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q",
          "down_blocks.0.resnets.1.time_emb_proj",
          "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.2.attentions.1.proj_out",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2",
          "up_blocks.3.resnets.2.conv2",
          "down_blocks.1.resnets.1.conv1",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.2.attentions.1.proj_in",
          "up_blocks.2.resnets.1.conv_shortcut",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v",
          "up_blocks.3.resnets.0.conv2",
          "mid_block.resnets.1.conv2",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.0.resnets.0.time_emb_proj",
          "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.2.resnets.2.conv_shortcut",
          "up_blocks.3.resnets.1.time_emb_proj",
          "up_blocks.3.resnets.2.time_emb_proj",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.resnets.2.conv_shortcut",
          "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.3.attentions.1.proj_out",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v",
          "down_blocks.1.attentions.0.proj_in",
          "up_blocks.2.resnets.0.conv_shortcut",
          "up_blocks.3.resnets.2.conv1",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.1.attentions.1.proj_in",
          "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v",
          "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.0.resnets.0.conv1",
          "down_blocks.0.resnets.0.time_emb_proj",
          "up_blocks.2.attentions.2.proj_out",
          "up_blocks.2.upsamplers.0.conv",
          "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k",
          "time_embedding.linear_2",
          "down_blocks.1.attentions.0.proj_out",
          "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.resnets.1.conv1",
          "up_blocks.2.attentions.1.proj_in",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.0.resnets.1.conv2",
          "up_blocks.3.resnets.0.conv1",
          "up_blocks.2.attentions.2.proj_in",
          "up_blocks.3.attentions.0.proj_out",
          "up_blocks.3.resnets.1.conv2",
          "down_blocks.0.attentions.0.proj_out",
          "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.0.attentions.1.proj_out",
          "up_blocks.3.attentions.2.proj_in",
          "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.3.attentions.0.proj_in",
          "up_blocks.3.resnets.0.conv_shortcut",
          "down_blocks.0.attentions.0.proj_in",
          "down_blocks.0.resnets.0.conv2",
          "down_blocks.1.resnets.0.conv_shortcut",
          "up_blocks.3.attentions.1.proj_in",
          "up_blocks.3.resnets.2.conv_shortcut",
          "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.3.resnets.1.conv_shortcut",
          "down_blocks.0.resnets.1.conv1",
          "down_blocks.0.attentions.1.proj_in",
          "down_blocks.0.downsamplers.0.conv"
        ],
        "sizes": [
          655360,
          6553600,
          1638400,
          1638400,
          1638400,
          655360,
          1638400,
          14745600,
          1638400,
          14745600,
          819200,
          409600,
          1310720,
          3276800,
          3276800,
          1638400,
          13107200,
          1638400,
          13107200,
          14745600,
          29491200,
          1638400,
          409600,
          1638400,
          1638400,
          1638400,
          14745600,
          1310720,
          327680,
          6553600,
          102400,
          1310720,
          14745600,
          1310720,
          3276800,
          1638400,
          409600,
          11059200,
          327680,
          655360,
          1638400,
          102400,
          1638400,
          29491200,
          1638400,
          3276800,
          819200,
          102400,
          655360,
          409600,
          1638400,
          1638400,
          6553600,
          14745600,
          14745600,
          1638400,
          655360,
          13107200,
          1638400,
          14745600,
          14745600,
          14745600,
          1638400,
          409600,
          1310720,
          1638400,
          1310720,
          1310720,
          1310720,
          29491200,
          409600,
          409600,
          655360,
          29491200,
          3276800,
          1310720,
          1638400,
          14745600,
          1310720,
          1638400,
          14745600,
          1638400,
          819200,
          1638400,
          6553600,
          14745600,
          409600,
          409600,
          1638400,
          1638400,
          1638400,
          1310720,
          14745600,
          22118400,
          409600,
          409600,
          409600,
          102400,
          1638400,
          409600,
          655360,
          1638400,
          1638400,
          14745600,
          14745600,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          5529600,
          409600,
          1638400,
          1638400,
          1638400,
          102400,
          102400,
          409600,
          1638400,
          7372800,
          1638400,
          6553600,
          1638400,
          409600,
          1638400,
          327680,
          655360,
          1638400,
          409600,
          409600,
          409600,
          409600,
          1638400,
          1638400,
          409600,
          409600,
          327680,
          3686400,
          409600,
          102400,
          13107200,
          1638400,
          655360,
          7372800,
          29491200,
          1638400,
          1843200,
          1638400,
          3686400,
          1638400,
          409600,
          819200,
          409600,
          409600,
          819200,
          6553600,
          1638400,
          3686400,
          1310720,
          1638400,
          409600,
          3276800,
          327680,
          102400,
          102400,
          3686400,
          1638400,
          1638400,
          14745600,
          1638400,
          3276800,
          102400,
          3686400,
          102400,
          3686400,
          409600,
          102400,
          1638400,
          409600,
          102400,
          655360,
          14745600,
          1638400,
          13107200,
          819200,
          1638400,
          1638400,
          14745600,
          1638400,
          102400,
          409600,
          102400,
          409600,
          1638400,
          409600,
          102400,
          409600,
          921600,
          3686400,
          102400,
          102400,
          102400,
          409600,
          409600,
          1638400,
          819200,
          327680,
          921600,
          14745600,
          102400,
          102400,
          102400,
          1638400,
          1638400,
          3276800,
          614400,
          409600,
          409600,
          3276800,
          2457600,
          409600,
          1638400,
          102400,
          102400,
          409600,
          1228800,
          1843200,
          327680,
          409600,
          819200,
          409600,
          102400,
          819200,
          409600,
          409600,
          102400,
          102400,
          102400,
          921600,
          409600,
          409600,
          3686400,
          102400,
          102400,
          819200,
          327680,
          327680,
          327680,
          1638400,
          409600,
          102400,
          819200,
          1843200,
          409600,
          3276800,
          409600,
          102400,
          921600,
          2764800,
          409600,
          102400,
          921600,
          102400,
          409600,
          102400,
          102400,
          409600,
          102400,
          307200,
          102400,
          921600,
          204800,
          102400,
          204800,
          819200,
          204800,
          921600,
          102400,
          921600
        ]
      },
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        88.9,
        88.7,
        88.9,
        88.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        89.1,
        88.7,
        89.2,
        88.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        88.9,
        88.8,
        88.8,
        88.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.7,
        88.8,
        89.2,
        89.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.8,
        88.6,
        88.7,
        88.0
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        88.9,
        88.5,
        88.9,
        88.3
      ],
      "mid_block.attentions.0.proj_in": [
        88.8,
        88.6,
        88.6,
        87.3
      ],
      "mid_block.resnets.0.conv2": [
        88.8,
        88.5,
        89.0,
        89.1
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        88.8,
        88.9,
        88.8,
        87.8
      ],
      "down_blocks.3.resnets.0.conv1": [
        89.1,
        88.5,
        88.7,
        88.2
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        89.1,
        88.6,
        88.6,
        88.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        88.9,
        88.5,
        88.9,
        88.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        89.0,
        88.5,
        88.7,
        88.5
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        88.8,
        88.6,
        88.5,
        88.5
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        88.6,
        88.8,
        88.9,
        87.4
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        88.8,
        88.6,
        88.7,
        88.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        88.9,
        88.7,
        88.9,
        88.5
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.8,
        88.7,
        88.4,
        88.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        88.9,
        88.5,
        88.8,
        88.8
      ],
      "down_blocks.3.resnets.1.conv1": [
        89.1,
        88.7,
        88.7,
        88.7
      ],
      "up_blocks.0.resnets.1.conv1": [
        89.1,
        88.8,
        88.7,
        88.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        88.9,
        88.8,
        88.9,
        87.8
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q": [
        88.9,
        88.8,
        88.7,
        88.8
      ],
      "mid_block.resnets.0.time_emb_proj": [
        88.7,
        88.6,
        88.8,
        88.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        88.9,
        88.7,
        88.0,
        88.4
      ],
      "down_blocks.3.resnets.1.time_emb_proj": [
        89.0,
        88.7,
        88.0,
        88.7
      ],
      "up_blocks.0.resnets.2.conv2": [
        89.0,
        88.4,
        88.9,
        88.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        89.1,
        88.8,
        88.2,
        88.7
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k": [
        88.6,
        88.5,
        88.1,
        88.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        88.7,
        88.7,
        88.5,
        88.8
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.8,
        88.5,
        88.7,
        88.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.9,
        88.5,
        88.5,
        88.9
      ],
      "up_blocks.0.resnets.0.conv2": [
        88.7,
        88.6,
        88.7,
        87.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        88.7,
        88.3,
        88.5,
        88.8
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        88.6,
        88.6,
        88.5,
        88.4
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        88.8,
        88.4,
        88.5,
        88.7
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.7,
        88.6,
        88.6,
        88.8
      ],
      "up_blocks.2.resnets.0.conv1": [
        89.0,
        88.6,
        88.5,
        88.6
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v": [
        88.5,
        88.5,
        88.8,
        88.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        88.7,
        88.3,
        88.5,
        88.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.8,
        88.4,
        88.5,
        87.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.5,
        88.2,
        88.4,
        87.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.8,
        88.3,
        88.7,
        88.4
      ],
      "up_blocks.0.resnets.0.conv1": [
        88.7,
        88.6,
        88.5,
        88.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        88.5,
        88.5,
        88.8,
        88.0
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        88.7,
        88.6,
        88.6,
        88.0
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        88.6,
        88.4,
        88.6,
        88.2
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.9,
        88.3,
        88.7,
        87.6
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.7,
        88.3,
        88.5,
        88.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.8,
        88.4,
        88.5,
        87.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.6,
        88.1,
        88.5,
        87.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.8,
        88.5,
        88.6,
        87.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        88.7,
        88.6,
        88.4,
        88.4
      ],
      "down_blocks.3.resnets.1.conv2": [
        88.9,
        88.4,
        88.5,
        88.1
      ],
      "up_blocks.0.upsamplers.0.conv": [
        88.7,
        88.3,
        88.6,
        88.3
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        88.6,
        88.6,
        88.6,
        88.2
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        88.7,
        88.1,
        88.5,
        88.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        88.9,
        88.1,
        88.6,
        88.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.8,
        88.5,
        88.3,
        87.6
      ],
      "down_blocks.2.resnets.1.conv1": [
        88.5,
        88.4,
        88.6,
        87.9
      ],
      "down_blocks.2.resnets.1.conv2": [
        89.0,
        88.1,
        88.5,
        87.5
      ],
      "down_blocks.3.resnets.0.conv2": [
        88.6,
        88.3,
        88.1,
        87.5
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        89.0,
        88.3,
        88.2,
        88.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.8,
        88.5,
        88.3,
        87.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        88.7,
        88.4,
        88.6,
        87.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        88.9,
        88.3,
        88.6,
        87.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        88.8,
        88.4,
        88.4,
        87.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        88.4,
        88.3,
        88.3,
        87.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        88.7,
        88.4,
        88.5,
        87.7
      ],
      "up_blocks.1.resnets.1.conv1": [
        88.5,
        88.2,
        88.8,
        87.6
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.6,
        88.3,
        88.6,
        87.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.8,
        88.3,
        88.2,
        87.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        88.5,
        88.3,
        88.5,
        87.7
      ],
      "up_blocks.0.resnets.2.conv1": [
        88.7,
        88.2,
        88.6,
        88.0
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        88.6,
        88.1,
        88.5,
        87.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        88.4,
        88.4,
        88.6,
        87.8
      ],
      "up_blocks.1.attentions.1.proj_out": [
        88.4,
        88.2,
        88.4,
        87.8
      ],
      "up_blocks.1.upsamplers.0.conv": [
        88.6,
        88.3,
        88.7,
        87.7
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.7,
        88.4,
        88.7,
        87.7
      ],
      "mid_block.attentions.0.proj_out": [
        88.6,
        88.4,
        88.4,
        87.6
      ],
      "mid_block.resnets.0.conv1": [
        88.3,
        88.3,
        88.6,
        87.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        88.3,
        88.3,
        88.4,
        87.7
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        88.5,
        88.4,
        88.6,
        87.7
      ],
      "up_blocks.1.attentions.0.proj_out": [
        88.7,
        88.3,
        88.5,
        87.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        88.5,
        88.1,
        88.8,
        87.7
      ],
      "mid_block.resnets.1.conv1": [
        88.5,
        88.3,
        88.5,
        87.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.4,
        88.4,
        88.8,
        87.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.5,
        88.2,
        88.5,
        87.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.4,
        88.3,
        88.5,
        87.8
      ],
      "down_blocks.2.attentions.1.proj_out": [
        88.3,
        88.2,
        88.4,
        87.7
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        88.4,
        88.3,
        88.6,
        87.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.5,
        88.5,
        88.2,
        87.6
      ],
      "up_blocks.1.resnets.1.conv2": [
        88.6,
        88.3,
        88.4,
        87.4
      ],
      "up_blocks.1.resnets.2.conv1": [
        88.4,
        88.3,
        88.4,
        87.8
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.2,
        88.2,
        88.8,
        87.4
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        88.5,
        88.5,
        88.5,
        87.6
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.6,
        88.5,
        88.5,
        87.5
      ],
      "up_blocks.3.attentions.2.proj_out": [
        88.3,
        88.0,
        88.0,
        87.3
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.2,
        88.0,
        88.2,
        87.5
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.4,
        88.2,
        87.9,
        87.5
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v": [
        88.2,
        88.2,
        88.3,
        87.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        88.3,
        87.9,
        88.2,
        87.7
      ],
      "down_blocks.2.attentions.0.proj_out": [
        88.4,
        88.0,
        88.4,
        87.7
      ],
      "down_blocks.2.resnets.0.conv2": [
        88.2,
        87.9,
        88.3,
        87.4
      ],
      "up_blocks.0.resnets.1.conv2": [
        88.3,
        88.0,
        88.6,
        87.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.3,
        87.9,
        87.9,
        87.3
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        88.4,
        88.1,
        88.1,
        87.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        88.3,
        88.1,
        87.9,
        87.5
      ],
      "up_blocks.1.attentions.1.proj_in": [
        88.4,
        87.9,
        87.9,
        87.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.3,
        87.9,
        87.8,
        87.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.2,
        87.9,
        87.6,
        87.6
      ],
      "up_blocks.2.resnets.2.conv1": [
        88.0,
        88.2,
        87.9,
        87.7
      ],
      "up_blocks.3.resnets.0.time_emb_proj": [
        88.3,
        88.0,
        88.0,
        87.4
      ],
      "mid_block.resnets.1.time_emb_proj": [
        88.4,
        87.6,
        88.1,
        87.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        88.1,
        87.9,
        87.8,
        87.6
      ],
      "up_blocks.1.attentions.0.proj_in": [
        88.4,
        87.9,
        88.2,
        87.4
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q": [
        88.3,
        87.9,
        88.0,
        87.3
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        88.2,
        88.0,
        87.7,
        87.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.2,
        87.7,
        88.3,
        87.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        88.1,
        87.8,
        88.3,
        87.6
      ],
      "down_blocks.2.resnets.0.conv1": [
        88.1,
        87.9,
        88.2,
        87.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        88.0,
        88.0,
        88.3,
        87.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        88.3,
        88.1,
        88.3,
        87.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        88.1,
        87.9,
        88.2,
        87.5
      ],
      "up_blocks.2.attentions.0.proj_out": [
        88.2,
        87.9,
        88.4,
        87.6
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        88.3,
        87.7,
        88.0,
        87.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v": [
        88.0,
        87.9,
        87.4,
        87.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        87.9,
        87.9,
        87.3,
        87.5
      ],
      "up_blocks.1.attentions.2.proj_out": [
        88.1,
        87.8,
        87.9,
        87.8
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.1,
        87.8,
        88.3,
        87.7
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k": [
        88.3,
        87.8,
        87.6,
        87.7
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        88.2,
        87.7,
        87.8,
        87.7
      ],
      "time_embedding.linear_1": [
        88.1,
        87.8,
        88.4,
        87.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        88.1,
        87.8,
        87.7,
        87.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        88.3,
        87.6,
        87.8,
        87.7
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        87.9,
        87.7,
        87.7,
        87.8
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        87.9,
        87.6,
        87.7,
        87.7
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        87.8,
        87.8,
        87.7,
        87.6
      ],
      "down_blocks.1.resnets.0.conv2": [
        88.0,
        87.9,
        87.2,
        87.5
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        87.9,
        88.0,
        87.2,
        87.3
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        88.0,
        87.4,
        87.3,
        87.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        88.1,
        87.7,
        87.2,
        87.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        87.9,
        87.6,
        87.4,
        87.1
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k": [
        88.1,
        87.6,
        87.4,
        87.2
      ],
      "up_blocks.2.resnets.1.conv1": [
        88.0,
        87.6,
        87.3,
        87.2
      ],
      "up_blocks.1.resnets.0.conv1": [
        87.8,
        87.9,
        87.2,
        87.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        88.0,
        87.7,
        87.4,
        87.3
      ],
      "down_blocks.1.resnets.0.conv1": [
        88.1,
        87.8,
        87.3,
        87.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.1,
        87.7,
        87.4,
        87.0
      ],
      "up_blocks.2.resnets.0.conv2": [
        88.2,
        87.8,
        87.4,
        87.5
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.2,
        87.7,
        87.7,
        87.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.1,
        87.7,
        87.3,
        87.3
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        88.1,
        87.7,
        87.8,
        87.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        88.1,
        87.6,
        87.4,
        87.0
      ],
      "down_blocks.1.attentions.1.proj_out": [
        87.9,
        87.6,
        87.0,
        87.0
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        88.1,
        87.8,
        87.0,
        86.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        87.9,
        87.6,
        86.7,
        86.7
      ],
      "up_blocks.1.attentions.2.proj_in": [
        88.3,
        87.7,
        87.3,
        86.7
      ],
      "up_blocks.2.resnets.2.conv2": [
        87.9,
        87.6,
        87.1,
        86.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        87.8,
        87.7,
        87.0,
        86.8
      ],
      "down_blocks.2.attentions.0.proj_in": [
        87.9,
        87.7,
        87.2,
        86.7
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        88.0,
        87.7,
        87.2,
        86.7
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        88.0,
        87.6,
        87.2,
        86.6
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k": [
        88.1,
        87.7,
        87.1,
        86.8
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q": [
        87.9,
        87.9,
        87.2,
        86.6
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        88.0,
        87.6,
        87.2,
        86.6
      ],
      "down_blocks.1.downsamplers.0.conv": [
        87.8,
        87.7,
        86.9,
        86.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        87.9,
        87.8,
        86.8,
        86.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        88.2,
        87.7,
        87.1,
        86.7
      ],
      "up_blocks.1.resnets.0.conv2": [
        87.8,
        87.8,
        87.1,
        86.4
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        88.0,
        87.8,
        86.8,
        86.5
      ],
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        88.1,
        87.6,
        87.0,
        86.6
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.0,
        87.9,
        86.9,
        86.8
      ],
      "up_blocks.2.resnets.1.conv2": [
        87.9,
        87.7,
        86.8,
        86.6
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        88.0,
        87.5,
        86.6,
        86.7
      ],
      "down_blocks.1.resnets.1.conv2": [
        87.9,
        87.4,
        86.7,
        86.8
      ],
      "up_blocks.2.attentions.0.proj_in": [
        87.9,
        87.8,
        87.0,
        86.5
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        88.0,
        87.8,
        86.6,
        86.5
      ],
      "down_blocks.3.resnets.0.time_emb_proj": [
        87.9,
        87.8,
        86.8,
        86.6
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q": [
        87.7,
        87.7,
        86.5,
        86.5
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.0,
        87.8,
        86.9,
        86.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        88.0,
        87.8,
        86.9,
        86.6
      ],
      "down_blocks.2.downsamplers.0.conv": [
        88.0,
        87.7,
        86.9,
        86.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        88.0,
        88.0,
        87.0,
        86.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        87.8,
        87.7,
        87.1,
        86.4
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        88.0,
        87.5,
        86.9,
        86.5
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        88.0,
        87.6,
        86.7,
        86.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        87.7,
        87.7,
        87.1,
        86.5
      ],
      "up_blocks.1.resnets.2.conv2": [
        87.8,
        87.7,
        86.9,
        86.4
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2": [
        87.8,
        87.8,
        87.1,
        86.4
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k": [
        87.9,
        87.8,
        86.8,
        86.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        88.0,
        87.5,
        86.7,
        86.7
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q": [
        87.9,
        87.6,
        86.7,
        86.5
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        87.7,
        87.6,
        86.2,
        86.7
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        87.9,
        87.6,
        86.1,
        86.6
      ],
      "up_blocks.2.attentions.1.proj_out": [
        87.9,
        87.6,
        85.8,
        86.6
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        87.4,
        87.4,
        86.0,
        86.5
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2": [
        87.5,
        87.3,
        86.2,
        86.6
      ],
      "up_blocks.3.resnets.2.conv2": [
        87.6,
        87.2,
        86.1,
        86.4
      ],
      "down_blocks.1.resnets.1.conv1": [
        87.6,
        87.1,
        86.4,
        86.3
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v": [
        87.4,
        87.5,
        86.4,
        86.1
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k": [
        87.4,
        87.2,
        86.6,
        86.3
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        87.5,
        87.3,
        86.8,
        86.5
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        87.5,
        87.2,
        86.4,
        86.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        87.5,
        87.2,
        86.8,
        86.4
      ],
      "down_blocks.2.attentions.1.proj_in": [
        87.4,
        87.5,
        87.1,
        86.3
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        87.5,
        87.2,
        86.9,
        86.6
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v": [
        87.4,
        87.2,
        87.1,
        86.6
      ],
      "up_blocks.3.resnets.0.conv2": [
        87.3,
        87.2,
        87.0,
        86.4
      ],
      "mid_block.resnets.1.conv2": [
        87.4,
        87.3,
        87.0,
        86.5
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        87.5,
        87.4,
        87.1,
        86.4
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k": [
        87.3,
        87.3,
        86.8,
        86.3
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        87.6,
        87.3,
        87.1,
        86.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        87.3,
        87.3,
        86.8,
        86.4
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        87.4,
        87.4,
        87.2,
        86.5
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        87.3,
        87.1,
        86.9,
        86.2
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        87.3,
        87.0,
        86.9,
        86.3
      ],
      "up_blocks.3.resnets.1.time_emb_proj": [
        87.3,
        87.3,
        86.9,
        86.5
      ],
      "up_blocks.3.resnets.2.time_emb_proj": [
        87.5,
        87.2,
        86.7,
        86.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        87.4,
        87.2,
        86.5,
        86.5
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        87.5,
        87.1,
        86.3,
        86.6
      ],
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        87.3,
        87.3,
        86.6,
        86.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        87.3,
        87.4,
        86.6,
        86.4
      ],
      "up_blocks.3.attentions.1.proj_out": [
        87.1,
        87.3,
        86.4,
        86.7
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v": [
        87.4,
        86.9,
        86.2,
        86.5
      ],
      "down_blocks.1.attentions.0.proj_in": [
        87.2,
        87.1,
        86.3,
        86.6
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        87.1,
        87.1,
        85.7,
        86.5
      ],
      "up_blocks.3.resnets.2.conv1": [
        87.1,
        87.0,
        86.0,
        86.5
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        87.0,
        87.0,
        86.1,
        86.8
      ],
      "down_blocks.1.attentions.1.proj_in": [
        86.8,
        87.1,
        85.1,
        86.4
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        87.0,
        87.0,
        85.3,
        86.5
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2": [
        87.1,
        87.0,
        85.2,
        86.6
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        86.9,
        86.7,
        85.4,
        86.6
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        86.8,
        86.5,
        85.4,
        86.5
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v": [
        86.8,
        86.5,
        84.8,
        86.4
      ],
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        86.8,
        86.7,
        85.3,
        86.6
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        86.9,
        86.9,
        85.2,
        86.2
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        86.8,
        86.6,
        85.0,
        86.5
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        86.8,
        86.7,
        86.2,
        86.4
      ],
      "down_blocks.0.resnets.0.conv1": [
        86.9,
        86.8,
        85.9,
        86.4
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        86.7,
        86.7,
        86.1,
        86.6
      ],
      "up_blocks.2.attentions.2.proj_out": [
        86.5,
        86.8,
        85.9,
        86.4
      ],
      "up_blocks.2.upsamplers.0.conv": [
        86.5,
        86.5,
        86.0,
        86.2
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v": [
        86.5,
        86.6,
        85.8,
        86.5
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        86.7,
        86.7,
        86.3,
        86.2
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        86.5,
        86.3,
        86.1,
        86.2
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        86.8,
        86.6,
        86.1,
        86.1
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        86.8,
        86.5,
        86.3,
        86.0
      ],
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k": [
        86.7,
        86.4,
        86.4,
        86.2
      ],
      "time_embedding.linear_2": [
        86.7,
        86.6,
        86.3,
        86.3
      ],
      "down_blocks.1.attentions.0.proj_out": [
        86.7,
        86.6,
        86.0,
        86.3
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        86.6,
        86.4,
        85.9,
        86.4
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        86.7,
        86.4,
        85.7,
        86.3
      ],
      "up_blocks.3.resnets.1.conv1": [
        86.7,
        86.6,
        85.7,
        86.3
      ],
      "up_blocks.2.attentions.1.proj_in": [
        86.5,
        86.4,
        85.7,
        86.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        86.5,
        86.5,
        85.4,
        86.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        86.5,
        86.2,
        82.3,
        86.3
      ],
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        86.3,
        86.2,
        82.3,
        86.0
      ],
      "down_blocks.0.resnets.1.conv2": [
        86.4,
        86.1,
        84.8,
        86.0
      ],
      "up_blocks.3.resnets.0.conv1": [
        86.6,
        86.2,
        85.0,
        85.9
      ],
      "up_blocks.2.attentions.2.proj_in": [
        86.2,
        85.9,
        85.0,
        85.8
      ],
      "up_blocks.3.attentions.0.proj_out": [
        86.2,
        86.1,
        84.9,
        85.8
      ],
      "up_blocks.3.resnets.1.conv2": [
        86.2,
        85.8,
        84.2,
        85.8
      ],
      "down_blocks.0.attentions.0.proj_out": [
        86.2,
        86.1,
        84.0,
        85.8
      ],
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2": [
        86.3,
        85.8,
        84.3,
        85.9
      ],
      "down_blocks.0.attentions.1.proj_out": [
        86.4,
        85.8,
        85.2,
        85.7
      ],
      "up_blocks.3.attentions.2.proj_in": [
        86.1,
        85.7,
        84.8,
        85.6
      ],
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        85.9,
        85.7,
        84.6,
        85.6
      ],
      "up_blocks.3.attentions.0.proj_in": [
        86.0,
        85.6,
        84.7,
        85.7
      ],
      "up_blocks.3.resnets.0.conv_shortcut": [
        86.0,
        85.7,
        84.7,
        85.3
      ],
      "down_blocks.0.attentions.0.proj_in": [
        86.0,
        85.7,
        85.0,
        85.4
      ],
      "down_blocks.0.resnets.0.conv2": [
        85.9,
        85.7,
        85.1,
        85.5
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        86.0,
        85.8,
        85.3,
        85.8
      ],
      "up_blocks.3.attentions.1.proj_in": [
        86.1,
        85.8,
        85.3,
        85.5
      ],
      "up_blocks.3.resnets.2.conv_shortcut": [
        85.9,
        85.6,
        85.1,
        85.6
      ],
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        85.9,
        85.7,
        85.4,
        85.3
      ],
      "up_blocks.3.resnets.1.conv_shortcut": [
        86.0,
        85.4,
        85.0,
        85.4
      ],
      "down_blocks.0.resnets.1.conv1": [
        85.6,
        85.4,
        84.8,
        84.3
      ],
      "down_blocks.0.attentions.1.proj_in": [
        85.8,
        85.4,
        80.7,
        85.2
      ],
      "down_blocks.0.downsamplers.0.conv": [
        85.8,
        85.3,
        82.1,
        82.5
      ]
    }
  },
  "baselines": {
    "original": 89.0,
    "linear_8bit": 79.3,
    "recipe_9.57_bit_mixedpalette": 87.3,
    "recipe_6.09_bit_mixedpalette": 84.8,
    "recipe_5.03_bit_mixedpalette": 81.5,
    "recipe_4.59_bit_mixedpalette": 81.4,
    "recipe_4.21_bit_mixedpalette": 78.6,
    "recipe_4.00_bit_mixedpalette": 78.85,
    "recipe_3.71_bit_mixedpalette": 77.0
  },
  "recipes": {
    "recipe_9.57_bit_mixedpalette": {
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.resnets.1.time_emb_proj": 1,
      "up_blocks.0.resnets.1.time_emb_proj": 1,
      "down_blocks.3.resnets.1.time_emb_proj": 2,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": 1,
      "up_blocks.0.resnets.0.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": 1,
      "up_blocks.0.resnets.2.time_emb_proj": 2,
      "down_blocks.3.resnets.0.time_emb_proj": 2,
      "mid_block.resnets.0.time_emb_proj": 2,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": 2,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 6,
      "mid_block.attentions.0.proj_in": 6,
      "up_blocks.0.resnets.2.conv_shortcut": 6,
      "mid_block.attentions.0.proj_out": 6,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": 2,
      "down_blocks.3.resnets.1.conv1": 6,
      "mid_block.resnets.0.conv1": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 8,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.0.resnets.0.conv_shortcut": 6,
      "down_blocks.3.resnets.1.conv2": 6,
      "up_blocks.0.resnets.0.conv1": 6,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.0.resnets.1.conv_shortcut": 6,
      "up_blocks.1.resnets.2.time_emb_proj": 6,
      "down_blocks.3.resnets.0.conv2": 6,
      "down_blocks.3.resnets.0.conv1": 6,
      "up_blocks.0.resnets.1.conv1": 8,
      "up_blocks.1.resnets.1.time_emb_proj": 6,
      "mid_block.resnets.0.conv2": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": 6,
      "mid_block.resnets.1.conv1": 6,
      "up_blocks.2.resnets.0.time_emb_proj": 6,
      "up_blocks.0.resnets.0.conv2": 4,
      "mid_block.resnets.1.conv2": 6,
      "up_blocks.0.resnets.2.conv1": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": 16,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 16,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 8,
      "up_blocks.0.resnets.1.conv2": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 8,
      "down_blocks.2.downsamplers.0.conv": 16,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": 8,
      "up_blocks.1.resnets.0.time_emb_proj": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.2.resnets.1.conv2": 6,
      "up_blocks.1.attentions.1.proj_out": 8,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 8,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": 16,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 16,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 8,
      "up_blocks.1.resnets.2.conv2": 16,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": 16,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.1.resnets.0.conv1": 16,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.2.resnets.1.conv1": 8,
      "down_blocks.2.resnets.0.time_emb_proj": 6,
      "down_blocks.2.resnets.1.time_emb_proj": 6,
      "up_blocks.1.attentions.0.proj_out": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": 16,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": 16,
      "up_blocks.1.resnets.2.conv1": 8,
      "up_blocks.1.resnets.1.conv1": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 16,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.1.resnets.1.conv2": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 8,
      "up_blocks.1.resnets.0.conv2": 16,
      "down_blocks.2.attentions.0.proj_in": 16,
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.2.resnets.0.conv1": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.0.resnets.2.conv2": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 16,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 16,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 16,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v": 16,
      "down_blocks.2.resnets.0.conv1": 16,
      "down_blocks.1.resnets.1.time_emb_proj": 16,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 16,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": 16,
      "down_blocks.1.attentions.1.proj_out": 16,
      "down_blocks.2.attentions.1.proj_in": 16,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 8,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.1.resnets.1.conv_shortcut": 8,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 16,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 16,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.1.upsamplers.0.conv": 8,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.2.resnets.1.conv1": 16,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 8,
      "up_blocks.1.resnets.0.conv_shortcut": 8,
      "up_blocks.1.resnets.2.conv_shortcut": 16,
      "down_blocks.2.attentions.1.proj_out": 8,
      "up_blocks.1.attentions.1.proj_in": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 16,
      "up_blocks.0.upsamplers.0.conv": 8,
      "up_blocks.2.resnets.1.time_emb_proj": 8,
      "down_blocks.2.attentions.0.proj_out": 8,
      "up_blocks.1.attentions.0.proj_in": 8,
      "up_blocks.2.resnets.1.conv2": 16,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": 8,
      "up_blocks.2.resnets.0.conv2": 16,
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 16,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k": 16,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 16,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 16,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q": 16,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 16,
      "up_blocks.1.attentions.2.proj_out": 16,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": 16,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 16,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0": 16,
      "up_blocks.1.attentions.2.proj_in": 16,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0": 16,
      "down_blocks.1.resnets.1.conv2": 16,
      "up_blocks.3.resnets.0.time_emb_proj": 16,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 16,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 16,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 16,
      "down_blocks.1.resnets.1.conv1": 16,
      "down_blocks.2.resnets.0.conv_shortcut": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 8,
      "down_blocks.2.resnets.0.conv2": 16,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": 16,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 8,
      "up_blocks.2.resnets.2.conv1": 16,
      "down_blocks.1.downsamplers.0.conv": 16,
      "down_blocks.1.resnets.0.conv2": 16,
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 16,
      "down_blocks.1.resnets.0.conv1": 16,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 16,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 16,
      "down_blocks.1.resnets.0.time_emb_proj": 16,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": 16,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": 16,
      "up_blocks.2.resnets.2.conv2": 16,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 16,
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2": 16,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 16,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": 16,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0": 16,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 16,
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 16,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": 16,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": 16,
      "down_blocks.0.resnets.1.time_emb_proj": 16,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q": 8,
      "down_blocks.1.attentions.0.proj_out": 16,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": 16,
      "up_blocks.2.attentions.1.proj_out": 16,
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": 16,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": 16,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": 16,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k": 16,
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2": 16,
      "up_blocks.2.attentions.1.proj_in": 16,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": 16,
      "down_blocks.1.attentions.0.proj_in": 16,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 16,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k": 16,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": 16,
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj": 16,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0": 16,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k": 16,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v": 16,
      "up_blocks.2.resnets.2.time_emb_proj": 16,
      "up_blocks.2.attentions.0.proj_in": 16,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 16,
      "down_blocks.0.attentions.0.proj_out": 16,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v": 16,
      "down_blocks.0.resnets.0.conv1": 16,
      "down_blocks.1.attentions.1.proj_in": 16,
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": 16,
      "down_blocks.0.resnets.1.conv2": 16,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 16,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q": 16,
      "down_blocks.0.attentions.1.proj_out": 16,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q": 16,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v": 16,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0": 16,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": 16,
      "up_blocks.2.attentions.0.proj_out": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q": 16,
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj": 16,
      "up_blocks.3.attentions.0.proj_out": 16,
      "up_blocks.2.resnets.1.conv_shortcut": 16,
      "down_blocks.0.attentions.0.proj_in": 16,
      "up_blocks.2.attentions.2.proj_out": 16,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k": 16,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q": 16,
      "down_blocks.0.attentions.1.proj_in": 16,
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": 16,
      "down_blocks.1.resnets.0.conv_shortcut": 16,
      "up_blocks.2.upsamplers.0.conv": 16,
      "down_blocks.0.resnets.1.conv1": 16,
      "up_blocks.2.attentions.2.proj_in": 16,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k": 16,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 8,
      "down_blocks.0.resnets.0.time_emb_proj": 16,
      "up_blocks.3.resnets.0.conv2": 16,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0": 16,
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2": 16,
      "up_blocks.2.resnets.2.conv_shortcut": 16,
      "up_blocks.3.resnets.0.conv1": 16,
      "down_blocks.0.downsamplers.0.conv": 16,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k": 16,
      "up_blocks.3.attentions.0.proj_in": 16,
      "up_blocks.3.resnets.1.time_emb_proj": 16,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k": 8,
      "up_blocks.3.attentions.1.proj_out": 16,
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2": 16,
      "down_blocks.0.resnets.0.conv2": 16,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v": 16,
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj": 16,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0": 16,
      "up_blocks.3.attentions.2.proj_out": 16,
      "up_blocks.2.resnets.0.conv_shortcut": 16,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v": 16,
      "up_blocks.3.attentions.1.proj_in": 16,
      "up_blocks.3.resnets.1.conv1": 16,
      "up_blocks.3.resnets.2.conv1": 16,
      "up_blocks.3.resnets.1.conv2": 16,
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj": 16,
      "up_blocks.3.resnets.1.conv_shortcut": 16,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0": 16,
      "up_blocks.3.resnets.2.time_emb_proj": 16,
      "up_blocks.3.attentions.2.proj_in": 16,
      "up_blocks.3.resnets.2.conv2": 16,
      "time_embedding.linear_1": 8,
      "time_embedding.linear_2": 16,
      "up_blocks.3.resnets.2.conv_shortcut": 16,
      "up_blocks.3.resnets.0.conv_shortcut": 16
    },
    "recipe_6.09_bit_mixedpalette": {
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.resnets.1.time_emb_proj": 1,
      "up_blocks.0.resnets.1.time_emb_proj": 1,
      "down_blocks.3.resnets.1.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": 1,
      "up_blocks.0.resnets.0.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": 1,
      "up_blocks.0.resnets.2.time_emb_proj": 1,
      "down_blocks.3.resnets.0.time_emb_proj": 1,
      "mid_block.resnets.0.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.attentions.0.proj_in": 1,
      "up_blocks.0.resnets.2.conv_shortcut": 4,
      "mid_block.attentions.0.proj_out": 2,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": 2,
      "down_blocks.3.resnets.1.conv1": 4,
      "mid_block.resnets.0.conv1": 2,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.0.resnets.0.conv_shortcut": 4,
      "down_blocks.3.resnets.1.conv2": 4,
      "up_blocks.0.resnets.0.conv1": 4,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.0.resnets.1.conv_shortcut": 4,
      "up_blocks.1.resnets.2.time_emb_proj": 4,
      "down_blocks.3.resnets.0.conv2": 4,
      "down_blocks.3.resnets.0.conv1": 4,
      "up_blocks.0.resnets.1.conv1": 4,
      "up_blocks.1.resnets.1.time_emb_proj": 6,
      "mid_block.resnets.0.conv2": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 2,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": 4,
      "mid_block.resnets.1.conv1": 4,
      "up_blocks.2.resnets.0.time_emb_proj": 4,
      "up_blocks.0.resnets.0.conv2": 4,
      "mid_block.resnets.1.conv2": 4,
      "up_blocks.0.resnets.2.conv1": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.0.resnets.1.conv2": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.2.downsamplers.0.conv": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": 8,
      "up_blocks.1.resnets.0.time_emb_proj": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 4,
      "down_blocks.2.resnets.1.conv2": 6,
      "up_blocks.1.attentions.1.proj_out": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.1.resnets.2.conv2": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.1.resnets.0.conv1": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.resnets.1.conv1": 6,
      "down_blocks.2.resnets.0.time_emb_proj": 6,
      "down_blocks.2.resnets.1.time_emb_proj": 4,
      "up_blocks.1.attentions.0.proj_out": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": 8,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.1.resnets.2.conv1": 6,
      "up_blocks.1.resnets.1.conv1": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.1.resnets.1.conv2": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.1.resnets.0.conv2": 6,
      "down_blocks.2.attentions.0.proj_in": 8,
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.2.resnets.0.conv1": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.0.resnets.2.conv2": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v": 8,
      "down_blocks.2.resnets.0.conv1": 8,
      "down_blocks.1.resnets.1.time_emb_proj": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.1.attentions.1.proj_out": 8,
      "down_blocks.2.attentions.1.proj_in": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.1.resnets.1.conv_shortcut": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.1.upsamplers.0.conv": 8,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.2.resnets.1.conv1": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.resnets.0.conv_shortcut": 6,
      "up_blocks.1.resnets.2.conv_shortcut": 6,
      "down_blocks.2.attentions.1.proj_out": 8,
      "up_blocks.1.attentions.1.proj_in": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 8,
      "up_blocks.0.upsamplers.0.conv": 8,
      "up_blocks.2.resnets.1.time_emb_proj": 6,
      "down_blocks.2.attentions.0.proj_out": 6,
      "up_blocks.1.attentions.0.proj_in": 8,
      "up_blocks.2.resnets.1.conv2": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": 8,
      "up_blocks.2.resnets.0.conv2": 8,
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q": 8,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.1.attentions.2.proj_out": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": 8,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0": 8,
      "up_blocks.1.attentions.2.proj_in": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0": 8,
      "down_blocks.1.resnets.1.conv2": 8,
      "up_blocks.3.resnets.0.time_emb_proj": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "down_blocks.1.resnets.1.conv1": 8,
      "down_blocks.2.resnets.0.conv_shortcut": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.resnets.0.conv2": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.2.resnets.2.conv1": 8,
      "down_blocks.1.downsamplers.0.conv": 8,
      "down_blocks.1.resnets.0.conv2": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.1.resnets.0.conv1": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.resnets.0.time_emb_proj": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": 8,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.2.resnets.2.conv2": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": 16,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0": 16,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": 16,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": 8,
      "down_blocks.0.resnets.1.time_emb_proj": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.1.attentions.0.proj_out": 16,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.2.attentions.1.proj_out": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": 16,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": 16,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2": 16,
      "up_blocks.2.attentions.1.proj_in": 16,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": 16,
      "down_blocks.1.attentions.0.proj_in": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": 16,
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v": 16,
      "up_blocks.2.resnets.2.time_emb_proj": 8,
      "up_blocks.2.attentions.0.proj_in": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 16,
      "down_blocks.0.attentions.0.proj_out": 16,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v": 8,
      "down_blocks.0.resnets.0.conv1": 16,
      "down_blocks.1.attentions.1.proj_in": 16,
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": 16,
      "down_blocks.0.resnets.1.conv2": 16,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 16,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q": 8,
      "down_blocks.0.attentions.1.proj_out": 16,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0": 16,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": 16,
      "up_blocks.2.attentions.0.proj_out": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj": 16,
      "up_blocks.3.attentions.0.proj_out": 16,
      "up_blocks.2.resnets.1.conv_shortcut": 8,
      "down_blocks.0.attentions.0.proj_in": 16,
      "up_blocks.2.attentions.2.proj_out": 16,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k": 16,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q": 8,
      "down_blocks.0.attentions.1.proj_in": 16,
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": 16,
      "down_blocks.1.resnets.0.conv_shortcut": 16,
      "up_blocks.2.upsamplers.0.conv": 16,
      "down_blocks.0.resnets.1.conv1": 16,
      "up_blocks.2.attentions.2.proj_in": 16,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 8,
      "down_blocks.0.resnets.0.time_emb_proj": 16,
      "up_blocks.3.resnets.0.conv2": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2": 16,
      "up_blocks.2.resnets.2.conv_shortcut": 8,
      "up_blocks.3.resnets.0.conv1": 16,
      "down_blocks.0.downsamplers.0.conv": 16,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k": 8,
      "up_blocks.3.attentions.0.proj_in": 16,
      "up_blocks.3.resnets.1.time_emb_proj": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k": 8,
      "up_blocks.3.attentions.1.proj_out": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.0.resnets.0.conv2": 16,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v": 16,
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj": 16,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0": 16,
      "up_blocks.3.attentions.2.proj_out": 8,
      "up_blocks.2.resnets.0.conv_shortcut": 16,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.3.attentions.1.proj_in": 16,
      "up_blocks.3.resnets.1.conv1": 16,
      "up_blocks.3.resnets.2.conv1": 16,
      "up_blocks.3.resnets.1.conv2": 16,
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj": 16,
      "up_blocks.3.resnets.1.conv_shortcut": 16,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0": 16,
      "up_blocks.3.resnets.2.time_emb_proj": 8,
      "up_blocks.3.attentions.2.proj_in": 16,
      "up_blocks.3.resnets.2.conv2": 8,
      "time_embedding.linear_1": 8,
      "time_embedding.linear_2": 16,
      "up_blocks.3.resnets.2.conv_shortcut": 16,
      "up_blocks.3.resnets.0.conv_shortcut": 16
    },
    "recipe_5.03_bit_mixedpalette": {
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.resnets.1.time_emb_proj": 1,
      "up_blocks.0.resnets.1.time_emb_proj": 1,
      "down_blocks.3.resnets.1.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": 1,
      "up_blocks.0.resnets.0.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": 1,
      "up_blocks.0.resnets.2.time_emb_proj": 1,
      "down_blocks.3.resnets.0.time_emb_proj": 1,
      "mid_block.resnets.0.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.attentions.0.proj_in": 1,
      "up_blocks.0.resnets.2.conv_shortcut": 1,
      "mid_block.attentions.0.proj_out": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": 1,
      "down_blocks.3.resnets.1.conv1": 2,
      "mid_block.resnets.0.conv1": 2,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.0.resnets.0.conv_shortcut": 2,
      "down_blocks.3.resnets.1.conv2": 4,
      "up_blocks.0.resnets.0.conv1": 2,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": 2,
      "up_blocks.0.resnets.1.conv_shortcut": 2,
      "up_blocks.1.resnets.2.time_emb_proj": 4,
      "down_blocks.3.resnets.0.conv2": 2,
      "down_blocks.3.resnets.0.conv1": 4,
      "up_blocks.0.resnets.1.conv1": 4,
      "up_blocks.1.resnets.1.time_emb_proj": 4,
      "mid_block.resnets.0.conv2": 2,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 2,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": 2,
      "mid_block.resnets.1.conv1": 2,
      "up_blocks.2.resnets.0.time_emb_proj": 4,
      "up_blocks.0.resnets.0.conv2": 2,
      "mid_block.resnets.1.conv2": 2,
      "up_blocks.0.resnets.2.conv1": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 2,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 2,
      "up_blocks.0.resnets.1.conv2": 2,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 4,
      "down_blocks.2.downsamplers.0.conv": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.resnets.0.time_emb_proj": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 4,
      "down_blocks.2.resnets.1.conv2": 4,
      "up_blocks.1.attentions.1.proj_out": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.1.resnets.2.conv2": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.1.resnets.0.conv1": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.resnets.1.conv1": 6,
      "down_blocks.2.resnets.0.time_emb_proj": 6,
      "down_blocks.2.resnets.1.time_emb_proj": 4,
      "up_blocks.1.attentions.0.proj_out": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.1.resnets.2.conv1": 4,
      "up_blocks.1.resnets.1.conv1": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.1.resnets.1.conv2": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.1.resnets.0.conv2": 6,
      "down_blocks.2.attentions.0.proj_in": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.2.resnets.0.conv1": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.0.resnets.2.conv2": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.2.resnets.0.conv1": 6,
      "down_blocks.1.resnets.1.time_emb_proj": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.1.attentions.1.proj_out": 8,
      "down_blocks.2.attentions.1.proj_in": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.1.resnets.1.conv_shortcut": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.1.upsamplers.0.conv": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.2.resnets.1.conv1": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.resnets.0.conv_shortcut": 6,
      "up_blocks.1.resnets.2.conv_shortcut": 6,
      "down_blocks.2.attentions.1.proj_out": 6,
      "up_blocks.1.attentions.1.proj_in": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.0.upsamplers.0.conv": 6,
      "up_blocks.2.resnets.1.time_emb_proj": 6,
      "down_blocks.2.attentions.0.proj_out": 6,
      "up_blocks.1.attentions.0.proj_in": 8,
      "up_blocks.2.resnets.1.conv2": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.2.resnets.0.conv2": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.1.attentions.2.proj_out": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.1.attentions.2.proj_in": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.1.resnets.1.conv2": 6,
      "up_blocks.3.resnets.0.time_emb_proj": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "down_blocks.1.resnets.1.conv1": 8,
      "down_blocks.2.resnets.0.conv_shortcut": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.resnets.0.conv2": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.2.resnets.2.conv1": 8,
      "down_blocks.1.downsamplers.0.conv": 8,
      "down_blocks.1.resnets.0.conv2": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.1.resnets.0.conv1": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.resnets.0.time_emb_proj": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": 8,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.2.resnets.2.conv2": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": 8,
      "down_blocks.0.resnets.1.time_emb_proj": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.1.attentions.0.proj_out": 8,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.2.attentions.1.proj_out": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.2.attentions.1.proj_in": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.1.attentions.0.proj_in": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.resnets.2.time_emb_proj": 8,
      "up_blocks.2.attentions.0.proj_in": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.0.attentions.0.proj_out": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v": 8,
      "down_blocks.0.resnets.0.conv1": 8,
      "down_blocks.1.attentions.1.proj_in": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.0.resnets.1.conv2": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 16,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.0.attentions.1.proj_out": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.attentions.0.proj_out": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.3.attentions.0.proj_out": 8,
      "up_blocks.2.resnets.1.conv_shortcut": 8,
      "down_blocks.0.attentions.0.proj_in": 8,
      "up_blocks.2.attentions.2.proj_out": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q": 8,
      "down_blocks.0.attentions.1.proj_in": 16,
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.resnets.0.conv_shortcut": 8,
      "up_blocks.2.upsamplers.0.conv": 8,
      "down_blocks.0.resnets.1.conv1": 16,
      "up_blocks.2.attentions.2.proj_in": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.resnets.0.time_emb_proj": 8,
      "up_blocks.3.resnets.0.conv2": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.2.resnets.2.conv_shortcut": 8,
      "up_blocks.3.resnets.0.conv1": 8,
      "down_blocks.0.downsamplers.0.conv": 16,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.3.attentions.0.proj_in": 8,
      "up_blocks.3.resnets.1.time_emb_proj": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k": 8,
      "up_blocks.3.attentions.1.proj_out": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.0.resnets.0.conv2": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0": 16,
      "up_blocks.3.attentions.2.proj_out": 8,
      "up_blocks.2.resnets.0.conv_shortcut": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.3.attentions.1.proj_in": 8,
      "up_blocks.3.resnets.1.conv1": 8,
      "up_blocks.3.resnets.2.conv1": 8,
      "up_blocks.3.resnets.1.conv2": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.3.resnets.1.conv_shortcut": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.resnets.2.time_emb_proj": 8,
      "up_blocks.3.attentions.2.proj_in": 8,
      "up_blocks.3.resnets.2.conv2": 8,
      "time_embedding.linear_1": 6,
      "time_embedding.linear_2": 8,
      "up_blocks.3.resnets.2.conv_shortcut": 8,
      "up_blocks.3.resnets.0.conv_shortcut": 8
    },
    "recipe_4.59_bit_mixedpalette": {
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.resnets.1.time_emb_proj": 1,
      "up_blocks.0.resnets.1.time_emb_proj": 1,
      "down_blocks.3.resnets.1.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": 1,
      "up_blocks.0.resnets.0.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": 1,
      "up_blocks.0.resnets.2.time_emb_proj": 1,
      "down_blocks.3.resnets.0.time_emb_proj": 1,
      "mid_block.resnets.0.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.attentions.0.proj_in": 1,
      "up_blocks.0.resnets.2.conv_shortcut": 1,
      "mid_block.attentions.0.proj_out": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": 1,
      "down_blocks.3.resnets.1.conv1": 2,
      "mid_block.resnets.0.conv1": 1,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": 1,
      "up_blocks.0.resnets.0.conv_shortcut": 2,
      "down_blocks.3.resnets.1.conv2": 2,
      "up_blocks.0.resnets.0.conv1": 2,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": 2,
      "up_blocks.0.resnets.1.conv_shortcut": 2,
      "up_blocks.1.resnets.2.time_emb_proj": 4,
      "down_blocks.3.resnets.0.conv2": 2,
      "down_blocks.3.resnets.0.conv1": 4,
      "up_blocks.0.resnets.1.conv1": 2,
      "up_blocks.1.resnets.1.time_emb_proj": 2,
      "mid_block.resnets.0.conv2": 2,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 2,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": 2,
      "mid_block.resnets.1.conv1": 2,
      "up_blocks.2.resnets.0.time_emb_proj": 4,
      "up_blocks.0.resnets.0.conv2": 2,
      "mid_block.resnets.1.conv2": 2,
      "up_blocks.0.resnets.2.conv1": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 2,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 2,
      "up_blocks.0.resnets.1.conv2": 2,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 4,
      "down_blocks.2.downsamplers.0.conv": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.resnets.0.time_emb_proj": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 2,
      "down_blocks.2.resnets.1.conv2": 4,
      "up_blocks.1.attentions.1.proj_out": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": 4,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.1.resnets.2.conv2": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.1.resnets.0.conv1": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.resnets.1.conv1": 6,
      "down_blocks.2.resnets.0.time_emb_proj": 6,
      "down_blocks.2.resnets.1.time_emb_proj": 4,
      "up_blocks.1.attentions.0.proj_out": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.1.resnets.2.conv1": 4,
      "up_blocks.1.resnets.1.conv1": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.1.resnets.1.conv2": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.1.resnets.0.conv2": 6,
      "down_blocks.2.attentions.0.proj_in": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.2.resnets.0.conv1": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.0.resnets.2.conv2": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.2.resnets.0.conv1": 6,
      "down_blocks.1.resnets.1.time_emb_proj": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.1.attentions.1.proj_out": 8,
      "down_blocks.2.attentions.1.proj_in": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.1.resnets.1.conv_shortcut": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.1.upsamplers.0.conv": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.2.resnets.1.conv1": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.resnets.0.conv_shortcut": 6,
      "up_blocks.1.resnets.2.conv_shortcut": 6,
      "down_blocks.2.attentions.1.proj_out": 6,
      "up_blocks.1.attentions.1.proj_in": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.0.upsamplers.0.conv": 6,
      "up_blocks.2.resnets.1.time_emb_proj": 6,
      "down_blocks.2.attentions.0.proj_out": 6,
      "up_blocks.1.attentions.0.proj_in": 6,
      "up_blocks.2.resnets.1.conv2": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.2.resnets.0.conv2": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.1.attentions.2.proj_out": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.1.attentions.2.proj_in": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.1.resnets.1.conv2": 6,
      "up_blocks.3.resnets.0.time_emb_proj": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "down_blocks.1.resnets.1.conv1": 8,
      "down_blocks.2.resnets.0.conv_shortcut": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.resnets.0.conv2": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.2.resnets.2.conv1": 8,
      "down_blocks.1.downsamplers.0.conv": 8,
      "down_blocks.1.resnets.0.conv2": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.1.resnets.0.conv1": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.resnets.0.time_emb_proj": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.2.resnets.2.conv2": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": 8,
      "down_blocks.0.resnets.1.time_emb_proj": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.1.attentions.0.proj_out": 8,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.2.attentions.1.proj_out": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.2.attentions.1.proj_in": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.1.attentions.0.proj_in": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.resnets.2.time_emb_proj": 6,
      "up_blocks.2.attentions.0.proj_in": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.0.attentions.0.proj_out": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v": 8,
      "down_blocks.0.resnets.0.conv1": 8,
      "down_blocks.1.attentions.1.proj_in": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.0.resnets.1.conv2": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.0.attentions.1.proj_out": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.attentions.0.proj_out": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.3.attentions.0.proj_out": 8,
      "up_blocks.2.resnets.1.conv_shortcut": 8,
      "down_blocks.0.attentions.0.proj_in": 8,
      "up_blocks.2.attentions.2.proj_out": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "down_blocks.0.attentions.1.proj_in": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.resnets.0.conv_shortcut": 8,
      "up_blocks.2.upsamplers.0.conv": 8,
      "down_blocks.0.resnets.1.conv1": 8,
      "up_blocks.2.attentions.2.proj_in": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.resnets.0.time_emb_proj": 8,
      "up_blocks.3.resnets.0.conv2": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.2.resnets.2.conv_shortcut": 8,
      "up_blocks.3.resnets.0.conv1": 8,
      "down_blocks.0.downsamplers.0.conv": 16,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.3.attentions.0.proj_in": 8,
      "up_blocks.3.resnets.1.time_emb_proj": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.1.proj_out": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.0.resnets.0.conv2": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.attentions.2.proj_out": 8,
      "up_blocks.2.resnets.0.conv_shortcut": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.3.attentions.1.proj_in": 8,
      "up_blocks.3.resnets.1.conv1": 8,
      "up_blocks.3.resnets.2.conv1": 8,
      "up_blocks.3.resnets.1.conv2": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.3.resnets.1.conv_shortcut": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.resnets.2.time_emb_proj": 6,
      "up_blocks.3.attentions.2.proj_in": 8,
      "up_blocks.3.resnets.2.conv2": 8,
      "time_embedding.linear_1": 6,
      "time_embedding.linear_2": 8,
      "up_blocks.3.resnets.2.conv_shortcut": 8,
      "up_blocks.3.resnets.0.conv_shortcut": 8
    },
    "recipe_4.21_bit_mixedpalette": {
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.resnets.1.time_emb_proj": 1,
      "up_blocks.0.resnets.1.time_emb_proj": 1,
      "down_blocks.3.resnets.1.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": 1,
      "up_blocks.0.resnets.0.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": 1,
      "up_blocks.0.resnets.2.time_emb_proj": 1,
      "down_blocks.3.resnets.0.time_emb_proj": 1,
      "mid_block.resnets.0.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.attentions.0.proj_in": 1,
      "up_blocks.0.resnets.2.conv_shortcut": 1,
      "mid_block.attentions.0.proj_out": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": 1,
      "down_blocks.3.resnets.1.conv1": 1,
      "mid_block.resnets.0.conv1": 1,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": 1,
      "up_blocks.0.resnets.0.conv_shortcut": 1,
      "down_blocks.3.resnets.1.conv2": 1,
      "up_blocks.0.resnets.0.conv1": 2,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": 2,
      "up_blocks.0.resnets.1.conv_shortcut": 2,
      "up_blocks.1.resnets.2.time_emb_proj": 4,
      "down_blocks.3.resnets.0.conv2": 2,
      "down_blocks.3.resnets.0.conv1": 4,
      "up_blocks.0.resnets.1.conv1": 1,
      "up_blocks.1.resnets.1.time_emb_proj": 1,
      "mid_block.resnets.0.conv2": 2,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": 1,
      "mid_block.resnets.1.conv1": 1,
      "up_blocks.2.resnets.0.time_emb_proj": 4,
      "up_blocks.0.resnets.0.conv2": 2,
      "mid_block.resnets.1.conv2": 2,
      "up_blocks.0.resnets.2.conv1": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 2,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 2,
      "up_blocks.0.resnets.1.conv2": 2,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 4,
      "down_blocks.2.downsamplers.0.conv": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.resnets.0.time_emb_proj": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 2,
      "down_blocks.2.resnets.1.conv2": 4,
      "up_blocks.1.attentions.1.proj_out": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": 4,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.1.resnets.2.conv2": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.1.resnets.0.conv1": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 2,
      "down_blocks.2.resnets.1.conv1": 2,
      "down_blocks.2.resnets.0.time_emb_proj": 6,
      "down_blocks.2.resnets.1.time_emb_proj": 4,
      "up_blocks.1.attentions.0.proj_out": 4,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.1.resnets.2.conv1": 4,
      "up_blocks.1.resnets.1.conv1": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.1.resnets.1.conv2": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.1.resnets.0.conv2": 6,
      "down_blocks.2.attentions.0.proj_in": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.2.resnets.0.conv1": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.0.resnets.2.conv2": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.2.resnets.0.conv1": 6,
      "down_blocks.1.resnets.1.time_emb_proj": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.1.attentions.1.proj_out": 8,
      "down_blocks.2.attentions.1.proj_in": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.1.resnets.1.conv_shortcut": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.1.upsamplers.0.conv": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.2.resnets.1.conv1": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.resnets.0.conv_shortcut": 6,
      "up_blocks.1.resnets.2.conv_shortcut": 6,
      "down_blocks.2.attentions.1.proj_out": 4,
      "up_blocks.1.attentions.1.proj_in": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.0.upsamplers.0.conv": 6,
      "up_blocks.2.resnets.1.time_emb_proj": 4,
      "down_blocks.2.attentions.0.proj_out": 6,
      "up_blocks.1.attentions.0.proj_in": 6,
      "up_blocks.2.resnets.1.conv2": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.2.resnets.0.conv2": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.1.attentions.2.proj_out": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.1.attentions.2.proj_in": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.1.resnets.1.conv2": 6,
      "up_blocks.3.resnets.0.time_emb_proj": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "down_blocks.1.resnets.1.conv1": 6,
      "down_blocks.2.resnets.0.conv_shortcut": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.resnets.0.conv2": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.2.resnets.2.conv1": 6,
      "down_blocks.1.downsamplers.0.conv": 8,
      "down_blocks.1.resnets.0.conv2": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.1.resnets.0.conv1": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.resnets.0.time_emb_proj": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.2.resnets.2.conv2": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": 8,
      "down_blocks.0.resnets.1.time_emb_proj": 6,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.1.attentions.0.proj_out": 8,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.2.attentions.1.proj_out": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.2.attentions.1.proj_in": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.1.attentions.0.proj_in": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.resnets.2.time_emb_proj": 6,
      "up_blocks.2.attentions.0.proj_in": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.0.attentions.0.proj_out": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v": 8,
      "down_blocks.0.resnets.0.conv1": 8,
      "down_blocks.1.attentions.1.proj_in": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.0.resnets.1.conv2": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.0.attentions.1.proj_out": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.attentions.0.proj_out": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.3.attentions.0.proj_out": 8,
      "up_blocks.2.resnets.1.conv_shortcut": 8,
      "down_blocks.0.attentions.0.proj_in": 8,
      "up_blocks.2.attentions.2.proj_out": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "down_blocks.0.attentions.1.proj_in": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.resnets.0.conv_shortcut": 8,
      "up_blocks.2.upsamplers.0.conv": 8,
      "down_blocks.0.resnets.1.conv1": 8,
      "up_blocks.2.attentions.2.proj_in": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.resnets.0.time_emb_proj": 8,
      "up_blocks.3.resnets.0.conv2": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.2.resnets.2.conv_shortcut": 8,
      "up_blocks.3.resnets.0.conv1": 8,
      "down_blocks.0.downsamplers.0.conv": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.3.attentions.0.proj_in": 8,
      "up_blocks.3.resnets.1.time_emb_proj": 6,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.1.proj_out": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.0.resnets.0.conv2": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.attentions.2.proj_out": 8,
      "up_blocks.2.resnets.0.conv_shortcut": 6,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.3.attentions.1.proj_in": 8,
      "up_blocks.3.resnets.1.conv1": 8,
      "up_blocks.3.resnets.2.conv1": 8,
      "up_blocks.3.resnets.1.conv2": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.3.resnets.1.conv_shortcut": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.resnets.2.time_emb_proj": 6,
      "up_blocks.3.attentions.2.proj_in": 8,
      "up_blocks.3.resnets.2.conv2": 8,
      "time_embedding.linear_1": 6,
      "time_embedding.linear_2": 8,
      "up_blocks.3.resnets.2.conv_shortcut": 8,
      "up_blocks.3.resnets.0.conv_shortcut": 8
    },
    "recipe_4.00_bit_mixedpalette": {
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.resnets.1.time_emb_proj": 1,
      "up_blocks.0.resnets.1.time_emb_proj": 1,
      "down_blocks.3.resnets.1.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": 1,
      "up_blocks.0.resnets.0.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": 1,
      "up_blocks.0.resnets.2.time_emb_proj": 1,
      "down_blocks.3.resnets.0.time_emb_proj": 1,
      "mid_block.resnets.0.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.attentions.0.proj_in": 1,
      "up_blocks.0.resnets.2.conv_shortcut": 1,
      "mid_block.attentions.0.proj_out": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": 1,
      "down_blocks.3.resnets.1.conv1": 1,
      "mid_block.resnets.0.conv1": 1,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": 1,
      "up_blocks.0.resnets.0.conv_shortcut": 1,
      "down_blocks.3.resnets.1.conv2": 1,
      "up_blocks.0.resnets.0.conv1": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": 1,
      "up_blocks.0.resnets.1.conv_shortcut": 1,
      "up_blocks.1.resnets.2.time_emb_proj": 1,
      "down_blocks.3.resnets.0.conv2": 1,
      "down_blocks.3.resnets.0.conv1": 4,
      "up_blocks.0.resnets.1.conv1": 1,
      "up_blocks.1.resnets.1.time_emb_proj": 1,
      "mid_block.resnets.0.conv2": 1,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": 1,
      "mid_block.resnets.1.conv1": 1,
      "up_blocks.2.resnets.0.time_emb_proj": 1,
      "up_blocks.0.resnets.0.conv2": 2,
      "mid_block.resnets.1.conv2": 2,
      "up_blocks.0.resnets.2.conv1": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 2,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 2,
      "up_blocks.0.resnets.1.conv2": 2,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 2,
      "down_blocks.2.downsamplers.0.conv": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.1.resnets.0.time_emb_proj": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 2,
      "down_blocks.2.resnets.1.conv2": 4,
      "up_blocks.1.attentions.1.proj_out": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 2,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 2,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 2,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": 4,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 2,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": 2,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 2,
      "up_blocks.1.resnets.2.conv2": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.1.resnets.0.conv1": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 2,
      "down_blocks.2.resnets.1.conv1": 2,
      "down_blocks.2.resnets.0.time_emb_proj": 4,
      "down_blocks.2.resnets.1.time_emb_proj": 4,
      "up_blocks.1.attentions.0.proj_out": 4,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.1.resnets.2.conv1": 4,
      "up_blocks.1.resnets.1.conv1": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.1.resnets.1.conv2": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.1.resnets.0.conv2": 6,
      "down_blocks.2.attentions.0.proj_in": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.2.resnets.0.conv1": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.0.resnets.2.conv2": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.2.resnets.0.conv1": 6,
      "down_blocks.1.resnets.1.time_emb_proj": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.1.attentions.1.proj_out": 6,
      "down_blocks.2.attentions.1.proj_in": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.1.resnets.1.conv_shortcut": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.1.upsamplers.0.conv": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.2.resnets.1.conv1": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.1.resnets.0.conv_shortcut": 6,
      "up_blocks.1.resnets.2.conv_shortcut": 6,
      "down_blocks.2.attentions.1.proj_out": 4,
      "up_blocks.1.attentions.1.proj_in": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.0.upsamplers.0.conv": 6,
      "up_blocks.2.resnets.1.time_emb_proj": 4,
      "down_blocks.2.attentions.0.proj_out": 4,
      "up_blocks.1.attentions.0.proj_in": 6,
      "up_blocks.2.resnets.1.conv2": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.2.resnets.0.conv2": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.1.attentions.2.proj_out": 4,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.1.attentions.2.proj_in": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.1.resnets.1.conv2": 6,
      "up_blocks.3.resnets.0.time_emb_proj": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 4,
      "down_blocks.1.resnets.1.conv1": 6,
      "down_blocks.2.resnets.0.conv_shortcut": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.resnets.0.conv2": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.2.resnets.2.conv1": 6,
      "down_blocks.1.downsamplers.0.conv": 6,
      "down_blocks.1.resnets.0.conv2": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.1.resnets.0.conv1": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 4,
      "down_blocks.1.resnets.0.time_emb_proj": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.2.resnets.2.conv2": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.0.resnets.1.time_emb_proj": 6,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.1.attentions.0.proj_out": 8,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.2.attentions.1.proj_out": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.2.attentions.1.proj_in": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.1.attentions.0.proj_in": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.resnets.2.time_emb_proj": 4,
      "up_blocks.2.attentions.0.proj_in": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.0.attentions.0.proj_out": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v": 8,
      "down_blocks.0.resnets.0.conv1": 8,
      "down_blocks.1.attentions.1.proj_in": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.0.resnets.1.conv2": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.0.attentions.1.proj_out": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.attentions.0.proj_out": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.3.attentions.0.proj_out": 8,
      "up_blocks.2.resnets.1.conv_shortcut": 8,
      "down_blocks.0.attentions.0.proj_in": 8,
      "up_blocks.2.attentions.2.proj_out": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "down_blocks.0.attentions.1.proj_in": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.resnets.0.conv_shortcut": 8,
      "up_blocks.2.upsamplers.0.conv": 8,
      "down_blocks.0.resnets.1.conv1": 8,
      "up_blocks.2.attentions.2.proj_in": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.resnets.0.time_emb_proj": 6,
      "up_blocks.3.resnets.0.conv2": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.2.resnets.2.conv_shortcut": 8,
      "up_blocks.3.resnets.0.conv1": 8,
      "down_blocks.0.downsamplers.0.conv": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.3.attentions.0.proj_in": 8,
      "up_blocks.3.resnets.1.time_emb_proj": 6,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.1.proj_out": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.0.resnets.0.conv2": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.attentions.2.proj_out": 8,
      "up_blocks.2.resnets.0.conv_shortcut": 6,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.3.attentions.1.proj_in": 8,
      "up_blocks.3.resnets.1.conv1": 8,
      "up_blocks.3.resnets.2.conv1": 8,
      "up_blocks.3.resnets.1.conv2": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.3.resnets.1.conv_shortcut": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.resnets.2.time_emb_proj": 6,
      "up_blocks.3.attentions.2.proj_in": 8,
      "up_blocks.3.resnets.2.conv2": 8,
      "time_embedding.linear_1": 6,
      "time_embedding.linear_2": 8,
      "up_blocks.3.resnets.2.conv_shortcut": 8,
      "up_blocks.3.resnets.0.conv_shortcut": 8
    },
    "recipe_3.71_bit_mixedpalette": {
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.resnets.1.time_emb_proj": 1,
      "up_blocks.0.resnets.1.time_emb_proj": 1,
      "down_blocks.3.resnets.1.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": 1,
      "up_blocks.0.resnets.0.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": 1,
      "up_blocks.0.resnets.2.time_emb_proj": 1,
      "down_blocks.3.resnets.0.time_emb_proj": 1,
      "mid_block.resnets.0.time_emb_proj": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.attentions.0.proj_in": 1,
      "up_blocks.0.resnets.2.conv_shortcut": 1,
      "mid_block.attentions.0.proj_out": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": 1,
      "down_blocks.3.resnets.1.conv1": 1,
      "mid_block.resnets.0.conv1": 1,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": 1,
      "up_blocks.0.resnets.0.conv_shortcut": 1,
      "down_blocks.3.resnets.1.conv2": 1,
      "up_blocks.0.resnets.0.conv1": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": 1,
      "up_blocks.0.resnets.1.conv_shortcut": 1,
      "up_blocks.1.resnets.2.time_emb_proj": 1,
      "down_blocks.3.resnets.0.conv2": 1,
      "down_blocks.3.resnets.0.conv1": 4,
      "up_blocks.0.resnets.1.conv1": 1,
      "up_blocks.1.resnets.1.time_emb_proj": 1,
      "mid_block.resnets.0.conv2": 1,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": 1,
      "mid_block.resnets.1.conv1": 1,
      "up_blocks.2.resnets.0.time_emb_proj": 1,
      "up_blocks.0.resnets.0.conv2": 1,
      "mid_block.resnets.1.conv2": 1,
      "up_blocks.0.resnets.2.conv1": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 2,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 2,
      "up_blocks.0.resnets.1.conv2": 2,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 2,
      "down_blocks.2.downsamplers.0.conv": 2,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.1.resnets.0.time_emb_proj": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 2,
      "down_blocks.2.resnets.1.conv2": 4,
      "up_blocks.1.attentions.1.proj_out": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 2,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 2,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 2,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": 4,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 2,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": 2,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 2,
      "up_blocks.1.resnets.2.conv2": 2,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.1.resnets.0.conv1": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 2,
      "down_blocks.2.resnets.1.conv1": 2,
      "down_blocks.2.resnets.0.time_emb_proj": 4,
      "down_blocks.2.resnets.1.time_emb_proj": 4,
      "up_blocks.1.attentions.0.proj_out": 4,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.1.resnets.2.conv1": 4,
      "up_blocks.1.resnets.1.conv1": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 2,
      "up_blocks.1.resnets.1.conv2": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.1.resnets.0.conv2": 6,
      "down_blocks.2.attentions.0.proj_in": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.2.resnets.0.conv1": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.0.resnets.2.conv2": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.2.resnets.0.conv1": 6,
      "down_blocks.1.resnets.1.time_emb_proj": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.1.attentions.1.proj_out": 6,
      "down_blocks.2.attentions.1.proj_in": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.1.resnets.1.conv_shortcut": 6,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.1.upsamplers.0.conv": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.2.resnets.1.conv1": 4,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.1.resnets.0.conv_shortcut": 6,
      "up_blocks.1.resnets.2.conv_shortcut": 6,
      "down_blocks.2.attentions.1.proj_out": 4,
      "up_blocks.1.attentions.1.proj_in": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.0.upsamplers.0.conv": 6,
      "up_blocks.2.resnets.1.time_emb_proj": 4,
      "down_blocks.2.attentions.0.proj_out": 4,
      "up_blocks.1.attentions.0.proj_in": 6,
      "up_blocks.2.resnets.1.conv2": 4,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.2.resnets.0.conv2": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.1.attentions.2.proj_out": 4,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.1.attentions.2.proj_in": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.1.resnets.1.conv2": 6,
      "up_blocks.3.resnets.0.time_emb_proj": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 4,
      "down_blocks.1.resnets.1.conv1": 6,
      "down_blocks.2.resnets.0.conv_shortcut": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.resnets.0.conv2": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.2.resnets.2.conv1": 6,
      "down_blocks.1.downsamplers.0.conv": 6,
      "down_blocks.1.resnets.0.conv2": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.1.resnets.0.conv1": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 4,
      "down_blocks.1.resnets.0.time_emb_proj": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.2.resnets.2.conv2": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.0.resnets.1.time_emb_proj": 6,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.1.attentions.0.proj_out": 8,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.2.attentions.1.proj_out": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.2.attentions.1.proj_in": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.1.attentions.0.proj_in": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.2.resnets.2.time_emb_proj": 4,
      "up_blocks.2.attentions.0.proj_in": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.0.attentions.0.proj_out": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v": 8,
      "down_blocks.0.resnets.0.conv1": 8,
      "down_blocks.1.attentions.1.proj_in": 6,
      "down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.0.resnets.1.conv2": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.0.attentions.1.proj_out": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.2.attentions.0.proj_out": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.3.attentions.0.proj_out": 8,
      "up_blocks.2.resnets.1.conv_shortcut": 6,
      "down_blocks.0.attentions.0.proj_in": 8,
      "up_blocks.2.attentions.2.proj_out": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "down_blocks.0.attentions.1.proj_in": 6,
      "down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.resnets.0.conv_shortcut": 8,
      "up_blocks.2.upsamplers.0.conv": 6,
      "down_blocks.0.resnets.1.conv1": 8,
      "up_blocks.2.attentions.2.proj_in": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.resnets.0.time_emb_proj": 6,
      "up_blocks.3.resnets.0.conv2": 8,
      "up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.2.resnets.2.conv_shortcut": 8,
      "up_blocks.3.resnets.0.conv1": 8,
      "down_blocks.0.downsamplers.0.conv": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.3.attentions.0.proj_in": 8,
      "up_blocks.3.resnets.1.time_emb_proj": 6,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.3.attentions.1.proj_out": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.0.resnets.0.conv2": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.attentions.2.proj_out": 8,
      "up_blocks.2.resnets.0.conv_shortcut": 6,
      "up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.3.attentions.1.proj_in": 8,
      "up_blocks.3.resnets.1.conv1": 8,
      "up_blocks.3.resnets.2.conv1": 8,
      "up_blocks.3.resnets.1.conv2": 8,
      "up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.3.resnets.1.conv_shortcut": 8,
      "up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.3.resnets.2.time_emb_proj": 6,
      "up_blocks.3.attentions.2.proj_in": 8,
      "up_blocks.3.resnets.2.conv2": 8,
      "time_embedding.linear_1": 4,
      "time_embedding.linear_2": 8,
      "up_blocks.3.resnets.2.conv_shortcut": 8,
      "up_blocks.3.resnets.0.conv_shortcut": 8
    }
  }
}