{
  "single_layer": {
    "1": {
      "time_embedding.linear_1": [
        48.0,
        47.9,
        48.2,
        48.1
      ],
      "time_embedding.linear_2": [
        48.5,
        48.4,
        48.7,
        48.6
      ],
      "add_embedding.linear_1": [
        69.2,
        69.8,
        71.0,
        70.8
      ],
      "add_embedding.linear_2": [
        73.4,
        68.1,
        71.1,
        72.5
      ],
      "down_blocks.0.resnets.0.conv1": [
        55.0,
        55.7,
        55.2,
        55.4
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        55.9,
        56.2,
        55.9,
        56.0
      ],
      "down_blocks.0.resnets.0.conv2": [
        52.3,
        52.3,
        52.5,
        52.4
      ],
      "down_blocks.0.resnets.1.conv1": [
        54.6,
        54.3,
        54.8,
        54.9
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        59.4,
        59.1,
        58.9,
        59.1
      ],
      "down_blocks.0.resnets.1.conv2": [
        55.5,
        55.3,
        55.4,
        55.6
      ],
      "down_blocks.0.downsamplers.0.conv": [
        51.9,
        51.6,
        52.1,
        51.9
      ],
      "down_blocks.1.attentions.0.proj_in": [
        58.6,
        58.2,
        58.8,
        59.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        67.7,
        68.1,
        68.4,
        68.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        68.3,
        67.9,
        68.4,
        68.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        64.4,
        64.9,
        65.3,
        65.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        62.7,
        62.5,
        63.0,
        63.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        78.7,
        77.7,
        77.8,
        78.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        64.4,
        64.8,
        66.3,
        66.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        72.4,
        75.3,
        71.2,
        71.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        75.8,
        73.6,
        75.3,
        75.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        59.8,
        59.3,
        60.3,
        60.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        62.6,
        61.4,
        62.7,
        63.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        66.3,
        66.4,
        66.4,
        66.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        66.5,
        66.4,
        66.8,
        66.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        62.5,
        61.7,
        62.3,
        62.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        62.3,
        62.0,
        62.4,
        62.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        64.8,
        66.7,
        64.6,
        64.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        60.0,
        60.7,
        60.1,
        60.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        66.4,
        64.9,
        66.6,
        67.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        65.9,
        64.7,
        65.6,
        66.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        59.4,
        58.7,
        59.3,
        59.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        62.7,
        61.9,
        62.7,
        63.0
      ],
      "down_blocks.1.attentions.0.proj_out": [
        61.4,
        61.1,
        61.6,
        61.7
      ],
      "down_blocks.1.attentions.1.proj_in": [
        65.1,
        64.6,
        65.5,
        65.3
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        70.4,
        69.8,
        70.6,
        70.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        70.6,
        70.1,
        70.8,
        70.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        69.3,
        68.6,
        69.4,
        69.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        68.2,
        67.5,
        68.2,
        67.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        77.2,
        76.3,
        76.5,
        76.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        70.6,
        70.8,
        71.1,
        70.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        78.8,
        78.4,
        78.5,
        78.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        78.0,
        77.7,
        78.4,
        78.3
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        64.6,
        64.1,
        64.6,
        64.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        68.1,
        67.9,
        68.3,
        68.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        74.6,
        74.1,
        74.6,
        74.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        74.6,
        74.1,
        74.7,
        74.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        72.3,
        71.8,
        72.2,
        72.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        71.7,
        71.2,
        71.7,
        71.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        76.8,
        76.6,
        77.2,
        76.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        72.9,
        72.3,
        72.7,
        72.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        77.6,
        76.8,
        77.5,
        77.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        76.1,
        74.9,
        76.2,
        75.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        66.3,
        65.9,
        66.3,
        66.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        69.3,
        69.0,
        69.2,
        69.2
      ],
      "down_blocks.1.attentions.1.proj_out": [
        66.9,
        66.4,
        67.1,
        66.9
      ],
      "down_blocks.1.resnets.0.conv1": [
        59.8,
        60.0,
        59.7,
        59.9
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        67.7,
        67.9,
        68.1,
        68.2
      ],
      "down_blocks.1.resnets.0.conv2": [
        65.5,
        65.2,
        65.6,
        65.7
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        54.2,
        53.8,
        54.2,
        54.2
      ],
      "down_blocks.1.resnets.1.conv1": [
        66.4,
        66.2,
        66.8,
        66.5
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        71.3,
        70.6,
        71.1,
        71.0
      ],
      "down_blocks.1.resnets.1.conv2": [
        68.0,
        67.7,
        68.2,
        67.9
      ],
      "down_blocks.1.downsamplers.0.conv": [
        65.0,
        62.9,
        63.4,
        63.1
      ],
      "down_blocks.2.attentions.0.proj_in": [
        76.7,
        76.2,
        76.9,
        76.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        80.9,
        81.1,
        80.9,
        81.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        80.9,
        81.3,
        81.3,
        81.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        79.9,
        79.6,
        79.9,
        79.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        80.1,
        80.0,
        80.1,
        80.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        81.9,
        82.1,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        81.4,
        81.7,
        81.4,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.7,
        81.7,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.8,
        81.7,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        79.6,
        79.4,
        79.7,
        79.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        80.0,
        80.0,
        80.3,
        80.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q": [
        81.5,
        81.8,
        82.0,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k": [
        81.7,
        81.8,
        82.0,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v": [
        81.4,
        81.3,
        81.6,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        81.2,
        81.3,
        81.3,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q": [
        81.4,
        82.1,
        81.9,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k": [
        77.2,
        76.8,
        77.2,
        77.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v": [
        81.7,
        81.9,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        81.7,
        82.0,
        82.3,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        80.0,
        80.1,
        80.1,
        80.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2": [
        80.5,
        80.4,
        80.7,
        80.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q": [
        81.0,
        81.1,
        81.4,
        81.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k": [
        81.3,
        81.1,
        81.4,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v": [
        80.7,
        80.9,
        80.9,
        80.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        81.2,
        80.9,
        81.2,
        80.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q": [
        81.7,
        82.0,
        82.0,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k": [
        81.4,
        81.8,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v": [
        81.8,
        82.0,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        81.8,
        82.0,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        79.9,
        79.9,
        79.8,
        79.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2": [
        80.3,
        80.3,
        80.4,
        80.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q": [
        80.1,
        80.4,
        80.4,
        80.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k": [
        80.1,
        80.2,
        80.6,
        80.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v": [
        79.7,
        79.8,
        79.8,
        79.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        79.5,
        79.8,
        80.0,
        79.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q": [
        81.8,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k": [
        81.6,
        81.8,
        81.8,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v": [
        82.0,
        82.1,
        81.8,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        82.0,
        81.9,
        82.1,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        80.4,
        80.6,
        80.8,
        80.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2": [
        81.0,
        81.0,
        81.2,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q": [
        80.9,
        81.2,
        81.2,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k": [
        80.9,
        81.2,
        81.2,
        80.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v": [
        80.4,
        80.5,
        80.3,
        80.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        80.7,
        80.7,
        80.4,
        80.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q": [
        81.9,
        82.0,
        82.1,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k": [
        81.8,
        82.0,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v": [
        81.8,
        82.1,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        82.0,
        82.1,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        80.2,
        80.0,
        80.5,
        80.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2": [
        80.6,
        80.5,
        81.1,
        80.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q": [
        80.4,
        80.8,
        80.9,
        80.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k": [
        80.6,
        80.8,
        80.9,
        80.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v": [
        80.5,
        80.5,
        80.6,
        80.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        80.6,
        80.2,
        80.3,
        80.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q": [
        82.1,
        82.0,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k": [
        81.7,
        81.9,
        82.1,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v": [
        81.8,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        78.9,
        78.4,
        78.6,
        78.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2": [
        80.0,
        79.9,
        79.9,
        79.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q": [
        81.1,
        81.3,
        81.6,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k": [
        81.3,
        81.3,
        81.7,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v": [
        80.9,
        81.0,
        81.4,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        80.5,
        80.8,
        81.0,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q": [
        81.9,
        82.3,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.2,
        82.1,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v": [
        82.0,
        82.1,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        81.9,
        82.2,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        79.5,
        79.5,
        79.7,
        79.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2": [
        80.2,
        80.4,
        80.2,
        80.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q": [
        81.5,
        81.7,
        81.7,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k": [
        81.6,
        81.5,
        81.8,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v": [
        80.8,
        80.7,
        81.0,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        80.9,
        80.9,
        81.4,
        80.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.1,
        82.1,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k": [
        81.8,
        82.1,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v": [
        81.9,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        80.1,
        80.2,
        80.6,
        80.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2": [
        80.7,
        80.8,
        80.9,
        80.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q": [
        81.2,
        81.5,
        81.7,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k": [
        81.2,
        81.4,
        81.8,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v": [
        80.9,
        81.1,
        81.2,
        81.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        81.2,
        81.1,
        81.5,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q": [
        81.9,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k": [
        81.9,
        82.1,
        82.0,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        81.9,
        82.1,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        80.5,
        80.6,
        80.7,
        80.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2": [
        80.8,
        80.9,
        81.2,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q": [
        81.7,
        82.1,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k": [
        81.9,
        81.9,
        82.0,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v": [
        80.8,
        81.0,
        81.2,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        81.0,
        81.2,
        81.1,
        81.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q": [
        81.8,
        82.2,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k": [
        80.9,
        81.4,
        81.3,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v": [
        81.2,
        81.5,
        81.6,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        81.4,
        81.5,
        81.6,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        80.1,
        80.1,
        80.4,
        80.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2": [
        80.6,
        80.5,
        80.9,
        80.5
      ],
      "down_blocks.2.attentions.0.proj_out": [
        76.2,
        75.9,
        76.3,
        76.2
      ],
      "down_blocks.2.attentions.1.proj_in": [
        80.1,
        79.6,
        80.4,
        80.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        81.5,
        81.5,
        81.9,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        81.7,
        81.5,
        81.9,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        80.7,
        80.3,
        80.8,
        80.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        80.6,
        80.5,
        80.3,
        80.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        81.8,
        82.0,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        81.6,
        81.5,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        81.8,
        81.9,
        81.9,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        81.6,
        81.9,
        81.9,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        80.0,
        79.3,
        79.8,
        80.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        80.4,
        80.0,
        80.5,
        80.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q": [
        81.6,
        81.5,
        81.8,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k": [
        81.6,
        81.6,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v": [
        81.4,
        81.1,
        81.4,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        81.2,
        81.3,
        81.5,
        81.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k": [
        80.4,
        79.7,
        79.3,
        80.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v": [
        81.7,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        82.0,
        82.1,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        80.7,
        80.3,
        80.8,
        81.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2": [
        81.0,
        80.8,
        81.3,
        81.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q": [
        81.7,
        81.7,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k": [
        81.5,
        81.6,
        82.1,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v": [
        81.3,
        81.0,
        81.4,
        81.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        81.4,
        81.2,
        81.7,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q": [
        82.1,
        82.1,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k": [
        82.0,
        82.0,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v": [
        81.9,
        82.0,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        81.9,
        81.9,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        81.1,
        80.9,
        81.2,
        81.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2": [
        81.4,
        81.2,
        81.5,
        81.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q": [
        82.1,
        82.1,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k": [
        82.0,
        82.2,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v": [
        81.6,
        81.6,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        81.4,
        81.2,
        81.7,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q": [
        82.1,
        82.2,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k": [
        82.0,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v": [
        82.1,
        82.1,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        81.2,
        80.9,
        81.2,
        81.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2": [
        81.4,
        81.2,
        81.4,
        81.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q": [
        82.0,
        82.3,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k": [
        82.0,
        82.3,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v": [
        81.5,
        81.6,
        81.8,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        81.5,
        81.6,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q": [
        82.0,
        82.1,
        82.2,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k": [
        81.9,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v": [
        82.0,
        82.1,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        81.9,
        82.0,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        81.2,
        81.1,
        81.3,
        81.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2": [
        81.4,
        81.5,
        81.6,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q": [
        82.1,
        82.1,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k": [
        82.0,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v": [
        81.7,
        81.4,
        81.8,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        81.6,
        82.0,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q": [
        81.9,
        82.2,
        82.4,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k": [
        81.9,
        82.1,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v": [
        81.9,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        81.9,
        82.2,
        82.4,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        81.2,
        81.0,
        81.4,
        81.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2": [
        81.4,
        81.4,
        81.6,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q": [
        82.2,
        82.0,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k": [
        81.9,
        82.1,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v": [
        81.5,
        81.8,
        81.9,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        81.7,
        81.7,
        81.8,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q": [
        82.0,
        82.1,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k": [
        82.0,
        82.1,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v": [
        81.9,
        82.4,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        82.3,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        81.4,
        81.1,
        81.4,
        81.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2": [
        81.3,
        80.9,
        81.5,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q": [
        82.1,
        82.2,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k": [
        82.0,
        82.2,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v": [
        81.7,
        81.7,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        81.7,
        81.7,
        81.7,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q": [
        82.1,
        82.3,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k": [
        82.0,
        82.3,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        81.2,
        81.2,
        81.3,
        81.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2": [
        81.4,
        81.4,
        81.6,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q": [
        82.0,
        82.0,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k": [
        82.1,
        82.0,
        82.3,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v": [
        81.4,
        81.5,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        81.7,
        81.8,
        81.9,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q": [
        82.2,
        82.1,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k": [
        82.0,
        81.9,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v": [
        82.1,
        82.1,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        81.0,
        80.8,
        81.4,
        81.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2": [
        81.4,
        81.3,
        81.4,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q": [
        82.0,
        81.6,
        82.1,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k": [
        82.0,
        81.8,
        82.0,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v": [
        81.3,
        81.5,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        81.4,
        81.7,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q": [
        82.1,
        82.2,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k": [
        82.1,
        82.2,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v": [
        82.0,
        82.2,
        82.4,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        80.6,
        80.1,
        80.8,
        80.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2": [
        81.1,
        80.9,
        81.4,
        81.0
      ],
      "down_blocks.2.attentions.1.proj_out": [
        79.9,
        79.5,
        78.9,
        79.5
      ],
      "down_blocks.2.resnets.0.conv1": [
        74.0,
        73.5,
        73.3,
        73.7
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        79.6,
        79.4,
        79.7,
        79.7
      ],
      "down_blocks.2.resnets.0.conv2": [
        78.4,
        77.4,
        76.8,
        78.3
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        67.6,
        64.9,
        65.7,
        65.0
      ],
      "down_blocks.2.resnets.1.conv1": [
        80.3,
        80.0,
        79.9,
        80.0
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        81.2,
        81.2,
        81.5,
        81.1
      ],
      "down_blocks.2.resnets.1.conv2": [
        80.1,
        79.7,
        80.1,
        80.1
      ],
      "up_blocks.0.attentions.0.proj_in": [
        78.0,
        74.2,
        76.8,
        77.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        79.9,
        79.6,
        78.9,
        80.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        80.1,
        79.4,
        78.8,
        80.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        80.0,
        79.6,
        78.8,
        80.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        80.2,
        79.7,
        79.2,
        80.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.1,
        82.0,
        82.4,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.0,
        82.0,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.9,
        82.1,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.9,
        82.2,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        74.2,
        71.0,
        72.1,
        71.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        79.6,
        77.3,
        75.5,
        77.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q": [
        81.4,
        81.2,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k": [
        81.5,
        81.3,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v": [
        80.6,
        80.9,
        81.0,
        81.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        81.0,
        80.6,
        80.9,
        81.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q": [
        81.8,
        82.0,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k": [
        81.9,
        82.0,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v": [
        81.9,
        82.1,
        82.0,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        80.8,
        80.5,
        80.8,
        81.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2": [
        81.1,
        81.1,
        81.3,
        81.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q": [
        81.4,
        81.1,
        81.6,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k": [
        81.4,
        81.5,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v": [
        81.2,
        80.9,
        81.2,
        81.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        80.9,
        80.9,
        81.1,
        81.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q": [
        81.6,
        82.1,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k": [
        81.6,
        82.0,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v": [
        81.7,
        82.0,
        81.7,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        81.9,
        82.2,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        81.0,
        81.1,
        81.2,
        80.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2": [
        81.2,
        81.3,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q": [
        80.7,
        80.6,
        80.4,
        80.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k": [
        80.8,
        80.6,
        80.3,
        80.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v": [
        80.8,
        80.6,
        80.4,
        80.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        80.7,
        80.5,
        80.0,
        80.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q": [
        81.9,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k": [
        81.9,
        82.2,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v": [
        82.0,
        82.0,
        82.2,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        82.0,
        82.0,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        81.2,
        81.2,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2": [
        81.3,
        81.2,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q": [
        81.3,
        81.7,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k": [
        81.5,
        81.4,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v": [
        81.2,
        81.3,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        81.4,
        81.4,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q": [
        81.9,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k": [
        81.9,
        82.1,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v": [
        82.0,
        82.0,
        82.1,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        82.0,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        81.3,
        81.4,
        81.2,
        81.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2": [
        81.4,
        81.3,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q": [
        81.8,
        81.8,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k": [
        81.7,
        81.8,
        82.0,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v": [
        80.9,
        81.2,
        81.6,
        81.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        81.4,
        81.3,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k": [
        81.8,
        82.1,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v": [
        82.1,
        82.2,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.0,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        81.1,
        81.2,
        81.2,
        81.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2": [
        81.3,
        81.5,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q": [
        81.8,
        82.0,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k": [
        82.0,
        82.1,
        82.1,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v": [
        81.2,
        81.3,
        81.3,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        81.5,
        81.3,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q": [
        81.8,
        82.3,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.0,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v": [
        81.9,
        82.1,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        82.0,
        82.1,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        81.4,
        81.4,
        81.4,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2": [
        81.4,
        81.6,
        81.6,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q": [
        81.8,
        81.6,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k": [
        81.7,
        81.8,
        82.0,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v": [
        81.4,
        81.4,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        81.4,
        81.5,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.1,
        82.3,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.0,
        82.0,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v": [
        81.9,
        82.3,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        82.0,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        81.2,
        81.3,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2": [
        81.4,
        81.4,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q": [
        81.8,
        82.0,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k": [
        81.9,
        82.0,
        81.9,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v": [
        81.5,
        81.6,
        81.7,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        81.6,
        81.5,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q": [
        82.0,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k": [
        82.1,
        82.3,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v": [
        81.9,
        82.1,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        82.0,
        82.3,
        82.5,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        80.9,
        80.8,
        81.2,
        80.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2": [
        81.3,
        81.3,
        81.5,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q": [
        81.8,
        81.6,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k": [
        81.8,
        82.0,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v": [
        81.2,
        80.9,
        81.3,
        81.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        81.2,
        81.2,
        81.3,
        81.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q": [
        82.0,
        82.2,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k": [
        81.9,
        82.2,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v": [
        81.9,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        82.1,
        82.1,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        80.2,
        80.5,
        80.8,
        80.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2": [
        80.8,
        80.9,
        80.9,
        80.8
      ],
      "up_blocks.0.attentions.0.proj_out": [
        78.3,
        77.8,
        78.2,
        78.6
      ],
      "up_blocks.0.attentions.1.proj_in": [
        71.4,
        69.8,
        70.7,
        70.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        80.2,
        80.3,
        80.3,
        80.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        80.2,
        80.4,
        80.3,
        80.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        76.1,
        75.2,
        75.8,
        75.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        75.2,
        75.6,
        75.4,
        75.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        81.5,
        81.8,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        81.4,
        81.5,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        81.1,
        80.9,
        80.8,
        81.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        80.3,
        81.1,
        81.2,
        80.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        72.9,
        71.4,
        72.1,
        71.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        76.5,
        76.1,
        76.5,
        76.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q": [
        81.1,
        81.3,
        81.3,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k": [
        81.2,
        81.5,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v": [
        79.0,
        78.7,
        78.6,
        78.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        79.0,
        78.9,
        79.2,
        79.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q": [
        81.4,
        81.2,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k": [
        74.6,
        75.2,
        73.3,
        73.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v": [
        79.9,
        80.4,
        80.4,
        80.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        80.4,
        81.2,
        81.7,
        81.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        73.5,
        73.0,
        74.2,
        73.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2": [
        76.0,
        75.4,
        76.5,
        76.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q": [
        80.9,
        81.0,
        81.0,
        80.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k": [
        80.7,
        80.6,
        80.7,
        80.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v": [
        79.6,
        79.6,
        79.8,
        79.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        79.7,
        79.9,
        80.3,
        79.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q": [
        81.5,
        81.7,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k": [
        70.6,
        69.7,
        70.4,
        70.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v": [
        79.4,
        80.1,
        79.2,
        78.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        81.5,
        81.1,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        76.4,
        76.4,
        76.7,
        76.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2": [
        78.5,
        78.2,
        78.5,
        78.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q": [
        80.6,
        80.8,
        81.0,
        80.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k": [
        80.6,
        80.7,
        80.9,
        80.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v": [
        79.6,
        79.6,
        79.8,
        79.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        79.2,
        78.9,
        79.1,
        78.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q": [
        81.0,
        81.2,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k": [
        80.5,
        81.3,
        81.2,
        81.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v": [
        79.4,
        79.7,
        79.0,
        78.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        81.0,
        81.1,
        81.7,
        81.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        78.6,
        78.1,
        78.6,
        78.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2": [
        80.0,
        79.5,
        80.0,
        79.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q": [
        80.7,
        80.4,
        80.5,
        80.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k": [
        80.5,
        80.4,
        80.4,
        80.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v": [
        79.3,
        78.8,
        79.4,
        79.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        78.6,
        79.0,
        79.1,
        78.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q": [
        81.5,
        81.5,
        81.9,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k": [
        81.1,
        81.3,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v": [
        80.3,
        80.3,
        79.8,
        79.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        81.6,
        81.1,
        81.8,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        76.6,
        76.2,
        76.6,
        75.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2": [
        78.4,
        78.3,
        78.5,
        78.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q": [
        80.5,
        80.4,
        80.7,
        80.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k": [
        80.6,
        80.5,
        80.7,
        80.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v": [
        79.0,
        78.6,
        78.6,
        78.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        79.4,
        79.4,
        79.5,
        79.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q": [
        81.6,
        81.9,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k": [
        81.4,
        81.5,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v": [
        80.2,
        80.7,
        79.8,
        79.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        81.5,
        81.5,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        77.5,
        77.1,
        77.4,
        77.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2": [
        78.0,
        77.9,
        78.2,
        78.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q": [
        80.0,
        80.1,
        80.6,
        80.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k": [
        80.3,
        80.2,
        80.5,
        80.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v": [
        79.0,
        79.1,
        79.3,
        79.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        78.9,
        79.1,
        79.4,
        78.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q": [
        81.7,
        81.8,
        81.7,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k": [
        81.4,
        81.3,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v": [
        81.0,
        81.1,
        80.3,
        80.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        81.3,
        81.4,
        82.0,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        78.6,
        78.9,
        78.9,
        78.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2": [
        79.5,
        79.2,
        79.8,
        79.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q": [
        80.2,
        80.3,
        80.5,
        80.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k": [
        80.1,
        80.2,
        80.5,
        80.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v": [
        79.2,
        78.6,
        79.5,
        78.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        79.4,
        79.0,
        79.7,
        79.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q": [
        81.7,
        81.9,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k": [
        81.0,
        81.2,
        80.9,
        80.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v": [
        80.9,
        81.1,
        81.2,
        81.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        81.4,
        81.7,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        76.4,
        77.2,
        76.9,
        76.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2": [
        78.4,
        78.7,
        78.7,
        78.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q": [
        80.3,
        80.2,
        80.6,
        80.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k": [
        80.7,
        80.4,
        80.8,
        80.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v": [
        79.3,
        79.4,
        79.7,
        79.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        79.0,
        79.0,
        79.4,
        78.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q": [
        81.7,
        81.6,
        81.9,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k": [
        81.5,
        81.4,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v": [
        81.3,
        81.4,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        81.5,
        81.5,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        77.1,
        77.6,
        77.3,
        77.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2": [
        78.5,
        78.8,
        78.8,
        78.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q": [
        80.4,
        80.9,
        81.0,
        80.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k": [
        80.8,
        80.7,
        80.8,
        81.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v": [
        78.0,
        78.5,
        78.8,
        78.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        78.3,
        78.4,
        78.7,
        78.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q": [
        81.3,
        81.1,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k": [
        81.3,
        81.1,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v": [
        81.5,
        81.8,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        81.6,
        81.9,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        72.1,
        71.4,
        72.0,
        71.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2": [
        75.6,
        74.9,
        75.7,
        75.1
      ],
      "up_blocks.0.attentions.1.proj_out": [
        68.9,
        68.4,
        69.0,
        68.6
      ],
      "up_blocks.0.attentions.2.proj_in": [
        74.1,
        73.5,
        74.2,
        74.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q": [
        80.6,
        80.4,
        80.7,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k": [
        80.6,
        80.5,
        80.8,
        80.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v": [
        77.7,
        77.0,
        77.3,
        77.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        77.1,
        76.1,
        77.1,
        77.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q": [
        81.6,
        82.0,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k": [
        79.6,
        78.5,
        78.3,
        78.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v": [
        81.1,
        81.5,
        81.4,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        81.6,
        81.7,
        81.1,
        81.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        76.1,
        75.5,
        75.9,
        76.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2": [
        77.3,
        76.8,
        77.0,
        77.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q": [
        81.2,
        81.0,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k": [
        81.3,
        81.2,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v": [
        78.8,
        78.6,
        79.1,
        78.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        78.9,
        78.7,
        79.3,
        78.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q": [
        81.8,
        81.8,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k": [
        81.4,
        81.5,
        81.9,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v": [
        80.5,
        79.6,
        80.1,
        80.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        80.9,
        80.7,
        81.5,
        80.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        77.4,
        77.0,
        77.4,
        77.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2": [
        78.2,
        78.3,
        78.8,
        78.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q": [
        81.0,
        81.1,
        81.2,
        80.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k": [
        80.9,
        81.0,
        81.3,
        81.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v": [
        79.4,
        79.3,
        79.3,
        79.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0": [
        79.4,
        79.1,
        79.4,
        79.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q": [
        81.6,
        81.9,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k": [
        81.1,
        81.6,
        82.0,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v": [
        80.7,
        80.0,
        79.9,
        79.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0": [
        81.1,
        81.4,
        81.8,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj": [
        78.7,
        77.9,
        78.3,
        78.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2": [
        80.0,
        79.9,
        79.6,
        79.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q": [
        81.2,
        81.1,
        81.2,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k": [
        81.1,
        81.2,
        81.2,
        81.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v": [
        79.6,
        79.1,
        79.7,
        79.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0": [
        79.4,
        79.1,
        79.3,
        79.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q": [
        81.7,
        81.8,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k": [
        81.6,
        81.4,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v": [
        81.2,
        81.3,
        81.0,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0": [
        81.6,
        81.7,
        82.0,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj": [
        78.8,
        79.1,
        78.7,
        78.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2": [
        79.7,
        79.6,
        79.3,
        79.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q": [
        80.5,
        80.3,
        80.7,
        80.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k": [
        80.7,
        80.5,
        80.9,
        80.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v": [
        79.3,
        78.9,
        78.8,
        78.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0": [
        78.5,
        77.3,
        77.5,
        77.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q": [
        81.5,
        81.8,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k": [
        81.5,
        81.8,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v": [
        81.4,
        81.6,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0": [
        81.5,
        81.9,
        82.1,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj": [
        78.9,
        78.4,
        78.6,
        78.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2": [
        79.8,
        79.7,
        80.1,
        80.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q": [
        81.2,
        81.1,
        81.2,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k": [
        81.3,
        81.4,
        81.1,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v": [
        79.3,
        79.4,
        79.6,
        79.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0": [
        80.0,
        80.1,
        80.0,
        79.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q": [
        81.5,
        82.0,
        82.1,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k": [
        81.3,
        81.6,
        81.9,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v": [
        81.5,
        81.6,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0": [
        81.9,
        82.0,
        82.2,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj": [
        78.8,
        78.2,
        78.7,
        78.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2": [
        79.7,
        79.3,
        79.8,
        79.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q": [
        80.9,
        81.0,
        81.2,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k": [
        81.0,
        81.0,
        81.0,
        80.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v": [
        79.8,
        79.5,
        79.6,
        79.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0": [
        79.9,
        79.5,
        79.7,
        79.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q": [
        81.7,
        82.0,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k": [
        81.6,
        81.9,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v": [
        81.8,
        81.9,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0": [
        81.6,
        81.9,
        81.9,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj": [
        79.5,
        79.2,
        79.4,
        79.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2": [
        79.9,
        79.5,
        80.0,
        79.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q": [
        80.8,
        80.8,
        81.1,
        81.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k": [
        80.9,
        80.8,
        81.2,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v": [
        80.0,
        79.7,
        80.0,
        79.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0": [
        80.1,
        79.7,
        80.1,
        79.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q": [
        81.7,
        81.9,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k": [
        81.7,
        82.0,
        82.0,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v": [
        82.0,
        81.8,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0": [
        81.7,
        82.1,
        82.1,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj": [
        79.4,
        78.8,
        79.1,
        79.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2": [
        80.0,
        79.8,
        80.0,
        79.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q": [
        80.8,
        80.4,
        80.6,
        80.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k": [
        80.9,
        80.6,
        80.7,
        80.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v": [
        79.6,
        79.4,
        79.7,
        79.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0": [
        79.6,
        79.1,
        79.7,
        79.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q": [
        82.0,
        82.0,
        81.8,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k": [
        82.0,
        81.7,
        81.8,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v": [
        81.9,
        81.9,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0": [
        82.0,
        82.1,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj": [
        77.2,
        75.8,
        76.2,
        76.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2": [
        79.0,
        78.2,
        78.4,
        78.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q": [
        80.9,
        81.0,
        81.0,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k": [
        81.0,
        81.0,
        81.3,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v": [
        78.9,
        78.5,
        78.7,
        78.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0": [
        79.5,
        78.9,
        79.2,
        79.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q": [
        81.6,
        81.8,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k": [
        82.1,
        81.7,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v": [
        81.9,
        81.8,
        82.0,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0": [
        81.7,
        82.2,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj": [
        77.6,
        76.8,
        77.2,
        77.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2": [
        79.0,
        78.6,
        79.0,
        78.8
      ],
      "up_blocks.0.attentions.2.proj_out": [
        72.2,
        71.1,
        71.1,
        71.4
      ],
      "up_blocks.0.resnets.0.conv1": [
        80.4,
        80.1,
        79.5,
        80.2
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        80.9,
        81.0,
        81.1,
        81.2
      ],
      "up_blocks.0.resnets.0.conv2": [
        78.8,
        79.0,
        78.0,
        78.8
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        77.9,
        77.4,
        77.5,
        77.7
      ],
      "up_blocks.0.resnets.1.conv1": [
        79.0,
        78.5,
        78.7,
        78.7
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        79.8,
        79.6,
        79.9,
        79.9
      ],
      "up_blocks.0.resnets.1.conv2": [
        79.6,
        79.6,
        79.6,
        79.4
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        70.6,
        70.2,
        71.1,
        70.7
      ],
      "up_blocks.0.resnets.2.conv1": [
        78.2,
        77.4,
        78.0,
        78.0
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        79.4,
        79.3,
        79.6,
        79.5
      ],
      "up_blocks.0.resnets.2.conv2": [
        78.8,
        78.5,
        78.4,
        78.4
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        64.4,
        64.1,
        64.6,
        64.3
      ],
      "up_blocks.0.upsamplers.0.conv": [
        62.2,
        61.7,
        62.1,
        62.0
      ],
      "up_blocks.1.attentions.0.proj_in": [
        64.1,
        63.8,
        63.9,
        64.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        71.2,
        70.5,
        71.1,
        71.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        71.0,
        70.4,
        70.7,
        70.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        66.2,
        66.0,
        66.5,
        66.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        63.5,
        63.0,
        63.6,
        63.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        75.1,
        74.7,
        75.1,
        74.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        74.9,
        74.2,
        75.3,
        74.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        80.3,
        80.5,
        79.8,
        79.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        80.1,
        80.2,
        80.6,
        80.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        65.6,
        65.3,
        65.7,
        65.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        67.0,
        66.7,
        67.3,
        67.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        69.4,
        69.1,
        69.5,
        69.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        69.3,
        69.2,
        69.5,
        69.3
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        66.0,
        65.7,
        66.2,
        66.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        66.4,
        66.0,
        66.5,
        66.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        74.7,
        74.7,
        74.4,
        74.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        76.7,
        76.8,
        76.5,
        76.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        80.1,
        80.1,
        79.8,
        79.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        78.7,
        78.0,
        79.1,
        78.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        65.5,
        65.1,
        65.4,
        65.3
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        69.0,
        68.7,
        69.1,
        69.0
      ],
      "up_blocks.1.attentions.0.proj_out": [
        66.9,
        66.6,
        67.0,
        67.0
      ],
      "up_blocks.1.attentions.1.proj_in": [
        60.1,
        59.6,
        60.2,
        60.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        68.0,
        67.4,
        68.1,
        67.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        68.2,
        67.8,
        68.5,
        68.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        61.4,
        60.6,
        61.9,
        61.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        58.0,
        57.1,
        58.1,
        57.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        66.0,
        65.7,
        65.2,
        65.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        55.7,
        55.7,
        54.5,
        54.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        78.5,
        77.6,
        78.8,
        78.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        75.4,
        74.9,
        75.2,
        74.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        62.1,
        61.7,
        62.2,
        62.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        64.0,
        63.5,
        64.0,
        63.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        66.1,
        65.7,
        66.2,
        66.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        66.9,
        66.3,
        66.8,
        66.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        64.4,
        63.8,
        64.4,
        64.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        63.2,
        62.5,
        63.1,
        63.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        74.3,
        74.6,
        73.7,
        73.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        62.4,
        62.2,
        62.2,
        62.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        76.3,
        76.6,
        76.5,
        77.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        73.6,
        73.5,
        73.3,
        73.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        61.6,
        60.7,
        61.4,
        61.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        64.5,
        63.6,
        64.3,
        64.1
      ],
      "up_blocks.1.attentions.1.proj_out": [
        62.9,
        62.3,
        63.0,
        62.9
      ],
      "up_blocks.1.attentions.2.proj_in": [
        59.6,
        59.5,
        59.9,
        59.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        66.1,
        66.0,
        66.2,
        66.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        67.1,
        66.8,
        66.9,
        67.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        61.7,
        60.8,
        61.7,
        61.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        58.7,
        58.3,
        59.4,
        59.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        52.1,
        52.7,
        52.7,
        52.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        62.6,
        63.0,
        61.5,
        62.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        70.3,
        70.5,
        70.2,
        70.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        70.8,
        70.1,
        71.3,
        71.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        59.3,
        59.4,
        59.3,
        59.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        62.3,
        62.5,
        62.6,
        62.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q": [
        65.4,
        65.3,
        65.5,
        65.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k": [
        66.5,
        66.2,
        66.4,
        66.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v": [
        62.1,
        61.9,
        62.1,
        62.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        62.4,
        62.1,
        62.4,
        62.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q": [
        71.6,
        71.5,
        71.3,
        71.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k": [
        55.8,
        55.3,
        55.8,
        55.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v": [
        77.4,
        77.9,
        77.3,
        77.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        74.2,
        74.6,
        73.8,
        73.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        58.0,
        57.9,
        58.0,
        58.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2": [
        63.3,
        63.2,
        63.3,
        63.2
      ],
      "up_blocks.1.attentions.2.proj_out": [
        60.1,
        60.1,
        60.2,
        60.2
      ],
      "up_blocks.1.resnets.0.conv1": [
        70.1,
        69.2,
        69.5,
        69.8
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        71.7,
        71.5,
        71.7,
        72.0
      ],
      "up_blocks.1.resnets.0.conv2": [
        69.3,
        68.9,
        69.4,
        69.1
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        55.8,
        55.5,
        55.9,
        55.8
      ],
      "up_blocks.1.resnets.1.conv1": [
        60.4,
        60.1,
        60.3,
        60.3
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        65.3,
        64.7,
        65.3,
        65.0
      ],
      "up_blocks.1.resnets.1.conv2": [
        61.5,
        61.0,
        61.3,
        61.4
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        55.2,
        55.2,
        55.2,
        55.4
      ],
      "up_blocks.1.resnets.2.conv1": [
        59.5,
        59.3,
        59.6,
        59.7
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        63.5,
        63.5,
        63.5,
        63.7
      ],
      "up_blocks.1.resnets.2.conv2": [
        61.4,
        61.0,
        61.2,
        61.1
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        51.6,
        51.5,
        51.5,
        51.5
      ],
      "up_blocks.1.upsamplers.0.conv": [
        48.4,
        48.5,
        48.5,
        48.4
      ],
      "up_blocks.2.resnets.0.conv1": [
        51.9,
        51.7,
        52.0,
        51.9
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        59.0,
        58.6,
        59.3,
        59.2
      ],
      "up_blocks.2.resnets.0.conv2": [
        51.5,
        51.4,
        51.7,
        51.6
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        47.8,
        47.8,
        47.9,
        47.8
      ],
      "up_blocks.2.resnets.1.conv1": [
        48.5,
        48.4,
        48.5,
        48.5
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        57.4,
        57.5,
        57.6,
        57.4
      ],
      "up_blocks.2.resnets.1.conv2": [
        48.8,
        48.9,
        48.9,
        48.9
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        45.1,
        45.1,
        45.1,
        45.0
      ],
      "up_blocks.2.resnets.2.conv1": [
        46.9,
        46.8,
        47.0,
        46.9
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        56.6,
        56.7,
        56.9,
        56.8
      ],
      "up_blocks.2.resnets.2.conv2": [
        44.6,
        44.8,
        44.7,
        44.7
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        43.1,
        43.2,
        43.2,
        43.2
      ],
      "mid_block.attentions.0.proj_in": [
        76.4,
        76.4,
        75.3,
        75.7
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        78.7,
        78.3,
        78.3,
        78.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        78.7,
        78.2,
        78.3,
        78.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        79.1,
        78.8,
        78.8,
        79.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        79.1,
        78.7,
        78.8,
        79.4
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        81.7,
        82.1,
        82.1,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        81.9,
        82.1,
        82.1,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.8,
        82.2,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.8,
        82.3,
        82.2,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        79.5,
        78.3,
        79.0,
        79.6
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        80.5,
        79.9,
        80.4,
        80.4
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_q": [
        79.5,
        78.6,
        79.0,
        79.4
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_k": [
        79.5,
        78.7,
        78.7,
        79.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_v": [
        79.0,
        78.3,
        78.7,
        78.7
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        79.2,
        78.2,
        78.5,
        78.9
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.1,
        82.1,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_k": [
        81.7,
        82.0,
        82.2,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_v": [
        81.8,
        82.3,
        82.0,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        82.1,
        82.1,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        79.6,
        79.8,
        79.9,
        80.0
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.2": [
        80.4,
        80.3,
        80.2,
        80.7
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_q": [
        80.8,
        80.6,
        81.0,
        80.8
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_k": [
        80.9,
        80.8,
        81.1,
        80.7
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_v": [
        80.7,
        80.1,
        80.9,
        80.6
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        80.6,
        80.5,
        80.8,
        80.7
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_q": [
        81.9,
        82.2,
        82.2,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_k": [
        81.8,
        82.2,
        82.2,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_v": [
        81.8,
        82.3,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        82.2,
        82.2,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        80.4,
        80.1,
        80.7,
        80.3
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.2": [
        81.1,
        80.9,
        81.2,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_q": [
        80.8,
        80.8,
        80.6,
        81.1
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_k": [
        80.7,
        80.7,
        80.4,
        81.1
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_v": [
        80.8,
        80.4,
        79.6,
        80.8
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        81.0,
        80.8,
        80.7,
        81.0
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_q": [
        81.9,
        82.3,
        82.4,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_k": [
        82.1,
        82.1,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_v": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        81.9,
        82.2,
        82.4,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        80.3,
        80.0,
        80.3,
        80.1
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.2": [
        80.8,
        80.8,
        80.8,
        80.7
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_q": [
        80.3,
        79.7,
        79.6,
        80.3
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_k": [
        80.6,
        79.7,
        79.8,
        80.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_v": [
        80.3,
        79.5,
        79.7,
        80.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        80.1,
        79.5,
        79.6,
        80.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_q": [
        82.1,
        82.1,
        82.3,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_k": [
        81.9,
        82.1,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_v": [
        81.9,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        80.8,
        80.7,
        80.6,
        80.9
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.2": [
        81.0,
        81.0,
        81.2,
        81.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_q": [
        81.0,
        80.9,
        81.1,
        81.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_k": [
        81.1,
        81.1,
        81.1,
        81.0
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_v": [
        80.9,
        80.8,
        80.9,
        80.9
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        80.8,
        80.8,
        80.7,
        80.8
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_q": [
        81.9,
        82.2,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_k": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_v": [
        81.9,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        80.8,
        80.4,
        80.7,
        80.9
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.2": [
        81.3,
        81.0,
        80.9,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_q": [
        81.6,
        81.6,
        81.8,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_k": [
        81.5,
        81.6,
        81.7,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_v": [
        81.3,
        81.5,
        81.5,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        81.4,
        81.4,
        81.5,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_q": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.0,
        82.2,
        82.2,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        80.9,
        80.9,
        81.1,
        80.9
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.2": [
        81.1,
        81.1,
        81.3,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_q": [
        81.4,
        81.5,
        81.7,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_k": [
        81.5,
        81.6,
        81.5,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_v": [
        81.2,
        81.2,
        81.6,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        80.9,
        81.1,
        81.3,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.2,
        82.2,
        82.2,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.2,
        82.0,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_v": [
        82.1,
        82.2,
        82.5,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        80.9,
        80.6,
        81.0,
        80.8
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.2": [
        81.2,
        81.2,
        81.3,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_q": [
        81.6,
        81.5,
        81.8,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_k": [
        81.7,
        81.5,
        81.6,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_v": [
        81.4,
        81.3,
        81.6,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        81.4,
        81.6,
        81.6,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_q": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_k": [
        81.9,
        82.3,
        82.2,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_v": [
        82.2,
        82.3,
        82.6,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        82.0,
        82.4,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        80.9,
        80.7,
        81.0,
        80.8
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.2": [
        81.1,
        81.1,
        81.4,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_q": [
        81.4,
        81.5,
        81.6,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_k": [
        81.5,
        81.7,
        81.6,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_v": [
        81.1,
        81.2,
        81.2,
        81.1
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        81.1,
        81.2,
        81.4,
        81.0
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_q": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_k": [
        82.1,
        82.2,
        82.4,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_v": [
        82.0,
        82.4,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        81.0,
        80.8,
        81.2,
        80.9
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.2": [
        80.8,
        81.0,
        81.3,
        80.9
      ],
      "mid_block.attentions.0.proj_out": [
        79.5,
        79.2,
        78.6,
        79.4
      ],
      "mid_block.resnets.0.conv1": [
        79.9,
        79.6,
        79.6,
        80.0
      ],
      "mid_block.resnets.0.time_emb_proj": [
        80.7,
        80.9,
        80.6,
        80.8
      ],
      "mid_block.resnets.0.conv2": [
        79.9,
        79.4,
        79.5,
        79.4
      ],
      "mid_block.resnets.1.conv1": [
        80.6,
        80.3,
        80.1,
        80.6
      ],
      "mid_block.resnets.1.time_emb_proj": [
        80.9,
        81.4,
        81.3,
        81.1
      ],
      "mid_block.resnets.1.conv2": [
        80.4,
        79.9,
        79.6,
        80.4
      ]
    },
    "2": {
      "time_embedding.linear_1": [
        55.5,
        55.5,
        55.4,
        55.4
      ],
      "time_embedding.linear_2": [
        50.9,
        50.7,
        50.9,
        50.8
      ],
      "add_embedding.linear_1": [
        70.1,
        72.3,
        70.6,
        71.3
      ],
      "add_embedding.linear_2": [
        75.6,
        74.3,
        74.1,
        74.7
      ],
      "down_blocks.0.resnets.0.conv1": [
        59.7,
        60.9,
        59.9,
        60.0
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        60.3,
        60.4,
        60.1,
        60.0
      ],
      "down_blocks.0.resnets.0.conv2": [
        56.2,
        56.3,
        56.4,
        56.4
      ],
      "down_blocks.0.resnets.1.conv1": [
        57.8,
        57.5,
        58.0,
        58.0
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        63.2,
        62.9,
        62.8,
        62.8
      ],
      "down_blocks.0.resnets.1.conv2": [
        58.3,
        57.9,
        58.1,
        58.5
      ],
      "down_blocks.0.downsamplers.0.conv": [
        59.0,
        58.7,
        59.4,
        59.3
      ],
      "down_blocks.1.attentions.0.proj_in": [
        63.7,
        63.7,
        64.1,
        64.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        75.0,
        75.2,
        75.3,
        75.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        75.7,
        75.6,
        76.3,
        75.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        70.8,
        71.4,
        71.9,
        71.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        68.4,
        67.9,
        68.3,
        68.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        80.9,
        80.3,
        81.1,
        81.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        79.9,
        79.1,
        80.4,
        80.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        75.8,
        77.7,
        74.9,
        74.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        78.5,
        77.0,
        77.5,
        77.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        63.7,
        63.9,
        64.1,
        64.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        68.0,
        67.4,
        67.6,
        68.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        74.4,
        74.1,
        74.5,
        74.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        74.9,
        75.0,
        74.8,
        75.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        69.7,
        69.6,
        70.1,
        69.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        70.2,
        69.8,
        70.2,
        69.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        72.6,
        74.6,
        74.1,
        74.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        62.2,
        63.0,
        62.2,
        62.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        70.0,
        68.8,
        69.7,
        70.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        69.5,
        68.6,
        69.0,
        69.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        63.7,
        62.9,
        63.8,
        64.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        67.8,
        66.9,
        67.8,
        68.0
      ],
      "down_blocks.1.attentions.0.proj_out": [
        68.6,
        68.0,
        68.6,
        68.6
      ],
      "down_blocks.1.attentions.1.proj_in": [
        70.9,
        70.5,
        71.1,
        71.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        77.3,
        77.1,
        77.6,
        77.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        77.4,
        77.2,
        77.6,
        77.3
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        74.8,
        74.4,
        75.2,
        74.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        74.9,
        74.6,
        75.0,
        74.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        80.7,
        80.3,
        80.7,
        80.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        80.2,
        79.6,
        80.8,
        80.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        80.7,
        80.6,
        80.5,
        80.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        79.4,
        79.0,
        79.7,
        79.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        71.1,
        70.6,
        71.1,
        71.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        72.9,
        72.6,
        72.7,
        72.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        78.9,
        78.7,
        78.9,
        78.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        79.2,
        79.0,
        79.2,
        79.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        77.2,
        77.1,
        77.4,
        77.3
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        76.3,
        76.3,
        76.4,
        76.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        80.7,
        80.5,
        81.1,
        80.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        80.1,
        79.8,
        80.7,
        80.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        79.7,
        79.5,
        79.7,
        79.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        78.1,
        76.8,
        78.3,
        77.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        71.6,
        71.1,
        71.4,
        71.3
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        74.5,
        74.0,
        74.4,
        74.4
      ],
      "down_blocks.1.attentions.1.proj_out": [
        72.6,
        72.1,
        72.9,
        72.7
      ],
      "down_blocks.1.resnets.0.conv1": [
        63.1,
        62.8,
        63.2,
        63.3
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        72.6,
        72.8,
        72.6,
        72.8
      ],
      "down_blocks.1.resnets.0.conv2": [
        71.2,
        72.0,
        71.8,
        71.9
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        60.9,
        60.6,
        61.1,
        61.1
      ],
      "down_blocks.1.resnets.1.conv1": [
        72.2,
        72.0,
        72.5,
        72.1
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        74.6,
        74.0,
        74.6,
        74.5
      ],
      "down_blocks.1.resnets.1.conv2": [
        74.0,
        73.7,
        74.1,
        73.9
      ],
      "down_blocks.1.downsamplers.0.conv": [
        70.4,
        68.2,
        69.0,
        68.6
      ],
      "down_blocks.2.attentions.0.proj_in": [
        79.5,
        79.4,
        80.0,
        79.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        81.6,
        81.3,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        81.5,
        81.4,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        81.0,
        81.0,
        81.3,
        81.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        80.8,
        80.8,
        81.0,
        80.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.0,
        82.1,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.0,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.9,
        82.1,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        82.0,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        80.8,
        80.7,
        80.8,
        80.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        80.9,
        81.0,
        81.3,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q": [
        81.8,
        81.9,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k": [
        81.9,
        82.1,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v": [
        81.3,
        81.5,
        81.8,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        81.3,
        81.4,
        81.5,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.0,
        82.2,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k": [
        81.7,
        82.1,
        82.1,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v": [
        81.9,
        82.0,
        82.1,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        82.0,
        82.3,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        80.9,
        80.9,
        81.1,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2": [
        81.3,
        81.3,
        81.3,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q": [
        81.5,
        81.4,
        82.0,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k": [
        81.5,
        81.3,
        82.0,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v": [
        81.2,
        81.4,
        81.5,
        81.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        81.5,
        81.4,
        81.5,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q": [
        82.0,
        82.3,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k": [
        81.9,
        82.1,
        82.4,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v": [
        81.8,
        82.1,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        82.0,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        81.1,
        81.0,
        81.1,
        81.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2": [
        81.0,
        81.2,
        81.2,
        81.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q": [
        81.0,
        81.2,
        81.5,
        81.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k": [
        81.0,
        81.1,
        81.4,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v": [
        80.9,
        81.0,
        81.2,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        80.7,
        81.0,
        81.2,
        80.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q": [
        82.1,
        82.1,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k": [
        82.0,
        82.1,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        82.2,
        82.0,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        81.2,
        81.4,
        81.7,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2": [
        81.3,
        81.3,
        81.7,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q": [
        81.7,
        81.7,
        81.8,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k": [
        81.7,
        81.5,
        81.7,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v": [
        81.0,
        81.2,
        81.2,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        81.4,
        81.3,
        81.5,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q": [
        82.2,
        82.1,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k": [
        82.1,
        82.2,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v": [
        81.9,
        82.2,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        82.2,
        82.1,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        80.7,
        80.7,
        81.0,
        80.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2": [
        81.2,
        81.1,
        81.6,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q": [
        81.3,
        81.5,
        81.6,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k": [
        81.3,
        81.4,
        81.6,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v": [
        81.3,
        81.4,
        81.7,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        80.9,
        81.0,
        81.1,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q": [
        82.2,
        82.3,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k": [
        82.0,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v": [
        82.2,
        82.3,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        80.4,
        80.3,
        80.5,
        80.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2": [
        80.4,
        80.9,
        81.1,
        80.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q": [
        81.6,
        81.8,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k": [
        81.6,
        81.9,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v": [
        81.4,
        81.3,
        81.7,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        81.2,
        81.3,
        81.6,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q": [
        82.0,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v": [
        82.0,
        82.5,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        82.3,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        80.5,
        80.6,
        80.8,
        80.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2": [
        81.0,
        81.1,
        81.3,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q": [
        81.9,
        81.8,
        82.1,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k": [
        81.7,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v": [
        81.1,
        81.3,
        81.5,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        81.4,
        81.5,
        81.7,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.1,
        82.1,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        81.8,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        80.7,
        80.9,
        80.9,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2": [
        81.2,
        81.1,
        81.6,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q": [
        81.6,
        81.9,
        82.0,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k": [
        81.6,
        81.7,
        82.1,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v": [
        81.5,
        81.4,
        81.7,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        81.3,
        81.3,
        81.7,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q": [
        82.0,
        82.3,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v": [
        82.3,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        81.9,
        82.3,
        82.4,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        81.2,
        81.0,
        81.3,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2": [
        81.3,
        81.4,
        81.7,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q": [
        81.8,
        82.1,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k": [
        82.0,
        82.3,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v": [
        81.1,
        81.3,
        81.6,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        81.2,
        81.4,
        81.6,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k": [
        81.5,
        81.9,
        81.8,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v": [
        81.5,
        81.6,
        81.8,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        81.4,
        81.6,
        81.8,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        81.0,
        80.9,
        81.4,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2": [
        81.3,
        81.3,
        81.5,
        81.3
      ],
      "down_blocks.2.attentions.0.proj_out": [
        79.6,
        79.7,
        79.7,
        79.5
      ],
      "down_blocks.2.attentions.1.proj_in": [
        81.0,
        80.7,
        80.9,
        81.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        81.8,
        81.9,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        81.8,
        82.1,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        81.3,
        81.2,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        81.1,
        81.1,
        81.6,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.2,
        82.6,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        82.1,
        82.2,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        82.1,
        82.1,
        82.2,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        81.9,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        80.9,
        80.7,
        81.0,
        80.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        81.4,
        81.3,
        81.6,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q": [
        81.9,
        82.1,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k": [
        82.0,
        82.1,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v": [
        81.7,
        81.7,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        81.6,
        81.7,
        81.9,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v": [
        81.9,
        82.1,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        82.0,
        82.2,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        81.4,
        81.0,
        81.3,
        81.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2": [
        81.6,
        81.8,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q": [
        82.1,
        82.1,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k": [
        82.0,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v": [
        81.4,
        81.4,
        81.9,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        81.6,
        81.8,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k": [
        82.0,
        82.1,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v": [
        82.0,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        81.4,
        81.7,
        82.0,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2": [
        81.5,
        81.8,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q": [
        82.0,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k": [
        82.1,
        82.3,
        82.5,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v": [
        81.7,
        81.9,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        81.8,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k": [
        82.1,
        82.2,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v": [
        82.1,
        82.3,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        81.5,
        81.5,
        81.9,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2": [
        81.8,
        82.0,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v": [
        81.7,
        82.1,
        82.0,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        82.0,
        82.1,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k": [
        81.8,
        82.4,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v": [
        82.0,
        82.1,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        82.1,
        82.1,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        81.6,
        81.4,
        81.7,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2": [
        81.9,
        81.9,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k": [
        82.1,
        82.3,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v": [
        81.8,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        81.8,
        81.8,
        82.2,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q": [
        82.0,
        82.3,
        82.3,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k": [
        81.9,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v": [
        82.2,
        82.3,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        82.2,
        82.1,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        81.5,
        81.2,
        81.9,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2": [
        81.5,
        82.0,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q": [
        82.1,
        82.1,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v": [
        81.7,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q": [
        82.2,
        82.2,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v": [
        82.1,
        82.2,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        81.6,
        81.7,
        81.8,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2": [
        81.6,
        81.7,
        81.8,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q": [
        82.2,
        82.3,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k": [
        82.2,
        82.3,
        82.2,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v": [
        81.8,
        82.0,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        81.7,
        82.1,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q": [
        82.2,
        82.4,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k": [
        82.1,
        82.2,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v": [
        82.2,
        82.4,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        81.6,
        81.6,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2": [
        81.6,
        81.9,
        82.1,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k": [
        82.2,
        82.3,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v": [
        81.8,
        82.0,
        81.9,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        81.8,
        82.0,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k": [
        82.1,
        82.1,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v": [
        82.1,
        82.3,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        81.6,
        81.7,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2": [
        81.5,
        81.7,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q": [
        82.1,
        82.2,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k": [
        82.1,
        82.1,
        82.5,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v": [
        81.7,
        81.9,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        81.8,
        81.5,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q": [
        82.1,
        82.1,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v": [
        81.9,
        82.1,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        82.2,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        81.1,
        81.1,
        81.4,
        81.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2": [
        81.6,
        81.4,
        82.0,
        81.6
      ],
      "down_blocks.2.attentions.1.proj_out": [
        80.8,
        80.7,
        80.8,
        80.8
      ],
      "down_blocks.2.resnets.0.conv1": [
        79.6,
        79.5,
        79.6,
        79.5
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        80.7,
        80.8,
        81.0,
        80.9
      ],
      "down_blocks.2.resnets.0.conv2": [
        80.3,
        79.9,
        79.9,
        80.3
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        70.9,
        68.8,
        69.0,
        68.8
      ],
      "down_blocks.2.resnets.1.conv1": [
        80.9,
        80.8,
        81.1,
        80.9
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        81.6,
        81.5,
        81.9,
        81.5
      ],
      "down_blocks.2.resnets.1.conv2": [
        80.9,
        80.7,
        81.0,
        81.0
      ],
      "up_blocks.0.attentions.0.proj_in": [
        80.6,
        79.6,
        78.0,
        80.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        80.8,
        80.7,
        80.1,
        81.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        80.9,
        80.6,
        80.0,
        81.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        80.7,
        80.6,
        80.3,
        80.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        81.1,
        80.8,
        80.2,
        81.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.3,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        82.0,
        82.1,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        80.3,
        78.7,
        76.9,
        79.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        81.0,
        80.6,
        78.5,
        80.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q": [
        81.9,
        81.9,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k": [
        81.7,
        81.9,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v": [
        81.3,
        81.3,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        81.2,
        81.4,
        81.5,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k": [
        82.0,
        82.4,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.0,
        82.1,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        82.0,
        82.2,
        82.2,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        81.3,
        81.4,
        81.5,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2": [
        81.6,
        81.7,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q": [
        81.8,
        81.9,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k": [
        81.7,
        82.0,
        82.0,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v": [
        81.3,
        81.4,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        81.4,
        81.3,
        81.5,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q": [
        81.9,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k": [
        81.9,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v": [
        81.9,
        82.2,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        81.7,
        82.1,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        81.4,
        81.6,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2": [
        81.6,
        81.6,
        82.0,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q": [
        81.3,
        81.3,
        81.1,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k": [
        81.3,
        81.3,
        81.5,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v": [
        81.4,
        81.3,
        81.4,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        81.2,
        81.2,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v": [
        82.1,
        82.1,
        82.0,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        82.0,
        82.3,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        81.4,
        81.6,
        81.8,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2": [
        81.5,
        81.5,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q": [
        81.8,
        82.0,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k": [
        81.7,
        82.1,
        82.2,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v": [
        81.7,
        81.8,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        81.4,
        81.4,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q": [
        82.1,
        82.4,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k": [
        81.9,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v": [
        82.0,
        82.1,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        81.6,
        81.5,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2": [
        81.6,
        81.6,
        81.9,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q": [
        82.0,
        82.0,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k": [
        81.9,
        82.1,
        82.2,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v": [
        81.6,
        81.7,
        81.8,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        81.4,
        81.5,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q": [
        82.1,
        82.3,
        82.5,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k": [
        82.1,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v": [
        82.1,
        82.2,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        81.4,
        81.6,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2": [
        81.6,
        81.9,
        81.9,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q": [
        82.2,
        82.2,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k": [
        82.2,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v": [
        81.6,
        81.8,
        81.8,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        81.5,
        81.6,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q": [
        82.0,
        82.2,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.0,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v": [
        81.9,
        82.4,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        81.9,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        81.6,
        81.9,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2": [
        81.7,
        81.8,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q": [
        82.0,
        82.1,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k": [
        81.9,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v": [
        81.6,
        81.6,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        81.7,
        81.6,
        82.0,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        81.4,
        81.7,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2": [
        81.5,
        81.7,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k": [
        82.1,
        82.1,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v": [
        81.6,
        81.5,
        82.1,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        81.8,
        81.7,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k": [
        82.0,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v": [
        82.0,
        82.1,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        81.4,
        81.3,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2": [
        81.5,
        81.8,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q": [
        81.9,
        82.2,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k": [
        81.8,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v": [
        81.6,
        81.5,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        81.5,
        81.8,
        81.9,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q": [
        81.9,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v": [
        82.3,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        81.0,
        81.2,
        81.3,
        81.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2": [
        81.4,
        81.5,
        81.8,
        81.5
      ],
      "up_blocks.0.attentions.0.proj_out": [
        80.3,
        80.2,
        80.5,
        80.4
      ],
      "up_blocks.0.attentions.1.proj_in": [
        77.3,
        77.1,
        77.9,
        77.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        81.3,
        81.2,
        81.4,
        81.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        81.3,
        81.3,
        81.5,
        81.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        80.3,
        80.1,
        80.6,
        79.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        80.0,
        79.5,
        80.2,
        80.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        82.0,
        82.3,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        81.4,
        81.8,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        81.1,
        81.8,
        81.8,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        77.9,
        77.0,
        77.8,
        77.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        79.3,
        79.2,
        79.6,
        79.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q": [
        81.7,
        81.7,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k": [
        81.6,
        81.8,
        81.9,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v": [
        80.5,
        80.7,
        80.9,
        80.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        80.3,
        80.5,
        80.5,
        80.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q": [
        81.8,
        82.0,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k": [
        76.6,
        77.3,
        75.2,
        75.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v": [
        81.2,
        81.4,
        81.3,
        81.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        81.5,
        81.7,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        78.3,
        77.9,
        79.0,
        78.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2": [
        79.8,
        79.5,
        80.2,
        80.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q": [
        81.6,
        81.3,
        81.6,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k": [
        81.4,
        81.5,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v": [
        80.6,
        80.7,
        81.0,
        80.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        80.4,
        80.8,
        80.9,
        80.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q": [
        81.9,
        82.0,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k": [
        81.6,
        81.7,
        81.6,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v": [
        81.1,
        81.2,
        81.1,
        80.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        81.5,
        81.8,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        78.5,
        78.4,
        78.9,
        78.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2": [
        80.1,
        80.4,
        80.8,
        80.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q": [
        81.3,
        81.3,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k": [
        81.1,
        81.4,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v": [
        80.5,
        80.5,
        80.8,
        80.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        80.3,
        80.4,
        80.3,
        80.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q": [
        81.8,
        81.9,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k": [
        81.6,
        82.0,
        82.1,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v": [
        81.0,
        81.2,
        80.7,
        80.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        81.6,
        81.7,
        82.0,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        80.1,
        79.8,
        80.0,
        79.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2": [
        80.4,
        80.5,
        80.7,
        80.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q": [
        81.0,
        81.3,
        81.4,
        81.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k": [
        81.1,
        81.2,
        81.2,
        81.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v": [
        80.2,
        80.3,
        80.7,
        80.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        80.1,
        80.0,
        80.4,
        79.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q": [
        81.7,
        82.2,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k": [
        81.8,
        82.0,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v": [
        81.4,
        81.4,
        81.3,
        81.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        82.0,
        82.0,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        79.7,
        79.5,
        79.8,
        79.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2": [
        80.0,
        79.6,
        80.2,
        79.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q": [
        81.1,
        81.3,
        81.6,
        81.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k": [
        81.2,
        81.3,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v": [
        80.4,
        80.6,
        80.5,
        80.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        80.6,
        80.6,
        80.7,
        80.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q": [
        81.8,
        82.2,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k": [
        81.6,
        82.0,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v": [
        81.3,
        81.7,
        81.1,
        80.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        81.8,
        81.7,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        79.6,
        79.6,
        79.8,
        79.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2": [
        80.1,
        80.1,
        80.4,
        80.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q": [
        81.0,
        81.0,
        81.2,
        80.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k": [
        81.1,
        81.2,
        81.3,
        81.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v": [
        80.5,
        80.4,
        80.8,
        80.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        80.2,
        80.4,
        80.4,
        80.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k": [
        81.9,
        82.0,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v": [
        81.6,
        81.6,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        81.8,
        81.9,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        79.9,
        80.0,
        80.0,
        79.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2": [
        80.5,
        80.5,
        80.6,
        80.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q": [
        81.2,
        81.3,
        81.3,
        80.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k": [
        81.1,
        81.3,
        81.3,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v": [
        80.5,
        80.0,
        80.6,
        80.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        80.5,
        80.4,
        80.5,
        80.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q": [
        82.0,
        82.3,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k": [
        82.0,
        82.2,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v": [
        81.5,
        81.8,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        81.9,
        81.7,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        79.2,
        79.3,
        79.4,
        78.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2": [
        79.8,
        80.5,
        80.4,
        80.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q": [
        81.2,
        81.0,
        81.1,
        81.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k": [
        81.2,
        81.2,
        81.6,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v": [
        80.5,
        80.5,
        80.7,
        80.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        80.5,
        80.5,
        80.8,
        80.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q": [
        82.1,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k": [
        81.9,
        82.1,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v": [
        81.5,
        81.8,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        81.9,
        82.1,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        78.9,
        79.3,
        79.2,
        79.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2": [
        80.0,
        80.3,
        80.4,
        80.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q": [
        81.3,
        81.3,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k": [
        81.2,
        81.5,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v": [
        80.1,
        80.2,
        80.5,
        80.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        80.2,
        80.4,
        80.8,
        80.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q": [
        81.8,
        82.1,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k": [
        81.6,
        81.6,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v": [
        81.8,
        82.1,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        77.1,
        76.6,
        77.3,
        77.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2": [
        79.6,
        79.1,
        79.4,
        79.1
      ],
      "up_blocks.0.attentions.1.proj_out": [
        75.3,
        74.8,
        75.3,
        75.0
      ],
      "up_blocks.0.attentions.2.proj_in": [
        78.3,
        78.1,
        78.3,
        78.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q": [
        81.1,
        81.1,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k": [
        81.3,
        81.4,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v": [
        80.0,
        79.8,
        80.1,
        80.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        77.7,
        77.6,
        77.9,
        77.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k": [
        81.8,
        82.2,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v": [
        81.8,
        81.9,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        81.9,
        82.1,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        78.9,
        78.5,
        78.8,
        78.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2": [
        79.7,
        79.1,
        79.5,
        79.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q": [
        81.5,
        81.6,
        81.6,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k": [
        81.7,
        82.0,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v": [
        80.5,
        80.5,
        80.8,
        80.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        80.3,
        80.4,
        80.1,
        80.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q": [
        82.0,
        82.3,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k": [
        81.1,
        81.5,
        80.8,
        81.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v": [
        81.1,
        80.8,
        81.2,
        81.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        81.7,
        81.7,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        79.6,
        79.4,
        79.8,
        79.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2": [
        80.1,
        80.0,
        80.1,
        79.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q": [
        81.5,
        81.6,
        81.8,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k": [
        81.6,
        81.6,
        81.9,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v": [
        80.4,
        80.2,
        80.5,
        80.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0": [
        80.4,
        80.3,
        80.6,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q": [
        81.8,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k": [
        81.7,
        82.2,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v": [
        81.3,
        81.6,
        81.6,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0": [
        81.5,
        81.8,
        81.9,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj": [
        80.3,
        80.1,
        80.4,
        79.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2": [
        80.7,
        80.6,
        80.8,
        80.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q": [
        81.6,
        81.7,
        81.8,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k": [
        81.6,
        81.8,
        82.0,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v": [
        80.4,
        80.6,
        80.6,
        80.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0": [
        80.1,
        80.2,
        80.6,
        80.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q": [
        82.0,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k": [
        81.9,
        82.2,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v": [
        81.5,
        81.7,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0": [
        81.6,
        81.9,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj": [
        80.0,
        80.2,
        80.3,
        80.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2": [
        80.7,
        80.6,
        80.8,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q": [
        81.2,
        81.2,
        81.6,
        81.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k": [
        81.4,
        81.1,
        81.6,
        81.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v": [
        80.4,
        80.2,
        80.2,
        80.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0": [
        79.4,
        79.0,
        79.4,
        79.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q": [
        82.0,
        82.3,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k": [
        81.9,
        82.2,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v": [
        81.7,
        81.9,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0": [
        81.8,
        81.9,
        82.0,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj": [
        80.5,
        80.3,
        80.3,
        80.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2": [
        80.6,
        80.6,
        80.8,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q": [
        81.6,
        81.8,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k": [
        81.7,
        81.9,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v": [
        80.8,
        80.8,
        80.8,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0": [
        80.9,
        80.9,
        81.0,
        80.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q": [
        81.9,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k": [
        81.9,
        82.1,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v": [
        82.0,
        82.0,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0": [
        81.8,
        82.2,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj": [
        80.0,
        79.9,
        80.2,
        80.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2": [
        80.4,
        80.5,
        80.8,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q": [
        81.4,
        81.6,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k": [
        81.6,
        81.6,
        81.8,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v": [
        80.6,
        80.7,
        80.8,
        80.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0": [
        80.8,
        80.7,
        80.7,
        80.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q": [
        82.0,
        82.3,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k": [
        82.0,
        82.1,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v": [
        82.0,
        82.1,
        82.1,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0": [
        81.9,
        82.1,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj": [
        80.4,
        80.3,
        80.3,
        80.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2": [
        80.7,
        80.7,
        80.8,
        80.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q": [
        81.4,
        81.4,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k": [
        81.6,
        81.4,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v": [
        80.7,
        80.6,
        80.9,
        80.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0": [
        80.6,
        80.4,
        80.7,
        80.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k": [
        81.9,
        82.3,
        82.2,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v": [
        81.8,
        82.0,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0": [
        82.1,
        82.1,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj": [
        80.5,
        80.0,
        80.6,
        80.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2": [
        80.7,
        80.7,
        81.0,
        80.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q": [
        81.3,
        81.5,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k": [
        81.2,
        81.2,
        81.4,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v": [
        80.7,
        80.6,
        80.7,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0": [
        80.5,
        80.5,
        80.8,
        80.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q": [
        82.3,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k": [
        82.1,
        82.0,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v": [
        82.0,
        82.1,
        82.3,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0": [
        82.0,
        82.0,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj": [
        79.7,
        78.5,
        79.0,
        78.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2": [
        80.4,
        80.3,
        80.4,
        80.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q": [
        81.6,
        81.6,
        81.8,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k": [
        81.4,
        81.6,
        82.0,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v": [
        80.3,
        80.3,
        80.5,
        80.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0": [
        80.6,
        80.6,
        80.5,
        80.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q": [
        81.9,
        82.1,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k": [
        82.1,
        82.2,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v": [
        82.0,
        82.3,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0": [
        81.9,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj": [
        79.8,
        79.6,
        79.8,
        79.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2": [
        80.2,
        80.1,
        80.5,
        80.3
      ],
      "up_blocks.0.attentions.2.proj_out": [
        77.1,
        76.6,
        77.1,
        76.9
      ],
      "up_blocks.0.resnets.0.conv1": [
        80.8,
        80.9,
        80.5,
        80.8
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        81.4,
        81.2,
        81.6,
        81.4
      ],
      "up_blocks.0.resnets.0.conv2": [
        80.3,
        80.3,
        80.0,
        80.2
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        79.8,
        79.7,
        79.7,
        79.8
      ],
      "up_blocks.0.resnets.1.conv1": [
        80.7,
        80.7,
        81.0,
        80.6
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        80.4,
        80.6,
        80.8,
        80.5
      ],
      "up_blocks.0.resnets.1.conv2": [
        80.6,
        80.3,
        80.7,
        80.4
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        75.1,
        74.1,
        75.1,
        74.6
      ],
      "up_blocks.0.resnets.2.conv1": [
        80.2,
        80.2,
        80.4,
        80.0
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        80.6,
        80.8,
        80.8,
        80.8
      ],
      "up_blocks.0.resnets.2.conv2": [
        80.2,
        80.1,
        80.4,
        80.1
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        68.5,
        68.0,
        68.9,
        68.4
      ],
      "up_blocks.0.upsamplers.0.conv": [
        66.6,
        66.1,
        66.6,
        66.3
      ],
      "up_blocks.1.attentions.0.proj_in": [
        70.0,
        69.9,
        70.0,
        69.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        77.0,
        76.5,
        77.0,
        76.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        76.7,
        76.4,
        76.7,
        76.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        72.6,
        72.4,
        72.6,
        72.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        68.4,
        67.9,
        68.5,
        68.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        80.0,
        79.4,
        79.9,
        79.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        79.7,
        79.2,
        79.9,
        79.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.1,
        81.5,
        81.4,
        81.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        80.9,
        80.8,
        81.2,
        80.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        71.7,
        71.2,
        71.3,
        71.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        71.2,
        71.2,
        71.6,
        71.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        75.9,
        75.5,
        75.9,
        75.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        75.3,
        75.2,
        75.2,
        75.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        72.3,
        71.7,
        72.5,
        72.3
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        70.3,
        70.0,
        70.5,
        70.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        80.0,
        80.1,
        80.0,
        79.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        81.2,
        81.0,
        81.3,
        81.3
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        81.0,
        81.2,
        81.0,
        80.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        79.3,
        78.3,
        79.3,
        79.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        70.6,
        70.4,
        70.5,
        70.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        74.5,
        74.3,
        74.6,
        74.4
      ],
      "up_blocks.1.attentions.0.proj_out": [
        72.5,
        72.3,
        72.7,
        72.6
      ],
      "up_blocks.1.attentions.1.proj_in": [
        65.9,
        65.2,
        66.0,
        65.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        74.9,
        74.6,
        75.0,
        75.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        74.9,
        74.7,
        75.1,
        75.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        69.8,
        69.1,
        69.8,
        69.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        61.8,
        61.0,
        61.7,
        61.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        70.9,
        70.8,
        70.3,
        70.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        76.8,
        75.5,
        76.9,
        77.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        80.4,
        80.3,
        80.7,
        80.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        75.7,
        75.1,
        75.3,
        74.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        66.1,
        66.0,
        66.2,
        66.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        68.3,
        67.9,
        68.6,
        68.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        72.9,
        72.7,
        73.0,
        73.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        73.3,
        73.1,
        73.2,
        73.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        69.5,
        69.2,
        69.5,
        69.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        69.4,
        68.9,
        69.4,
        69.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        79.8,
        79.7,
        79.7,
        79.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        66.0,
        65.9,
        66.4,
        66.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        79.2,
        79.8,
        79.2,
        79.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        74.5,
        74.2,
        74.1,
        74.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        66.9,
        66.0,
        66.7,
        66.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        70.8,
        70.1,
        70.7,
        70.8
      ],
      "up_blocks.1.attentions.1.proj_out": [
        68.7,
        68.4,
        68.6,
        68.9
      ],
      "up_blocks.1.attentions.2.proj_in": [
        65.0,
        64.6,
        64.9,
        64.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        74.3,
        74.0,
        74.5,
        74.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        74.4,
        74.4,
        74.3,
        74.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        71.2,
        70.7,
        71.4,
        71.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        61.8,
        60.9,
        62.1,
        62.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        64.2,
        63.6,
        63.6,
        64.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        78.4,
        78.4,
        78.3,
        78.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        78.5,
        77.5,
        78.6,
        78.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        72.7,
        71.7,
        73.0,
        73.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        63.6,
        63.4,
        63.2,
        63.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        65.4,
        65.4,
        65.7,
        65.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q": [
        73.2,
        73.0,
        73.1,
        73.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k": [
        74.2,
        73.9,
        74.2,
        74.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v": [
        69.5,
        69.1,
        69.5,
        69.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        67.2,
        67.4,
        67.4,
        67.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q": [
        74.8,
        74.9,
        74.6,
        74.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k": [
        68.6,
        66.4,
        69.3,
        68.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v": [
        79.7,
        79.7,
        79.7,
        79.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        74.1,
        74.9,
        73.5,
        73.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        66.0,
        65.8,
        66.2,
        66.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2": [
        70.6,
        70.5,
        70.8,
        70.7
      ],
      "up_blocks.1.attentions.2.proj_out": [
        65.7,
        65.5,
        65.8,
        65.6
      ],
      "up_blocks.1.resnets.0.conv1": [
        74.0,
        73.3,
        73.5,
        73.6
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        73.8,
        73.5,
        73.7,
        74.0
      ],
      "up_blocks.1.resnets.0.conv2": [
        74.1,
        73.7,
        74.2,
        74.0
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        61.5,
        60.7,
        61.5,
        61.5
      ],
      "up_blocks.1.resnets.1.conv1": [
        65.0,
        64.7,
        64.8,
        64.8
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        69.7,
        69.1,
        70.1,
        69.8
      ],
      "up_blocks.1.resnets.1.conv2": [
        67.6,
        67.2,
        67.5,
        67.7
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        60.6,
        60.4,
        60.5,
        60.7
      ],
      "up_blocks.1.resnets.2.conv1": [
        64.5,
        64.0,
        64.6,
        64.6
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        71.5,
        71.6,
        71.4,
        71.5
      ],
      "up_blocks.1.resnets.2.conv2": [
        67.7,
        67.2,
        67.5,
        67.4
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        58.2,
        58.0,
        58.0,
        58.1
      ],
      "up_blocks.1.upsamplers.0.conv": [
        54.6,
        54.5,
        54.7,
        54.6
      ],
      "up_blocks.2.resnets.0.conv1": [
        55.4,
        55.3,
        55.4,
        55.4
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        62.6,
        62.2,
        62.7,
        62.7
      ],
      "up_blocks.2.resnets.0.conv2": [
        54.8,
        54.5,
        54.9,
        54.8
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        52.3,
        52.4,
        52.4,
        52.4
      ],
      "up_blocks.2.resnets.1.conv1": [
        51.8,
        51.8,
        52.0,
        52.0
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        61.5,
        61.6,
        61.9,
        61.8
      ],
      "up_blocks.2.resnets.1.conv2": [
        51.4,
        51.5,
        51.5,
        51.6
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        52.5,
        52.5,
        52.6,
        52.5
      ],
      "up_blocks.2.resnets.2.conv1": [
        50.1,
        50.1,
        50.3,
        50.2
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        60.3,
        60.3,
        60.6,
        60.4
      ],
      "up_blocks.2.resnets.2.conv2": [
        48.8,
        48.9,
        48.8,
        48.8
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        54.5,
        54.6,
        54.6,
        54.6
      ],
      "mid_block.attentions.0.proj_in": [
        80.2,
        79.8,
        79.6,
        79.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        80.3,
        80.2,
        80.4,
        80.5
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        80.3,
        80.3,
        80.3,
        80.3
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        80.6,
        80.6,
        80.7,
        80.6
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        80.5,
        80.5,
        80.6,
        80.7
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        81.9,
        82.1,
        82.4,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        82.0,
        82.3,
        82.5,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        80.9,
        80.5,
        80.8,
        80.7
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        81.2,
        81.0,
        81.2,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_q": [
        80.9,
        80.4,
        80.6,
        80.6
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_k": [
        80.8,
        80.5,
        80.5,
        80.7
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_v": [
        80.3,
        80.1,
        80.5,
        80.6
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        80.4,
        80.3,
        80.7,
        80.6
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_k": [
        82.0,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.1,
        82.4,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        82.0,
        82.4,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        81.2,
        80.9,
        81.3,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.2": [
        81.3,
        81.2,
        81.3,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_q": [
        81.4,
        81.4,
        81.5,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_k": [
        81.5,
        81.3,
        81.7,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_v": [
        81.2,
        81.2,
        81.5,
        81.1
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        81.4,
        81.5,
        81.7,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_q": [
        82.0,
        82.4,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_k": [
        81.9,
        82.3,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_v": [
        81.9,
        82.3,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        81.8,
        82.3,
        82.2,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        81.3,
        81.1,
        81.3,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.2": [
        81.6,
        81.6,
        81.7,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_q": [
        81.4,
        81.4,
        81.7,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_k": [
        81.6,
        81.7,
        81.7,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_v": [
        81.5,
        81.4,
        81.1,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        81.4,
        81.4,
        81.6,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_k": [
        82.0,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        81.0,
        81.1,
        81.1,
        81.1
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.2": [
        81.3,
        81.4,
        81.6,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_q": [
        81.0,
        80.8,
        80.9,
        81.1
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_k": [
        81.2,
        81.0,
        80.9,
        80.9
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_v": [
        81.2,
        80.6,
        80.8,
        81.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        81.0,
        80.4,
        80.8,
        81.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_k": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_v": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        82.2,
        82.4,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        81.2,
        81.3,
        81.6,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.2": [
        81.4,
        81.4,
        81.9,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_q": [
        81.6,
        81.8,
        81.8,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_k": [
        81.6,
        81.8,
        81.7,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_v": [
        81.5,
        81.6,
        81.8,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        81.6,
        81.5,
        81.6,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_q": [
        82.3,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_k": [
        82.1,
        82.4,
        82.4,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        81.3,
        81.1,
        81.4,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.2": [
        81.6,
        81.9,
        81.9,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_q": [
        81.9,
        81.9,
        82.2,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_k": [
        81.8,
        82.0,
        82.1,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_v": [
        81.7,
        82.0,
        82.1,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        81.8,
        81.8,
        81.9,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_q": [
        82.1,
        82.4,
        82.4,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_v": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        82.0,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        81.3,
        81.4,
        81.7,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.2": [
        81.6,
        81.7,
        82.0,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_q": [
        81.7,
        81.9,
        82.0,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_k": [
        81.8,
        81.8,
        81.8,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_v": [
        81.9,
        81.9,
        82.2,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        81.5,
        81.6,
        81.7,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.0,
        82.1,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_v": [
        82.1,
        82.4,
        82.2,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        81.3,
        81.3,
        81.6,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.2": [
        81.5,
        81.7,
        81.7,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_q": [
        81.9,
        82.1,
        82.1,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_k": [
        81.9,
        82.0,
        82.1,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_v": [
        81.7,
        81.6,
        81.9,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        81.7,
        81.8,
        82.0,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_q": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_k": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_v": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        82.1,
        82.2,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        81.3,
        81.6,
        81.7,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.2": [
        81.6,
        81.7,
        81.8,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_q": [
        81.8,
        82.0,
        82.1,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_k": [
        81.8,
        82.1,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_v": [
        81.4,
        81.4,
        81.7,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        81.6,
        81.9,
        81.9,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_q": [
        82.0,
        82.1,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_k": [
        82.3,
        82.4,
        82.3,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_v": [
        81.9,
        82.2,
        82.3,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        81.4,
        81.1,
        81.6,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.2": [
        81.5,
        81.9,
        81.8,
        81.7
      ],
      "mid_block.attentions.0.proj_out": [
        80.5,
        80.5,
        80.4,
        80.5
      ],
      "mid_block.resnets.0.conv1": [
        80.8,
        80.7,
        80.8,
        81.0
      ],
      "mid_block.resnets.0.time_emb_proj": [
        81.3,
        81.3,
        81.1,
        81.0
      ],
      "mid_block.resnets.0.conv2": [
        80.7,
        80.9,
        80.9,
        80.7
      ],
      "mid_block.resnets.1.conv1": [
        81.1,
        81.0,
        80.6,
        81.3
      ],
      "mid_block.resnets.1.time_emb_proj": [
        81.4,
        81.6,
        81.5,
        81.4
      ],
      "mid_block.resnets.1.conv2": [
        81.1,
        81.1,
        81.2,
        81.2
      ]
    },
    "4": {
      "time_embedding.linear_1": [
        79.1,
        78.9,
        79.2,
        79.0
      ],
      "time_embedding.linear_2": [
        69.8,
        69.6,
        69.3,
        69.4
      ],
      "add_embedding.linear_1": [
        79.5,
        79.6,
        79.3,
        79.8
      ],
      "add_embedding.linear_2": [
        79.6,
        79.4,
        79.8,
        79.3
      ],
      "down_blocks.0.resnets.0.conv1": [
        73.7,
        74.2,
        73.6,
        74.2
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        76.5,
        77.2,
        76.6,
        76.1
      ],
      "down_blocks.0.resnets.0.conv2": [
        66.7,
        66.6,
        66.8,
        66.8
      ],
      "down_blocks.0.resnets.1.conv1": [
        68.8,
        68.1,
        68.7,
        68.6
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        77.7,
        77.2,
        78.1,
        77.6
      ],
      "down_blocks.0.resnets.1.conv2": [
        67.6,
        67.2,
        67.3,
        67.8
      ],
      "down_blocks.0.downsamplers.0.conv": [
        71.3,
        70.9,
        71.3,
        71.1
      ],
      "down_blocks.1.attentions.0.proj_in": [
        74.2,
        73.8,
        73.8,
        74.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        80.7,
        80.7,
        81.0,
        80.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        80.7,
        80.9,
        80.7,
        80.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        79.0,
        78.9,
        79.0,
        79.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        76.8,
        77.3,
        76.9,
        77.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        81.7,
        82.2,
        82.1,
        82.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        81.7,
        81.8,
        82.0,
        81.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        77.9,
        80.2,
        77.4,
        77.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.4,
        80.4,
        81.1,
        81.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        74.2,
        73.5,
        74.3,
        74.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        75.6,
        75.6,
        75.2,
        75.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        80.1,
        80.3,
        80.4,
        80.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        80.4,
        80.4,
        80.5,
        80.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        78.6,
        78.7,
        78.9,
        79.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        77.9,
        78.0,
        77.7,
        78.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        80.7,
        80.4,
        80.5,
        80.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        76.8,
        76.2,
        76.2,
        76.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        79.4,
        79.2,
        79.4,
        79.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        79.0,
        78.4,
        79.1,
        79.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        74.1,
        73.9,
        74.2,
        74.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        78.0,
        78.0,
        78.2,
        78.3
      ],
      "down_blocks.1.attentions.0.proj_out": [
        77.4,
        76.9,
        77.2,
        77.3
      ],
      "down_blocks.1.attentions.1.proj_in": [
        78.5,
        78.6,
        78.8,
        78.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        80.8,
        81.1,
        81.4,
        81.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        81.0,
        81.1,
        81.4,
        81.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        80.6,
        80.4,
        80.7,
        80.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        80.0,
        79.9,
        80.1,
        79.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        81.7,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        82.0,
        81.8,
        82.2,
        82.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        81.5,
        81.8,
        81.8,
        81.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        81.5,
        81.8,
        82.0,
        81.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        78.9,
        78.7,
        78.8,
        78.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        79.6,
        79.4,
        79.7,
        79.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        81.2,
        81.2,
        81.5,
        81.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        81.3,
        81.3,
        81.4,
        81.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        80.5,
        80.9,
        80.9,
        80.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        80.1,
        80.1,
        80.5,
        80.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        81.8,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        81.9,
        82.0,
        82.3,
        81.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        81.6,
        81.6,
        81.9,
        81.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        81.4,
        80.9,
        81.7,
        81.3
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        79.0,
        78.8,
        79.1,
        79.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        79.8,
        79.9,
        80.1,
        79.9
      ],
      "down_blocks.1.attentions.1.proj_out": [
        79.5,
        79.5,
        79.6,
        79.5
      ],
      "down_blocks.1.resnets.0.conv1": [
        75.7,
        75.6,
        75.4,
        75.5
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        80.7,
        80.5,
        81.0,
        80.7
      ],
      "down_blocks.1.resnets.0.conv2": [
        79.8,
        79.6,
        79.7,
        79.6
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        72.1,
        71.9,
        72.3,
        72.4
      ],
      "down_blocks.1.resnets.1.conv1": [
        78.2,
        78.2,
        78.6,
        78.2
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        80.0,
        79.7,
        80.1,
        79.9
      ],
      "down_blocks.1.resnets.1.conv2": [
        79.9,
        79.9,
        80.1,
        79.9
      ],
      "down_blocks.1.downsamplers.0.conv": [
        76.9,
        75.9,
        76.4,
        75.9
      ],
      "down_blocks.2.attentions.0.proj_in": [
        80.9,
        81.0,
        81.4,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        82.0,
        81.9,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        81.7,
        81.7,
        82.0,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        82.1,
        82.2,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        82.1,
        82.2,
        82.5,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        81.7,
        81.8,
        81.8,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        81.9,
        82.1,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q": [
        82.3,
        82.4,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k": [
        82.2,
        82.1,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v": [
        81.9,
        82.2,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        81.7,
        82.0,
        82.1,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.1,
        82.2,
        82.2,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        81.9,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        81.7,
        81.7,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2": [
        81.8,
        82.0,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k": [
        82.0,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v": [
        81.7,
        82.0,
        82.1,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        81.8,
        82.1,
        82.1,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q": [
        82.2,
        82.2,
        82.5,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k": [
        82.2,
        82.1,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v": [
        81.9,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        81.7,
        81.8,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2": [
        81.7,
        81.8,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q": [
        81.8,
        82.0,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k": [
        81.9,
        82.2,
        82.5,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v": [
        81.7,
        82.0,
        82.1,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        81.8,
        81.9,
        82.0,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q": [
        82.1,
        82.2,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        81.6,
        82.1,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2": [
        81.7,
        81.7,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k": [
        82.0,
        82.1,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v": [
        81.8,
        82.1,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        81.9,
        82.0,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q": [
        82.2,
        82.4,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        81.7,
        82.1,
        81.9,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2": [
        81.7,
        81.8,
        82.1,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q": [
        82.0,
        82.0,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k": [
        82.0,
        82.1,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v": [
        81.8,
        82.0,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        81.8,
        82.1,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k": [
        82.2,
        82.4,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.1,
        82.4,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        81.7,
        81.8,
        81.9,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2": [
        81.7,
        82.0,
        82.2,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v": [
        81.8,
        82.2,
        82.2,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        81.9,
        82.2,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v": [
        82.1,
        82.3,
        82.5,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        81.6,
        81.8,
        81.8,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2": [
        81.8,
        81.6,
        82.1,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v": [
        82.0,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        81.8,
        82.1,
        82.1,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.2,
        82.2,
        82.2,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        81.8,
        81.9,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2": [
        81.6,
        81.8,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q": [
        82.2,
        82.2,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k": [
        82.1,
        82.2,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v": [
        82.0,
        81.9,
        82.1,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        82.1,
        82.1,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v": [
        82.1,
        82.3,
        82.3,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        81.7,
        81.8,
        82.1,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2": [
        82.1,
        82.2,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v": [
        81.8,
        82.2,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        81.8,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k": [
        82.1,
        82.4,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v": [
        81.9,
        81.9,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        81.9,
        81.6,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        81.6,
        81.8,
        82.0,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2": [
        81.7,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.proj_out": [
        81.3,
        81.4,
        81.6,
        81.2
      ],
      "down_blocks.2.attentions.1.proj_in": [
        81.8,
        82.0,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        81.9,
        82.5,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        81.9,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.4,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        82.0,
        82.3,
        82.5,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        81.7,
        81.9,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        82.1,
        82.2,
        82.1,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q": [
        81.9,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v": [
        81.8,
        82.3,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        82.1,
        82.1,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k": [
        82.3,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v": [
        82.0,
        82.2,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        81.8,
        82.1,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2": [
        81.9,
        82.1,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v": [
        82.0,
        82.2,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k": [
        82.4,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        82.0,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2": [
        81.9,
        82.1,
        82.3,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k": [
        82.3,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        81.7,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        82.0,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        81.9,
        82.3,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2": [
        82.1,
        82.3,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v": [
        82.0,
        82.3,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        81.9,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        82.3,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        82.0,
        82.1,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2": [
        81.9,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q": [
        82.1,
        82.3,
        82.6,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v": [
        82.2,
        82.1,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        81.7,
        81.9,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2": [
        81.9,
        82.3,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v": [
        82.0,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        82.1,
        82.3,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q": [
        81.8,
        82.4,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        82.0,
        82.1,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        81.7,
        82.2,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2": [
        81.8,
        82.1,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q": [
        82.3,
        82.2,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v": [
        82.2,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        82.1,
        82.1,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q": [
        81.8,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        82.2,
        82.2,
        82.4,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        82.0,
        82.2,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2": [
        82.1,
        82.2,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v": [
        82.1,
        82.3,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v": [
        82.2,
        82.3,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        81.9,
        82.1,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2": [
        81.8,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v": [
        82.1,
        82.4,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        81.9,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v": [
        82.1,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        81.7,
        82.0,
        82.0,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.proj_out": [
        81.7,
        81.8,
        82.0,
        81.5
      ],
      "down_blocks.2.resnets.0.conv1": [
        81.5,
        81.5,
        81.8,
        81.6
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        81.7,
        81.5,
        82.1,
        81.7
      ],
      "down_blocks.2.resnets.0.conv2": [
        81.5,
        81.6,
        81.9,
        81.5
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        78.4,
        77.8,
        78.5,
        77.9
      ],
      "down_blocks.2.resnets.1.conv1": [
        81.5,
        81.7,
        81.9,
        81.5
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        82.2,
        82.1,
        82.1,
        82.1
      ],
      "down_blocks.2.resnets.1.conv2": [
        81.9,
        81.9,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.0.proj_in": [
        81.7,
        81.7,
        81.4,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        81.7,
        82.0,
        81.9,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        81.7,
        82.1,
        82.1,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        82.0,
        82.1,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        81.7,
        81.8,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.3,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        82.1,
        82.3,
        82.5,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        81.8,
        81.9,
        81.5,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        81.8,
        81.9,
        81.8,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k": [
        82.2,
        82.3,
        82.2,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v": [
        81.9,
        82.0,
        82.2,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        81.9,
        82.2,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k": [
        82.0,
        82.2,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        81.9,
        82.0,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2": [
        82.0,
        82.1,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k": [
        82.1,
        82.4,
        82.5,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v": [
        81.7,
        82.1,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        81.8,
        81.7,
        81.9,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q": [
        82.0,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k": [
        81.9,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v": [
        81.8,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        81.8,
        82.3,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2": [
        82.0,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q": [
        81.9,
        82.3,
        82.2,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k": [
        81.7,
        82.3,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v": [
        81.8,
        81.9,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        81.7,
        81.7,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        81.9,
        82.1,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v": [
        81.7,
        82.3,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        81.6,
        81.5,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k": [
        82.0,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v": [
        82.3,
        82.3,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        81.8,
        82.1,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2": [
        82.1,
        82.3,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v": [
        81.8,
        82.2,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        81.8,
        81.6,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        81.9,
        82.0,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2": [
        82.0,
        82.1,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k": [
        82.3,
        82.1,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v": [
        81.7,
        82.2,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        81.7,
        81.9,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v": [
        82.3,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        82.1,
        82.0,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2": [
        81.9,
        82.3,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k": [
        82.3,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v": [
        81.9,
        82.1,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        81.8,
        82.0,
        82.2,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.2,
        82.4,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v": [
        82.1,
        82.4,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        81.8,
        82.0,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2": [
        81.9,
        82.0,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q": [
        82.2,
        82.2,
        82.5,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v": [
        81.9,
        82.3,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        81.9,
        82.1,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2": [
        82.0,
        82.3,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v": [
        81.9,
        82.1,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        81.6,
        81.9,
        82.0,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v": [
        82.3,
        82.3,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        81.7,
        81.9,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2": [
        81.7,
        81.9,
        82.1,
        81.6
      ],
      "up_blocks.0.attentions.0.proj_out": [
        81.3,
        81.5,
        81.8,
        81.4
      ],
      "up_blocks.0.attentions.1.proj_in": [
        80.1,
        80.1,
        80.3,
        80.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        82.1,
        82.1,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        82.1,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        81.3,
        81.4,
        81.6,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        80.6,
        80.6,
        80.8,
        80.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        82.3,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        81.8,
        82.0,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        81.8,
        82.2,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        81.2,
        81.2,
        81.4,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        81.1,
        81.2,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q": [
        82.1,
        82.1,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v": [
        81.5,
        81.4,
        81.8,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        81.4,
        81.2,
        81.4,
        81.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v": [
        82.0,
        82.1,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        81.0,
        81.1,
        81.4,
        81.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2": [
        81.2,
        81.2,
        81.4,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k": [
        81.9,
        82.0,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v": [
        81.3,
        81.4,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        81.3,
        81.5,
        81.8,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v": [
        82.0,
        82.1,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        81.0,
        81.2,
        81.3,
        80.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2": [
        81.1,
        81.0,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q": [
        81.7,
        82.0,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k": [
        82.0,
        82.1,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v": [
        81.5,
        81.4,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        81.4,
        81.4,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k": [
        82.0,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v": [
        82.0,
        82.1,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        82.0,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        81.2,
        81.1,
        81.4,
        81.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2": [
        81.6,
        81.5,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q": [
        81.9,
        82.1,
        82.4,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k": [
        81.8,
        82.1,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v": [
        81.3,
        81.4,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        81.4,
        81.4,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q": [
        82.1,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k": [
        82.0,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v": [
        82.0,
        82.1,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        81.3,
        81.3,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2": [
        81.4,
        81.4,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q": [
        82.1,
        82.1,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k": [
        81.9,
        81.9,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v": [
        81.2,
        81.3,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        81.3,
        81.5,
        81.9,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k": [
        81.9,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        82.0,
        82.2,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        81.3,
        81.3,
        81.3,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2": [
        81.2,
        81.3,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q": [
        81.7,
        82.1,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k": [
        82.0,
        82.1,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v": [
        81.5,
        81.4,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        81.3,
        81.5,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v": [
        82.3,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        82.1,
        82.2,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        81.1,
        81.1,
        81.3,
        81.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2": [
        81.4,
        81.3,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q": [
        81.9,
        82.0,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k": [
        81.9,
        82.0,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v": [
        81.3,
        81.3,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        81.4,
        81.3,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q": [
        81.9,
        82.2,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v": [
        81.8,
        82.3,
        82.5,
        82.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        80.9,
        81.1,
        81.1,
        80.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2": [
        81.0,
        81.4,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q": [
        81.8,
        82.2,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k": [
        81.9,
        82.1,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v": [
        81.3,
        81.2,
        81.4,
        81.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        81.6,
        81.5,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k": [
        81.9,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        81.0,
        81.1,
        81.4,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2": [
        81.3,
        81.5,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q": [
        81.8,
        82.0,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k": [
        81.7,
        82.0,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v": [
        81.3,
        81.4,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        81.2,
        81.1,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v": [
        81.8,
        82.3,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        80.3,
        80.6,
        80.8,
        80.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2": [
        81.2,
        80.6,
        81.1,
        80.9
      ],
      "up_blocks.0.attentions.1.proj_out": [
        80.3,
        80.4,
        80.8,
        80.3
      ],
      "up_blocks.0.attentions.2.proj_in": [
        80.7,
        80.6,
        80.6,
        80.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q": [
        81.9,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k": [
        81.9,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v": [
        81.1,
        81.2,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        80.0,
        80.3,
        80.4,
        80.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k": [
        82.0,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v": [
        81.7,
        82.2,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        81.9,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        81.1,
        81.2,
        81.4,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2": [
        81.0,
        80.9,
        81.3,
        81.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q": [
        81.7,
        82.3,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k": [
        82.0,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v": [
        81.3,
        81.3,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        81.1,
        81.3,
        81.5,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v": [
        81.8,
        82.1,
        82.0,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        81.9,
        82.1,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        81.3,
        81.4,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2": [
        81.1,
        81.1,
        81.5,
        81.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q": [
        81.9,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k": [
        81.7,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v": [
        81.3,
        81.3,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0": [
        81.2,
        81.5,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v": [
        81.8,
        82.2,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0": [
        81.7,
        82.2,
        82.4,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj": [
        81.3,
        81.5,
        81.6,
        81.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2": [
        81.4,
        81.6,
        81.5,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q": [
        82.1,
        82.0,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k": [
        81.8,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v": [
        81.2,
        81.3,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0": [
        81.0,
        81.0,
        81.4,
        81.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k": [
        82.2,
        82.2,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v": [
        82.0,
        82.1,
        82.1,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj": [
        81.3,
        81.5,
        81.5,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2": [
        81.6,
        81.4,
        81.8,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q": [
        82.2,
        82.1,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k": [
        82.0,
        82.2,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v": [
        81.5,
        81.3,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0": [
        80.9,
        81.0,
        81.1,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k": [
        82.0,
        82.1,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0": [
        82.0,
        82.3,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj": [
        81.4,
        81.4,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2": [
        81.5,
        81.6,
        81.9,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q": [
        82.0,
        82.3,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k": [
        82.0,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v": [
        81.3,
        81.3,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0": [
        81.4,
        81.4,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj": [
        81.2,
        81.2,
        81.5,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2": [
        81.4,
        81.2,
        81.7,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q": [
        81.9,
        82.2,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k": [
        81.9,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v": [
        81.4,
        81.3,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0": [
        81.5,
        81.6,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v": [
        82.1,
        82.1,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj": [
        81.3,
        81.1,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2": [
        81.6,
        81.5,
        81.9,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q": [
        82.1,
        82.2,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v": [
        81.6,
        81.6,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0": [
        81.2,
        81.2,
        81.2,
        81.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q": [
        82.2,
        82.2,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k": [
        81.8,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0": [
        82.2,
        82.4,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj": [
        81.3,
        81.5,
        81.5,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2": [
        81.5,
        81.5,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q": [
        81.9,
        82.1,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k": [
        82.0,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v": [
        81.3,
        81.3,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0": [
        81.5,
        81.5,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k": [
        82.0,
        82.1,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v": [
        82.1,
        82.3,
        82.6,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0": [
        82.1,
        82.1,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj": [
        81.2,
        81.2,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2": [
        81.4,
        81.5,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q": [
        81.9,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k": [
        81.9,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v": [
        81.2,
        81.2,
        81.6,
        81.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0": [
        81.5,
        81.5,
        81.5,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k": [
        82.1,
        82.3,
        82.5,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v": [
        82.2,
        82.4,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj": [
        81.0,
        80.9,
        81.4,
        80.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2": [
        81.0,
        81.0,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.2.proj_out": [
        80.5,
        80.4,
        80.7,
        80.3
      ],
      "up_blocks.0.resnets.0.conv1": [
        81.4,
        81.7,
        81.6,
        81.5
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        81.9,
        82.2,
        82.2,
        81.9
      ],
      "up_blocks.0.resnets.0.conv2": [
        81.7,
        81.9,
        81.8,
        81.8
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        81.4,
        81.3,
        81.5,
        81.3
      ],
      "up_blocks.0.resnets.1.conv1": [
        81.7,
        81.6,
        81.9,
        81.5
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        81.7,
        81.5,
        81.8,
        81.5
      ],
      "up_blocks.0.resnets.1.conv2": [
        81.5,
        81.4,
        81.7,
        81.5
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        79.3,
        79.4,
        79.5,
        79.4
      ],
      "up_blocks.0.resnets.2.conv1": [
        81.2,
        81.3,
        81.3,
        81.2
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        81.6,
        81.7,
        81.8,
        81.4
      ],
      "up_blocks.0.resnets.2.conv2": [
        81.4,
        81.5,
        81.6,
        81.4
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        75.1,
        75.1,
        75.4,
        74.9
      ],
      "up_blocks.0.upsamplers.0.conv": [
        74.5,
        74.1,
        74.4,
        74.0
      ],
      "up_blocks.1.attentions.0.proj_in": [
        77.6,
        77.3,
        77.6,
        77.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        80.8,
        80.6,
        81.2,
        80.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        80.8,
        80.7,
        81.0,
        80.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        79.4,
        79.4,
        79.5,
        79.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        74.6,
        74.4,
        75.0,
        74.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        81.9,
        82.1,
        82.2,
        82.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        81.7,
        81.5,
        82.1,
        81.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        82.0,
        82.1,
        82.0,
        81.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.7,
        81.8,
        82.1,
        81.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        78.7,
        78.6,
        78.8,
        78.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        77.0,
        76.9,
        77.6,
        77.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        80.4,
        80.5,
        80.5,
        80.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        80.3,
        80.3,
        80.5,
        80.3
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        79.2,
        79.3,
        79.5,
        79.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        76.5,
        76.3,
        77.0,
        76.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        81.8,
        82.2,
        82.1,
        82.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        82.0,
        82.3,
        82.3,
        81.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        81.9,
        82.1,
        82.1,
        81.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        81.7,
        81.6,
        82.0,
        81.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        78.7,
        78.4,
        78.9,
        78.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        79.9,
        79.7,
        79.8,
        79.6
      ],
      "up_blocks.1.attentions.0.proj_out": [
        79.0,
        79.2,
        79.3,
        79.1
      ],
      "up_blocks.1.attentions.1.proj_in": [
        75.1,
        75.2,
        75.2,
        75.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        80.4,
        80.3,
        80.6,
        80.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        80.5,
        80.4,
        80.7,
        80.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        78.6,
        78.3,
        78.6,
        78.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        68.5,
        67.9,
        68.5,
        68.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        81.8,
        81.5,
        81.9,
        81.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        80.9,
        80.7,
        80.9,
        80.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        81.8,
        82.0,
        81.9,
        81.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        81.0,
        81.1,
        81.1,
        80.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        75.7,
        75.6,
        75.8,
        75.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        75.5,
        75.3,
        75.8,
        75.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        79.6,
        79.6,
        79.8,
        79.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        79.6,
        79.7,
        79.9,
        79.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        78.2,
        78.1,
        78.2,
        78.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        76.5,
        76.3,
        76.6,
        76.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        81.8,
        82.1,
        82.1,
        81.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        81.7,
        81.7,
        81.9,
        81.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        81.6,
        81.6,
        81.4,
        81.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        80.9,
        81.0,
        81.1,
        80.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        76.8,
        76.5,
        76.8,
        77.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        78.6,
        78.7,
        78.9,
        78.8
      ],
      "up_blocks.1.attentions.1.proj_out": [
        77.5,
        77.5,
        77.8,
        77.7
      ],
      "up_blocks.1.attentions.2.proj_in": [
        74.1,
        74.1,
        74.2,
        74.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        80.3,
        80.2,
        80.7,
        80.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        80.7,
        80.5,
        80.7,
        80.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        78.6,
        78.6,
        78.8,
        78.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        74.6,
        74.1,
        74.7,
        74.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        79.6,
        79.0,
        80.0,
        79.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        81.1,
        80.8,
        81.5,
        81.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        81.0,
        80.6,
        81.4,
        81.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        79.6,
        79.6,
        80.2,
        80.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        75.0,
        74.5,
        75.0,
        75.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        74.7,
        74.7,
        75.1,
        75.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q": [
        79.5,
        79.4,
        79.8,
        79.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k": [
        79.7,
        80.2,
        80.3,
        80.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v": [
        76.8,
        76.6,
        77.0,
        76.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        77.0,
        76.8,
        77.1,
        77.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q": [
        81.1,
        81.2,
        81.3,
        81.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k": [
        74.1,
        71.3,
        75.7,
        75.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v": [
        81.5,
        81.9,
        82.0,
        81.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        81.0,
        81.3,
        81.1,
        80.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        75.4,
        75.3,
        75.5,
        75.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2": [
        77.6,
        77.8,
        78.1,
        78.0
      ],
      "up_blocks.1.attentions.2.proj_out": [
        75.9,
        75.8,
        76.0,
        75.9
      ],
      "up_blocks.1.resnets.0.conv1": [
        80.2,
        80.2,
        80.2,
        80.1
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        79.8,
        79.7,
        79.9,
        79.8
      ],
      "up_blocks.1.resnets.0.conv2": [
        79.7,
        79.9,
        79.9,
        79.7
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        73.9,
        73.5,
        73.9,
        73.7
      ],
      "up_blocks.1.resnets.1.conv1": [
        74.0,
        73.8,
        74.0,
        73.9
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        79.9,
        79.8,
        80.2,
        79.8
      ],
      "up_blocks.1.resnets.1.conv2": [
        76.9,
        76.9,
        76.9,
        76.6
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        71.1,
        70.9,
        70.7,
        71.0
      ],
      "up_blocks.1.resnets.2.conv1": [
        73.2,
        73.0,
        73.1,
        73.5
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        80.2,
        80.2,
        80.3,
        80.1
      ],
      "up_blocks.1.resnets.2.conv2": [
        77.5,
        77.3,
        77.7,
        77.5
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        72.0,
        71.7,
        71.8,
        72.0
      ],
      "up_blocks.1.upsamplers.0.conv": [
        67.6,
        67.7,
        67.6,
        67.5
      ],
      "up_blocks.2.resnets.0.conv1": [
        64.7,
        64.2,
        64.7,
        64.7
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        78.3,
        78.2,
        78.5,
        78.4
      ],
      "up_blocks.2.resnets.0.conv2": [
        64.8,
        64.3,
        65.0,
        65.0
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        66.6,
        66.6,
        66.7,
        66.4
      ],
      "up_blocks.2.resnets.1.conv1": [
        60.4,
        60.3,
        60.5,
        60.3
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        77.6,
        77.6,
        76.8,
        77.0
      ],
      "up_blocks.2.resnets.1.conv2": [
        59.8,
        59.9,
        59.9,
        59.9
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        66.0,
        66.0,
        66.1,
        66.0
      ],
      "up_blocks.2.resnets.2.conv1": [
        61.7,
        61.7,
        61.9,
        61.9
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        75.9,
        75.0,
        75.7,
        75.4
      ],
      "up_blocks.2.resnets.2.conv2": [
        65.7,
        65.5,
        65.7,
        65.7
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        71.5,
        71.7,
        71.6,
        71.6
      ],
      "mid_block.attentions.0.proj_in": [
        81.4,
        81.7,
        81.7,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        81.8,
        81.9,
        81.9,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        81.8,
        81.8,
        81.9,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        81.7,
        81.8,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        81.9,
        81.9,
        82.0,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.3,
        82.2,
        82.5,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        81.8,
        82.0,
        82.2,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        81.9,
        82.0,
        82.0,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_q": [
        81.7,
        82.0,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_k": [
        81.8,
        81.8,
        81.9,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_v": [
        81.7,
        82.1,
        82.0,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        81.7,
        81.9,
        81.8,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.1,
        82.4,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_k": [
        82.3,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        81.7,
        82.0,
        82.1,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.2": [
        82.2,
        81.9,
        82.3,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_q": [
        81.8,
        82.1,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_k": [
        82.0,
        82.2,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_v": [
        81.9,
        82.0,
        82.1,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        81.9,
        82.2,
        82.0,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_q": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_k": [
        82.1,
        82.4,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_v": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        81.9,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        82.0,
        81.9,
        82.2,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.2": [
        81.9,
        82.2,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_q": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_k": [
        82.1,
        82.1,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_v": [
        81.9,
        82.0,
        82.2,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        81.8,
        82.3,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_k": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_v": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        81.9,
        82.1,
        82.4,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        81.7,
        82.0,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.2": [
        81.9,
        81.9,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_q": [
        81.9,
        81.9,
        82.2,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_k": [
        81.9,
        82.1,
        82.0,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_v": [
        81.9,
        81.9,
        82.0,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        81.8,
        81.7,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_q": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_v": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        81.7,
        82.0,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.2": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_q": [
        82.2,
        82.4,
        82.5,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_k": [
        82.1,
        82.2,
        82.2,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_v": [
        81.9,
        82.2,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        82.0,
        82.2,
        82.3,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_q": [
        82.1,
        82.4,
        82.4,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_k": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_v": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        81.8,
        82.1,
        82.2,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.2": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_k": [
        82.1,
        82.3,
        82.4,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_v": [
        82.1,
        82.2,
        82.2,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        81.9,
        82.1,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_q": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        82.3,
        82.3,
        82.3,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        81.9,
        82.1,
        82.1,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.2": [
        82.2,
        82.2,
        82.4,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_q": [
        82.0,
        82.2,
        82.5,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_k": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_v": [
        82.0,
        82.2,
        82.3,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.0,
        82.3,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_v": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        82.0,
        82.0,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.2": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_q": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_v": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        81.8,
        82.1,
        82.3,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_q": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_k": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_v": [
        82.1,
        82.4,
        82.4,
        82.4
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        81.8,
        82.2,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.2": [
        81.9,
        82.3,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_q": [
        81.9,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_k": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_v": [
        81.9,
        82.1,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_k": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        81.7,
        81.9,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.2": [
        81.9,
        82.2,
        82.3,
        81.9
      ],
      "mid_block.attentions.0.proj_out": [
        81.8,
        81.9,
        82.0,
        81.8
      ],
      "mid_block.resnets.0.conv1": [
        81.8,
        81.7,
        82.0,
        81.7
      ],
      "mid_block.resnets.0.time_emb_proj": [
        81.8,
        81.9,
        82.1,
        81.8
      ],
      "mid_block.resnets.0.conv2": [
        81.9,
        81.8,
        82.0,
        81.7
      ],
      "mid_block.resnets.1.conv1": [
        81.6,
        82.0,
        81.9,
        81.8
      ],
      "mid_block.resnets.1.time_emb_proj": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "mid_block.resnets.1.conv2": [
        81.7,
        81.7,
        81.9,
        81.7
      ]
    },
    "6": {
      "time_embedding.linear_1": [
        80.6,
        80.6,
        80.9,
        80.5
      ],
      "time_embedding.linear_2": [
        80.2,
        80.3,
        80.5,
        80.3
      ],
      "add_embedding.linear_1": [
        80.8,
        81.0,
        80.5,
        80.6
      ],
      "add_embedding.linear_2": [
        81.0,
        80.7,
        80.7,
        81.0
      ],
      "down_blocks.0.resnets.0.conv1": [
        79.5,
        79.5,
        79.8,
        79.8
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        80.1,
        80.4,
        80.4,
        80.5
      ],
      "down_blocks.0.resnets.0.conv2": [
        76.2,
        76.2,
        76.2,
        76.2
      ],
      "down_blocks.0.resnets.1.conv1": [
        76.4,
        76.5,
        76.6,
        76.9
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        80.8,
        80.7,
        81.2,
        80.4
      ],
      "down_blocks.0.resnets.1.conv2": [
        77.7,
        77.3,
        77.8,
        77.6
      ],
      "down_blocks.0.downsamplers.0.conv": [
        79.5,
        79.3,
        79.5,
        79.3
      ],
      "down_blocks.1.attentions.0.proj_in": [
        80.0,
        79.8,
        79.9,
        79.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        81.7,
        81.9,
        82.0,
        81.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        81.8,
        81.9,
        82.0,
        81.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        81.2,
        81.1,
        81.4,
        81.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        80.9,
        80.8,
        81.1,
        80.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        81.9,
        82.1,
        82.4,
        82.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.7,
        82.0,
        81.9,
        81.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        82.1,
        82.2,
        82.2,
        82.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        79.4,
        79.8,
        80.0,
        79.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        80.7,
        80.4,
        80.9,
        80.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        81.3,
        81.4,
        81.7,
        81.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        81.4,
        81.7,
        81.8,
        81.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        81.0,
        81.3,
        81.1,
        81.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        81.0,
        81.1,
        81.2,
        80.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        81.8,
        82.1,
        82.1,
        81.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        81.5,
        81.5,
        81.6,
        81.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        81.8,
        82.1,
        82.1,
        81.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        81.5,
        81.8,
        81.8,
        81.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        79.8,
        79.8,
        80.1,
        79.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        80.8,
        81.1,
        81.3,
        80.9
      ],
      "down_blocks.1.attentions.0.proj_out": [
        80.5,
        80.8,
        80.7,
        80.5
      ],
      "down_blocks.1.attentions.1.proj_in": [
        81.0,
        81.1,
        81.3,
        81.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        81.9,
        82.1,
        82.1,
        81.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        82.1,
        82.1,
        82.1,
        81.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        81.7,
        81.7,
        81.8,
        81.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        81.1,
        81.4,
        81.6,
        81.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        81.7,
        82.1,
        82.3,
        82.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        81.1,
        81.1,
        81.2,
        81.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        81.1,
        81.1,
        81.2,
        81.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        81.8,
        82.0,
        82.1,
        82.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        81.9,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        81.5,
        81.5,
        81.8,
        81.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        81.4,
        81.4,
        81.8,
        81.3
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        82.2,
        82.4,
        82.3,
        82.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        82.0,
        82.3,
        82.2,
        81.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        82.0,
        81.9,
        82.3,
        81.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        80.8,
        81.2,
        81.3,
        80.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        81.3,
        81.2,
        81.6,
        81.2
      ],
      "down_blocks.1.attentions.1.proj_out": [
        81.2,
        81.0,
        81.4,
        81.2
      ],
      "down_blocks.1.resnets.0.conv1": [
        80.8,
        81.0,
        81.1,
        80.9
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        81.6,
        81.8,
        81.9,
        81.6
      ],
      "down_blocks.1.resnets.0.conv2": [
        81.4,
        81.4,
        81.5,
        81.2
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        79.4,
        79.3,
        79.5,
        79.1
      ],
      "down_blocks.1.resnets.1.conv1": [
        80.9,
        81.0,
        81.2,
        81.0
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        81.9,
        81.9,
        82.3,
        81.8
      ],
      "down_blocks.1.resnets.1.conv2": [
        81.4,
        81.3,
        81.6,
        81.3
      ],
      "down_blocks.1.downsamplers.0.conv": [
        81.0,
        81.2,
        81.4,
        81.1
      ],
      "down_blocks.2.attentions.0.proj_in": [
        81.9,
        82.2,
        82.1,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        82.2,
        82.2,
        82.5,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        81.8,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        82.0,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.1,
        82.4,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        81.9,
        82.3,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        81.9,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v": [
        82.0,
        82.4,
        82.4,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        82.1,
        82.1,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k": [
        82.2,
        82.3,
        82.2,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        82.3,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        82.0,
        82.3,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q": [
        82.3,
        82.4,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k": [
        82.3,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        82.0,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        82.0,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k": [
        82.1,
        82.4,
        82.3,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v": [
        82.1,
        82.3,
        82.5,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v": [
        82.1,
        82.4,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2": [
        82.0,
        82.0,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k": [
        82.3,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k": [
        82.3,
        82.4,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        81.8,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2": [
        82.2,
        82.1,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k": [
        82.1,
        82.3,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v": [
        82.0,
        82.0,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        82.2,
        82.2,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v": [
        82.2,
        82.2,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.2,
        82.3,
        82.6,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        82.0,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q": [
        82.3,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v": [
        82.2,
        82.1,
        82.4,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v": [
        82.3,
        82.4,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2": [
        82.1,
        82.1,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q": [
        82.3,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v": [
        82.0,
        82.2,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        82.0,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v": [
        82.2,
        82.2,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        82.3,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        82.1,
        82.3,
        82.1,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2": [
        81.9,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q": [
        82.0,
        82.2,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k": [
        82.2,
        82.3,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        82.1,
        82.2,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        81.9,
        82.0,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q": [
        82.3,
        82.1,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v": [
        82.0,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k": [
        82.1,
        82.2,
        82.6,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v": [
        82.1,
        82.4,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        81.9,
        82.3,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2": [
        81.8,
        82.2,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.0.proj_out": [
        81.9,
        81.9,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.1.proj_in": [
        82.1,
        82.2,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        82.2,
        82.2,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        82.0,
        82.4,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        82.1,
        82.1,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k": [
        82.3,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q": [
        82.3,
        82.5,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k": [
        82.1,
        82.4,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        82.2,
        82.4,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        82.1,
        82.5,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k": [
        82.3,
        82.3,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v": [
        82.1,
        82.4,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        82.3,
        82.4,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k": [
        82.2,
        82.4,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v": [
        82.2,
        82.2,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q": [
        82.1,
        82.2,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k": [
        82.1,
        82.4,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v": [
        82.2,
        82.4,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q": [
        82.2,
        82.3,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v": [
        82.2,
        82.4,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.4,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v": [
        82.0,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        82.2,
        82.2,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2": [
        82.3,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q": [
        82.1,
        82.4,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k": [
        82.2,
        82.4,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q": [
        82.2,
        82.4,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v": [
        82.1,
        82.2,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2": [
        82.0,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q": [
        82.1,
        82.3,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v": [
        81.9,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        82.1,
        82.3,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q": [
        82.2,
        82.3,
        82.3,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k": [
        81.9,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v": [
        82.1,
        82.3,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.proj_out": [
        81.9,
        82.3,
        82.1,
        82.1
      ],
      "down_blocks.2.resnets.0.conv1": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.resnets.0.conv2": [
        81.9,
        82.4,
        82.2,
        82.0
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        81.5,
        81.3,
        81.5,
        81.3
      ],
      "down_blocks.2.resnets.1.conv1": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.resnets.1.conv2": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.proj_in": [
        82.1,
        82.2,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        82.3,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.0,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        82.0,
        82.1,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k": [
        82.3,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        82.0,
        82.4,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.3,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k": [
        82.2,
        82.2,
        82.5,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.2,
        82.2,
        82.5,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        82.2,
        82.2,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k": [
        82.3,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        82.0,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q": [
        82.1,
        82.4,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v": [
        81.9,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k": [
        82.2,
        82.4,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v": [
        82.1,
        82.4,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        82.3,
        82.2,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        82.0,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q": [
        82.3,
        82.4,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v": [
        82.0,
        82.4,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        82.1,
        82.1,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2": [
        82.1,
        82.2,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k": [
        82.3,
        82.4,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q": [
        82.2,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v": [
        82.3,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        82.0,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k": [
        82.3,
        82.1,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.3,
        82.5,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q": [
        82.1,
        82.4,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        82.2,
        82.2,
        82.2,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v": [
        82.1,
        82.2,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        82.2,
        82.2,
        82.5,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        81.7,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        82.3,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        81.9,
        82.4,
        82.4,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2": [
        82.1,
        82.2,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q": [
        82.0,
        82.1,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k": [
        82.2,
        82.3,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        82.0,
        82.3,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2": [
        81.9,
        82.2,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.0.proj_out": [
        81.8,
        82.0,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.1.proj_in": [
        81.7,
        81.6,
        81.8,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        81.7,
        82.0,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        81.3,
        81.6,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        81.9,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        81.9,
        81.8,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        81.7,
        81.6,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q": [
        82.0,
        82.4,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k": [
        82.1,
        82.3,
        82.5,
        82.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v": [
        81.6,
        81.6,
        82.0,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        81.7,
        81.8,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q": [
        82.3,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k": [
        81.9,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        82.2,
        82.3,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        81.8,
        81.9,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2": [
        81.8,
        81.9,
        82.2,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k": [
        82.3,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v": [
        81.8,
        82.0,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        81.7,
        81.8,
        82.0,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        81.8,
        82.2,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2": [
        81.8,
        82.0,
        82.1,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q": [
        81.9,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k": [
        82.1,
        82.3,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v": [
        81.7,
        82.0,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        81.9,
        82.0,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        82.2,
        82.1,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        81.9,
        82.0,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2": [
        81.8,
        82.2,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q": [
        82.1,
        82.4,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v": [
        81.8,
        82.0,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        81.9,
        82.1,
        81.9,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q": [
        82.3,
        82.3,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k": [
        81.9,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        81.7,
        81.7,
        82.2,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2": [
        81.7,
        81.6,
        82.0,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k": [
        82.1,
        82.4,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v": [
        81.9,
        82.1,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        81.8,
        81.8,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q": [
        82.3,
        82.2,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k": [
        82.3,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v": [
        82.2,
        82.2,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        81.6,
        81.9,
        82.1,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2": [
        81.6,
        81.9,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k": [
        82.1,
        82.2,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v": [
        82.1,
        82.2,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        81.7,
        81.8,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k": [
        82.0,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        82.0,
        82.2,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        81.8,
        82.0,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2": [
        82.0,
        82.1,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v": [
        81.9,
        82.2,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        81.8,
        81.9,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v": [
        82.3,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        81.7,
        82.0,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2": [
        81.8,
        82.0,
        82.2,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k": [
        82.1,
        82.2,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v": [
        81.8,
        82.0,
        82.1,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        81.9,
        82.1,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k": [
        82.3,
        82.4,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v": [
        82.2,
        82.2,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        81.6,
        81.8,
        81.9,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2": [
        81.8,
        82.0,
        82.3,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q": [
        81.9,
        82.4,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v": [
        81.7,
        81.8,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        81.7,
        82.0,
        82.1,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q": [
        82.1,
        82.2,
        82.6,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v": [
        82.1,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        81.4,
        81.5,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2": [
        81.8,
        81.9,
        82.1,
        81.5
      ],
      "up_blocks.0.attentions.1.proj_out": [
        81.2,
        81.4,
        81.7,
        81.2
      ],
      "up_blocks.0.attentions.2.proj_in": [
        81.6,
        81.6,
        81.9,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v": [
        81.8,
        81.9,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        81.6,
        81.8,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q": [
        82.3,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v": [
        82.3,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        81.8,
        82.1,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2": [
        81.6,
        81.7,
        82.1,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k": [
        82.3,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v": [
        81.6,
        81.9,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        81.6,
        82.0,
        81.9,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k": [
        82.1,
        82.2,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v": [
        82.0,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        81.8,
        82.0,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2": [
        81.5,
        82.0,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q": [
        81.9,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v": [
        81.7,
        82.0,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0": [
        81.7,
        81.9,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q": [
        81.9,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0": [
        82.1,
        82.4,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj": [
        82.0,
        82.2,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2": [
        81.8,
        82.0,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q": [
        82.3,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v": [
        81.7,
        81.9,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0": [
        81.7,
        82.0,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0": [
        82.3,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj": [
        81.7,
        81.8,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2": [
        81.8,
        82.0,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v": [
        81.6,
        81.9,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0": [
        81.6,
        81.7,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k": [
        82.3,
        82.4,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v": [
        82.2,
        82.4,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0": [
        82.2,
        82.4,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj": [
        81.7,
        82.0,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2": [
        81.8,
        82.1,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q": [
        82.0,
        82.2,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v": [
        82.0,
        82.1,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0": [
        81.9,
        82.0,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q": [
        82.2,
        82.5,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v": [
        81.9,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0": [
        82.2,
        82.4,
        82.5,
        82.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj": [
        81.7,
        82.0,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2": [
        81.8,
        81.7,
        82.2,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v": [
        81.7,
        82.0,
        82.2,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0": [
        81.9,
        82.0,
        82.2,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v": [
        82.3,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj": [
        81.8,
        81.9,
        82.0,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2": [
        82.0,
        82.2,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v": [
        81.7,
        81.9,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0": [
        81.9,
        82.2,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k": [
        82.2,
        82.1,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj": [
        81.8,
        82.0,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2": [
        81.8,
        82.0,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v": [
        81.8,
        82.1,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0": [
        81.6,
        82.0,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q": [
        82.1,
        82.4,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj": [
        81.8,
        81.7,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2": [
        81.7,
        82.0,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q": [
        81.9,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k": [
        82.0,
        82.1,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v": [
        82.0,
        82.0,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0": [
        81.8,
        82.0,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q": [
        82.2,
        82.4,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0": [
        82.3,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj": [
        81.9,
        81.6,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2": [
        81.7,
        81.8,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.proj_out": [
        81.5,
        81.5,
        81.7,
        81.5
      ],
      "up_blocks.0.resnets.0.conv1": [
        81.9,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.resnets.0.conv2": [
        82.0,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        81.8,
        81.8,
        82.0,
        81.8
      ],
      "up_blocks.0.resnets.1.conv1": [
        82.1,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        82.0,
        82.2,
        82.2,
        82.1
      ],
      "up_blocks.0.resnets.1.conv2": [
        82.0,
        82.2,
        82.0,
        82.0
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        81.5,
        81.4,
        81.9,
        81.5
      ],
      "up_blocks.0.resnets.2.conv1": [
        82.0,
        82.0,
        82.3,
        82.1
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        82.0,
        82.3,
        82.3,
        82.3
      ],
      "up_blocks.0.resnets.2.conv2": [
        81.7,
        82.0,
        82.0,
        81.9
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        80.2,
        80.3,
        80.5,
        80.5
      ],
      "up_blocks.0.upsamplers.0.conv": [
        80.3,
        80.4,
        80.8,
        80.4
      ],
      "up_blocks.1.attentions.0.proj_in": [
        80.6,
        80.6,
        80.9,
        80.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        81.8,
        81.9,
        82.1,
        81.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        81.7,
        81.7,
        81.9,
        81.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        81.1,
        81.2,
        81.6,
        81.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        80.6,
        80.6,
        80.9,
        80.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.0,
        82.1,
        82.4,
        82.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.9,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.7,
        82.3,
        82.3,
        82.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        80.7,
        80.9,
        81.0,
        80.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        80.8,
        80.8,
        81.1,
        80.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        81.4,
        81.6,
        81.7,
        81.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        81.6,
        81.5,
        81.6,
        81.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        80.9,
        80.9,
        81.2,
        80.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        80.7,
        80.9,
        81.0,
        80.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.0,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        80.9,
        81.0,
        81.3,
        80.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        81.3,
        81.4,
        81.4,
        81.2
      ],
      "up_blocks.1.attentions.0.proj_out": [
        81.0,
        81.0,
        81.4,
        81.0
      ],
      "up_blocks.1.attentions.1.proj_in": [
        80.1,
        80.2,
        80.3,
        80.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        81.6,
        81.5,
        81.8,
        81.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        81.5,
        81.8,
        81.9,
        81.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        80.8,
        80.9,
        80.9,
        80.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        80.1,
        80.1,
        80.4,
        80.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        81.9,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        82.3,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        82.1,
        82.1,
        82.2,
        82.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        82.1,
        82.0,
        82.2,
        81.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        80.2,
        80.4,
        80.5,
        80.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        80.6,
        80.5,
        80.5,
        80.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        81.2,
        81.5,
        81.6,
        81.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        81.3,
        81.3,
        81.6,
        81.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        80.7,
        80.8,
        80.8,
        80.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        80.6,
        80.6,
        80.9,
        80.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        82.1,
        82.3,
        82.3,
        82.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        82.0,
        82.0,
        82.3,
        82.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        81.9,
        82.1,
        82.2,
        82.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        80.4,
        80.1,
        80.3,
        80.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        80.9,
        80.9,
        81.2,
        80.9
      ],
      "up_blocks.1.attentions.1.proj_out": [
        80.8,
        80.7,
        81.0,
        80.7
      ],
      "up_blocks.1.attentions.2.proj_in": [
        80.0,
        79.7,
        79.9,
        79.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        81.5,
        81.4,
        81.7,
        81.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        81.4,
        81.5,
        81.9,
        81.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        80.8,
        80.6,
        81.1,
        80.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        80.1,
        80.1,
        80.2,
        80.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        81.9,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        82.1,
        82.1,
        82.2,
        82.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        82.0,
        82.3,
        82.0,
        81.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        82.0,
        81.8,
        82.1,
        81.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        79.8,
        79.7,
        80.0,
        79.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        80.2,
        80.3,
        80.7,
        80.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q": [
        81.3,
        81.4,
        81.6,
        81.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k": [
        81.1,
        81.2,
        81.5,
        81.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v": [
        80.7,
        80.8,
        81.0,
        80.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        80.1,
        80.0,
        80.4,
        80.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q": [
        82.1,
        82.4,
        82.4,
        82.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k": [
        82.2,
        82.3,
        82.3,
        82.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v": [
        81.8,
        82.2,
        82.4,
        81.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        81.9,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        80.4,
        80.6,
        80.6,
        80.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2": [
        80.7,
        80.7,
        81.0,
        80.5
      ],
      "up_blocks.1.attentions.2.proj_out": [
        80.1,
        80.1,
        80.3,
        79.9
      ],
      "up_blocks.1.resnets.0.conv1": [
        81.5,
        81.4,
        81.6,
        81.5
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        82.0,
        82.2,
        82.2,
        82.0
      ],
      "up_blocks.1.resnets.0.conv2": [
        81.2,
        81.2,
        81.6,
        81.3
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        79.9,
        79.8,
        80.0,
        79.9
      ],
      "up_blocks.1.resnets.1.conv1": [
        80.4,
        80.4,
        80.7,
        80.3
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        81.3,
        81.3,
        81.6,
        81.1
      ],
      "up_blocks.1.resnets.1.conv2": [
        80.8,
        80.7,
        81.0,
        80.7
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        78.9,
        78.8,
        79.0,
        78.9
      ],
      "up_blocks.1.resnets.2.conv1": [
        80.3,
        80.1,
        80.5,
        80.2
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        81.1,
        81.3,
        81.5,
        81.2
      ],
      "up_blocks.1.resnets.2.conv2": [
        80.4,
        80.6,
        80.9,
        80.5
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        79.4,
        79.4,
        79.5,
        79.5
      ],
      "up_blocks.1.upsamplers.0.conv": [
        77.2,
        77.5,
        77.6,
        77.5
      ],
      "up_blocks.2.resnets.0.conv1": [
        77.6,
        77.6,
        77.8,
        77.5
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        80.7,
        81.1,
        81.0,
        81.0
      ],
      "up_blocks.2.resnets.0.conv2": [
        77.6,
        77.3,
        77.6,
        77.7
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        75.8,
        75.9,
        75.8,
        75.8
      ],
      "up_blocks.2.resnets.1.conv1": [
        73.7,
        73.7,
        74.1,
        73.9
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        80.8,
        81.0,
        81.0,
        80.8
      ],
      "up_blocks.2.resnets.1.conv2": [
        72.4,
        72.3,
        72.4,
        72.5
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        76.0,
        76.1,
        76.3,
        76.0
      ],
      "up_blocks.2.resnets.2.conv1": [
        73.0,
        73.3,
        73.3,
        73.3
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        81.3,
        81.4,
        81.5,
        81.3
      ],
      "up_blocks.2.resnets.2.conv2": [
        75.0,
        75.2,
        75.4,
        75.2
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        79.5,
        79.9,
        79.9,
        79.7
      ],
      "mid_block.attentions.0.proj_in": [
        81.9,
        82.2,
        82.2,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        81.9,
        82.3,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.4,
        82.4,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.2,
        82.3,
        82.5,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        82.0,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_q": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_k": [
        82.2,
        82.2,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_v": [
        82.1,
        82.3,
        82.2,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        82.0,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        81.9,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.2": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_v": [
        81.8,
        82.4,
        82.4,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_k": [
        82.2,
        82.3,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_v": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.2": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_q": [
        82.2,
        82.1,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_k": [
        82.1,
        82.2,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_v": [
        82.0,
        82.3,
        82.4,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_q": [
        82.1,
        82.2,
        82.3,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_k": [
        82.3,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_v": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        82.2,
        82.4,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.2": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_q": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_k": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_v": [
        82.1,
        82.3,
        82.3,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        82.1,
        82.2,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_k": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_v": [
        82.3,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        82.4,
        82.4,
        82.4,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        82.2,
        82.2,
        82.4,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.2": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_q": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_k": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_v": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_q": [
        82.2,
        82.2,
        82.4,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_k": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_v": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.2,
        82.4,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        81.9,
        82.1,
        82.5,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.2": [
        82.0,
        82.1,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_k": [
        82.2,
        82.0,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_v": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_q": [
        81.9,
        82.3,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.3,
        82.3,
        82.5,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_v": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        82.3,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.2": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_q": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_k": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_v": [
        82.0,
        82.1,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.3,
        82.2,
        82.4,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        82.2,
        82.2,
        82.4,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.2": [
        82.0,
        82.2,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_v": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_q": [
        82.2,
        82.2,
        82.5,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_v": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        82.0,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.2": [
        82.2,
        82.1,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_q": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_k": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_v": [
        82.1,
        82.1,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_q": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_k": [
        82.0,
        82.2,
        82.4,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_v": [
        82.2,
        82.4,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        82.2,
        82.1,
        82.4,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        82.0,
        82.1,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.2": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.proj_out": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.resnets.0.conv1": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.resnets.0.time_emb_proj": [
        82.2,
        82.4,
        82.3,
        82.3
      ],
      "mid_block.resnets.0.conv2": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.resnets.1.conv1": [
        81.9,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.resnets.1.time_emb_proj": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.resnets.1.conv2": [
        82.1,
        82.3,
        82.4,
        82.0
      ]
    },
    "8": {
      "time_embedding.linear_1": [
        81.3,
        81.4,
        81.7,
        81.5
      ],
      "time_embedding.linear_2": [
        81.0,
        80.8,
        81.1,
        81.0
      ],
      "add_embedding.linear_1": [
        81.5,
        81.3,
        81.7,
        81.2
      ],
      "add_embedding.linear_2": [
        81.2,
        81.2,
        81.7,
        81.3
      ],
      "down_blocks.0.resnets.0.conv1": [
        80.7,
        80.9,
        81.0,
        80.9
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        81.1,
        80.8,
        81.0,
        81.1
      ],
      "down_blocks.0.resnets.0.conv2": [
        80.4,
        80.3,
        80.5,
        80.3
      ],
      "down_blocks.0.resnets.1.conv1": [
        79.6,
        79.7,
        79.6,
        79.9
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        81.0,
        81.0,
        81.4,
        81.1
      ],
      "down_blocks.0.resnets.1.conv2": [
        80.3,
        80.6,
        80.9,
        80.3
      ],
      "down_blocks.0.downsamplers.0.conv": [
        81.2,
        81.4,
        81.4,
        81.2
      ],
      "down_blocks.1.attentions.0.proj_in": [
        81.5,
        81.4,
        81.8,
        81.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        82.1,
        82.1,
        82.3,
        81.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        81.9,
        82.1,
        82.1,
        82.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        81.7,
        82.1,
        82.0,
        82.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        81.7,
        81.9,
        82.0,
        81.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.9,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        81.4,
        81.4,
        81.8,
        81.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        81.5,
        81.6,
        81.9,
        81.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        82.0,
        82.1,
        82.2,
        82.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        81.9,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        81.7,
        81.8,
        81.9,
        81.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        81.7,
        81.7,
        82.0,
        81.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        82.0,
        82.1,
        82.3,
        82.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.0,
        82.3,
        82.4,
        81.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        82.2,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        81.1,
        81.3,
        81.6,
        81.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        81.8,
        81.9,
        81.9,
        81.8
      ],
      "down_blocks.1.attentions.0.proj_out": [
        81.5,
        81.6,
        81.8,
        81.5
      ],
      "down_blocks.1.attentions.1.proj_in": [
        81.7,
        81.9,
        82.1,
        81.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        82.0,
        82.1,
        82.1,
        81.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        82.0,
        82.1,
        82.1,
        81.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        82.0,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        81.8,
        81.9,
        81.9,
        81.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        81.8,
        81.9,
        82.1,
        81.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        82.0,
        82.1,
        82.4,
        82.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        82.0,
        82.2,
        82.4,
        82.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        81.9,
        82.1,
        82.2,
        82.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        81.9,
        82.1,
        82.2,
        82.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        81.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        82.2,
        82.2,
        82.5,
        82.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        81.6,
        82.0,
        82.0,
        81.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        81.9,
        82.1,
        82.0,
        81.8
      ],
      "down_blocks.1.attentions.1.proj_out": [
        81.8,
        81.8,
        82.0,
        81.7
      ],
      "down_blocks.1.resnets.0.conv1": [
        81.7,
        81.8,
        81.7,
        81.5
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        81.7,
        82.2,
        82.1,
        81.9
      ],
      "down_blocks.1.resnets.0.conv2": [
        82.0,
        82.1,
        81.9,
        81.6
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        81.2,
        81.2,
        81.4,
        81.2
      ],
      "down_blocks.1.resnets.1.conv1": [
        81.8,
        82.0,
        82.2,
        82.1
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.1.resnets.1.conv2": [
        82.0,
        82.1,
        82.1,
        81.8
      ],
      "down_blocks.1.downsamplers.0.conv": [
        81.8,
        82.0,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.proj_in": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        82.0,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        81.9,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.1,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.2,
        82.2,
        82.5,
        82.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2": [
        82.2,
        82.2,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k": [
        82.2,
        82.3,
        82.5,
        82.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v": [
        82.2,
        82.4,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2": [
        82.0,
        82.2,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k": [
        81.8,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v": [
        82.0,
        82.4,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q": [
        82.2,
        82.2,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        82.0,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        82.2,
        82.2,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        81.9,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q": [
        82.2,
        82.4,
        82.6,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q": [
        82.3,
        82.3,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k": [
        82.3,
        82.2,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v": [
        82.1,
        82.2,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        82.3,
        82.3,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        82.3,
        82.3,
        82.3,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v": [
        82.0,
        82.4,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        82.1,
        82.4,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        82.0,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        82.2,
        82.2,
        82.3,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2": [
        82.1,
        82.3,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q": [
        82.0,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k": [
        82.3,
        82.1,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v": [
        82.2,
        82.3,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        82.4,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k": [
        82.3,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        82.3,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        81.9,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.proj_out": [
        81.9,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.proj_in": [
        82.1,
        82.2,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        82.1,
        82.3,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        82.1,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q": [
        82.3,
        82.4,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v": [
        82.0,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q": [
        82.1,
        82.3,
        82.2,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v": [
        82.3,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2": [
        82.1,
        82.2,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k": [
        82.3,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        82.1,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v": [
        82.3,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        82.3,
        82.2,
        82.3,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k": [
        82.2,
        82.4,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        82.3,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2": [
        82.2,
        82.2,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k": [
        82.3,
        82.3,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v": [
        82.3,
        82.2,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        82.2,
        82.4,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        82.0,
        82.2,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2": [
        82.0,
        82.3,
        82.3,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q": [
        82.0,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v": [
        82.2,
        82.3,
        82.6,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        82.0,
        82.3,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k": [
        82.3,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        82.2,
        82.2,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2": [
        82.2,
        82.4,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k": [
        82.1,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        82.1,
        82.4,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q": [
        82.3,
        82.3,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v": [
        82.3,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2": [
        82.3,
        82.4,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2": [
        82.3,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q": [
        81.9,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k": [
        82.3,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v": [
        82.1,
        82.3,
        82.5,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        82.3,
        82.4,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2": [
        82.1,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q": [
        81.9,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        82.3,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.proj_out": [
        82.2,
        82.4,
        82.6,
        82.1
      ],
      "down_blocks.2.resnets.0.conv1": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.resnets.0.conv2": [
        81.9,
        82.2,
        82.4,
        82.3
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        81.9,
        82.0,
        82.2,
        81.7
      ],
      "down_blocks.2.resnets.1.conv1": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.resnets.1.conv2": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.proj_in": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        81.9,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        82.1,
        82.3,
        82.5,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.4,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.0,
        82.2,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.9,
        82.4,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.9,
        82.4,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q": [
        82.3,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k": [
        82.1,
        82.3,
        82.5,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v": [
        82.3,
        82.4,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k": [
        82.2,
        82.2,
        82.5,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        82.2,
        82.4,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        82.0,
        82.4,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k": [
        82.3,
        82.2,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q": [
        82.2,
        82.3,
        82.2,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k": [
        82.2,
        82.2,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        82.3,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        82.3,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2": [
        82.3,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        81.9,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k": [
        82.3,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v": [
        82.2,
        82.4,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        82.3,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q": [
        82.2,
        82.4,
        82.5,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        82.2,
        82.4,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q": [
        82.3,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k": [
        82.0,
        82.3,
        82.6,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.2,
        82.3,
        82.5,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        82.3,
        82.4,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2": [
        81.9,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q": [
        82.0,
        82.1,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        82.3,
        82.4,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2": [
        82.3,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q": [
        82.0,
        82.4,
        82.5,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k": [
        82.2,
        82.2,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v": [
        82.3,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.2,
        82.4,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.3,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v": [
        82.1,
        82.3,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        82.3,
        82.3,
        82.2,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q": [
        82.2,
        82.4,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k": [
        82.2,
        82.2,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        82.3,
        82.4,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        82.0,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q": [
        81.9,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k": [
        82.3,
        82.2,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q": [
        82.3,
        82.3,
        82.6,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v": [
        82.2,
        82.2,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        82.3,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.proj_out": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.proj_in": [
        81.9,
        82.3,
        82.2,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        82.2,
        82.2,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        82.0,
        82.1,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        81.9,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        82.0,
        82.3,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q": [
        82.2,
        82.2,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        81.9,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q": [
        82.1,
        82.4,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k": [
        82.1,
        82.4,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        82.1,
        82.3,
        82.4,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2": [
        82.0,
        82.2,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v": [
        82.0,
        82.1,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        81.9,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q": [
        82.3,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        82.0,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q": [
        82.4,
        82.4,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v": [
        82.0,
        82.4,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        81.9,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        81.9,
        82.2,
        82.4,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q": [
        82.0,
        82.4,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k": [
        82.2,
        82.3,
        82.5,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        82.3,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q": [
        82.0,
        82.4,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k": [
        82.2,
        82.1,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v": [
        82.3,
        82.4,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        82.1,
        82.2,
        82.4,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q": [
        82.0,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v": [
        81.9,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k": [
        82.1,
        82.2,
        82.2,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v": [
        82.3,
        82.4,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        82.0,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        81.9,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2": [
        82.1,
        82.1,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q": [
        82.0,
        82.2,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q": [
        82.2,
        82.2,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        82.1,
        82.2,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2": [
        82.2,
        82.4,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q": [
        82.2,
        82.2,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        82.0,
        82.2,
        82.2,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k": [
        82.2,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        81.9,
        82.1,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2": [
        81.9,
        82.3,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q": [
        82.2,
        82.4,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        82.1,
        82.2,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q": [
        82.2,
        82.2,
        82.6,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k": [
        82.2,
        82.3,
        82.6,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v": [
        82.2,
        82.3,
        82.5,
        82.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        82.0,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2": [
        82.1,
        82.3,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q": [
        82.3,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        82.1,
        82.0,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2": [
        82.0,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.1.proj_out": [
        81.7,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.proj_in": [
        81.7,
        82.2,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q": [
        82.0,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k": [
        82.0,
        82.4,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v": [
        82.0,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        81.9,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.4,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k": [
        82.2,
        82.4,
        82.6,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        82.0,
        82.2,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2": [
        81.9,
        82.1,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v": [
        82.0,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v": [
        82.2,
        82.4,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2": [
        82.0,
        82.1,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q": [
        82.1,
        82.4,
        82.5,
        82.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k": [
        82.3,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v": [
        81.9,
        82.2,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0": [
        81.9,
        82.2,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k": [
        82.4,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v": [
        82.3,
        82.4,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj": [
        82.2,
        82.1,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q": [
        82.0,
        82.1,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k": [
        82.2,
        82.3,
        82.5,
        82.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v": [
        82.1,
        82.3,
        82.4,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0": [
        82.2,
        82.4,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0": [
        82.3,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj": [
        81.8,
        82.2,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2": [
        81.9,
        82.3,
        82.4,
        82.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k": [
        82.2,
        82.3,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v": [
        81.8,
        82.1,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q": [
        82.2,
        82.4,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k": [
        82.1,
        82.4,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj": [
        82.0,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2": [
        82.0,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k": [
        82.3,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0": [
        81.9,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q": [
        82.1,
        82.4,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj": [
        81.9,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q": [
        82.2,
        82.4,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v": [
        82.1,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj": [
        81.9,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2": [
        82.1,
        82.3,
        82.5,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q": [
        82.3,
        82.3,
        82.6,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k": [
        82.4,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v": [
        82.1,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0": [
        82.0,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v": [
        82.2,
        82.3,
        82.6,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj": [
        82.1,
        82.4,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q": [
        82.3,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k": [
        82.3,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v": [
        82.3,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0": [
        82.0,
        82.1,
        82.4,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q": [
        82.1,
        82.4,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj": [
        82.0,
        82.2,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2": [
        82.1,
        82.3,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k": [
        82.2,
        82.2,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v": [
        81.9,
        82.1,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q": [
        82.2,
        82.4,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v": [
        81.9,
        82.4,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj": [
        82.2,
        82.3,
        82.4,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.proj_out": [
        81.8,
        82.0,
        82.0,
        81.6
      ],
      "up_blocks.0.resnets.0.conv1": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.resnets.0.conv2": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        82.0,
        82.1,
        82.3,
        82.1
      ],
      "up_blocks.0.resnets.1.conv1": [
        81.9,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.resnets.1.conv2": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        81.9,
        82.0,
        82.3,
        82.0
      ],
      "up_blocks.0.resnets.2.conv1": [
        81.9,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.resnets.2.conv2": [
        82.0,
        82.2,
        82.2,
        82.2
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        81.4,
        81.5,
        81.8,
        81.6
      ],
      "up_blocks.0.upsamplers.0.conv": [
        81.5,
        81.2,
        81.7,
        81.2
      ],
      "up_blocks.1.attentions.0.proj_in": [
        81.2,
        81.4,
        81.8,
        81.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        82.0,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        82.2,
        82.1,
        82.4,
        82.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        81.7,
        82.0,
        82.0,
        81.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        81.6,
        81.7,
        81.7,
        81.3
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        82.2,
        82.3,
        82.3,
        82.3
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        81.7,
        81.7,
        82.0,
        81.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        81.6,
        81.9,
        81.9,
        81.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        81.7,
        82.1,
        82.1,
        82.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        81.9,
        82.1,
        82.2,
        82.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        81.6,
        81.8,
        82.1,
        81.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        81.6,
        81.5,
        81.9,
        81.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        81.7,
        81.8,
        81.8,
        81.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        81.7,
        81.5,
        82.0,
        81.9
      ],
      "up_blocks.1.attentions.0.proj_out": [
        81.8,
        81.8,
        81.9,
        81.6
      ],
      "up_blocks.1.attentions.1.proj_in": [
        81.3,
        81.3,
        81.5,
        81.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        81.9,
        82.1,
        82.3,
        81.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        82.0,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        81.5,
        81.2,
        81.9,
        81.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        81.5,
        81.4,
        81.7,
        81.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.1,
        82.4,
        82.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        81.5,
        81.3,
        81.6,
        81.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        81.6,
        81.7,
        81.8,
        81.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        81.9,
        82.0,
        82.0,
        81.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        82.0,
        81.9,
        82.3,
        82.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        81.6,
        81.8,
        81.8,
        81.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        81.5,
        81.7,
        81.8,
        81.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        82.0,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        82.0,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        81.4,
        81.6,
        81.7,
        81.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        81.8,
        81.7,
        81.9,
        81.7
      ],
      "up_blocks.1.attentions.1.proj_out": [
        81.6,
        81.4,
        81.8,
        81.6
      ],
      "up_blocks.1.attentions.2.proj_in": [
        81.3,
        81.1,
        81.4,
        81.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        82.0,
        82.1,
        82.2,
        82.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        82.1,
        82.2,
        82.4,
        81.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        81.4,
        81.4,
        81.7,
        81.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        81.1,
        81.1,
        81.7,
        81.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        82.1,
        82.1,
        82.3,
        82.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        81.2,
        81.4,
        81.6,
        81.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        81.5,
        81.4,
        81.9,
        81.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q": [
        81.8,
        82.0,
        82.0,
        81.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k": [
        81.9,
        82.2,
        82.2,
        81.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v": [
        81.1,
        81.3,
        81.4,
        81.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        81.2,
        81.4,
        81.8,
        81.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q": [
        82.3,
        82.3,
        82.2,
        82.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        82.0,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        81.5,
        81.5,
        81.6,
        81.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2": [
        81.3,
        81.6,
        81.6,
        81.7
      ],
      "up_blocks.1.attentions.2.proj_out": [
        81.3,
        81.3,
        81.4,
        81.2
      ],
      "up_blocks.1.resnets.0.conv1": [
        81.9,
        82.1,
        82.2,
        82.1
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.1.resnets.0.conv2": [
        81.6,
        82.0,
        82.1,
        81.7
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        81.3,
        81.3,
        81.5,
        81.2
      ],
      "up_blocks.1.resnets.1.conv1": [
        81.4,
        81.7,
        81.7,
        81.5
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        81.8,
        81.9,
        82.3,
        81.8
      ],
      "up_blocks.1.resnets.1.conv2": [
        81.8,
        81.8,
        81.8,
        81.6
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        80.9,
        81.1,
        81.2,
        81.0
      ],
      "up_blocks.1.resnets.2.conv1": [
        81.4,
        81.6,
        81.6,
        81.3
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        81.8,
        81.7,
        82.1,
        81.6
      ],
      "up_blocks.1.resnets.2.conv2": [
        81.4,
        81.5,
        81.7,
        81.3
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        81.0,
        80.9,
        81.0,
        80.8
      ],
      "up_blocks.1.upsamplers.0.conv": [
        80.9,
        80.8,
        81.0,
        80.9
      ],
      "up_blocks.2.resnets.0.conv1": [
        80.3,
        80.3,
        80.7,
        80.5
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        81.4,
        81.5,
        81.7,
        81.5
      ],
      "up_blocks.2.resnets.0.conv2": [
        80.7,
        80.9,
        80.9,
        80.8
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        80.4,
        80.5,
        80.5,
        80.3
      ],
      "up_blocks.2.resnets.1.conv1": [
        80.3,
        80.4,
        80.2,
        80.4
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        81.5,
        81.6,
        81.6,
        81.5
      ],
      "up_blocks.2.resnets.1.conv2": [
        80.6,
        80.8,
        81.0,
        80.9
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        80.6,
        81.0,
        80.9,
        80.7
      ],
      "up_blocks.2.resnets.2.conv1": [
        80.2,
        80.2,
        80.3,
        80.3
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        81.7,
        81.8,
        82.1,
        81.8
      ],
      "up_blocks.2.resnets.2.conv2": [
        81.0,
        81.0,
        81.2,
        81.0
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        81.7,
        81.6,
        82.0,
        81.5
      ],
      "mid_block.attentions.0.proj_in": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        82.0,
        82.3,
        82.6,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        82.2,
        82.2,
        82.3,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        82.0,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.3,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        82.1,
        82.4,
        82.3,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        82.1,
        82.2,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_q": [
        82.0,
        82.4,
        82.3,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_k": [
        82.3,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_v": [
        82.2,
        82.2,
        82.5,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.2,
        82.2,
        82.2,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        82.3,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.2": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_q": [
        81.9,
        82.3,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_k": [
        82.2,
        82.4,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_v": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        82.3,
        82.4,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_q": [
        82.3,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        82.2,
        82.4,
        82.4,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        82.2,
        82.2,
        82.4,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.2": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_q": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_k": [
        82.1,
        82.2,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_v": [
        82.3,
        82.2,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_q": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_k": [
        82.2,
        82.2,
        82.5,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_v": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        82.3,
        82.3,
        82.4,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.2": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_k": [
        82.0,
        82.4,
        82.7,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_v": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        82.1,
        82.4,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_q": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_k": [
        81.9,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_v": [
        82.3,
        82.4,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        82.1,
        82.4,
        82.6,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        82.2,
        82.3,
        82.5,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.2": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_k": [
        82.3,
        82.4,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_v": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_q": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_k": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_v": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        82.3,
        82.4,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.2": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_q": [
        82.3,
        82.4,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_k": [
        82.3,
        82.4,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_v": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        82.2,
        82.3,
        82.5,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_q": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_v": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.2": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_q": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_k": [
        82.2,
        82.4,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_v": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        82.3,
        82.2,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.3,
        82.4,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_v": [
        82.1,
        82.4,
        82.5,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.2": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_q": [
        82.3,
        82.3,
        82.6,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_k": [
        82.2,
        82.2,
        82.3,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_v": [
        82.0,
        82.2,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_q": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_k": [
        82.1,
        82.2,
        82.5,
        82.3
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        82.1,
        82.4,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        82.2,
        82.2,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.2": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_q": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_k": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_v": [
        82.3,
        82.4,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_q": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_v": [
        82.2,
        82.2,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        82.3,
        82.2,
        82.4,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.2": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.proj_out": [
        81.9,
        82.3,
        82.5,
        82.1
      ],
      "mid_block.resnets.0.conv1": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.resnets.0.time_emb_proj": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.resnets.0.conv2": [
        82.3,
        82.3,
        82.4,
        82.0
      ],
      "mid_block.resnets.1.conv1": [
        82.2,
        82.3,
        82.5,
        82.0
      ],
      "mid_block.resnets.1.time_emb_proj": [
        82.1,
        82.4,
        82.5,
        82.0
      ],
      "mid_block.resnets.1.conv2": [
        82.3,
        82.3,
        82.2,
        82.1
      ]
    }
  },
  "cumulative": {
    "1": {
      "metadata": {
        "nbits": 1,
        "out_ngroups": 1,
        "in_ngroups": 1,
        "cumulative": true,
        "candidates": [
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "down_blocks.2.resnets.1.time_emb_proj",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.7.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.8.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0",
          "mid_block.resnets.1.time_emb_proj",
          "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.6.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.5.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.2.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.4.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v",
          "up_blocks.0.resnets.0.time_emb_proj",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.9.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.3.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj",
          "mid_block.resnets.0.time_emb_proj",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.1.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_v",
          "mid_block.resnets.1.conv1",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v",
          "down_blocks.2.attentions.1.proj_in",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_k",
          "mid_block.resnets.1.conv2",
          "down_blocks.2.resnets.1.conv1",
          "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v",
          "up_blocks.0.resnets.0.conv1",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v",
          "down_blocks.2.resnets.1.conv2",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2",
          "up_blocks.0.resnets.1.time_emb_proj",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2",
          "mid_block.resnets.0.conv1",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2",
          "down_blocks.2.resnets.0.time_emb_proj",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj",
          "mid_block.resnets.0.conv2",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.resnets.1.conv2",
          "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v",
          "down_blocks.2.attentions.1.proj_out",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0",
          "up_blocks.0.resnets.2.time_emb_proj",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v",
          "mid_block.attentions.0.proj_out",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.resnets.1.conv1",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.resnets.0.conv2",
          "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2",
          "up_blocks.0.resnets.2.conv2",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj",
          "up_blocks.0.attentions.0.proj_out",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2",
          "up_blocks.0.resnets.2.conv1",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0",
          "down_blocks.2.resnets.0.conv2",
          "up_blocks.0.resnets.0.conv_shortcut",
          "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj",
          "down_blocks.2.attentions.0.proj_in",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.0.proj_in",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.2.attentions.0.proj_out",
          "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2",
          "mid_block.attentions.0.proj_in",
          "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q",
          "up_blocks.0.attentions.2.proj_in",
          "down_blocks.2.resnets.0.conv1",
          "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj",
          "up_blocks.1.resnets.0.time_emb_proj",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.2.proj_out",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q",
          "add_embedding.linear_2",
          "down_blocks.1.resnets.1.time_emb_proj",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.2.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.attn1.to_k",
          "up_blocks.0.resnets.1.conv_shortcut",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.1.proj_in",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
          "add_embedding.linear_1",
          "up_blocks.1.resnets.0.conv1",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2",
          "up_blocks.1.resnets.0.conv2",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2",
          "up_blocks.0.attentions.1.proj_out",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "down_blocks.1.resnets.0.time_emb_proj",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.resnets.1.conv2",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.0.proj_out",
          "down_blocks.1.attentions.1.proj_out",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k",
          "down_blocks.1.resnets.1.conv1",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v",
          "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v",
          "down_blocks.2.resnets.0.conv_shortcut",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "down_blocks.1.resnets.0.conv2",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q",
          "down_blocks.1.attentions.1.proj_in",
          "up_blocks.1.resnets.1.time_emb_proj",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.resnets.2.conv_shortcut",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v",
          "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2",
          "up_blocks.1.attentions.0.proj_in",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.1.downsamplers.0.conv",
          "up_blocks.1.resnets.2.time_emb_proj",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.1.proj_out",
          "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.upsamplers.0.conv",
          "down_blocks.1.attentions.0.proj_out",
          "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.resnets.1.conv2",
          "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.1.resnets.2.conv2",
          "up_blocks.1.resnets.1.conv1",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k",
          "up_blocks.1.attentions.2.proj_out",
          "up_blocks.1.attentions.1.proj_in",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.resnets.0.conv1",
          "up_blocks.1.attentions.2.proj_in",
          "up_blocks.1.resnets.2.conv1",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "down_blocks.0.resnets.1.time_emb_proj",
          "up_blocks.2.resnets.0.time_emb_proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.attentions.0.proj_in",
          "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.2.resnets.1.time_emb_proj",
          "up_blocks.2.resnets.2.time_emb_proj",
          "down_blocks.0.resnets.0.time_emb_proj",
          "up_blocks.1.resnets.0.conv_shortcut",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k",
          "down_blocks.0.resnets.1.conv2",
          "down_blocks.0.resnets.0.conv1",
          "up_blocks.1.resnets.1.conv_shortcut",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.0.resnets.1.conv1",
          "down_blocks.1.resnets.0.conv_shortcut",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q",
          "down_blocks.0.resnets.0.conv2",
          "down_blocks.0.downsamplers.0.conv",
          "up_blocks.2.resnets.0.conv1",
          "up_blocks.2.resnets.0.conv2",
          "up_blocks.1.resnets.2.conv_shortcut",
          "up_blocks.2.resnets.1.conv2",
          "time_embedding.linear_2",
          "up_blocks.2.resnets.1.conv1",
          "up_blocks.1.upsamplers.0.conv",
          "time_embedding.linear_1",
          "up_blocks.2.resnets.0.conv_shortcut",
          "up_blocks.2.resnets.2.conv1",
          "up_blocks.2.resnets.1.conv_shortcut",
          "up_blocks.2.resnets.2.conv2",
          "up_blocks.2.resnets.2.conv_shortcut"
        ],
        "sizes": [
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          6553600,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          6553600,
          1638400,
          6553600,
          6553600,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          6553600,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          2621440,
          2621440,
          6553600,
          6553600,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          6553600,
          13107200,
          13107200,
          1638400,
          13107200,
          1638400,
          2621440,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          2621440,
          1638400,
          13107200,
          1638400,
          13107200,
          1638400,
          1638400,
          6553600,
          6553600,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          13107200,
          6553600,
          13107200,
          6553600,
          1638400,
          1638400,
          6553600,
          6553600,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          13107200,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          2621440,
          1638400,
          13107200,
          6553600,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          6553600,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          6553600,
          1638400,
          13107200,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          13107200,
          13107200,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          6553600,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          13107200,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          14745600,
          1638400,
          1638400,
          13107200,
          6553600,
          1638400,
          6553600,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          409600,
          6553600,
          13107200,
          1638400,
          2621440,
          13107200,
          13107200,
          1638400,
          1638400,
          13107200,
          2621440,
          1638400,
          6553600,
          2621440,
          1638400,
          1638400,
          14745600,
          14745600,
          13107200,
          2621440,
          29491200,
          2621440,
          14745600,
          1638400,
          1310720,
          1638400,
          1638400,
          1638400,
          1310720,
          6553600,
          2621440,
          1638400,
          6553600,
          1638400,
          13107200,
          1638400,
          6553600,
          13107200,
          13107200,
          1638400,
          6553600,
          6553600,
          1638400,
          1638400,
          6553600,
          14745600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          14745600,
          13107200,
          14745600,
          6553600,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          13107200,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          6553600,
          1638400,
          13107200,
          6553600,
          2621440,
          29491200,
          1638400,
          1638400,
          1638400,
          1638400,
          409600,
          1638400,
          14745600,
          13107200,
          13107200,
          6553600,
          14745600,
          1638400,
          1638400,
          13107200,
          6553600,
          1638400,
          1310720,
          1638400,
          6553600,
          13107200,
          6553600,
          6553600,
          1310720,
          13107200,
          1638400,
          409600,
          409600,
          6553600,
          22118400,
          1638400,
          14745600,
          3276800,
          6553600,
          1310720,
          1638400,
          13107200,
          13107200,
          1310720,
          13107200,
          13107200,
          2621440,
          6553600,
          409600,
          1638400,
          13107200,
          1638400,
          409600,
          1638400,
          1310720,
          1310720,
          13107200,
          13107200,
          13107200,
          6553600,
          1638400,
          6553600,
          1638400,
          13107200,
          409600,
          1638400,
          1638400,
          6553600,
          409600,
          409600,
          409600,
          1310720,
          409600,
          409600,
          409600,
          2621440,
          409600,
          409600,
          1638400,
          7372800,
          13107200,
          409600,
          1310720,
          1310720,
          13107200,
          409600,
          13107200,
          13107200,
          819200,
          409600,
          1638400,
          409600,
          1638400,
          819200,
          409600,
          409600,
          1310720,
          409600,
          3276800,
          409600,
          1638400,
          2621440,
          1310720,
          409600,
          3604480,
          11059200,
          409600,
          409600,
          1638400,
          3686400,
          409600,
          1638400,
          1638400,
          409600,
          1638400,
          409600,
          409600,
          819200,
          409600,
          3686400,
          409600,
          1638400,
          409600,
          409600,
          409600,
          409600,
          409600,
          3686400,
          409600,
          409600,
          409600,
          409600,
          1310720,
          3276800,
          409600,
          409600,
          409600,
          819200,
          409600,
          3276800,
          409600,
          3686400,
          1310720,
          409600,
          3276800,
          409600,
          409600,
          819200,
          409600,
          3276800,
          2457600,
          409600,
          1638400,
          409600,
          1638400,
          3686400,
          819200,
          409600,
          1638400,
          409600,
          409600,
          409600,
          1638400,
          1638400,
          1638400,
          409600,
          409600,
          1310720,
          1310720,
          409600,
          3276800,
          409600,
          14745600,
          409600,
          409600,
          409600,
          3686400,
          3276800,
          3686400,
          7372800,
          1310720,
          409600,
          409600,
          3276800,
          1843200,
          409600,
          5529600,
          3276800,
          3276800,
          409600,
          409600,
          409600,
          409600,
          3276800,
          409600,
          409600,
          409600,
          409600,
          1228800,
          1310720,
          921600,
          921600,
          819200,
          1310720,
          921600,
          204800,
          409600,
          921600,
          921600,
          2764800,
          921600,
          614400,
          921600,
          1638400,
          1843200,
          3686400,
          409600,
          307200,
          1843200,
          204800,
          921600,
          204800
        ]
      },
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_v": [
        82.0,
        82.5,
        82.4,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        81.9,
        82.1,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.1,
        82.1,
        82.2,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_v": [
        82.2,
        82.1,
        82.2,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_q": [
        81.9,
        82.1,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_v": [
        82.0,
        82.0,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        81.8,
        81.9,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v": [
        81.9,
        82.0,
        82.3,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        81.8,
        82.0,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_q": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        81.9,
        81.9,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_q": [
        82.0,
        82.0,
        82.2,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        81.8,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        81.8,
        81.9,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v": [
        81.9,
        82.0,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        81.8,
        82.1,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        81.7,
        82.0,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        81.8,
        81.8,
        82.2,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q": [
        81.7,
        81.8,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_v": [
        81.7,
        81.8,
        82.2,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_v": [
        81.7,
        81.8,
        82.1,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_v": [
        81.7,
        81.8,
        82.1,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        81.7,
        81.8,
        82.0,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        81.8,
        81.9,
        82.0,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_k": [
        81.7,
        81.8,
        82.1,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q": [
        81.7,
        81.8,
        81.8,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q": [
        81.8,
        81.6,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q": [
        81.7,
        81.3,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q": [
        81.6,
        81.6,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        81.6,
        81.5,
        82.1,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v": [
        81.5,
        81.4,
        81.9,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_q": [
        81.6,
        81.4,
        82.0,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        81.6,
        81.4,
        82.0,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q": [
        81.5,
        81.3,
        81.9,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k": [
        81.5,
        81.3,
        81.9,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v": [
        81.6,
        81.3,
        81.9,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v": [
        81.4,
        81.3,
        81.9,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        81.5,
        81.3,
        81.8,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_v": [
        81.5,
        81.4,
        81.8,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q": [
        81.4,
        81.4,
        81.8,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q": [
        81.4,
        81.3,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v": [
        81.5,
        81.5,
        81.8,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k": [
        81.5,
        81.3,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v": [
        81.4,
        81.3,
        81.6,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q": [
        81.4,
        81.4,
        81.6,
        81.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v": [
        81.4,
        81.3,
        81.5,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q": [
        81.4,
        81.4,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        81.4,
        81.3,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k": [
        81.4,
        81.3,
        81.3,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        81.3,
        81.3,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        81.4,
        81.1,
        81.3,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_q": [
        81.4,
        81.2,
        81.3,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_k": [
        81.3,
        81.2,
        81.3,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_q": [
        81.2,
        81.2,
        81.3,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_k": [
        81.3,
        81.2,
        81.2,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_k": [
        81.3,
        81.2,
        81.2,
        81.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        81.0,
        81.2,
        81.1,
        81.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q": [
        81.2,
        81.2,
        81.2,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        81.2,
        81.0,
        81.2,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k": [
        81.2,
        81.1,
        81.1,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v": [
        81.0,
        81.1,
        81.1,
        81.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k": [
        81.2,
        80.7,
        81.2,
        81.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q": [
        81.0,
        80.7,
        81.0,
        80.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q": [
        81.1,
        80.8,
        81.0,
        80.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v": [
        81.0,
        80.5,
        81.0,
        81.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q": [
        80.9,
        80.6,
        81.0,
        81.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q": [
        81.1,
        80.5,
        81.0,
        80.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k": [
        80.9,
        80.5,
        80.8,
        80.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q": [
        80.9,
        80.6,
        80.9,
        80.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k": [
        80.8,
        80.5,
        80.8,
        80.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v": [
        80.9,
        80.4,
        80.7,
        80.8
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        80.8,
        80.4,
        80.7,
        80.7
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        80.8,
        80.5,
        80.8,
        80.7
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_k": [
        80.7,
        80.5,
        80.7,
        80.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k": [
        80.7,
        80.4,
        80.7,
        80.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q": [
        80.6,
        80.4,
        80.6,
        80.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q": [
        80.6,
        80.3,
        80.6,
        80.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        80.6,
        80.3,
        80.6,
        80.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q": [
        80.5,
        80.4,
        80.3,
        80.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q": [
        80.6,
        80.4,
        80.5,
        80.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k": [
        80.6,
        80.4,
        80.6,
        80.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k": [
        80.5,
        80.3,
        80.5,
        80.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k": [
        80.6,
        80.3,
        80.5,
        80.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        80.5,
        80.3,
        80.4,
        80.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v": [
        80.6,
        80.2,
        80.3,
        80.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k": [
        80.7,
        80.1,
        80.4,
        80.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v": [
        80.5,
        80.2,
        80.1,
        80.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q": [
        80.6,
        80.0,
        80.2,
        80.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k": [
        80.5,
        80.1,
        80.2,
        80.3
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        80.5,
        80.1,
        80.1,
        80.3
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_v": [
        80.5,
        80.2,
        80.2,
        80.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_q": [
        80.5,
        80.2,
        80.1,
        80.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_q": [
        80.5,
        80.1,
        80.1,
        80.2
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_k": [
        80.4,
        80.0,
        80.0,
        80.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v": [
        80.4,
        79.9,
        80.0,
        80.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q": [
        80.3,
        80.1,
        79.9,
        80.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        80.4,
        80.1,
        80.0,
        80.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k": [
        80.4,
        80.0,
        79.9,
        80.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v": [
        80.3,
        79.9,
        79.8,
        80.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v": [
        80.4,
        80.0,
        79.9,
        80.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v": [
        80.4,
        80.0,
        79.9,
        80.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        80.4,
        80.1,
        80.0,
        80.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        80.3,
        80.0,
        79.8,
        80.1
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_q": [
        80.3,
        80.0,
        79.7,
        80.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_k": [
        80.4,
        80.0,
        79.9,
        80.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k": [
        80.3,
        80.1,
        79.8,
        80.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k": [
        80.4,
        80.0,
        79.8,
        80.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        80.3,
        80.0,
        79.9,
        80.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k": [
        80.4,
        80.0,
        79.9,
        80.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k": [
        80.4,
        80.0,
        80.1,
        80.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k": [
        80.4,
        80.1,
        79.9,
        80.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        80.3,
        79.9,
        79.9,
        80.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        80.2,
        79.9,
        79.7,
        80.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k": [
        80.1,
        80.1,
        79.8,
        80.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q": [
        80.1,
        80.1,
        79.7,
        80.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        80.2,
        79.9,
        79.7,
        79.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q": [
        80.1,
        80.0,
        79.9,
        80.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k": [
        80.1,
        79.9,
        79.8,
        80.0
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_k": [
        80.0,
        79.8,
        79.7,
        80.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q": [
        80.1,
        79.9,
        79.6,
        79.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v": [
        79.8,
        79.8,
        79.4,
        79.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        79.8,
        79.9,
        79.3,
        79.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        79.8,
        79.6,
        79.5,
        79.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k": [
        79.9,
        79.5,
        79.5,
        79.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        79.9,
        79.5,
        79.5,
        79.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k": [
        79.9,
        79.5,
        79.3,
        79.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k": [
        79.9,
        79.6,
        79.4,
        79.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k": [
        79.9,
        79.5,
        79.4,
        79.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        79.9,
        79.5,
        79.5,
        79.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q": [
        79.9,
        79.4,
        79.4,
        79.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        79.8,
        79.4,
        79.3,
        79.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q": [
        79.8,
        79.4,
        79.3,
        79.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q": [
        79.8,
        79.5,
        79.2,
        79.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        79.8,
        79.4,
        79.3,
        79.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q": [
        79.7,
        79.3,
        79.4,
        79.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q": [
        79.7,
        79.3,
        79.4,
        79.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0": [
        79.7,
        79.3,
        79.4,
        79.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0": [
        79.6,
        79.3,
        79.2,
        79.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0": [
        79.5,
        79.2,
        79.3,
        79.4
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        79.5,
        79.1,
        79.3,
        79.4
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_k": [
        79.5,
        79.3,
        79.2,
        79.3
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_v": [
        79.3,
        79.2,
        79.1,
        79.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v": [
        79.2,
        79.1,
        79.1,
        79.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v": [
        79.1,
        78.9,
        79.0,
        79.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q": [
        79.1,
        79.0,
        78.8,
        79.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k": [
        79.1,
        79.0,
        78.9,
        79.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        79.0,
        79.0,
        78.9,
        79.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        79.0,
        79.0,
        78.9,
        78.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q": [
        79.0,
        79.0,
        78.9,
        79.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k": [
        79.0,
        79.0,
        78.9,
        79.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0": [
        78.9,
        78.8,
        79.0,
        79.0
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        79.0,
        78.9,
        78.9,
        79.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v": [
        78.8,
        78.9,
        78.8,
        78.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q": [
        78.8,
        78.9,
        78.8,
        78.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        78.7,
        79.0,
        78.5,
        78.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q": [
        78.7,
        79.0,
        78.5,
        78.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v": [
        78.7,
        79.0,
        78.5,
        78.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v": [
        78.7,
        78.9,
        78.3,
        78.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        78.8,
        78.9,
        78.2,
        78.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q": [
        78.6,
        78.7,
        78.0,
        78.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k": [
        78.2,
        78.6,
        77.5,
        77.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k": [
        78.3,
        78.6,
        77.6,
        77.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v": [
        78.2,
        78.3,
        77.3,
        77.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k": [
        78.2,
        78.3,
        77.3,
        77.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k": [
        78.1,
        78.3,
        77.2,
        77.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q": [
        78.1,
        78.2,
        77.3,
        77.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v": [
        78.0,
        78.1,
        77.1,
        77.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v": [
        77.9,
        78.1,
        77.1,
        77.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k": [
        78.0,
        78.0,
        77.0,
        77.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        77.9,
        77.8,
        76.8,
        77.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k": [
        77.9,
        77.8,
        76.8,
        77.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q": [
        77.8,
        77.7,
        77.0,
        77.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v": [
        77.8,
        77.8,
        77.1,
        77.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q": [
        77.9,
        77.7,
        77.0,
        77.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q": [
        77.8,
        77.6,
        77.0,
        77.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v": [
        77.9,
        77.7,
        77.0,
        77.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        77.9,
        77.6,
        76.9,
        76.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k": [
        77.9,
        77.6,
        76.8,
        76.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q": [
        77.8,
        77.5,
        76.9,
        76.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q": [
        77.6,
        77.4,
        76.9,
        76.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q": [
        77.6,
        77.3,
        76.8,
        76.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q": [
        77.6,
        77.4,
        76.7,
        76.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v": [
        77.5,
        77.1,
        76.4,
        76.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k": [
        77.6,
        77.1,
        76.5,
        76.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        77.7,
        77.3,
        76.5,
        76.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k": [
        77.6,
        77.4,
        76.6,
        76.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q": [
        77.5,
        77.3,
        76.5,
        76.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        77.5,
        77.4,
        76.5,
        76.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q": [
        77.4,
        77.2,
        76.4,
        76.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k": [
        77.4,
        77.2,
        76.3,
        76.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k": [
        77.4,
        77.2,
        76.3,
        76.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q": [
        77.2,
        77.2,
        76.2,
        76.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q": [
        77.2,
        77.2,
        76.2,
        76.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q": [
        77.3,
        77.2,
        76.1,
        76.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0": [
        77.1,
        77.2,
        76.2,
        76.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k": [
        77.2,
        77.2,
        76.1,
        76.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q": [
        77.2,
        77.1,
        76.2,
        76.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q": [
        77.2,
        77.1,
        76.2,
        76.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k": [
        77.0,
        77.1,
        76.1,
        76.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q": [
        77.1,
        77.1,
        76.1,
        76.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k": [
        77.2,
        77.0,
        76.1,
        76.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v": [
        76.1,
        76.0,
        75.1,
        75.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        76.2,
        76.0,
        75.1,
        75.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k": [
        76.0,
        75.6,
        74.7,
        74.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q": [
        76.0,
        75.6,
        74.8,
        74.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q": [
        75.8,
        75.5,
        74.6,
        74.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q": [
        75.9,
        75.4,
        74.6,
        74.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0": [
        75.7,
        75.4,
        74.6,
        74.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        75.7,
        75.2,
        74.8,
        74.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k": [
        75.7,
        75.2,
        74.8,
        74.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k": [
        75.7,
        75.1,
        74.8,
        74.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q": [
        75.6,
        74.9,
        74.7,
        74.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q": [
        75.4,
        74.8,
        74.4,
        74.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0": [
        75.2,
        74.7,
        74.4,
        74.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q": [
        75.2,
        74.8,
        74.3,
        74.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        75.1,
        74.7,
        74.2,
        74.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        75.1,
        74.6,
        74.3,
        74.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        75.1,
        74.5,
        74.3,
        74.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v": [
        74.9,
        74.4,
        74.1,
        74.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        74.8,
        74.2,
        73.9,
        74.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v": [
        74.9,
        74.3,
        73.9,
        74.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        74.9,
        74.5,
        73.9,
        74.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v": [
        75.0,
        74.4,
        74.0,
        74.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        75.1,
        74.3,
        74.0,
        74.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k": [
        75.0,
        74.2,
        73.9,
        74.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q": [
        75.0,
        74.2,
        73.9,
        74.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k": [
        74.9,
        74.0,
        73.8,
        74.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v": [
        74.9,
        74.0,
        73.7,
        73.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v": [
        74.8,
        74.0,
        73.9,
        74.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v": [
        74.8,
        74.0,
        73.7,
        74.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v": [
        74.7,
        73.9,
        73.6,
        73.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        74.8,
        73.9,
        73.7,
        73.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k": [
        74.7,
        73.7,
        73.6,
        73.7
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_q": [
        74.8,
        73.8,
        73.6,
        73.8
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_q": [
        74.8,
        73.9,
        73.7,
        73.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        74.6,
        73.8,
        73.6,
        73.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        74.6,
        73.9,
        73.5,
        73.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        74.7,
        73.9,
        73.6,
        73.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q": [
        74.8,
        73.9,
        73.8,
        73.7
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_k": [
        74.8,
        73.9,
        73.8,
        73.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2": [
        74.7,
        73.9,
        73.8,
        73.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        74.7,
        73.9,
        73.8,
        73.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k": [
        74.7,
        73.9,
        73.8,
        73.7
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_q": [
        74.7,
        73.8,
        73.8,
        73.6
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_k": [
        74.6,
        73.8,
        73.8,
        73.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q": [
        74.5,
        73.7,
        73.7,
        73.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k": [
        74.5,
        73.7,
        73.6,
        73.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        74.4,
        73.7,
        73.6,
        73.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k": [
        74.3,
        73.6,
        73.7,
        73.4
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_k": [
        74.2,
        73.6,
        73.7,
        73.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q": [
        74.5,
        73.2,
        73.5,
        73.3
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_k": [
        74.6,
        73.2,
        73.5,
        73.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        74.6,
        73.2,
        73.5,
        73.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        74.5,
        73.3,
        73.5,
        73.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v": [
        74.5,
        73.4,
        73.5,
        73.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v": [
        74.4,
        73.3,
        73.4,
        73.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q": [
        74.2,
        73.3,
        73.4,
        73.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        74.2,
        73.0,
        73.3,
        72.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k": [
        74.1,
        72.9,
        73.4,
        73.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k": [
        73.9,
        72.8,
        73.2,
        72.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k": [
        73.8,
        72.7,
        73.2,
        72.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2": [
        73.9,
        72.9,
        73.3,
        73.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        73.9,
        72.8,
        73.3,
        73.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        73.9,
        72.7,
        73.1,
        72.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v": [
        73.7,
        72.6,
        73.0,
        72.8
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_q": [
        73.7,
        72.5,
        72.9,
        72.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        73.7,
        72.6,
        72.8,
        72.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        72.3,
        71.8,
        71.9,
        71.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v": [
        72.3,
        71.8,
        71.9,
        71.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k": [
        72.2,
        71.8,
        71.9,
        71.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k": [
        72.2,
        71.9,
        72.0,
        71.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        72.1,
        71.7,
        71.9,
        71.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v": [
        71.9,
        71.7,
        71.8,
        71.6
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        71.9,
        71.7,
        71.9,
        71.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2": [
        72.0,
        71.8,
        72.0,
        71.6
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        72.0,
        71.8,
        72.0,
        71.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2": [
        72.0,
        71.8,
        72.0,
        71.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2": [
        71.9,
        71.8,
        72.0,
        71.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q": [
        72.0,
        71.7,
        72.0,
        71.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k": [
        71.9,
        71.8,
        71.9,
        71.5
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_v": [
        71.9,
        71.7,
        71.9,
        71.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k": [
        71.8,
        71.7,
        71.7,
        71.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k": [
        71.7,
        71.6,
        71.7,
        71.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        71.8,
        71.7,
        71.7,
        71.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        71.8,
        71.8,
        71.8,
        71.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q": [
        71.6,
        71.7,
        71.7,
        71.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k": [
        71.6,
        71.6,
        71.6,
        71.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        71.6,
        71.6,
        71.7,
        71.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2": [
        71.7,
        71.6,
        71.7,
        71.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        71.7,
        71.6,
        71.6,
        71.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0": [
        71.5,
        71.7,
        71.6,
        71.3
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_v": [
        71.5,
        71.6,
        71.7,
        71.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        71.3,
        71.4,
        71.4,
        71.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q": [
        71.3,
        71.3,
        71.4,
        71.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2": [
        71.3,
        71.3,
        71.4,
        71.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2": [
        71.3,
        71.4,
        71.4,
        71.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v": [
        71.2,
        71.3,
        71.4,
        71.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2": [
        71.2,
        71.2,
        71.2,
        70.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k": [
        71.3,
        71.1,
        71.3,
        70.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q": [
        71.3,
        71.2,
        71.3,
        70.8
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_v": [
        71.4,
        71.2,
        71.4,
        70.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v": [
        71.4,
        71.3,
        71.4,
        71.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q": [
        71.4,
        71.3,
        71.4,
        70.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2": [
        71.4,
        71.3,
        71.5,
        71.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v": [
        71.3,
        71.3,
        71.5,
        71.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k": [
        71.3,
        71.4,
        71.5,
        71.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2": [
        71.2,
        71.3,
        71.4,
        71.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2": [
        71.2,
        71.4,
        71.6,
        71.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v": [
        71.2,
        71.4,
        71.6,
        71.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        71.3,
        71.4,
        71.4,
        70.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k": [
        71.3,
        71.4,
        71.3,
        70.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2": [
        71.4,
        71.4,
        71.3,
        70.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v": [
        71.4,
        71.4,
        71.3,
        70.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2": [
        71.2,
        71.3,
        71.2,
        70.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        71.3,
        71.4,
        71.3,
        70.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        71.1,
        71.1,
        71.0,
        70.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        71.1,
        71.2,
        71.1,
        70.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        71.1,
        71.2,
        71.0,
        70.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v": [
        71.1,
        71.1,
        71.0,
        70.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v": [
        70.9,
        71.1,
        70.8,
        70.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        71.0,
        71.1,
        70.8,
        70.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q": [
        71.1,
        71.2,
        71.0,
        70.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        71.0,
        71.2,
        71.0,
        70.8
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        71.0,
        71.4,
        71.0,
        70.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q": [
        71.0,
        71.4,
        70.9,
        70.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q": [
        71.0,
        71.3,
        70.9,
        70.6
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.2": [
        71.0,
        71.3,
        70.9,
        70.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v": [
        71.0,
        71.3,
        70.9,
        70.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k": [
        71.1,
        71.4,
        71.0,
        70.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        71.0,
        71.4,
        70.9,
        70.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        70.8,
        71.6,
        70.8,
        70.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k": [
        70.8,
        71.6,
        70.8,
        70.6
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.2": [
        70.8,
        71.5,
        70.8,
        70.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k": [
        70.9,
        71.5,
        70.7,
        70.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v": [
        70.8,
        71.4,
        70.6,
        70.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        70.8,
        71.4,
        70.6,
        70.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        70.7,
        71.3,
        70.7,
        70.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        70.8,
        71.4,
        70.6,
        70.4
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        70.7,
        71.3,
        70.6,
        70.4
      ],
      "mid_block.resnets.1.time_emb_proj": [
        70.6,
        71.2,
        70.5,
        70.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2": [
        70.6,
        71.3,
        70.5,
        70.3
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.2": [
        70.6,
        71.3,
        70.5,
        70.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        70.7,
        71.4,
        70.6,
        70.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q": [
        70.7,
        71.4,
        70.6,
        70.4
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_v": [
        70.7,
        71.4,
        70.6,
        70.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        70.8,
        71.5,
        70.8,
        70.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q": [
        70.9,
        71.6,
        70.8,
        70.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        70.9,
        71.6,
        70.8,
        70.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v": [
        70.8,
        71.5,
        70.7,
        70.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k": [
        70.8,
        71.5,
        70.7,
        70.5
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        70.9,
        71.4,
        70.7,
        70.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        70.9,
        71.4,
        70.6,
        70.6
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.2": [
        70.9,
        71.5,
        70.7,
        70.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        70.9,
        71.5,
        70.7,
        70.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2": [
        70.6,
        71.2,
        70.5,
        70.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v": [
        70.9,
        71.5,
        70.7,
        70.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k": [
        70.9,
        71.4,
        70.7,
        70.5
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.2": [
        70.9,
        71.5,
        70.7,
        70.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2": [
        71.0,
        71.5,
        70.6,
        70.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        71.0,
        71.7,
        71.0,
        70.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        70.9,
        71.7,
        71.0,
        71.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.2": [
        70.9,
        71.6,
        71.1,
        71.0
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_k": [
        70.9,
        71.4,
        71.0,
        70.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q": [
        70.9,
        71.4,
        71.0,
        70.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v": [
        70.9,
        71.5,
        71.2,
        71.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        70.9,
        71.5,
        71.3,
        71.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q": [
        70.9,
        71.4,
        71.1,
        71.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q": [
        70.7,
        71.3,
        71.0,
        71.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v": [
        70.7,
        71.2,
        71.1,
        71.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        70.5,
        71.2,
        71.0,
        71.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v": [
        70.4,
        70.9,
        70.9,
        70.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k": [
        70.3,
        70.9,
        70.8,
        70.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q": [
        70.2,
        70.8,
        70.7,
        70.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v": [
        70.2,
        70.8,
        70.8,
        70.7
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        70.1,
        70.7,
        70.7,
        70.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        70.1,
        70.7,
        70.6,
        70.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2": [
        70.0,
        70.7,
        70.6,
        70.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k": [
        70.1,
        70.6,
        70.6,
        70.5
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_q": [
        70.0,
        70.6,
        70.6,
        70.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        69.9,
        70.5,
        70.6,
        70.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k": [
        69.7,
        70.4,
        70.4,
        70.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k": [
        69.6,
        70.3,
        70.3,
        70.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        69.6,
        70.5,
        70.3,
        70.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k": [
        69.5,
        70.4,
        70.2,
        70.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q": [
        69.5,
        70.3,
        70.1,
        70.1
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.2": [
        69.4,
        70.3,
        70.1,
        70.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        69.5,
        70.3,
        70.1,
        70.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v": [
        69.5,
        70.2,
        70.2,
        70.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        69.6,
        70.4,
        70.2,
        70.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k": [
        69.4,
        70.2,
        70.1,
        70.1
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        69.4,
        70.2,
        70.0,
        70.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        67.1,
        67.7,
        67.5,
        67.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q": [
        67.1,
        67.8,
        67.6,
        67.3
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        67.2,
        67.8,
        67.6,
        67.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2": [
        67.1,
        67.7,
        67.4,
        67.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        67.2,
        67.6,
        67.5,
        67.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q": [
        67.1,
        67.6,
        67.4,
        67.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        66.9,
        67.4,
        67.3,
        67.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v": [
        66.9,
        67.5,
        67.3,
        67.2
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_k": [
        66.9,
        67.4,
        67.2,
        67.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        66.9,
        67.4,
        67.2,
        67.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_v": [
        67.0,
        67.5,
        67.3,
        67.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        66.9,
        67.5,
        67.2,
        67.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v": [
        66.8,
        67.2,
        67.1,
        66.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2": [
        66.2,
        66.7,
        66.5,
        66.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k": [
        66.0,
        66.6,
        66.3,
        66.2
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        66.0,
        66.5,
        66.4,
        66.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_q": [
        66.0,
        66.5,
        66.3,
        66.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v": [
        65.9,
        66.4,
        66.4,
        66.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        66.0,
        66.4,
        66.3,
        66.2
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_q": [
        65.9,
        66.4,
        66.3,
        66.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        66.1,
        66.4,
        66.3,
        66.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        66.1,
        66.5,
        66.4,
        66.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q": [
        65.9,
        66.3,
        66.1,
        65.9
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        66.0,
        66.4,
        66.2,
        66.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k": [
        66.0,
        66.3,
        66.1,
        66.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q": [
        65.9,
        66.2,
        66.0,
        65.9
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.2": [
        65.8,
        66.1,
        65.9,
        65.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2": [
        65.5,
        66.0,
        65.8,
        65.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k": [
        65.4,
        65.9,
        65.7,
        65.5
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        65.4,
        65.8,
        65.6,
        65.4
      ],
      "mid_block.resnets.0.time_emb_proj": [
        65.3,
        65.8,
        65.5,
        65.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k": [
        65.2,
        65.7,
        65.4,
        65.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2": [
        65.2,
        65.6,
        65.5,
        65.4
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_k": [
        65.3,
        65.7,
        65.5,
        65.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q": [
        65.2,
        65.7,
        65.5,
        65.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        65.1,
        65.5,
        65.3,
        65.3
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        65.1,
        65.5,
        65.3,
        65.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v": [
        65.0,
        65.2,
        65.2,
        65.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k": [
        64.8,
        65.1,
        65.1,
        65.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k": [
        64.9,
        65.1,
        65.2,
        65.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v": [
        64.8,
        65.0,
        65.2,
        65.2
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        64.8,
        65.0,
        65.2,
        65.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2": [
        65.1,
        65.3,
        65.4,
        65.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k": [
        65.1,
        65.3,
        65.4,
        65.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q": [
        65.1,
        65.2,
        65.4,
        65.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        65.1,
        65.3,
        65.4,
        65.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        65.1,
        65.3,
        65.4,
        65.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q": [
        65.2,
        65.3,
        65.5,
        65.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q": [
        65.0,
        65.2,
        65.3,
        65.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        65.0,
        65.2,
        65.3,
        65.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        64.4,
        64.6,
        64.8,
        64.8
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_v": [
        64.4,
        64.6,
        64.7,
        64.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k": [
        64.4,
        64.5,
        64.8,
        64.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        64.3,
        64.5,
        64.8,
        64.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k": [
        64.4,
        64.5,
        64.8,
        64.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k": [
        64.1,
        64.2,
        64.5,
        64.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q": [
        64.0,
        64.1,
        64.4,
        64.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q": [
        64.1,
        64.1,
        64.4,
        64.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2": [
        64.1,
        64.1,
        64.4,
        64.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v": [
        64.1,
        64.2,
        64.5,
        64.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        64.0,
        64.0,
        64.3,
        64.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        63.4,
        63.6,
        63.9,
        64.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        63.3,
        63.4,
        63.8,
        63.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k": [
        63.2,
        63.3,
        63.7,
        63.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q": [
        63.0,
        63.2,
        63.6,
        63.6
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.2": [
        63.0,
        63.1,
        63.5,
        63.6
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_v": [
        63.1,
        63.1,
        63.5,
        63.6
      ],
      "mid_block.resnets.1.conv1": [
        62.9,
        63.0,
        63.4,
        63.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v": [
        62.9,
        63.0,
        63.5,
        63.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        62.8,
        63.0,
        63.4,
        63.4
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        62.9,
        63.0,
        63.5,
        63.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        62.9,
        63.1,
        63.5,
        63.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k": [
        63.0,
        63.0,
        63.5,
        63.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2": [
        63.0,
        63.1,
        63.5,
        63.5
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        63.0,
        63.1,
        63.5,
        63.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q": [
        63.0,
        63.1,
        63.5,
        63.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k": [
        63.0,
        63.1,
        63.5,
        63.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q": [
        62.8,
        62.8,
        63.2,
        63.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        62.8,
        62.9,
        63.2,
        63.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q": [
        62.9,
        63.0,
        63.2,
        63.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        62.9,
        62.9,
        63.2,
        63.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2": [
        63.2,
        63.3,
        63.6,
        63.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        63.2,
        63.4,
        63.6,
        63.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k": [
        63.3,
        63.4,
        63.6,
        63.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v": [
        63.2,
        63.3,
        63.6,
        63.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        63.1,
        63.3,
        63.7,
        63.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        63.1,
        63.3,
        63.7,
        63.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        63.2,
        63.3,
        63.7,
        63.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q": [
        63.2,
        63.3,
        63.7,
        63.6
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        63.1,
        63.3,
        63.6,
        63.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v": [
        63.0,
        63.2,
        63.5,
        63.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        63.0,
        63.2,
        63.4,
        63.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        63.0,
        63.2,
        63.4,
        63.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v": [
        63.1,
        63.2,
        63.4,
        63.3
      ],
      "down_blocks.2.attentions.1.proj_in": [
        63.2,
        63.3,
        63.5,
        63.4
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_k": [
        63.2,
        63.4,
        63.4,
        63.4
      ],
      "mid_block.resnets.1.conv2": [
        63.3,
        63.4,
        63.5,
        63.4
      ],
      "down_blocks.2.resnets.1.conv1": [
        63.2,
        63.4,
        63.5,
        63.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        63.3,
        63.5,
        63.6,
        63.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v": [
        63.2,
        63.4,
        63.4,
        63.4
      ],
      "up_blocks.0.resnets.0.conv1": [
        63.0,
        63.2,
        63.2,
        63.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v": [
        62.9,
        63.2,
        63.2,
        63.1
      ],
      "down_blocks.2.resnets.1.conv2": [
        63.0,
        63.2,
        63.2,
        63.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0": [
        62.9,
        63.1,
        63.1,
        62.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        62.9,
        63.0,
        63.1,
        62.9
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_q": [
        62.9,
        63.0,
        63.0,
        62.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        62.8,
        63.0,
        63.0,
        63.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0": [
        63.0,
        63.1,
        63.2,
        63.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        62.9,
        63.0,
        63.1,
        63.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2": [
        62.9,
        63.0,
        63.0,
        63.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k": [
        62.9,
        63.0,
        63.0,
        63.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v": [
        62.9,
        63.0,
        63.0,
        63.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2": [
        63.0,
        63.1,
        63.0,
        62.9
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_v": [
        62.9,
        63.1,
        63.0,
        63.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        63.0,
        63.1,
        63.1,
        63.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        63.0,
        63.1,
        63.0,
        63.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2": [
        62.9,
        63.0,
        63.0,
        62.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        62.9,
        63.1,
        62.8,
        62.8
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        62.9,
        63.0,
        62.8,
        62.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        62.9,
        63.0,
        62.8,
        62.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2": [
        62.9,
        63.0,
        62.6,
        62.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2": [
        62.9,
        63.1,
        62.6,
        62.8
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        62.8,
        63.0,
        62.5,
        62.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        62.5,
        62.6,
        62.2,
        62.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2": [
        62.3,
        62.5,
        62.1,
        62.3
      ],
      "mid_block.resnets.0.conv1": [
        62.3,
        62.5,
        62.0,
        62.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        62.3,
        62.5,
        62.1,
        62.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v": [
        62.3,
        62.4,
        62.0,
        62.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0": [
        62.1,
        62.2,
        61.8,
        62.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v": [
        62.1,
        62.1,
        61.7,
        61.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v": [
        62.1,
        62.2,
        61.7,
        62.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v": [
        61.9,
        62.0,
        61.5,
        61.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2": [
        61.9,
        61.9,
        61.4,
        61.8
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        61.8,
        61.9,
        61.5,
        61.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        61.8,
        61.9,
        61.5,
        61.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        61.8,
        61.8,
        61.4,
        61.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v": [
        61.7,
        61.7,
        61.3,
        61.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        61.7,
        61.7,
        61.3,
        61.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        61.8,
        61.7,
        61.4,
        61.7
      ],
      "mid_block.resnets.0.conv2": [
        61.8,
        61.8,
        61.5,
        61.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        61.8,
        61.7,
        61.4,
        61.7
      ],
      "up_blocks.0.resnets.1.conv2": [
        62.1,
        62.1,
        61.7,
        61.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2": [
        62.1,
        62.0,
        61.7,
        61.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2": [
        62.1,
        61.9,
        61.7,
        61.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v": [
        62.1,
        62.0,
        61.6,
        61.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v": [
        62.1,
        61.9,
        61.6,
        61.8
      ],
      "down_blocks.2.attentions.1.proj_out": [
        62.2,
        62.0,
        61.8,
        61.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0": [
        62.2,
        61.9,
        61.7,
        61.8
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        62.1,
        61.8,
        61.6,
        61.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v": [
        62.1,
        61.9,
        61.6,
        61.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        62.4,
        62.2,
        62.0,
        62.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        62.5,
        62.4,
        62.0,
        62.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v": [
        62.3,
        62.2,
        61.8,
        61.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0": [
        62.4,
        62.2,
        61.9,
        62.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj": [
        62.2,
        62.2,
        61.8,
        62.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0": [
        62.3,
        62.2,
        61.9,
        62.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v": [
        62.4,
        62.3,
        62.0,
        62.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0": [
        62.4,
        62.3,
        61.9,
        62.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v": [
        62.6,
        62.5,
        62.1,
        62.3
      ],
      "mid_block.attentions.0.proj_out": [
        62.6,
        62.5,
        62.1,
        62.4
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_q": [
        62.6,
        62.5,
        62.1,
        62.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v": [
        62.5,
        62.3,
        61.9,
        62.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v": [
        62.7,
        62.5,
        62.1,
        62.4
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        62.7,
        62.5,
        62.1,
        62.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        62.7,
        62.6,
        62.1,
        62.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj": [
        62.7,
        62.5,
        62.1,
        62.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        62.6,
        62.4,
        61.9,
        62.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v": [
        62.6,
        62.5,
        62.1,
        62.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        62.7,
        62.7,
        62.2,
        62.4
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_k": [
        62.7,
        62.6,
        62.2,
        62.4
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        62.6,
        62.6,
        62.2,
        62.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        62.6,
        62.7,
        62.2,
        62.4
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        62.6,
        62.7,
        62.3,
        62.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v": [
        62.7,
        62.7,
        62.2,
        62.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        62.7,
        62.7,
        62.3,
        62.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        62.7,
        62.8,
        62.3,
        62.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj": [
        62.6,
        62.8,
        62.2,
        62.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2": [
        62.7,
        62.8,
        62.3,
        62.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v": [
        62.7,
        62.9,
        62.3,
        62.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        62.6,
        62.7,
        62.2,
        62.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2": [
        62.6,
        62.8,
        62.3,
        62.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k": [
        62.6,
        62.7,
        62.3,
        62.6
      ],
      "up_blocks.0.resnets.1.conv1": [
        62.6,
        62.7,
        62.2,
        62.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v": [
        62.5,
        62.7,
        62.1,
        62.5
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        62.5,
        62.6,
        62.1,
        62.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v": [
        62.2,
        62.4,
        61.8,
        62.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v": [
        61.8,
        62.2,
        61.5,
        61.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        61.8,
        62.1,
        61.5,
        61.8
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_v": [
        61.8,
        62.1,
        61.4,
        61.8
      ],
      "up_blocks.0.resnets.0.conv2": [
        61.6,
        62.2,
        61.2,
        61.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj": [
        61.7,
        62.3,
        61.4,
        62.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj": [
        61.8,
        62.3,
        61.4,
        62.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2": [
        61.7,
        62.2,
        61.4,
        61.9
      ],
      "up_blocks.0.resnets.2.conv2": [
        61.7,
        62.2,
        61.4,
        61.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        61.6,
        62.2,
        61.4,
        61.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        61.2,
        61.9,
        61.0,
        61.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        61.4,
        61.9,
        61.2,
        61.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2": [
        61.4,
        61.9,
        61.1,
        61.7
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        61.4,
        61.9,
        61.1,
        61.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        61.4,
        61.9,
        61.1,
        61.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v": [
        61.1,
        61.6,
        60.9,
        61.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2": [
        61.1,
        61.6,
        60.9,
        61.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        60.9,
        61.3,
        60.7,
        61.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2": [
        61.1,
        61.4,
        60.9,
        61.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2": [
        61.1,
        61.5,
        60.9,
        61.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        61.1,
        61.4,
        60.8,
        61.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj": [
        61.0,
        61.4,
        60.8,
        61.4
      ],
      "up_blocks.0.attentions.0.proj_out": [
        61.6,
        61.9,
        61.2,
        61.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        61.5,
        61.8,
        61.2,
        61.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        61.5,
        61.7,
        61.2,
        61.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2": [
        61.7,
        61.8,
        61.4,
        61.9
      ],
      "up_blocks.0.resnets.2.conv1": [
        61.7,
        61.8,
        61.4,
        61.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0": [
        61.7,
        61.8,
        61.4,
        61.9
      ],
      "down_blocks.2.resnets.0.conv2": [
        61.5,
        61.7,
        61.2,
        61.7
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        61.7,
        61.8,
        61.4,
        61.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        61.7,
        61.8,
        61.4,
        61.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v": [
        61.6,
        61.7,
        61.3,
        61.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v": [
        61.3,
        61.3,
        61.1,
        61.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        61.4,
        61.5,
        61.2,
        61.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        61.6,
        61.5,
        61.3,
        61.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        61.3,
        61.2,
        61.1,
        61.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        61.4,
        61.2,
        61.2,
        61.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj": [
        61.2,
        61.0,
        61.0,
        61.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k": [
        61.1,
        60.9,
        61.0,
        61.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2": [
        61.1,
        60.8,
        60.9,
        61.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        61.3,
        61.0,
        61.0,
        61.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        60.5,
        60.5,
        60.4,
        60.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        60.6,
        60.6,
        60.5,
        60.7
      ],
      "down_blocks.2.attentions.0.proj_in": [
        60.3,
        60.4,
        60.3,
        60.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        60.3,
        60.4,
        60.2,
        60.5
      ],
      "up_blocks.0.attentions.0.proj_in": [
        60.6,
        60.5,
        60.4,
        60.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        60.4,
        60.3,
        60.2,
        60.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        60.4,
        60.3,
        60.2,
        60.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        60.2,
        60.2,
        60.0,
        60.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj": [
        60.4,
        60.3,
        60.1,
        60.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        60.2,
        60.2,
        60.0,
        60.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        60.5,
        60.3,
        60.3,
        60.4
      ],
      "down_blocks.2.attentions.0.proj_out": [
        60.3,
        60.1,
        60.1,
        60.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2": [
        60.3,
        59.9,
        60.0,
        60.1
      ],
      "mid_block.attentions.0.proj_in": [
        60.2,
        59.9,
        60.0,
        60.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        60.0,
        59.7,
        59.8,
        59.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        59.9,
        59.5,
        59.7,
        59.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        60.1,
        59.6,
        59.8,
        59.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        60.2,
        59.6,
        59.8,
        59.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2": [
        60.2,
        59.9,
        59.8,
        60.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        60.1,
        59.8,
        59.7,
        59.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        60.0,
        59.8,
        59.7,
        59.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        60.0,
        59.8,
        59.7,
        59.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        57.0,
        56.9,
        56.6,
        56.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        56.9,
        56.9,
        56.6,
        56.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        57.1,
        57.0,
        56.7,
        56.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        57.1,
        57.0,
        56.8,
        56.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k": [
        57.0,
        56.9,
        56.6,
        56.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        56.9,
        56.8,
        56.5,
        56.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        56.9,
        56.7,
        56.4,
        56.5
      ],
      "up_blocks.0.attentions.2.proj_in": [
        56.6,
        56.6,
        56.3,
        56.4
      ],
      "down_blocks.2.resnets.0.conv1": [
        56.5,
        56.5,
        56.1,
        56.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        56.3,
        56.2,
        55.8,
        56.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        56.2,
        56.1,
        55.7,
        55.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        56.2,
        56.1,
        55.6,
        55.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        56.1,
        56.1,
        55.5,
        55.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        55.9,
        55.8,
        55.4,
        55.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        55.8,
        55.8,
        55.4,
        55.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        55.4,
        55.3,
        54.9,
        55.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        55.2,
        55.2,
        54.8,
        54.9
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        54.8,
        54.9,
        54.6,
        54.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        54.8,
        54.9,
        54.6,
        54.7
      ],
      "up_blocks.0.attentions.2.proj_out": [
        54.7,
        54.7,
        54.4,
        54.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q": [
        54.7,
        54.7,
        54.3,
        54.5
      ],
      "add_embedding.linear_2": [
        54.7,
        53.9,
        54.4,
        54.6
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        54.1,
        53.3,
        53.8,
        54.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        54.0,
        53.1,
        53.7,
        53.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        53.9,
        53.1,
        53.6,
        53.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        53.7,
        52.9,
        53.3,
        53.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        53.7,
        53.0,
        53.4,
        53.7
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        53.9,
        53.2,
        53.6,
        53.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        54.0,
        53.2,
        53.8,
        54.0
      ],
      "up_blocks.0.attentions.1.proj_in": [
        53.4,
        52.9,
        53.3,
        53.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k": [
        52.2,
        51.6,
        52.3,
        52.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        52.2,
        51.5,
        52.2,
        52.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        52.4,
        51.7,
        52.5,
        52.4
      ],
      "add_embedding.linear_1": [
        52.2,
        51.8,
        52.7,
        52.5
      ],
      "up_blocks.1.resnets.0.conv1": [
        52.6,
        52.0,
        52.9,
        52.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        52.4,
        51.9,
        52.7,
        52.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        52.3,
        51.7,
        52.6,
        52.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        51.8,
        51.2,
        52.1,
        51.9
      ],
      "up_blocks.1.resnets.0.conv2": [
        51.9,
        51.3,
        52.2,
        52.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        51.8,
        51.2,
        52.1,
        52.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        51.9,
        51.2,
        52.2,
        52.0
      ],
      "up_blocks.0.attentions.1.proj_out": [
        52.4,
        51.7,
        52.8,
        52.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        52.3,
        51.5,
        52.6,
        52.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        52.1,
        51.4,
        52.5,
        52.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        52.0,
        51.3,
        52.3,
        52.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        51.7,
        51.1,
        52.0,
        51.9
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        51.5,
        50.7,
        51.7,
        51.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        51.4,
        50.7,
        51.7,
        51.5
      ],
      "down_blocks.1.resnets.1.conv2": [
        51.3,
        50.5,
        51.6,
        51.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        51.2,
        50.4,
        51.5,
        51.3
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        51.1,
        50.3,
        51.4,
        51.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        51.0,
        50.3,
        51.4,
        51.1
      ],
      "up_blocks.1.attentions.0.proj_out": [
        51.0,
        50.2,
        51.3,
        51.0
      ],
      "down_blocks.1.attentions.1.proj_out": [
        50.8,
        50.1,
        51.3,
        50.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        50.6,
        49.9,
        51.0,
        50.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        50.4,
        49.7,
        50.8,
        50.4
      ],
      "down_blocks.1.resnets.1.conv1": [
        50.4,
        49.7,
        50.8,
        50.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        50.1,
        49.4,
        50.4,
        50.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        49.9,
        49.2,
        50.2,
        49.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k": [
        49.6,
        48.8,
        49.9,
        49.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        49.6,
        48.9,
        49.9,
        49.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        49.7,
        48.9,
        50.0,
        49.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        49.6,
        48.9,
        50.0,
        49.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        49.5,
        48.8,
        49.8,
        49.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        49.2,
        48.5,
        49.6,
        49.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        49.0,
        48.3,
        49.5,
        49.1
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        48.7,
        48.0,
        49.1,
        48.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        48.7,
        48.1,
        49.1,
        48.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        48.6,
        48.0,
        49.0,
        48.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        48.6,
        48.0,
        49.0,
        48.7
      ],
      "down_blocks.1.resnets.0.conv2": [
        48.4,
        47.8,
        48.7,
        48.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        48.1,
        47.5,
        48.4,
        48.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q": [
        47.8,
        47.2,
        48.1,
        47.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        47.7,
        47.1,
        48.1,
        47.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        47.7,
        47.1,
        48.0,
        47.8
      ],
      "down_blocks.1.attentions.1.proj_in": [
        47.5,
        47.0,
        47.9,
        47.7
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        46.5,
        46.0,
        46.9,
        46.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        46.3,
        45.9,
        46.8,
        46.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        46.2,
        45.8,
        46.7,
        46.5
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        46.5,
        46.0,
        46.9,
        46.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        45.9,
        45.5,
        46.3,
        46.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        46.1,
        45.8,
        46.5,
        46.4
      ],
      "up_blocks.1.attentions.0.proj_in": [
        46.4,
        45.9,
        46.8,
        46.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        46.4,
        46.0,
        46.8,
        46.6
      ],
      "down_blocks.1.downsamplers.0.conv": [
        46.4,
        46.1,
        46.9,
        46.7
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        45.7,
        45.5,
        46.1,
        46.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        45.7,
        45.5,
        46.2,
        46.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2": [
        45.3,
        45.1,
        45.7,
        45.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        45.0,
        44.8,
        45.4,
        45.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        44.9,
        44.7,
        45.3,
        45.1
      ],
      "up_blocks.1.attentions.1.proj_out": [
        45.0,
        44.9,
        45.5,
        45.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        45.2,
        45.1,
        45.7,
        45.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        44.8,
        44.6,
        45.2,
        45.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        44.7,
        44.6,
        45.2,
        44.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        44.5,
        44.4,
        45.0,
        44.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        44.5,
        44.4,
        44.9,
        44.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        44.5,
        44.3,
        44.9,
        44.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        44.5,
        44.3,
        44.9,
        44.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        44.5,
        44.3,
        44.9,
        44.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        44.5,
        44.3,
        44.9,
        44.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v": [
        44.1,
        43.9,
        44.5,
        44.3
      ],
      "up_blocks.0.upsamplers.0.conv": [
        44.3,
        44.1,
        44.7,
        44.5
      ],
      "down_blocks.1.attentions.0.proj_out": [
        44.4,
        44.2,
        44.8,
        44.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        44.4,
        44.1,
        44.7,
        44.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        44.3,
        44.1,
        44.7,
        44.4
      ],
      "up_blocks.1.resnets.1.conv2": [
        44.2,
        44.0,
        44.6,
        44.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        44.0,
        43.8,
        44.3,
        44.1
      ],
      "up_blocks.1.resnets.2.conv2": [
        43.5,
        43.3,
        43.8,
        43.6
      ],
      "up_blocks.1.resnets.1.conv1": [
        43.5,
        43.4,
        43.9,
        43.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        43.5,
        43.3,
        43.9,
        43.6
      ],
      "up_blocks.1.attentions.2.proj_out": [
        43.7,
        43.5,
        44.0,
        43.8
      ],
      "up_blocks.1.attentions.1.proj_in": [
        43.7,
        43.5,
        44.0,
        43.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        43.7,
        43.5,
        44.0,
        43.8
      ],
      "down_blocks.1.resnets.0.conv1": [
        43.6,
        43.4,
        43.9,
        43.7
      ],
      "up_blocks.1.attentions.2.proj_in": [
        43.7,
        43.4,
        44.0,
        43.8
      ],
      "up_blocks.1.resnets.2.conv1": [
        43.6,
        43.4,
        44.0,
        43.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        43.1,
        42.8,
        43.4,
        43.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        43.1,
        42.8,
        43.4,
        43.2
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        43.0,
        42.8,
        43.4,
        43.1
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        42.4,
        42.3,
        42.8,
        42.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        42.7,
        42.6,
        43.1,
        42.8
      ],
      "down_blocks.1.attentions.0.proj_in": [
        42.7,
        42.6,
        43.1,
        42.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        42.3,
        42.1,
        42.6,
        42.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        42.3,
        42.2,
        42.7,
        42.4
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        42.2,
        42.1,
        42.5,
        42.2
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        42.8,
        42.8,
        43.1,
        42.9
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        42.4,
        42.5,
        42.6,
        42.5
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        42.3,
        42.3,
        42.4,
        42.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k": [
        42.0,
        42.0,
        42.2,
        41.9
      ],
      "down_blocks.0.resnets.1.conv2": [
        41.9,
        41.9,
        42.1,
        41.8
      ],
      "down_blocks.0.resnets.0.conv1": [
        41.9,
        41.9,
        42.0,
        41.8
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        41.6,
        41.6,
        41.7,
        41.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        41.7,
        41.7,
        41.8,
        41.6
      ],
      "down_blocks.0.resnets.1.conv1": [
        41.4,
        41.4,
        41.5,
        41.3
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        41.1,
        41.2,
        41.3,
        41.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        41.1,
        41.1,
        41.3,
        41.1
      ],
      "down_blocks.0.resnets.0.conv2": [
        41.2,
        41.1,
        41.4,
        41.1
      ],
      "down_blocks.0.downsamplers.0.conv": [
        41.1,
        40.9,
        41.2,
        40.8
      ],
      "up_blocks.2.resnets.0.conv1": [
        41.0,
        41.0,
        41.0,
        40.8
      ],
      "up_blocks.2.resnets.0.conv2": [
        41.4,
        41.3,
        41.3,
        41.1
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        41.1,
        41.1,
        41.0,
        40.9
      ],
      "up_blocks.2.resnets.1.conv2": [
        40.8,
        40.7,
        40.6,
        40.6
      ],
      "time_embedding.linear_2": [
        40.6,
        40.5,
        40.6,
        40.6
      ],
      "up_blocks.2.resnets.1.conv1": [
        40.4,
        40.3,
        40.4,
        40.4
      ],
      "up_blocks.1.upsamplers.0.conv": [
        40.5,
        40.5,
        40.5,
        40.5
      ],
      "time_embedding.linear_1": [
        40.7,
        40.6,
        40.7,
        40.7
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        40.7,
        40.6,
        40.7,
        40.8
      ],
      "up_blocks.2.resnets.2.conv1": [
        40.8,
        40.7,
        40.8,
        40.9
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        40.5,
        40.3,
        40.4,
        40.5
      ],
      "up_blocks.2.resnets.2.conv2": [
        39.6,
        39.5,
        39.6,
        39.7
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        41.5,
        41.5,
        41.5,
        41.5
      ]
    },
    "2": {
      "metadata": {
        "nbits": 2,
        "out_ngroups": 1,
        "in_ngroups": 1,
        "cumulative": true,
        "candidates": [
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.5.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.6.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.9.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.8.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2",
          "down_blocks.2.resnets.1.time_emb_proj",
          "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.2.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.7.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.4.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0",
          "mid_block.resnets.1.time_emb_proj",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.3.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k",
          "up_blocks.0.resnets.0.time_emb_proj",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.1.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q",
          "mid_block.resnets.0.time_emb_proj",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.2",
          "mid_block.resnets.1.conv2",
          "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_k",
          "mid_block.resnets.1.conv1",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_q",
          "down_blocks.2.resnets.1.conv1",
          "down_blocks.2.attentions.1.proj_in",
          "down_blocks.2.resnets.1.conv2",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.2.resnets.0.time_emb_proj",
          "mid_block.resnets.0.conv1",
          "mid_block.resnets.0.conv2",
          "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2",
          "down_blocks.2.attentions.1.proj_out",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0",
          "up_blocks.0.resnets.0.conv1",
          "up_blocks.0.resnets.1.conv1",
          "up_blocks.0.resnets.2.time_emb_proj",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0",
          "up_blocks.0.resnets.1.time_emb_proj",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v",
          "up_blocks.0.resnets.1.conv2",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0",
          "mid_block.attentions.0.proj_out",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v",
          "up_blocks.0.attentions.0.proj_out",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2",
          "up_blocks.0.resnets.0.conv2",
          "up_blocks.0.resnets.2.conv1",
          "up_blocks.0.resnets.2.conv2",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2",
          "down_blocks.2.resnets.0.conv2",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2",
          "mid_block.attentions.0.proj_in",
          "up_blocks.0.resnets.0.conv_shortcut",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.proj_in",
          "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj",
          "down_blocks.2.attentions.0.proj_out",
          "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.0.proj_in",
          "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj",
          "down_blocks.2.resnets.0.conv1",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.2.proj_in",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.attentions.1.proj_in",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj",
          "up_blocks.0.attentions.2.proj_out",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.1.proj_out",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_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.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.resnets.1.conv_shortcut",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q",
          "add_embedding.linear_2",
          "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2",
          "down_blocks.1.resnets.1.time_emb_proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.1.resnets.0.conv2",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q",
          "down_blocks.1.resnets.1.conv2",
          "up_blocks.1.resnets.0.time_emb_proj",
          "up_blocks.1.resnets.0.conv1",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.1.resnets.0.time_emb_proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.1.attentions.1.proj_out",
          "up_blocks.1.attentions.0.proj_out",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v",
          "down_blocks.1.resnets.1.conv1",
          "down_blocks.1.resnets.0.conv2",
          "up_blocks.1.resnets.2.time_emb_proj",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v",
          "add_embedding.linear_1",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.attentions.1.proj_in",
          "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2",
          "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.1.attentions.0.proj_in",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v",
          "up_blocks.1.resnets.1.time_emb_proj",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v",
          "down_blocks.2.resnets.0.conv_shortcut",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.1.downsamplers.0.conv",
          "up_blocks.1.attentions.1.proj_out",
          "up_blocks.0.resnets.2.conv_shortcut",
          "down_blocks.1.attentions.0.proj_out",
          "down_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.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2",
          "up_blocks.1.resnets.1.conv2",
          "up_blocks.1.resnets.2.conv2",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.0.upsamplers.0.conv",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k",
          "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.1.attentions.1.proj_in",
          "up_blocks.1.attentions.2.proj_out",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.2.proj_in",
          "up_blocks.1.resnets.1.conv1",
          "up_blocks.1.resnets.2.conv1",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.attentions.0.proj_in",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q",
          "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.resnets.0.conv1",
          "down_blocks.0.resnets.1.time_emb_proj",
          "up_blocks.2.resnets.0.time_emb_proj",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.2.resnets.1.time_emb_proj",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.resnets.0.conv_shortcut",
          "down_blocks.1.resnets.0.conv_shortcut",
          "up_blocks.1.resnets.1.conv_shortcut",
          "up_blocks.2.resnets.2.time_emb_proj",
          "down_blocks.0.resnets.0.time_emb_proj",
          "down_blocks.0.resnets.0.conv1",
          "down_blocks.0.downsamplers.0.conv",
          "down_blocks.0.resnets.1.conv2",
          "up_blocks.1.resnets.2.conv_shortcut",
          "down_blocks.0.resnets.1.conv1",
          "down_blocks.0.resnets.0.conv2",
          "time_embedding.linear_1",
          "up_blocks.2.resnets.0.conv1",
          "up_blocks.2.resnets.0.conv2",
          "up_blocks.1.upsamplers.0.conv",
          "up_blocks.2.resnets.2.conv_shortcut",
          "up_blocks.2.resnets.1.conv_shortcut",
          "up_blocks.2.resnets.0.conv_shortcut",
          "up_blocks.2.resnets.1.conv1",
          "up_blocks.2.resnets.1.conv2",
          "time_embedding.linear_2",
          "up_blocks.2.resnets.2.conv1",
          "up_blocks.2.resnets.2.conv2"
        ],
        "sizes": [
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          2621440,
          2621440,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          6553600,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          6553600,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          2621440,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          6553600,
          1638400,
          6553600,
          6553600,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          6553600,
          6553600,
          13107200,
          6553600,
          1638400,
          6553600,
          2621440,
          6553600,
          13107200,
          1638400,
          1638400,
          6553600,
          1638400,
          6553600,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          13107200,
          13107200,
          6553600,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          6553600,
          1638400,
          6553600,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          13107200,
          2621440,
          1638400,
          1638400,
          6553600,
          1638400,
          6553600,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          13107200,
          13107200,
          6553600,
          1638400,
          1638400,
          1638400,
          13107200,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          6553600,
          13107200,
          1638400,
          1638400,
          13107200,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          13107200,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          13107200,
          6553600,
          1638400,
          13107200,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          6553600,
          6553600,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          2621440,
          1310720,
          1638400,
          13107200,
          2621440,
          13107200,
          1638400,
          1638400,
          13107200,
          1638400,
          1310720,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          13107200,
          1638400,
          13107200,
          1638400,
          2621440,
          6553600,
          14745600,
          6553600,
          6553600,
          1638400,
          1638400,
          2621440,
          13107200,
          1638400,
          13107200,
          13107200,
          6553600,
          1638400,
          1310720,
          1638400,
          14745600,
          1638400,
          2621440,
          13107200,
          1638400,
          14745600,
          1638400,
          14745600,
          1638400,
          1638400,
          1638400,
          409600,
          13107200,
          2621440,
          13107200,
          409600,
          1638400,
          1638400,
          14745600,
          14745600,
          13107200,
          1638400,
          13107200,
          6553600,
          1638400,
          409600,
          1638400,
          1638400,
          1638400,
          29491200,
          29491200,
          1638400,
          1638400,
          6553600,
          13107200,
          1638400,
          6553600,
          1638400,
          6553600,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          409600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1310720,
          1638400,
          1638400,
          14745600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1310720,
          6553600,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          13107200,
          1638400,
          13107200,
          1310720,
          1638400,
          13107200,
          1638400,
          1638400,
          6553600,
          6553600,
          1638400,
          1638400,
          6553600,
          6553600,
          14745600,
          22118400,
          14745600,
          1638400,
          1638400,
          13107200,
          13107200,
          1310720,
          6553600,
          14745600,
          1638400,
          13107200,
          6553600,
          409600,
          1310720,
          1638400,
          6553600,
          1638400,
          13107200,
          13107200,
          6553600,
          1638400,
          3276800,
          409600,
          1638400,
          13107200,
          1310720,
          409600,
          13107200,
          1638400,
          13107200,
          1310720,
          1310720,
          1638400,
          13107200,
          7372800,
          409600,
          6553600,
          1310720,
          6553600,
          6553600,
          1638400,
          13107200,
          409600,
          13107200,
          13107200,
          409600,
          409600,
          13107200,
          13107200,
          13107200,
          1310720,
          13107200,
          1310720,
          1638400,
          409600,
          1638400,
          409600,
          13107200,
          1638400,
          409600,
          409600,
          409600,
          13107200,
          1638400,
          409600,
          409600,
          1310720,
          409600,
          2621440,
          1310720,
          409600,
          409600,
          409600,
          409600,
          409600,
          1638400,
          409600,
          409600,
          409600,
          409600,
          409600,
          3276800,
          409600,
          1638400,
          1638400,
          819200,
          409600,
          409600,
          1638400,
          409600,
          409600,
          409600,
          409600,
          3686400,
          409600,
          3686400,
          819200,
          11059200,
          409600,
          409600,
          409600,
          1638400,
          819200,
          409600,
          409600,
          409600,
          409600,
          409600,
          3686400,
          3686400,
          819200,
          3276800,
          409600,
          1638400,
          3276800,
          409600,
          3604480,
          3276800,
          409600,
          1638400,
          409600,
          1638400,
          3276800,
          409600,
          409600,
          409600,
          1310720,
          409600,
          819200,
          409600,
          409600,
          819200,
          409600,
          409600,
          409600,
          3686400,
          409600,
          2457600,
          409600,
          409600,
          1638400,
          409600,
          1310720,
          1638400,
          1638400,
          3686400,
          3686400,
          409600,
          3276800,
          14745600,
          3276800,
          1310720,
          3276800,
          409600,
          409600,
          1638400,
          409600,
          7372800,
          5529600,
          3276800,
          409600,
          409600,
          3276800,
          3276800,
          1843200,
          409600,
          409600,
          1310720,
          409600,
          409600,
          409600,
          1228800,
          204800,
          819200,
          409600,
          409600,
          921600,
          921600,
          921600,
          614400,
          921600,
          921600,
          409600,
          2764800,
          921600,
          3686400,
          204800,
          204800,
          307200,
          1843200,
          921600,
          1638400,
          1843200,
          921600
        ]
      },
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_k": [
        82.2,
        82.2,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        82.1,
        82.1,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.0,
        82.1,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_k": [
        81.9,
        82.1,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v": [
        82.0,
        82.2,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q": [
        81.9,
        82.1,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v": [
        82.0,
        82.0,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q": [
        81.9,
        82.1,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k": [
        81.9,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q": [
        81.9,
        82.0,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q": [
        81.9,
        82.1,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        81.9,
        82.0,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.0,
        82.1,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k": [
        81.9,
        82.0,
        82.2,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        81.9,
        82.0,
        82.3,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_q": [
        81.8,
        82.0,
        82.1,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        81.9,
        82.0,
        82.1,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        82.1,
        82.0,
        82.3,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.0,
        82.0,
        82.0,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        81.8,
        82.0,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k": [
        81.8,
        81.9,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q": [
        81.9,
        82.1,
        82.1,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q": [
        81.9,
        82.1,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        81.8,
        82.0,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q": [
        81.8,
        82.1,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        81.9,
        82.0,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v": [
        81.8,
        82.0,
        81.9,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q": [
        81.6,
        82.1,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v": [
        81.7,
        81.8,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q": [
        81.8,
        81.9,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v": [
        81.9,
        82.0,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_q": [
        81.8,
        82.1,
        82.0,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_q": [
        81.7,
        81.9,
        82.0,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_q": [
        81.8,
        82.0,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_v": [
        81.7,
        81.9,
        81.9,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_k": [
        81.6,
        81.9,
        82.0,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        81.8,
        81.8,
        81.9,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q": [
        81.6,
        82.0,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q": [
        81.7,
        82.0,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        81.7,
        82.0,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k": [
        81.7,
        81.9,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        81.7,
        82.0,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v": [
        81.7,
        82.0,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_v": [
        81.6,
        81.9,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k": [
        81.7,
        81.9,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        81.6,
        81.8,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q": [
        81.6,
        81.9,
        81.9,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        81.6,
        81.9,
        81.9,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k": [
        81.7,
        81.7,
        81.9,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q": [
        81.7,
        81.8,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        81.8,
        81.8,
        81.8,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k": [
        81.7,
        81.8,
        81.9,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q": [
        81.9,
        81.8,
        81.8,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        81.9,
        81.8,
        81.8,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k": [
        81.7,
        81.8,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.6,
        81.7,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q": [
        81.6,
        81.8,
        81.8,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q": [
        81.7,
        81.6,
        81.8,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q": [
        81.6,
        81.7,
        81.7,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_q": [
        81.5,
        81.7,
        81.7,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_v": [
        81.7,
        81.6,
        81.8,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_q": [
        81.6,
        81.6,
        81.6,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_q": [
        81.7,
        81.8,
        81.9,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        81.7,
        81.8,
        81.7,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v": [
        81.7,
        81.6,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q": [
        81.5,
        81.6,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q": [
        81.7,
        81.8,
        81.8,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k": [
        81.7,
        81.6,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        81.7,
        81.7,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        81.6,
        81.6,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k": [
        81.6,
        81.6,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q": [
        81.8,
        81.6,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q": [
        81.6,
        81.6,
        81.5,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0": [
        81.6,
        81.5,
        81.7,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.5,
        81.5,
        81.7,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_k": [
        81.5,
        81.5,
        81.6,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_v": [
        81.4,
        81.5,
        81.6,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_k": [
        81.3,
        81.5,
        81.6,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_v": [
        81.4,
        81.5,
        81.5,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v": [
        81.4,
        81.6,
        81.5,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v": [
        81.4,
        81.5,
        81.5,
        81.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q": [
        81.4,
        81.4,
        81.5,
        81.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k": [
        81.4,
        81.4,
        81.5,
        81.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q": [
        81.5,
        81.3,
        81.6,
        81.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k": [
        81.5,
        81.4,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        81.3,
        81.4,
        81.6,
        81.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q": [
        81.3,
        81.4,
        81.4,
        81.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q": [
        81.4,
        81.4,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        81.3,
        81.3,
        81.4,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q": [
        81.3,
        81.3,
        81.4,
        81.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_v": [
        81.3,
        81.4,
        81.5,
        81.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_v": [
        81.3,
        81.6,
        81.5,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k": [
        81.5,
        81.3,
        81.5,
        81.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        81.5,
        81.4,
        81.5,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k": [
        81.4,
        81.3,
        81.6,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k": [
        81.4,
        81.3,
        81.5,
        81.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v": [
        81.2,
        81.4,
        81.4,
        81.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k": [
        81.3,
        81.3,
        81.4,
        81.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k": [
        81.3,
        81.3,
        81.3,
        81.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k": [
        81.3,
        81.3,
        81.4,
        81.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q": [
        81.2,
        81.2,
        81.4,
        81.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k": [
        81.3,
        81.3,
        81.4,
        81.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v": [
        81.3,
        81.2,
        81.3,
        81.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k": [
        81.3,
        81.3,
        81.5,
        81.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        81.2,
        81.3,
        81.4,
        80.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        81.3,
        81.2,
        81.4,
        81.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q": [
        81.2,
        81.3,
        81.1,
        80.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        81.1,
        81.1,
        81.2,
        80.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q": [
        81.1,
        81.2,
        81.3,
        80.9
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_k": [
        81.2,
        81.1,
        81.1,
        80.9
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        81.3,
        81.1,
        81.2,
        81.0
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_q": [
        81.2,
        81.1,
        81.2,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.2,
        81.1,
        81.1,
        80.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q": [
        81.2,
        81.0,
        81.0,
        80.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        81.2,
        81.1,
        81.1,
        80.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k": [
        81.1,
        81.1,
        81.0,
        80.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k": [
        81.1,
        81.1,
        81.0,
        80.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        81.2,
        81.0,
        81.1,
        80.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v": [
        81.0,
        80.9,
        81.1,
        80.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v": [
        81.1,
        80.9,
        81.0,
        80.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q": [
        81.0,
        80.9,
        81.0,
        80.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        81.1,
        80.8,
        81.1,
        80.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k": [
        81.0,
        81.0,
        81.0,
        80.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k": [
        81.0,
        81.0,
        81.0,
        80.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q": [
        81.0,
        80.8,
        81.0,
        81.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v": [
        81.1,
        81.0,
        81.1,
        80.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        81.1,
        81.0,
        81.0,
        80.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.0,
        80.8,
        81.0,
        80.9
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_k": [
        81.0,
        81.0,
        81.1,
        80.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v": [
        81.0,
        80.9,
        81.2,
        80.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k": [
        81.1,
        80.8,
        81.1,
        80.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        81.0,
        80.7,
        81.1,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        81.0,
        80.9,
        81.1,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q": [
        81.1,
        80.9,
        81.0,
        80.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        81.1,
        80.8,
        81.0,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k": [
        80.9,
        80.8,
        81.0,
        80.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        80.9,
        80.8,
        81.0,
        80.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v": [
        80.9,
        80.8,
        81.1,
        80.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        80.9,
        80.9,
        81.1,
        80.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q": [
        80.9,
        80.7,
        81.2,
        80.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q": [
        80.9,
        80.9,
        81.1,
        80.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k": [
        80.9,
        80.9,
        81.1,
        80.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k": [
        81.0,
        80.8,
        81.1,
        80.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q": [
        80.8,
        80.8,
        81.0,
        80.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q": [
        80.8,
        80.7,
        81.1,
        80.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q": [
        80.8,
        80.7,
        81.0,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q": [
        80.9,
        80.8,
        80.9,
        80.6
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        81.0,
        80.9,
        81.0,
        80.6
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_k": [
        81.0,
        80.9,
        81.0,
        80.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        80.8,
        80.8,
        81.0,
        80.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        81.0,
        80.8,
        81.0,
        80.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v": [
        80.9,
        80.7,
        81.0,
        80.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k": [
        80.9,
        80.7,
        80.9,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k": [
        80.8,
        80.7,
        81.0,
        80.6
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_q": [
        80.9,
        80.8,
        80.9,
        80.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q": [
        80.8,
        80.7,
        81.0,
        80.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q": [
        80.8,
        80.7,
        81.0,
        80.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        80.8,
        80.8,
        80.9,
        80.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        80.9,
        80.7,
        80.9,
        80.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        80.8,
        80.8,
        80.8,
        80.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q": [
        80.7,
        80.7,
        80.8,
        80.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k": [
        80.7,
        80.7,
        80.8,
        80.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k": [
        80.7,
        80.6,
        80.8,
        80.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v": [
        80.7,
        80.7,
        80.9,
        80.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        80.7,
        80.7,
        80.7,
        80.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v": [
        80.8,
        80.7,
        80.8,
        80.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q": [
        80.7,
        80.7,
        80.7,
        80.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        80.7,
        80.7,
        80.6,
        80.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q": [
        80.8,
        80.6,
        80.7,
        80.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k": [
        80.6,
        80.6,
        80.6,
        80.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        80.6,
        80.5,
        80.7,
        80.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k": [
        80.5,
        80.5,
        80.6,
        80.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q": [
        80.6,
        80.4,
        80.5,
        80.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q": [
        80.6,
        80.4,
        80.5,
        80.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0": [
        80.5,
        80.5,
        80.5,
        80.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k": [
        80.5,
        80.5,
        80.5,
        80.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v": [
        80.5,
        80.4,
        80.5,
        80.4
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_k": [
        80.4,
        80.5,
        80.5,
        80.4
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_v": [
        80.5,
        80.4,
        80.4,
        80.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k": [
        80.4,
        80.6,
        80.5,
        80.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k": [
        80.4,
        80.5,
        80.4,
        80.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q": [
        80.4,
        80.5,
        80.4,
        80.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        80.4,
        80.5,
        80.4,
        80.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v": [
        80.4,
        80.5,
        80.4,
        80.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k": [
        80.4,
        80.5,
        80.3,
        80.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q": [
        80.2,
        80.3,
        80.4,
        80.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v": [
        80.2,
        80.3,
        80.4,
        80.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k": [
        80.3,
        80.4,
        80.4,
        80.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v": [
        80.2,
        80.2,
        80.2,
        80.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k": [
        80.2,
        80.3,
        80.1,
        80.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k": [
        80.1,
        80.2,
        80.1,
        79.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q": [
        80.0,
        80.2,
        80.1,
        80.0
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        80.0,
        80.3,
        80.1,
        80.0
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_v": [
        80.0,
        80.1,
        80.1,
        80.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k": [
        79.9,
        80.2,
        80.0,
        80.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q": [
        79.9,
        80.2,
        80.1,
        80.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k": [
        79.9,
        80.1,
        80.0,
        80.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k": [
        79.9,
        80.1,
        80.0,
        80.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q": [
        79.9,
        80.1,
        80.0,
        80.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k": [
        79.9,
        79.8,
        79.8,
        79.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        79.6,
        79.7,
        79.9,
        79.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0": [
        79.6,
        79.8,
        79.9,
        79.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k": [
        79.7,
        79.7,
        79.8,
        79.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v": [
        79.6,
        79.6,
        79.9,
        79.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k": [
        79.6,
        79.7,
        79.9,
        79.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        79.7,
        79.6,
        79.9,
        79.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q": [
        79.7,
        79.6,
        79.8,
        79.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k": [
        79.5,
        79.6,
        79.7,
        79.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q": [
        79.5,
        79.5,
        79.6,
        79.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k": [
        79.5,
        79.6,
        79.6,
        79.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0": [
        79.5,
        79.5,
        79.6,
        79.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0": [
        79.5,
        79.5,
        79.6,
        79.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v": [
        79.3,
        79.5,
        79.5,
        79.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q": [
        79.5,
        79.5,
        79.5,
        79.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        79.4,
        79.5,
        79.5,
        79.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        79.4,
        79.5,
        79.5,
        79.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v": [
        79.3,
        79.5,
        79.5,
        79.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k": [
        79.3,
        79.4,
        79.4,
        79.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k": [
        79.1,
        79.3,
        79.5,
        79.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v": [
        79.2,
        79.3,
        79.5,
        79.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k": [
        79.2,
        79.4,
        79.6,
        79.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v": [
        79.2,
        79.3,
        79.5,
        79.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q": [
        79.1,
        79.3,
        79.4,
        79.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        79.2,
        79.3,
        79.4,
        79.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        79.1,
        79.4,
        79.3,
        79.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q": [
        79.3,
        79.3,
        79.4,
        79.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        79.1,
        79.3,
        79.5,
        79.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k": [
        79.2,
        79.3,
        79.4,
        79.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v": [
        79.3,
        79.3,
        79.3,
        79.4
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_q": [
        79.2,
        79.3,
        79.2,
        79.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k": [
        79.0,
        79.2,
        79.2,
        79.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        79.0,
        79.1,
        79.3,
        79.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        79.0,
        79.3,
        79.2,
        79.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v": [
        79.1,
        79.3,
        79.2,
        79.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q": [
        79.1,
        79.2,
        79.2,
        79.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q": [
        79.0,
        79.2,
        79.2,
        79.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q": [
        79.1,
        79.2,
        79.2,
        79.4
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_k": [
        79.0,
        79.1,
        79.2,
        79.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v": [
        79.1,
        79.2,
        79.3,
        79.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        79.1,
        79.2,
        79.3,
        79.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q": [
        79.2,
        79.2,
        79.2,
        79.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        79.1,
        79.2,
        79.3,
        79.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v": [
        79.0,
        79.1,
        79.2,
        79.4
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_q": [
        79.2,
        79.1,
        79.4,
        79.5
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_k": [
        79.0,
        79.2,
        79.4,
        79.4
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_v": [
        79.1,
        79.1,
        79.3,
        79.3
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_q": [
        79.0,
        79.0,
        79.3,
        79.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q": [
        79.1,
        79.0,
        79.3,
        79.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k": [
        79.0,
        79.0,
        79.2,
        79.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v": [
        79.0,
        79.0,
        79.3,
        79.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        79.0,
        78.9,
        79.3,
        79.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v": [
        78.9,
        79.0,
        79.2,
        79.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_k": [
        79.0,
        78.9,
        79.3,
        79.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v": [
        79.0,
        79.0,
        79.3,
        79.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v": [
        79.0,
        78.9,
        79.2,
        79.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        79.0,
        78.9,
        79.0,
        79.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        79.0,
        78.9,
        79.1,
        79.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k": [
        78.9,
        78.9,
        79.1,
        79.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q": [
        79.0,
        78.8,
        79.2,
        79.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2": [
        79.0,
        78.7,
        79.3,
        79.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2": [
        79.1,
        78.7,
        79.3,
        79.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q": [
        79.0,
        78.8,
        79.1,
        79.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k": [
        79.0,
        78.7,
        79.1,
        79.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k": [
        78.9,
        78.6,
        79.1,
        79.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        78.9,
        78.5,
        79.2,
        79.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0": [
        78.7,
        78.4,
        79.1,
        79.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0": [
        78.6,
        78.3,
        79.0,
        79.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v": [
        78.4,
        78.1,
        78.9,
        79.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_q": [
        78.4,
        78.2,
        79.0,
        79.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        78.4,
        78.2,
        79.0,
        79.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k": [
        78.3,
        78.2,
        78.9,
        78.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v": [
        78.3,
        78.2,
        78.8,
        79.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        78.4,
        78.3,
        78.9,
        78.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v": [
        78.2,
        78.3,
        78.7,
        78.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v": [
        78.2,
        78.3,
        78.6,
        78.8
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_v": [
        78.2,
        78.3,
        78.8,
        78.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v": [
        78.2,
        78.4,
        78.6,
        78.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k": [
        78.1,
        78.3,
        78.6,
        78.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        78.2,
        78.0,
        78.4,
        78.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q": [
        78.1,
        78.1,
        78.4,
        78.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2": [
        78.1,
        77.9,
        78.5,
        78.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2": [
        77.9,
        78.0,
        78.5,
        78.6
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_k": [
        78.0,
        78.0,
        78.5,
        78.5
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        78.1,
        78.1,
        78.4,
        78.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2": [
        78.1,
        77.9,
        78.5,
        78.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q": [
        78.1,
        77.9,
        78.5,
        78.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k": [
        78.0,
        78.0,
        78.4,
        78.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q": [
        78.0,
        78.0,
        78.4,
        78.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k": [
        77.8,
        77.9,
        78.4,
        78.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2": [
        77.6,
        78.0,
        78.4,
        78.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k": [
        77.6,
        77.9,
        78.3,
        78.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k": [
        77.8,
        77.9,
        78.4,
        78.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v": [
        77.8,
        77.9,
        78.3,
        78.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        77.7,
        77.8,
        78.3,
        78.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        77.7,
        77.9,
        78.3,
        78.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        77.8,
        77.9,
        78.3,
        78.5
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        77.8,
        78.0,
        78.3,
        78.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        77.7,
        77.9,
        78.3,
        78.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q": [
        77.9,
        77.9,
        78.3,
        78.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k": [
        77.9,
        78.0,
        78.3,
        78.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v": [
        77.8,
        77.8,
        78.3,
        78.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        77.7,
        77.9,
        78.4,
        78.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q": [
        77.8,
        77.9,
        78.3,
        78.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v": [
        77.6,
        77.9,
        78.1,
        78.4
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.2": [
        77.7,
        77.9,
        78.2,
        78.4
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_v": [
        77.7,
        77.9,
        78.2,
        78.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2": [
        77.7,
        77.8,
        78.2,
        78.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2": [
        77.7,
        77.9,
        78.2,
        78.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q": [
        77.7,
        77.8,
        78.2,
        78.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        77.5,
        77.8,
        77.9,
        78.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k": [
        77.5,
        77.7,
        77.9,
        78.2
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.2": [
        77.5,
        77.8,
        77.9,
        78.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        77.5,
        77.8,
        77.9,
        78.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.2": [
        77.6,
        77.8,
        78.0,
        78.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2": [
        77.6,
        77.7,
        77.9,
        78.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        77.7,
        77.8,
        77.8,
        78.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2": [
        77.7,
        77.7,
        77.8,
        78.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        77.7,
        77.8,
        77.7,
        78.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.2": [
        77.6,
        77.7,
        77.8,
        78.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k": [
        77.6,
        77.7,
        77.7,
        78.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2": [
        77.5,
        77.7,
        77.7,
        77.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        77.5,
        77.7,
        77.7,
        77.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        77.5,
        77.7,
        77.7,
        78.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0": [
        77.2,
        77.5,
        77.8,
        78.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2": [
        77.2,
        77.5,
        77.7,
        78.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v": [
        77.2,
        77.6,
        77.7,
        78.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2": [
        77.1,
        77.6,
        77.7,
        78.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v": [
        77.1,
        77.6,
        77.7,
        77.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        77.2,
        77.5,
        77.7,
        77.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k": [
        77.1,
        77.5,
        77.7,
        77.9
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_q": [
        77.1,
        77.4,
        77.7,
        78.0
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_k": [
        76.9,
        77.5,
        77.7,
        77.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        77.0,
        77.4,
        77.7,
        77.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        76.9,
        77.4,
        77.6,
        77.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2": [
        76.9,
        77.4,
        77.7,
        77.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2": [
        76.9,
        77.3,
        77.7,
        77.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v": [
        77.0,
        77.4,
        77.7,
        77.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v": [
        77.0,
        77.4,
        77.7,
        77.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q": [
        77.0,
        77.4,
        77.7,
        77.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k": [
        77.0,
        77.3,
        77.7,
        77.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v": [
        77.0,
        77.4,
        77.5,
        77.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        77.0,
        77.5,
        77.4,
        77.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k": [
        77.1,
        77.4,
        77.5,
        77.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2": [
        77.0,
        77.4,
        77.4,
        77.5
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        77.0,
        77.4,
        77.3,
        77.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2": [
        76.9,
        77.3,
        77.2,
        77.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        77.0,
        77.2,
        77.2,
        77.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k": [
        76.8,
        77.2,
        77.3,
        77.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k": [
        76.7,
        77.2,
        77.2,
        77.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k": [
        76.8,
        77.2,
        77.2,
        77.1
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_k": [
        76.7,
        77.2,
        77.2,
        77.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q": [
        76.6,
        77.1,
        77.2,
        77.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        76.5,
        77.2,
        77.2,
        77.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        76.3,
        77.1,
        77.1,
        77.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        76.5,
        77.1,
        77.1,
        77.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v": [
        76.3,
        77.1,
        76.8,
        76.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q": [
        76.3,
        77.1,
        76.8,
        76.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q": [
        76.4,
        77.0,
        76.8,
        76.9
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.2": [
        76.4,
        77.2,
        76.7,
        76.9
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_v": [
        76.3,
        77.1,
        76.7,
        76.9
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.2": [
        76.4,
        77.0,
        76.7,
        76.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v": [
        76.4,
        77.1,
        76.7,
        76.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q": [
        76.5,
        77.1,
        76.6,
        76.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v": [
        76.4,
        77.1,
        76.8,
        76.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k": [
        76.3,
        77.0,
        76.7,
        76.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v": [
        76.3,
        76.9,
        76.7,
        76.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        76.2,
        76.9,
        76.6,
        76.9
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        76.2,
        76.9,
        76.6,
        77.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        76.1,
        77.0,
        76.6,
        76.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        76.2,
        76.9,
        76.7,
        76.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        76.2,
        76.9,
        76.6,
        76.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        76.2,
        76.9,
        76.6,
        76.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2": [
        76.2,
        77.0,
        76.7,
        77.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q": [
        76.2,
        77.0,
        76.6,
        77.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q": [
        76.1,
        77.0,
        76.5,
        77.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k": [
        76.0,
        77.0,
        76.5,
        77.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        76.1,
        76.9,
        76.5,
        77.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        76.2,
        76.9,
        76.5,
        77.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        76.0,
        76.8,
        76.5,
        76.9
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_k": [
        76.1,
        76.9,
        76.5,
        76.9
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        76.1,
        76.8,
        76.6,
        77.0
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        76.1,
        76.7,
        76.6,
        76.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v": [
        76.1,
        76.7,
        76.6,
        76.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        76.0,
        76.7,
        76.5,
        76.7
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.2": [
        75.9,
        76.7,
        76.4,
        76.8
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        76.0,
        76.7,
        76.3,
        76.7
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_v": [
        76.0,
        76.7,
        76.5,
        76.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v": [
        75.9,
        76.7,
        76.3,
        76.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        76.0,
        76.7,
        76.5,
        76.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        75.8,
        76.7,
        76.4,
        76.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        76.0,
        76.6,
        76.6,
        76.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v": [
        75.8,
        76.5,
        76.1,
        76.5
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        75.8,
        76.5,
        76.1,
        76.5
      ],
      "mid_block.resnets.1.time_emb_proj": [
        75.8,
        76.5,
        76.1,
        76.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        75.7,
        76.6,
        76.0,
        76.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        75.5,
        76.5,
        76.1,
        76.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q": [
        75.6,
        76.5,
        76.1,
        76.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q": [
        75.5,
        76.4,
        76.0,
        76.5
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_q": [
        75.4,
        76.3,
        76.0,
        76.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v": [
        75.4,
        76.4,
        76.0,
        76.5
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        75.4,
        76.4,
        76.0,
        76.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2": [
        75.3,
        76.4,
        75.9,
        76.6
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_q": [
        75.3,
        76.4,
        76.0,
        76.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        75.3,
        76.3,
        75.9,
        76.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q": [
        75.2,
        76.2,
        75.9,
        76.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k": [
        75.1,
        76.2,
        75.8,
        76.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2": [
        75.1,
        76.2,
        75.8,
        76.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        75.0,
        76.0,
        75.8,
        76.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        74.8,
        75.9,
        75.7,
        76.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q": [
        74.7,
        75.9,
        75.5,
        76.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k": [
        74.8,
        75.9,
        75.7,
        76.1
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.2": [
        74.7,
        75.9,
        75.7,
        76.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        74.7,
        75.9,
        75.7,
        76.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        74.7,
        75.9,
        75.7,
        76.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        74.7,
        75.8,
        75.6,
        76.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        74.7,
        75.7,
        75.7,
        76.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        74.6,
        75.7,
        75.5,
        76.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k": [
        74.7,
        75.8,
        75.4,
        76.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k": [
        74.7,
        75.8,
        75.5,
        76.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k": [
        74.7,
        75.7,
        75.3,
        76.0
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        74.6,
        75.5,
        75.2,
        76.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v": [
        74.6,
        75.3,
        75.2,
        75.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        74.4,
        75.3,
        75.1,
        75.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v": [
        74.2,
        75.2,
        75.1,
        75.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k": [
        74.3,
        75.1,
        75.0,
        75.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v": [
        74.4,
        75.2,
        74.9,
        75.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        74.2,
        75.0,
        74.8,
        75.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v": [
        74.3,
        75.1,
        74.8,
        75.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k": [
        74.2,
        75.0,
        74.8,
        75.6
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        74.1,
        74.9,
        74.7,
        75.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2": [
        74.2,
        75.0,
        74.8,
        75.6
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_v": [
        74.3,
        75.1,
        74.8,
        75.6
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        74.2,
        75.1,
        74.7,
        75.6
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        74.3,
        75.0,
        74.7,
        75.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        74.2,
        75.0,
        74.7,
        75.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v": [
        74.2,
        74.9,
        74.6,
        75.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q": [
        74.0,
        74.7,
        74.5,
        75.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q": [
        74.1,
        74.6,
        74.4,
        75.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k": [
        73.9,
        74.7,
        74.3,
        75.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        73.9,
        74.6,
        74.4,
        75.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q": [
        74.0,
        74.6,
        74.4,
        75.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v": [
        73.8,
        74.4,
        74.1,
        74.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k": [
        73.7,
        74.3,
        74.0,
        74.6
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.2": [
        73.7,
        74.3,
        73.9,
        74.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2": [
        73.7,
        74.2,
        73.9,
        74.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2": [
        73.5,
        74.2,
        73.9,
        74.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v": [
        73.5,
        74.1,
        73.9,
        74.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        73.6,
        74.0,
        73.8,
        74.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k": [
        73.7,
        74.1,
        73.8,
        74.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q": [
        73.4,
        74.0,
        73.7,
        74.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v": [
        73.4,
        73.9,
        73.6,
        74.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2": [
        73.4,
        73.8,
        73.6,
        74.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        73.4,
        73.8,
        73.6,
        74.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v": [
        73.3,
        73.7,
        73.5,
        74.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        73.2,
        73.7,
        73.5,
        74.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q": [
        73.2,
        73.5,
        73.4,
        74.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        73.2,
        73.5,
        73.4,
        74.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        73.3,
        73.5,
        73.5,
        74.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v": [
        73.0,
        73.3,
        73.2,
        73.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        73.1,
        73.1,
        73.2,
        73.6
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_v": [
        73.2,
        73.1,
        73.1,
        73.6
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        73.1,
        73.1,
        73.2,
        73.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v": [
        73.3,
        73.3,
        73.1,
        73.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        73.3,
        73.3,
        73.0,
        73.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k": [
        72.9,
        73.1,
        72.9,
        73.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q": [
        73.0,
        73.1,
        73.0,
        73.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        72.9,
        73.1,
        72.9,
        73.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q": [
        72.9,
        72.9,
        72.7,
        73.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        72.9,
        72.8,
        72.7,
        73.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k": [
        72.9,
        72.7,
        72.6,
        73.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q": [
        72.7,
        72.6,
        72.5,
        73.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q": [
        72.4,
        72.5,
        72.4,
        72.9
      ],
      "mid_block.resnets.0.time_emb_proj": [
        72.2,
        72.4,
        72.2,
        72.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k": [
        72.1,
        72.3,
        72.0,
        72.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        72.2,
        72.3,
        72.1,
        72.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        72.2,
        72.3,
        72.1,
        72.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k": [
        72.1,
        72.2,
        71.9,
        72.5
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        72.1,
        72.1,
        71.9,
        72.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v": [
        71.9,
        72.0,
        71.8,
        72.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k": [
        71.9,
        71.9,
        71.5,
        71.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        71.8,
        71.8,
        71.4,
        71.9
      ],
      "mid_block.resnets.1.conv2": [
        71.9,
        71.9,
        71.5,
        72.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2": [
        72.1,
        71.9,
        71.7,
        72.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2": [
        72.1,
        71.9,
        71.6,
        72.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        72.2,
        72.0,
        71.7,
        72.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q": [
        72.2,
        72.0,
        71.6,
        72.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v": [
        71.2,
        71.3,
        70.8,
        71.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        71.2,
        71.3,
        70.7,
        71.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v": [
        71.1,
        71.3,
        70.7,
        71.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        71.1,
        71.3,
        70.7,
        71.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        71.2,
        71.1,
        70.6,
        71.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        71.2,
        71.2,
        70.6,
        71.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q": [
        71.2,
        71.1,
        70.6,
        71.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        71.1,
        71.0,
        70.6,
        71.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_k": [
        71.2,
        71.1,
        70.6,
        71.0
      ],
      "mid_block.resnets.1.conv1": [
        71.2,
        71.0,
        70.6,
        71.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        71.0,
        70.9,
        70.5,
        70.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v": [
        70.9,
        70.9,
        70.4,
        70.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        70.7,
        70.7,
        70.2,
        70.6
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_q": [
        70.8,
        70.8,
        70.1,
        70.6
      ],
      "down_blocks.2.resnets.1.conv1": [
        70.8,
        70.7,
        70.1,
        70.5
      ],
      "down_blocks.2.attentions.1.proj_in": [
        70.9,
        70.8,
        70.2,
        70.6
      ],
      "down_blocks.2.resnets.1.conv2": [
        70.8,
        70.7,
        70.1,
        70.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0": [
        70.7,
        70.6,
        70.0,
        70.5
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_v": [
        70.8,
        70.7,
        70.1,
        70.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        70.8,
        70.7,
        70.1,
        70.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        70.8,
        70.7,
        70.1,
        70.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        70.7,
        70.5,
        70.0,
        70.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v": [
        70.6,
        70.3,
        69.8,
        70.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        70.6,
        70.3,
        69.7,
        70.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        70.6,
        70.3,
        69.6,
        70.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        70.6,
        70.3,
        69.7,
        70.2
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        70.3,
        70.0,
        69.4,
        69.8
      ],
      "mid_block.resnets.0.conv1": [
        70.2,
        70.0,
        69.4,
        69.8
      ],
      "mid_block.resnets.0.conv2": [
        70.3,
        70.0,
        69.4,
        69.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        70.3,
        70.1,
        69.5,
        69.9
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        70.4,
        70.1,
        69.5,
        70.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        70.4,
        70.1,
        69.5,
        69.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2": [
        70.2,
        70.1,
        69.4,
        69.9
      ],
      "down_blocks.2.attentions.1.proj_out": [
        70.3,
        70.0,
        69.3,
        69.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        70.3,
        70.0,
        69.3,
        69.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        70.4,
        70.2,
        69.4,
        69.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v": [
        70.3,
        70.1,
        69.3,
        69.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0": [
        70.2,
        70.1,
        69.2,
        69.7
      ],
      "up_blocks.0.resnets.0.conv1": [
        70.1,
        70.1,
        69.1,
        69.7
      ],
      "up_blocks.0.resnets.1.conv1": [
        70.4,
        70.4,
        69.3,
        70.0
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        70.0,
        70.0,
        69.0,
        69.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v": [
        69.8,
        69.8,
        68.9,
        69.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2": [
        70.0,
        69.9,
        69.1,
        69.7
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        69.9,
        69.8,
        69.1,
        69.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        70.0,
        69.8,
        69.1,
        69.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2": [
        70.1,
        69.8,
        69.2,
        69.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v": [
        69.8,
        69.7,
        69.0,
        69.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2": [
        69.8,
        69.7,
        68.9,
        69.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v": [
        69.9,
        69.8,
        68.9,
        69.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2": [
        70.0,
        69.8,
        69.1,
        69.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v": [
        70.0,
        69.8,
        69.0,
        69.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        70.0,
        69.8,
        69.0,
        69.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v": [
        70.0,
        69.8,
        68.9,
        69.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        69.9,
        69.7,
        68.8,
        69.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        70.0,
        69.8,
        68.8,
        69.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        70.1,
        69.9,
        68.9,
        69.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2": [
        70.0,
        69.7,
        68.8,
        69.4
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        70.0,
        69.8,
        68.9,
        69.5
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_q": [
        70.0,
        69.8,
        68.9,
        69.5
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_k": [
        70.1,
        69.8,
        68.9,
        69.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2": [
        70.0,
        69.8,
        68.9,
        69.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0": [
        69.9,
        69.7,
        68.8,
        69.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v": [
        69.9,
        69.7,
        68.7,
        69.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2": [
        69.8,
        69.6,
        68.7,
        69.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        69.8,
        69.6,
        68.8,
        69.3
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        69.4,
        69.2,
        68.4,
        68.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        69.4,
        69.2,
        68.4,
        68.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        69.4,
        69.2,
        68.4,
        68.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        69.2,
        69.0,
        68.2,
        68.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v": [
        69.4,
        69.1,
        68.3,
        68.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v": [
        69.4,
        69.2,
        68.3,
        68.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v": [
        69.3,
        69.0,
        68.2,
        68.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0": [
        69.3,
        69.0,
        68.2,
        68.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2": [
        69.3,
        69.0,
        68.2,
        68.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0": [
        69.1,
        69.0,
        68.1,
        68.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        69.1,
        68.9,
        68.1,
        68.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0": [
        68.9,
        68.8,
        67.9,
        68.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v": [
        69.0,
        68.8,
        68.0,
        68.4
      ],
      "up_blocks.0.resnets.1.conv2": [
        69.1,
        69.0,
        68.1,
        68.5
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        69.2,
        69.1,
        68.1,
        68.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v": [
        69.1,
        68.9,
        68.0,
        68.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        68.8,
        68.7,
        67.9,
        68.2
      ],
      "mid_block.attentions.0.proj_out": [
        68.8,
        68.6,
        67.8,
        68.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        68.7,
        68.7,
        67.8,
        68.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        68.8,
        68.7,
        67.9,
        68.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2": [
        68.9,
        68.8,
        68.0,
        68.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        68.6,
        68.4,
        67.7,
        68.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v": [
        68.6,
        68.5,
        67.8,
        68.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2": [
        68.7,
        68.5,
        67.7,
        68.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_v": [
        68.7,
        68.4,
        67.8,
        68.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v": [
        68.7,
        68.4,
        67.7,
        68.1
      ],
      "up_blocks.0.attentions.0.proj_out": [
        68.7,
        68.4,
        67.8,
        68.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v": [
        68.6,
        68.4,
        67.7,
        68.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj": [
        68.8,
        68.4,
        67.9,
        68.3
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        68.8,
        68.4,
        67.9,
        68.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        68.8,
        68.5,
        68.0,
        68.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj": [
        68.9,
        68.5,
        68.0,
        68.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v": [
        68.9,
        68.5,
        67.9,
        68.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        68.6,
        68.3,
        67.7,
        68.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        68.6,
        68.2,
        67.7,
        68.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        68.4,
        68.2,
        67.6,
        68.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj": [
        68.3,
        68.1,
        67.4,
        67.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        68.3,
        68.1,
        67.5,
        68.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v": [
        68.1,
        67.9,
        67.2,
        67.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2": [
        68.1,
        67.9,
        67.3,
        67.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        68.1,
        67.9,
        67.3,
        67.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0": [
        68.2,
        67.9,
        67.3,
        67.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v": [
        68.2,
        67.8,
        67.3,
        67.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2": [
        68.1,
        67.9,
        67.3,
        67.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2": [
        68.1,
        67.9,
        67.4,
        68.0
      ],
      "up_blocks.0.resnets.0.conv2": [
        68.1,
        67.9,
        67.3,
        68.0
      ],
      "up_blocks.0.resnets.2.conv1": [
        68.3,
        68.0,
        67.4,
        68.1
      ],
      "up_blocks.0.resnets.2.conv2": [
        68.4,
        68.1,
        67.6,
        68.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        68.4,
        68.2,
        67.6,
        68.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        68.3,
        68.2,
        67.6,
        68.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj": [
        68.4,
        68.2,
        67.8,
        68.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj": [
        68.6,
        68.3,
        67.8,
        68.3
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        68.5,
        68.3,
        67.9,
        68.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2": [
        68.6,
        68.2,
        67.8,
        68.3
      ],
      "down_blocks.2.resnets.0.conv2": [
        68.5,
        68.2,
        67.8,
        68.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        68.5,
        68.2,
        67.8,
        68.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj": [
        68.4,
        68.1,
        67.7,
        68.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2": [
        68.2,
        67.9,
        67.6,
        68.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        68.1,
        67.8,
        67.5,
        67.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        67.9,
        67.5,
        67.3,
        67.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v": [
        67.6,
        67.3,
        67.1,
        67.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2": [
        67.5,
        67.0,
        67.0,
        67.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        67.4,
        67.1,
        67.1,
        67.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        67.4,
        67.2,
        67.1,
        67.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        67.3,
        67.0,
        67.0,
        67.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2": [
        67.2,
        66.7,
        67.0,
        67.3
      ],
      "mid_block.attentions.0.proj_in": [
        67.1,
        66.5,
        66.9,
        67.3
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        66.9,
        66.2,
        66.8,
        67.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        66.8,
        66.2,
        66.8,
        67.2
      ],
      "down_blocks.2.attentions.0.proj_in": [
        66.8,
        66.2,
        66.8,
        67.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj": [
        67.0,
        66.3,
        67.0,
        67.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        67.0,
        66.2,
        67.0,
        67.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        66.9,
        66.2,
        66.9,
        67.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        66.8,
        66.0,
        66.7,
        67.1
      ],
      "down_blocks.2.attentions.0.proj_out": [
        66.6,
        65.8,
        66.7,
        67.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        66.7,
        65.8,
        66.8,
        67.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v": [
        66.7,
        65.9,
        66.7,
        67.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        66.6,
        65.7,
        66.5,
        66.9
      ],
      "up_blocks.0.attentions.0.proj_in": [
        66.8,
        65.9,
        66.7,
        67.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        66.6,
        65.7,
        66.6,
        66.9
      ],
      "down_blocks.2.resnets.0.conv1": [
        66.5,
        65.7,
        66.6,
        66.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        66.4,
        65.6,
        66.5,
        66.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2": [
        66.3,
        65.4,
        66.4,
        66.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        66.3,
        65.5,
        66.5,
        66.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        66.5,
        65.4,
        66.4,
        66.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2": [
        66.6,
        65.8,
        66.6,
        67.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0": [
        66.5,
        65.6,
        66.4,
        66.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        66.4,
        65.5,
        66.3,
        66.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        66.4,
        65.5,
        66.3,
        66.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        66.2,
        65.5,
        66.3,
        66.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj": [
        66.4,
        65.5,
        66.3,
        66.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        66.4,
        65.5,
        66.3,
        66.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        66.2,
        65.4,
        66.3,
        66.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        66.3,
        65.4,
        66.3,
        66.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        66.2,
        65.3,
        66.0,
        66.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        65.9,
        65.0,
        65.8,
        66.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        65.8,
        64.9,
        65.7,
        65.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        65.6,
        64.5,
        65.4,
        65.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        65.5,
        64.4,
        65.4,
        65.6
      ],
      "up_blocks.0.attentions.2.proj_in": [
        65.3,
        64.3,
        65.2,
        65.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        65.2,
        64.1,
        65.0,
        65.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        64.6,
        63.5,
        64.4,
        64.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        64.7,
        63.6,
        64.3,
        64.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        64.6,
        63.5,
        64.2,
        64.3
      ],
      "up_blocks.0.attentions.1.proj_in": [
        65.1,
        63.9,
        64.3,
        64.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        64.9,
        63.8,
        64.3,
        64.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        64.7,
        63.5,
        64.2,
        64.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        64.6,
        63.5,
        64.0,
        64.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        64.6,
        63.7,
        64.1,
        64.2
      ],
      "up_blocks.0.attentions.2.proj_out": [
        64.4,
        63.5,
        63.9,
        64.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        64.3,
        63.4,
        63.9,
        63.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        64.1,
        63.2,
        63.6,
        63.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        64.0,
        63.2,
        63.6,
        63.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        63.8,
        62.9,
        63.4,
        63.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k": [
        63.8,
        63.0,
        63.3,
        63.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        63.7,
        62.9,
        63.1,
        63.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        63.5,
        62.7,
        63.0,
        63.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        63.3,
        62.6,
        62.9,
        62.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        63.2,
        62.5,
        62.7,
        62.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        62.9,
        62.2,
        62.4,
        62.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        62.7,
        62.0,
        62.2,
        62.3
      ],
      "up_blocks.0.attentions.1.proj_out": [
        62.5,
        62.0,
        62.1,
        62.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        62.3,
        61.7,
        61.9,
        61.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        62.3,
        61.6,
        61.8,
        61.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        62.0,
        61.4,
        61.5,
        61.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        61.7,
        61.1,
        61.2,
        61.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        61.5,
        60.9,
        61.1,
        61.0
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        61.5,
        60.9,
        61.0,
        60.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q": [
        61.6,
        61.0,
        61.0,
        61.0
      ],
      "add_embedding.linear_2": [
        61.5,
        60.6,
        60.9,
        60.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        61.4,
        60.5,
        60.8,
        60.8
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        61.0,
        60.2,
        60.5,
        60.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        60.7,
        60.0,
        60.3,
        60.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        60.5,
        59.8,
        60.0,
        60.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        60.4,
        59.7,
        60.0,
        60.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        60.1,
        59.4,
        59.8,
        59.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        60.1,
        59.4,
        59.8,
        59.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k": [
        59.7,
        58.9,
        59.4,
        59.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        59.4,
        58.6,
        59.1,
        59.0
      ],
      "up_blocks.1.resnets.0.conv2": [
        59.3,
        58.5,
        58.9,
        58.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        59.1,
        58.4,
        58.8,
        58.7
      ],
      "down_blocks.1.resnets.1.conv2": [
        59.0,
        58.2,
        58.6,
        58.6
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        58.8,
        58.1,
        58.4,
        58.4
      ],
      "up_blocks.1.resnets.0.conv1": [
        58.7,
        57.9,
        58.3,
        58.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        58.4,
        57.7,
        58.1,
        58.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q": [
        58.1,
        57.4,
        57.8,
        57.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        57.9,
        57.1,
        57.6,
        57.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        57.9,
        57.1,
        57.6,
        57.6
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        57.9,
        57.1,
        57.6,
        57.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        57.5,
        56.7,
        57.3,
        57.2
      ],
      "down_blocks.1.attentions.1.proj_out": [
        57.5,
        56.7,
        57.3,
        57.2
      ],
      "up_blocks.1.attentions.0.proj_out": [
        57.4,
        56.5,
        57.1,
        57.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        57.3,
        56.5,
        57.1,
        57.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        57.2,
        56.3,
        57.0,
        56.9
      ],
      "down_blocks.1.resnets.1.conv1": [
        57.1,
        56.2,
        56.9,
        56.8
      ],
      "down_blocks.1.resnets.0.conv2": [
        56.8,
        55.9,
        56.7,
        56.5
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        56.8,
        55.9,
        56.5,
        56.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        56.8,
        55.9,
        56.5,
        56.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        56.8,
        55.8,
        56.5,
        56.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        56.5,
        55.6,
        56.3,
        56.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        56.3,
        55.4,
        56.2,
        56.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        56.2,
        55.4,
        56.0,
        55.9
      ],
      "add_embedding.linear_1": [
        55.8,
        55.3,
        55.6,
        55.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        55.8,
        55.3,
        55.7,
        55.6
      ],
      "down_blocks.1.attentions.1.proj_in": [
        55.7,
        55.2,
        55.7,
        55.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2": [
        55.3,
        54.9,
        55.3,
        55.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        55.4,
        54.9,
        55.3,
        55.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        55.2,
        54.7,
        55.1,
        55.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        55.0,
        54.6,
        54.9,
        54.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        55.1,
        54.7,
        55.0,
        54.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        55.1,
        54.7,
        55.0,
        54.9
      ],
      "up_blocks.1.attentions.0.proj_in": [
        54.8,
        54.4,
        54.7,
        54.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        54.7,
        54.3,
        54.7,
        54.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        54.6,
        54.1,
        54.5,
        54.4
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        54.1,
        53.7,
        54.1,
        53.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        53.9,
        53.5,
        53.9,
        53.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        53.6,
        53.2,
        53.6,
        53.4
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        53.4,
        52.9,
        53.3,
        53.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v": [
        53.0,
        52.6,
        52.9,
        52.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        52.7,
        52.3,
        52.6,
        52.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        52.6,
        52.2,
        52.6,
        52.4
      ],
      "down_blocks.1.downsamplers.0.conv": [
        52.4,
        51.9,
        52.3,
        52.1
      ],
      "up_blocks.1.attentions.1.proj_out": [
        52.2,
        51.7,
        52.1,
        51.8
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        51.7,
        51.4,
        51.7,
        51.5
      ],
      "down_blocks.1.attentions.0.proj_out": [
        51.6,
        51.2,
        51.6,
        51.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        51.5,
        51.1,
        51.5,
        51.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        51.4,
        51.0,
        51.4,
        51.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        51.3,
        50.9,
        51.4,
        51.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k": [
        51.3,
        50.9,
        51.3,
        51.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        51.4,
        50.9,
        51.3,
        51.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        51.3,
        50.8,
        51.3,
        51.2
      ],
      "up_blocks.1.resnets.1.conv2": [
        51.2,
        50.8,
        51.2,
        51.2
      ],
      "up_blocks.1.resnets.2.conv2": [
        50.9,
        50.5,
        51.0,
        50.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        50.6,
        50.1,
        50.6,
        50.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        50.4,
        50.0,
        50.4,
        50.3
      ],
      "up_blocks.0.upsamplers.0.conv": [
        50.2,
        49.8,
        50.2,
        50.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        50.1,
        49.8,
        50.2,
        50.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        50.1,
        49.7,
        50.2,
        50.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        49.8,
        49.5,
        49.9,
        49.8
      ],
      "up_blocks.1.attentions.1.proj_in": [
        49.7,
        49.4,
        49.8,
        49.7
      ],
      "up_blocks.1.attentions.2.proj_out": [
        49.4,
        49.2,
        49.5,
        49.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        49.2,
        49.0,
        49.3,
        49.2
      ],
      "up_blocks.1.attentions.2.proj_in": [
        49.1,
        48.9,
        49.2,
        49.1
      ],
      "up_blocks.1.resnets.1.conv1": [
        49.1,
        49.0,
        49.2,
        49.2
      ],
      "up_blocks.1.resnets.2.conv1": [
        49.1,
        49.0,
        49.2,
        49.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        49.1,
        49.0,
        49.2,
        49.1
      ],
      "down_blocks.1.attentions.0.proj_in": [
        49.0,
        48.9,
        49.1,
        49.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        49.0,
        48.9,
        49.1,
        49.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        49.0,
        48.8,
        49.1,
        49.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        48.9,
        48.8,
        49.0,
        48.9
      ],
      "down_blocks.1.resnets.0.conv1": [
        48.6,
        48.5,
        48.6,
        48.5
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        48.5,
        48.4,
        48.5,
        48.4
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        48.1,
        48.1,
        48.2,
        48.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        48.0,
        47.9,
        48.1,
        48.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        48.0,
        47.9,
        48.0,
        48.0
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        47.8,
        47.8,
        47.9,
        47.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        47.7,
        47.7,
        47.8,
        47.7
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        47.7,
        47.8,
        47.9,
        47.8
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        47.7,
        47.7,
        47.9,
        47.7
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        47.2,
        47.2,
        47.4,
        47.3
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        46.7,
        46.8,
        47.0,
        46.9
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        46.4,
        46.5,
        46.6,
        46.5
      ],
      "down_blocks.0.resnets.0.conv1": [
        46.4,
        46.5,
        46.6,
        46.5
      ],
      "down_blocks.0.downsamplers.0.conv": [
        46.0,
        46.0,
        46.3,
        46.1
      ],
      "down_blocks.0.resnets.1.conv2": [
        45.8,
        45.8,
        46.0,
        45.8
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        44.8,
        44.9,
        45.0,
        44.8
      ],
      "down_blocks.0.resnets.1.conv1": [
        44.6,
        44.7,
        44.8,
        44.6
      ],
      "down_blocks.0.resnets.0.conv2": [
        44.4,
        44.4,
        44.6,
        44.3
      ],
      "time_embedding.linear_1": [
        43.8,
        43.8,
        43.9,
        43.5
      ],
      "up_blocks.2.resnets.0.conv1": [
        43.8,
        43.7,
        43.8,
        43.5
      ],
      "up_blocks.2.resnets.0.conv2": [
        44.1,
        44.1,
        44.1,
        43.9
      ],
      "up_blocks.1.upsamplers.0.conv": [
        43.7,
        43.7,
        43.7,
        43.5
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        44.2,
        44.2,
        44.2,
        44.1
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        43.7,
        43.7,
        43.8,
        43.6
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        43.3,
        43.2,
        43.3,
        43.1
      ],
      "up_blocks.2.resnets.1.conv1": [
        42.9,
        42.7,
        42.9,
        42.7
      ],
      "up_blocks.2.resnets.1.conv2": [
        42.7,
        42.6,
        42.8,
        42.7
      ],
      "time_embedding.linear_2": [
        43.1,
        42.9,
        43.1,
        43.1
      ],
      "up_blocks.2.resnets.2.conv1": [
        42.6,
        42.4,
        42.6,
        42.6
      ],
      "up_blocks.2.resnets.2.conv2": [
        42.1,
        42.0,
        42.0,
        42.0
      ]
    },
    "4": {
      "metadata": {
        "nbits": 4,
        "out_ngroups": 1,
        "in_ngroups": 1,
        "cumulative": true,
        "candidates": [
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.6.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.8.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.1.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.7.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.4.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.5.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj",
          "down_blocks.2.resnets.1.time_emb_proj",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.8.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.2.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_v",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.9.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q",
          "up_blocks.0.resnets.0.time_emb_proj",
          "mid_block.attentions.0.transformer_blocks.1.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_q",
          "mid_block.resnets.1.time_emb_proj",
          "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.3.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v",
          "down_blocks.2.attentions.1.proj_in",
          "down_blocks.2.resnets.1.conv2",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj",
          "mid_block.resnets.0.time_emb_proj",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_q",
          "mid_block.attentions.0.proj_out",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "mid_block.resnets.0.conv2",
          "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0",
          "mid_block.resnets.1.conv1",
          "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2",
          "up_blocks.0.resnets.0.conv2",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_k",
          "mid_block.resnets.0.conv1",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj",
          "down_blocks.2.attentions.1.proj_out",
          "down_blocks.2.resnets.0.time_emb_proj",
          "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "mid_block.resnets.1.conv2",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.resnets.1.conv1",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.resnets.1.conv1",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0",
          "down_blocks.2.resnets.0.conv2",
          "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2",
          "up_blocks.0.resnets.1.time_emb_proj",
          "up_blocks.0.resnets.2.time_emb_proj",
          "down_blocks.2.resnets.0.conv1",
          "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2",
          "up_blocks.0.attentions.0.proj_in",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0",
          "up_blocks.0.resnets.0.conv1",
          "mid_block.attentions.0.proj_in",
          "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2",
          "up_blocks.0.resnets.1.conv2",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.0.proj_out",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0",
          "up_blocks.0.resnets.2.conv2",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v",
          "down_blocks.2.attentions.0.proj_out",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v",
          "up_blocks.0.resnets.0.conv_shortcut",
          "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.resnets.2.conv1",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.attentions.0.proj_in",
          "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "down_blocks.1.resnets.0.time_emb_proj",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.0.attentions.2.proj_in",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.2.proj_out",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.1.proj_out",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.1.resnets.2.time_emb_proj",
          "up_blocks.1.resnets.0.conv1",
          "up_blocks.0.attentions.1.proj_in",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.resnets.1.conv2",
          "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2",
          "down_blocks.1.resnets.1.time_emb_proj",
          "up_blocks.1.resnets.1.time_emb_proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.resnets.0.conv2",
          "up_blocks.1.resnets.0.time_emb_proj",
          "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k",
          "down_blocks.1.resnets.0.conv2",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q",
          "add_embedding.linear_1",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q",
          "add_embedding.linear_2",
          "down_blocks.1.attentions.1.proj_out",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.resnets.1.conv_shortcut",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v",
          "up_blocks.1.attentions.0.proj_out",
          "time_embedding.linear_1",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "down_blocks.1.attentions.1.proj_in",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.2.resnets.0.time_emb_proj",
          "down_blocks.1.resnets.1.conv1",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.resnets.0.conv_shortcut",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v",
          "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2",
          "down_blocks.0.resnets.1.time_emb_proj",
          "up_blocks.1.attentions.1.proj_out",
          "up_blocks.1.attentions.0.proj_in",
          "up_blocks.1.resnets.2.conv2",
          "up_blocks.2.resnets.1.time_emb_proj",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "down_blocks.1.attentions.0.proj_out",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.1.resnets.1.conv2",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v",
          "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "down_blocks.0.resnets.0.time_emb_proj",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "down_blocks.1.downsamplers.0.conv",
          "up_blocks.1.attentions.2.proj_out",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.1.resnets.0.conv1",
          "up_blocks.2.resnets.2.time_emb_proj",
          "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.1.proj_in",
          "up_blocks.0.resnets.2.conv_shortcut",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.0.upsamplers.0.conv",
          "up_blocks.1.attentions.2.proj_in",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "down_blocks.1.attentions.0.proj_in",
          "up_blocks.1.resnets.1.conv1",
          "down_blocks.0.resnets.0.conv1",
          "up_blocks.1.resnets.0.conv_shortcut",
          "up_blocks.1.resnets.2.conv1",
          "down_blocks.1.resnets.0.conv_shortcut",
          "up_blocks.1.resnets.2.conv_shortcut",
          "up_blocks.2.resnets.2.conv_shortcut",
          "down_blocks.0.downsamplers.0.conv",
          "up_blocks.1.resnets.1.conv_shortcut",
          "time_embedding.linear_2",
          "down_blocks.0.resnets.1.conv1",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.upsamplers.0.conv",
          "down_blocks.0.resnets.1.conv2",
          "down_blocks.0.resnets.0.conv2",
          "up_blocks.2.resnets.0.conv_shortcut",
          "up_blocks.2.resnets.1.conv_shortcut",
          "up_blocks.2.resnets.2.conv2",
          "up_blocks.2.resnets.0.conv2",
          "up_blocks.2.resnets.0.conv1",
          "up_blocks.2.resnets.2.conv1",
          "up_blocks.2.resnets.1.conv1",
          "up_blocks.2.resnets.1.conv2"
        ],
        "sizes": [
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          2621440,
          2621440,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          6553600,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          6553600,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          2621440,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          6553600,
          6553600,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          6553600,
          13107200,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          2621440,
          6553600,
          6553600,
          1638400,
          6553600,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1310720,
          1638400,
          13107200,
          6553600,
          1638400,
          1638400,
          6553600,
          1638400,
          409600,
          1638400,
          13107200,
          6553600,
          13107200,
          2621440,
          13107200,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          13107200,
          1638400,
          13107200,
          1638400,
          1638400,
          409600,
          1638400,
          1310720,
          1638400,
          6553600,
          6553600,
          13107200,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          409600,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          409600,
          1310720,
          409600,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1310720,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          13107200,
          6553600,
          13107200,
          13107200,
          1638400,
          2621440,
          409600,
          6553600,
          13107200,
          13107200,
          6553600,
          1638400,
          1638400,
          13107200,
          1638400,
          2621440,
          1638400,
          1638400,
          1310720,
          13107200,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          6553600,
          1638400,
          13107200,
          13107200,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          14745600,
          1638400,
          13107200,
          13107200,
          6553600,
          13107200,
          13107200,
          1638400,
          1638400,
          13107200,
          1638400,
          13107200,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          6553600,
          6553600,
          1638400,
          13107200,
          14745600,
          13107200,
          6553600,
          13107200,
          6553600,
          409600,
          1310720,
          1638400,
          1638400,
          1638400,
          14745600,
          6553600,
          14745600,
          1310720,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          14745600,
          409600,
          13107200,
          6553600,
          1638400,
          6553600,
          1310720,
          1638400,
          13107200,
          13107200,
          1638400,
          1638400,
          13107200,
          1310720,
          14745600,
          409600,
          13107200,
          13107200,
          1638400,
          1638400,
          409600,
          1310720,
          29491200,
          1310720,
          1310720,
          14745600,
          1638400,
          14745600,
          6553600,
          6553600,
          1638400,
          1638400,
          7372800,
          6553600,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          29491200,
          1638400,
          6553600,
          1638400,
          6553600,
          6553600,
          14745600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          14745600,
          1638400,
          13107200,
          1310720,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          6553600,
          6553600,
          13107200,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          3276800,
          6553600,
          1638400,
          13107200,
          1638400,
          13107200,
          6553600,
          409600,
          6553600,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          13107200,
          409600,
          409600,
          13107200,
          6553600,
          1638400,
          6553600,
          1638400,
          22118400,
          409600,
          1638400,
          6553600,
          13107200,
          1310720,
          13107200,
          409600,
          13107200,
          6553600,
          6553600,
          13107200,
          1638400,
          13107200,
          1310720,
          1638400,
          13107200,
          1638400,
          409600,
          6553600,
          409600,
          13107200,
          409600,
          409600,
          13107200,
          6553600,
          409600,
          1638400,
          409600,
          409600,
          1310720,
          409600,
          819200,
          409600,
          409600,
          409600,
          1638400,
          1638400,
          409600,
          13107200,
          409600,
          409600,
          1638400,
          409600,
          1638400,
          409600,
          409600,
          409600,
          409600,
          409600,
          1638400,
          409600,
          819200,
          11059200,
          1638400,
          409600,
          409600,
          3686400,
          1638400,
          819200,
          819200,
          409600,
          3686400,
          819200,
          1638400,
          409600,
          409600,
          3686400,
          409600,
          3604480,
          1638400,
          409600,
          1638400,
          409600,
          1310720,
          409600,
          3276800,
          409600,
          409600,
          409600,
          409600,
          3276800,
          409600,
          409600,
          3276800,
          1638400,
          3276800,
          409600,
          3276800,
          409600,
          409600,
          409600,
          3686400,
          1310720,
          819200,
          409600,
          1638400,
          409600,
          1638400,
          409600,
          409600,
          409600,
          3686400,
          409600,
          1638400,
          409600,
          409600,
          409600,
          3686400,
          409600,
          3276800,
          409600,
          409600,
          1310720,
          409600,
          3686400,
          409600,
          3276800,
          1638400,
          1843200,
          409600,
          3276800,
          1638400,
          409600,
          2457600,
          3276800,
          1638400,
          409600,
          409600,
          14745600,
          409600,
          1310720,
          3276800,
          3276800,
          409600,
          7372800,
          921600,
          1228800,
          5529600,
          204800,
          614400,
          204800,
          921600,
          819200,
          1638400,
          921600,
          409600,
          3686400,
          921600,
          921600,
          307200,
          204800,
          921600,
          921600,
          2764800,
          1843200,
          1843200,
          921600
        ]
      },
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v": [
        82.0,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.4,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v": [
        82.3,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.0,
        82.3,
        82.2,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_v": [
        81.9,
        82.2,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        81.8,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        81.9,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q": [
        81.9,
        82.3,
        82.4,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_q": [
        81.8,
        82.4,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q": [
        81.9,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q": [
        81.9,
        82.3,
        82.5,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.2,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q": [
        81.9,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        81.9,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q": [
        82.0,
        82.1,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q": [
        81.9,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k": [
        81.9,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q": [
        81.9,
        82.3,
        82.2,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k": [
        81.9,
        82.1,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q": [
        81.9,
        82.1,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v": [
        81.9,
        82.1,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        81.9,
        82.2,
        82.2,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q": [
        81.9,
        82.3,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v": [
        81.9,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q": [
        82.0,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v": [
        82.0,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        82.0,
        82.3,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_k": [
        81.9,
        82.2,
        82.3,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_q": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_v": [
        81.7,
        82.3,
        82.3,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        81.9,
        82.1,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        82.0,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_k": [
        81.9,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k": [
        82.1,
        82.2,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k": [
        82.0,
        82.1,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k": [
        82.1,
        82.3,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q": [
        82.2,
        82.1,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        82.0,
        82.2,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k": [
        81.9,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k": [
        82.0,
        82.2,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q": [
        82.0,
        82.1,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k": [
        82.0,
        82.2,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v": [
        82.1,
        82.3,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q": [
        81.8,
        82.2,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k": [
        82.1,
        82.2,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        82.0,
        82.1,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k": [
        82.0,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q": [
        82.1,
        82.2,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.2,
        82.2,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        82.2,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.1,
        82.2,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        82.0,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        82.0,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q": [
        82.0,
        82.1,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q": [
        82.0,
        82.3,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q": [
        82.1,
        82.0,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v": [
        82.1,
        82.0,
        82.1,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_q": [
        81.9,
        82.0,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_k": [
        82.0,
        81.9,
        82.1,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_v": [
        82.0,
        81.9,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_k": [
        82.1,
        82.1,
        82.2,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_q": [
        82.0,
        82.0,
        82.1,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_k": [
        82.0,
        81.9,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_q": [
        82.1,
        81.8,
        82.3,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_v": [
        82.0,
        81.9,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q": [
        81.7,
        81.9,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        82.0,
        81.9,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q": [
        81.9,
        82.0,
        82.2,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_v": [
        81.9,
        81.9,
        82.2,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        81.9,
        81.9,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_q": [
        82.0,
        81.9,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_k": [
        81.8,
        81.9,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        81.8,
        81.8,
        82.3,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.8,
        82.0,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        81.8,
        81.9,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q": [
        82.0,
        81.9,
        82.1,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k": [
        82.0,
        81.7,
        82.1,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v": [
        81.7,
        82.1,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        81.8,
        81.9,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q": [
        81.9,
        81.9,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q": [
        81.9,
        81.9,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k": [
        82.0,
        82.0,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v": [
        81.9,
        82.0,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        81.9,
        82.0,
        81.9,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q": [
        81.8,
        82.1,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        81.9,
        81.7,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k": [
        81.9,
        81.8,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k": [
        82.0,
        82.1,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q": [
        81.9,
        81.9,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k": [
        81.9,
        82.1,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v": [
        81.9,
        81.9,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        81.9,
        82.1,
        82.2,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v": [
        81.9,
        81.9,
        82.1,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v": [
        81.8,
        82.0,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k": [
        81.9,
        82.0,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q": [
        81.8,
        81.9,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v": [
        81.9,
        81.7,
        82.3,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q": [
        81.8,
        81.8,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q": [
        81.9,
        81.9,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        81.9,
        81.9,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q": [
        81.8,
        81.7,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q": [
        81.9,
        81.9,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k": [
        81.9,
        81.7,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q": [
        81.8,
        81.9,
        81.9,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k": [
        81.9,
        81.8,
        81.8,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k": [
        81.9,
        81.6,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.0,
        81.8,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v": [
        82.0,
        82.0,
        82.2,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q": [
        81.9,
        81.8,
        81.8,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q": [
        82.0,
        81.9,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        81.9,
        81.8,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        81.9,
        81.8,
        82.1,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        81.8,
        81.6,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k": [
        81.7,
        82.0,
        82.0,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k": [
        81.7,
        82.0,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q": [
        81.8,
        81.8,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v": [
        81.9,
        82.0,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k": [
        81.7,
        81.7,
        82.0,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q": [
        81.7,
        81.8,
        82.0,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q": [
        81.8,
        81.5,
        82.0,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k": [
        81.8,
        81.7,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k": [
        81.8,
        81.6,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q": [
        81.8,
        81.6,
        82.0,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0": [
        81.7,
        81.6,
        81.9,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q": [
        81.7,
        81.7,
        81.9,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.8,
        81.8,
        82.1,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_k": [
        81.8,
        81.8,
        81.8,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_k": [
        81.7,
        81.7,
        82.0,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_q": [
        81.8,
        81.8,
        81.9,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_q": [
        81.7,
        81.7,
        81.9,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_k": [
        81.7,
        81.7,
        82.0,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_v": [
        81.9,
        81.8,
        81.8,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_k": [
        81.8,
        81.6,
        82.0,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_q": [
        81.9,
        81.8,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k": [
        81.9,
        81.8,
        81.9,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k": [
        81.8,
        81.7,
        81.9,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v": [
        81.7,
        81.7,
        81.9,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q": [
        81.8,
        81.7,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v": [
        81.8,
        81.7,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k": [
        81.7,
        81.6,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k": [
        81.7,
        81.8,
        82.0,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k": [
        81.8,
        81.7,
        81.8,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k": [
        81.8,
        81.8,
        81.9,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_v": [
        81.8,
        81.8,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        81.8,
        81.7,
        81.9,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_v": [
        81.7,
        81.7,
        81.8,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        81.7,
        81.8,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        81.8,
        81.7,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q": [
        81.8,
        81.7,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q": [
        81.8,
        81.7,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q": [
        81.7,
        81.8,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k": [
        81.7,
        81.7,
        82.1,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q": [
        81.6,
        81.6,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v": [
        81.7,
        81.7,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k": [
        81.7,
        81.8,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q": [
        81.5,
        81.7,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k": [
        81.6,
        81.7,
        82.1,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v": [
        81.6,
        81.6,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        81.5,
        81.6,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        81.6,
        81.7,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q": [
        81.7,
        81.7,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k": [
        81.7,
        81.7,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v": [
        81.6,
        81.7,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k": [
        81.6,
        81.8,
        81.9,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.6,
        81.7,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q": [
        81.6,
        81.6,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k": [
        81.5,
        81.5,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k": [
        81.6,
        81.6,
        81.8,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k": [
        81.5,
        81.7,
        81.9,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q": [
        81.5,
        81.6,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k": [
        81.6,
        81.8,
        81.8,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q": [
        81.6,
        81.7,
        81.7,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k": [
        81.7,
        81.5,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k": [
        81.6,
        81.7,
        81.8,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v": [
        81.7,
        81.6,
        81.7,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k": [
        81.8,
        81.5,
        81.8,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q": [
        81.7,
        81.6,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q": [
        81.6,
        81.5,
        81.8,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        81.6,
        81.6,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k": [
        81.6,
        81.6,
        81.9,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        81.6,
        81.6,
        81.9,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v": [
        81.6,
        81.8,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v": [
        81.6,
        81.6,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q": [
        81.5,
        81.6,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q": [
        81.4,
        81.5,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k": [
        81.4,
        81.7,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v": [
        81.4,
        81.6,
        81.9,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0": [
        81.4,
        81.7,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k": [
        81.5,
        81.6,
        81.8,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_v": [
        81.6,
        81.5,
        81.9,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_q": [
        81.5,
        81.7,
        81.9,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_k": [
        81.5,
        81.6,
        81.7,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_k": [
        81.6,
        81.7,
        81.8,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        81.5,
        81.6,
        81.7,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        81.6,
        81.6,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k": [
        81.5,
        81.6,
        81.9,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v": [
        81.5,
        81.6,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q": [
        81.5,
        81.5,
        81.8,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v": [
        81.5,
        81.6,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        81.4,
        81.6,
        81.8,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k": [
        81.4,
        81.5,
        81.9,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        81.4,
        81.5,
        81.8,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        81.4,
        81.6,
        81.8,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        81.5,
        81.5,
        81.7,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q": [
        81.5,
        81.5,
        81.7,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        81.4,
        81.6,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q": [
        81.3,
        81.5,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        81.4,
        81.5,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q": [
        81.4,
        81.6,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k": [
        81.3,
        81.6,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k": [
        81.3,
        81.5,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q": [
        81.3,
        81.5,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        81.3,
        81.5,
        81.6,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k": [
        81.3,
        81.4,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        81.3,
        81.5,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k": [
        81.3,
        81.5,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k": [
        81.3,
        81.4,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0": [
        81.3,
        81.4,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q": [
        81.3,
        81.4,
        81.7,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_q": [
        81.4,
        81.4,
        81.6,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_q": [
        81.3,
        81.4,
        81.8,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        81.3,
        81.4,
        81.6,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.2": [
        81.2,
        81.5,
        81.7,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        81.2,
        81.4,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k": [
        81.2,
        81.4,
        81.8,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        81.3,
        81.4,
        81.7,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q": [
        81.3,
        81.4,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        81.3,
        81.4,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        81.2,
        81.4,
        81.7,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q": [
        81.3,
        81.3,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v": [
        81.2,
        81.3,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2": [
        81.1,
        81.4,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k": [
        81.2,
        81.3,
        81.6,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k": [
        81.2,
        81.1,
        81.6,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        81.3,
        81.3,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.2,
        81.3,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k": [
        81.2,
        81.4,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k": [
        81.2,
        81.2,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q": [
        81.3,
        81.2,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k": [
        81.3,
        81.3,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v": [
        81.2,
        81.2,
        81.5,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0": [
        81.3,
        81.1,
        81.5,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v": [
        81.2,
        81.2,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0": [
        81.1,
        81.1,
        81.6,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        81.1,
        81.2,
        81.6,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.2": [
        81.2,
        81.1,
        81.5,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_v": [
        81.1,
        81.1,
        81.6,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_k": [
        81.1,
        81.1,
        81.7,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.1,
        81.3,
        81.6,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k": [
        81.2,
        81.1,
        81.5,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        81.1,
        81.2,
        81.7,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        81.3,
        81.1,
        81.7,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v": [
        81.2,
        81.1,
        81.6,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        81.2,
        81.1,
        81.6,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2": [
        81.1,
        81.1,
        81.4,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        81.0,
        81.2,
        81.6,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2": [
        81.2,
        81.2,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v": [
        81.0,
        81.3,
        81.5,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        81.1,
        81.2,
        81.4,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        81.1,
        80.9,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q": [
        81.0,
        80.8,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v": [
        81.0,
        80.9,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        81.0,
        80.9,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k": [
        81.1,
        80.9,
        81.6,
        81.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k": [
        81.1,
        80.9,
        81.6,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.2": [
        81.0,
        81.0,
        81.6,
        81.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q": [
        81.0,
        81.0,
        81.5,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v": [
        80.9,
        81.1,
        81.5,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        80.9,
        81.0,
        81.4,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        80.9,
        81.2,
        81.5,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k": [
        81.0,
        81.0,
        81.5,
        81.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        80.9,
        80.9,
        81.4,
        81.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v": [
        81.1,
        80.9,
        81.5,
        81.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        80.9,
        81.2,
        81.4,
        81.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v": [
        80.9,
        81.0,
        81.4,
        81.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q": [
        80.9,
        81.0,
        81.4,
        81.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v": [
        80.9,
        81.0,
        81.5,
        81.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2": [
        80.8,
        81.1,
        81.4,
        81.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q": [
        80.8,
        80.9,
        81.4,
        81.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q": [
        80.9,
        81.2,
        81.3,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        80.9,
        81.0,
        81.5,
        81.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q": [
        81.0,
        81.0,
        81.4,
        81.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k": [
        81.0,
        80.8,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k": [
        81.0,
        80.9,
        81.4,
        81.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v": [
        81.0,
        80.9,
        81.5,
        81.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k": [
        81.0,
        80.8,
        81.4,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0": [
        81.0,
        80.8,
        81.5,
        81.0
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_q": [
        80.9,
        81.0,
        81.4,
        81.1
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_k": [
        80.9,
        80.7,
        81.4,
        81.0
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        81.0,
        80.9,
        81.4,
        81.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.2": [
        80.9,
        81.0,
        81.4,
        81.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_q": [
        81.0,
        81.1,
        81.4,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k": [
        81.0,
        80.9,
        81.4,
        81.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2": [
        80.9,
        81.0,
        81.4,
        81.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        80.9,
        81.0,
        81.3,
        81.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v": [
        81.0,
        81.0,
        81.3,
        81.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        80.9,
        80.9,
        81.3,
        81.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v": [
        81.0,
        80.9,
        81.4,
        81.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2": [
        80.9,
        80.8,
        81.4,
        81.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v": [
        81.0,
        80.8,
        81.2,
        81.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2": [
        80.9,
        81.1,
        81.2,
        81.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2": [
        80.9,
        81.0,
        81.3,
        81.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        80.8,
        81.0,
        81.1,
        81.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k": [
        80.8,
        81.0,
        81.2,
        81.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k": [
        80.9,
        80.9,
        81.2,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k": [
        81.0,
        80.9,
        81.1,
        81.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0": [
        80.8,
        80.9,
        81.2,
        80.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q": [
        80.8,
        80.8,
        81.3,
        80.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k": [
        80.8,
        80.9,
        81.3,
        80.9
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_k": [
        80.8,
        81.0,
        81.2,
        81.0
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_v": [
        80.8,
        80.8,
        81.2,
        80.9
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        81.0,
        80.9,
        81.2,
        80.8
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_q": [
        81.0,
        80.9,
        81.2,
        81.0
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        81.0,
        80.9,
        81.2,
        80.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        80.9,
        80.8,
        81.3,
        81.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2": [
        80.9,
        80.9,
        81.2,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k": [
        80.8,
        80.9,
        81.3,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k": [
        80.8,
        80.8,
        81.2,
        81.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v": [
        80.8,
        81.0,
        81.2,
        81.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k": [
        80.9,
        81.0,
        81.2,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        80.8,
        80.9,
        81.2,
        80.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2": [
        80.7,
        80.8,
        81.2,
        80.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        80.8,
        80.6,
        81.1,
        80.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        80.8,
        80.7,
        81.2,
        81.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        80.8,
        80.8,
        81.1,
        80.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q": [
        80.8,
        80.8,
        81.2,
        80.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        80.8,
        80.8,
        81.1,
        80.8
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        80.8,
        80.7,
        81.0,
        80.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k": [
        80.8,
        80.7,
        81.2,
        80.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2": [
        80.9,
        80.7,
        81.0,
        80.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2": [
        80.8,
        80.8,
        81.1,
        80.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v": [
        80.8,
        80.9,
        81.0,
        80.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2": [
        80.8,
        80.7,
        81.2,
        80.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q": [
        80.8,
        80.7,
        81.1,
        80.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v": [
        80.9,
        80.8,
        81.0,
        80.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k": [
        80.9,
        80.7,
        81.0,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k": [
        80.9,
        80.8,
        81.1,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q": [
        80.8,
        80.8,
        80.9,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q": [
        80.8,
        80.8,
        81.0,
        80.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q": [
        80.9,
        80.9,
        81.0,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k": [
        80.7,
        80.8,
        81.0,
        80.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        80.8,
        80.7,
        81.1,
        80.7
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_k": [
        80.8,
        80.7,
        81.0,
        80.8
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        80.9,
        80.8,
        81.0,
        80.8
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.2": [
        80.7,
        80.8,
        81.1,
        80.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        80.7,
        80.7,
        81.1,
        80.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q": [
        80.9,
        80.7,
        81.1,
        80.8
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.2": [
        80.7,
        80.7,
        80.9,
        80.9
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_v": [
        80.8,
        80.6,
        81.1,
        80.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        80.8,
        80.5,
        80.9,
        80.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v": [
        80.8,
        80.7,
        81.0,
        80.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        80.9,
        80.6,
        81.0,
        80.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2": [
        80.9,
        80.8,
        80.9,
        80.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        80.8,
        80.7,
        80.9,
        80.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v": [
        80.8,
        80.8,
        80.9,
        80.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        80.8,
        80.7,
        80.9,
        80.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v": [
        80.8,
        80.7,
        81.0,
        80.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k": [
        80.7,
        80.7,
        80.9,
        80.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v": [
        80.7,
        80.7,
        80.8,
        80.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q": [
        80.7,
        80.6,
        80.9,
        80.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k": [
        80.8,
        80.6,
        80.9,
        80.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v": [
        80.7,
        80.7,
        80.8,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k": [
        80.8,
        80.7,
        80.9,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k": [
        80.7,
        80.6,
        81.0,
        80.8
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_v": [
        80.7,
        80.6,
        81.0,
        80.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q": [
        80.7,
        80.5,
        81.0,
        80.7
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.2": [
        80.8,
        80.7,
        80.9,
        80.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k": [
        80.7,
        80.8,
        80.8,
        80.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        80.7,
        80.9,
        80.9,
        80.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        80.6,
        80.8,
        80.9,
        80.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        80.8,
        80.8,
        80.9,
        80.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v": [
        80.7,
        80.7,
        81.0,
        80.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q": [
        80.7,
        80.7,
        80.9,
        80.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v": [
        80.7,
        80.7,
        80.9,
        80.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q": [
        80.7,
        80.7,
        81.0,
        80.7
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        80.7,
        80.7,
        81.0,
        80.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v": [
        80.7,
        80.8,
        80.9,
        80.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k": [
        80.6,
        80.9,
        80.8,
        80.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        80.7,
        80.6,
        80.8,
        80.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v": [
        80.6,
        80.5,
        80.9,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v": [
        80.7,
        80.6,
        80.9,
        80.6
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_q": [
        80.7,
        80.6,
        80.8,
        80.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q": [
        80.7,
        80.7,
        80.7,
        80.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q": [
        80.5,
        80.6,
        80.8,
        80.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v": [
        80.6,
        80.6,
        80.8,
        80.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2": [
        80.6,
        80.6,
        80.9,
        80.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q": [
        80.6,
        80.4,
        80.8,
        80.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v": [
        80.6,
        80.5,
        80.8,
        80.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0": [
        80.6,
        80.4,
        80.6,
        80.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q": [
        80.6,
        80.4,
        80.7,
        80.5
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        80.6,
        80.3,
        80.7,
        80.5
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.2": [
        80.6,
        80.5,
        80.7,
        80.4
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        80.5,
        80.4,
        80.7,
        80.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        80.6,
        80.4,
        80.7,
        80.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v": [
        80.6,
        80.5,
        80.6,
        80.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        80.6,
        80.4,
        80.6,
        80.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        80.6,
        80.4,
        80.6,
        80.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q": [
        80.6,
        80.5,
        80.6,
        80.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        80.5,
        80.5,
        80.5,
        80.5
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_k": [
        80.5,
        80.4,
        80.6,
        80.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        80.4,
        80.2,
        80.5,
        80.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        80.5,
        80.4,
        80.5,
        80.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        80.5,
        80.4,
        80.5,
        80.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2": [
        80.5,
        80.3,
        80.4,
        80.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        80.5,
        80.3,
        80.5,
        80.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        80.5,
        80.4,
        80.5,
        80.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        80.5,
        80.3,
        80.3,
        80.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v": [
        80.5,
        80.2,
        80.3,
        80.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        80.4,
        80.1,
        80.4,
        80.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k": [
        80.4,
        80.3,
        80.3,
        80.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q": [
        80.4,
        80.3,
        80.3,
        80.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k": [
        80.4,
        80.2,
        80.4,
        80.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k": [
        80.3,
        80.3,
        80.3,
        80.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        80.3,
        80.2,
        80.3,
        80.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_v": [
        80.3,
        80.2,
        80.3,
        80.3
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        80.3,
        80.2,
        80.3,
        80.3
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        80.3,
        80.1,
        80.3,
        80.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_v": [
        80.3,
        80.1,
        80.2,
        80.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        80.4,
        80.2,
        80.3,
        80.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        80.2,
        80.2,
        80.0,
        80.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        80.2,
        80.1,
        80.1,
        80.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        80.1,
        80.0,
        80.0,
        79.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        80.0,
        80.0,
        80.1,
        80.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        80.1,
        80.0,
        79.9,
        80.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v": [
        80.1,
        80.0,
        80.0,
        80.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        80.1,
        79.9,
        80.0,
        79.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v": [
        80.2,
        80.0,
        79.9,
        79.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v": [
        80.1,
        80.0,
        79.9,
        79.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        80.1,
        79.9,
        79.9,
        80.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k": [
        80.0,
        79.9,
        80.0,
        79.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q": [
        80.0,
        80.0,
        79.9,
        79.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v": [
        80.0,
        80.0,
        79.7,
        79.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        80.0,
        79.9,
        79.8,
        79.8
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        80.0,
        79.9,
        79.8,
        79.9
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_q": [
        79.9,
        79.9,
        79.8,
        79.8
      ],
      "mid_block.resnets.1.time_emb_proj": [
        80.0,
        79.9,
        79.7,
        79.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2": [
        79.9,
        80.0,
        79.7,
        79.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k": [
        80.0,
        79.9,
        79.8,
        79.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        79.9,
        79.9,
        79.7,
        79.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        79.9,
        79.9,
        79.7,
        79.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2": [
        79.9,
        80.0,
        79.6,
        79.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        79.9,
        79.8,
        79.7,
        79.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        80.0,
        79.8,
        79.6,
        79.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q": [
        79.9,
        79.8,
        79.7,
        79.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v": [
        79.8,
        79.7,
        79.5,
        79.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        79.8,
        79.7,
        79.5,
        79.6
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.2": [
        79.8,
        79.7,
        79.4,
        79.6
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        79.7,
        79.7,
        79.4,
        79.6
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        79.9,
        79.7,
        79.5,
        79.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2": [
        79.7,
        79.7,
        79.5,
        79.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        79.8,
        79.7,
        79.5,
        79.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v": [
        79.7,
        79.7,
        79.4,
        79.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        79.7,
        79.7,
        79.4,
        79.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v": [
        79.8,
        79.7,
        79.3,
        79.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        79.7,
        79.6,
        79.2,
        79.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q": [
        79.7,
        79.6,
        79.2,
        79.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k": [
        79.7,
        79.6,
        79.3,
        79.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        79.7,
        79.6,
        79.4,
        79.3
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        79.6,
        79.5,
        79.2,
        79.1
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_v": [
        79.6,
        79.5,
        79.3,
        79.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_v": [
        79.6,
        79.5,
        79.3,
        79.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        79.7,
        79.4,
        79.2,
        79.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v": [
        79.5,
        79.4,
        79.1,
        79.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v": [
        79.6,
        79.4,
        79.1,
        79.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2": [
        79.5,
        79.4,
        79.1,
        79.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v": [
        79.6,
        79.5,
        79.2,
        79.1
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        79.5,
        79.3,
        79.1,
        79.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        79.4,
        79.4,
        79.1,
        79.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        79.4,
        79.5,
        79.2,
        79.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        79.5,
        79.4,
        79.0,
        79.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        79.3,
        79.4,
        79.2,
        79.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v": [
        79.4,
        79.4,
        79.1,
        79.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v": [
        79.4,
        79.5,
        79.1,
        79.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        79.4,
        79.4,
        79.1,
        79.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        79.4,
        79.4,
        79.1,
        79.1
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_v": [
        79.5,
        79.4,
        79.1,
        79.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v": [
        79.6,
        79.4,
        79.1,
        79.0
      ],
      "down_blocks.2.attentions.1.proj_in": [
        79.5,
        79.5,
        79.0,
        79.1
      ],
      "down_blocks.2.resnets.1.conv2": [
        79.5,
        79.5,
        79.0,
        79.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        79.6,
        79.5,
        78.9,
        79.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        79.4,
        79.4,
        79.0,
        79.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        79.5,
        79.4,
        79.0,
        79.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2": [
        79.5,
        79.4,
        79.0,
        79.0
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        79.5,
        79.3,
        79.0,
        78.9
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        79.6,
        79.4,
        79.0,
        79.0
      ],
      "mid_block.resnets.0.time_emb_proj": [
        79.4,
        79.4,
        78.9,
        78.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        79.4,
        79.3,
        78.9,
        78.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        79.4,
        79.3,
        78.7,
        78.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v": [
        79.4,
        79.3,
        78.6,
        78.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        79.4,
        79.2,
        78.6,
        78.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        79.4,
        79.2,
        78.8,
        78.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        79.4,
        79.3,
        78.7,
        78.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        79.2,
        79.1,
        78.6,
        78.7
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_q": [
        79.2,
        79.1,
        78.6,
        78.7
      ],
      "mid_block.attentions.0.proj_out": [
        79.4,
        79.1,
        78.6,
        78.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        79.2,
        79.0,
        78.5,
        78.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2": [
        79.3,
        79.0,
        78.5,
        78.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        79.1,
        78.9,
        78.5,
        78.6
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        79.1,
        79.1,
        78.5,
        78.6
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        79.1,
        78.9,
        78.4,
        78.6
      ],
      "mid_block.resnets.0.conv2": [
        79.0,
        78.8,
        78.5,
        78.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        79.0,
        78.9,
        78.5,
        78.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2": [
        79.1,
        78.9,
        78.5,
        78.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        79.0,
        78.8,
        78.5,
        78.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2": [
        79.1,
        78.8,
        78.4,
        78.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        79.1,
        78.7,
        78.5,
        78.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        78.9,
        78.6,
        78.3,
        78.4
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        79.0,
        78.7,
        78.5,
        78.5
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        79.1,
        78.7,
        78.4,
        78.5
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        79.2,
        78.6,
        78.3,
        78.6
      ],
      "mid_block.resnets.1.conv1": [
        79.2,
        78.8,
        78.3,
        78.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2": [
        79.1,
        78.8,
        78.4,
        78.7
      ],
      "up_blocks.0.resnets.0.conv2": [
        79.1,
        78.6,
        78.3,
        78.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        79.1,
        78.5,
        78.2,
        78.4
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        79.1,
        78.5,
        78.3,
        78.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        79.1,
        78.5,
        78.3,
        78.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        79.1,
        78.5,
        78.2,
        78.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        79.0,
        78.5,
        78.2,
        78.5
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_k": [
        78.9,
        78.5,
        78.2,
        78.4
      ],
      "mid_block.resnets.0.conv1": [
        78.9,
        78.5,
        78.1,
        78.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        78.8,
        78.4,
        78.1,
        78.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        78.9,
        78.3,
        78.2,
        78.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2": [
        78.9,
        78.3,
        78.1,
        78.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        78.8,
        78.3,
        78.0,
        78.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2": [
        78.9,
        78.2,
        78.1,
        78.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        78.8,
        78.3,
        78.0,
        78.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        78.8,
        78.3,
        78.1,
        78.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        78.7,
        78.2,
        78.0,
        78.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        78.6,
        78.3,
        78.0,
        78.2
      ],
      "down_blocks.2.attentions.1.proj_out": [
        78.7,
        78.2,
        77.9,
        78.2
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        78.6,
        78.2,
        78.0,
        78.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        78.6,
        78.3,
        78.1,
        78.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        78.5,
        78.1,
        78.0,
        78.0
      ],
      "mid_block.resnets.1.conv2": [
        78.5,
        78.0,
        78.0,
        78.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        78.5,
        78.1,
        77.8,
        77.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        78.6,
        78.1,
        77.9,
        78.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        78.5,
        78.1,
        78.0,
        78.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        78.3,
        78.0,
        77.8,
        77.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        78.4,
        77.9,
        77.9,
        78.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        78.2,
        77.9,
        77.8,
        77.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v": [
        78.2,
        77.9,
        77.8,
        77.9
      ],
      "up_blocks.0.resnets.1.conv1": [
        78.1,
        77.7,
        77.7,
        77.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        78.2,
        77.8,
        77.7,
        77.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        78.1,
        77.8,
        77.7,
        77.8
      ],
      "down_blocks.2.resnets.1.conv1": [
        78.1,
        77.8,
        77.8,
        77.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        78.1,
        77.6,
        77.8,
        77.8
      ],
      "down_blocks.2.resnets.0.conv2": [
        78.1,
        77.6,
        77.8,
        77.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2": [
        77.9,
        77.8,
        77.7,
        77.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2": [
        78.0,
        77.7,
        77.7,
        77.7
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        78.0,
        77.7,
        77.8,
        77.6
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        77.9,
        77.6,
        77.7,
        77.7
      ],
      "down_blocks.2.resnets.0.conv1": [
        77.9,
        77.6,
        77.7,
        77.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2": [
        77.8,
        77.6,
        77.6,
        77.5
      ],
      "up_blocks.0.attentions.0.proj_in": [
        77.8,
        77.6,
        77.7,
        77.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0": [
        77.8,
        77.6,
        77.7,
        77.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v": [
        77.8,
        77.6,
        77.6,
        77.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2": [
        77.8,
        77.5,
        77.6,
        77.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0": [
        77.7,
        77.6,
        77.5,
        77.5
      ],
      "up_blocks.0.resnets.0.conv1": [
        77.7,
        77.6,
        77.6,
        77.6
      ],
      "mid_block.attentions.0.proj_in": [
        77.8,
        77.6,
        77.6,
        77.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2": [
        77.8,
        77.5,
        77.6,
        77.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        77.8,
        77.6,
        77.6,
        77.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2": [
        77.9,
        77.5,
        77.5,
        77.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2": [
        77.8,
        77.4,
        77.5,
        77.5
      ],
      "up_blocks.0.resnets.1.conv2": [
        77.8,
        77.5,
        77.5,
        77.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v": [
        77.7,
        77.5,
        77.5,
        77.6
      ],
      "up_blocks.0.attentions.0.proj_out": [
        77.8,
        77.5,
        77.5,
        77.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v": [
        77.7,
        77.3,
        77.4,
        77.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        77.6,
        77.3,
        77.4,
        77.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        77.7,
        77.3,
        77.4,
        77.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v": [
        77.7,
        77.2,
        77.5,
        77.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj": [
        77.7,
        77.2,
        77.5,
        77.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0": [
        77.7,
        77.3,
        77.4,
        77.5
      ],
      "up_blocks.0.resnets.2.conv2": [
        77.7,
        77.3,
        77.4,
        77.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0": [
        77.6,
        77.1,
        77.3,
        77.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj": [
        77.7,
        77.2,
        77.2,
        77.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        77.6,
        77.1,
        77.2,
        77.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v": [
        77.6,
        77.2,
        77.2,
        77.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        77.6,
        77.1,
        77.3,
        77.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0": [
        77.5,
        77.0,
        77.1,
        77.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj": [
        77.5,
        77.0,
        77.2,
        77.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v": [
        77.4,
        77.1,
        77.2,
        77.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        77.4,
        77.1,
        77.2,
        77.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2": [
        77.5,
        77.0,
        77.2,
        77.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2": [
        77.4,
        77.0,
        77.2,
        77.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj": [
        77.4,
        77.1,
        77.2,
        77.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v": [
        77.3,
        77.0,
        77.1,
        77.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2": [
        77.4,
        77.1,
        77.1,
        77.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v": [
        77.3,
        76.9,
        77.0,
        77.0
      ],
      "down_blocks.2.attentions.0.proj_out": [
        77.3,
        77.0,
        77.1,
        77.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        77.3,
        77.0,
        77.1,
        76.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        77.2,
        77.0,
        77.1,
        77.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        77.2,
        76.9,
        77.0,
        77.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v": [
        77.2,
        77.0,
        77.1,
        76.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        77.1,
        76.9,
        77.0,
        76.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v": [
        77.2,
        76.9,
        77.1,
        76.8
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        77.1,
        76.9,
        77.1,
        76.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2": [
        77.2,
        76.9,
        77.1,
        77.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v": [
        77.2,
        76.9,
        77.2,
        77.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        77.2,
        76.9,
        77.2,
        77.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v": [
        77.2,
        76.9,
        77.0,
        77.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj": [
        77.2,
        77.0,
        77.1,
        77.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2": [
        77.4,
        76.9,
        77.1,
        77.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        77.2,
        76.5,
        76.8,
        76.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2": [
        77.2,
        76.6,
        76.9,
        77.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v": [
        77.3,
        76.6,
        76.8,
        77.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v": [
        77.2,
        76.5,
        76.8,
        76.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v": [
        77.2,
        76.5,
        76.8,
        76.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        77.2,
        76.4,
        76.7,
        76.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        77.1,
        76.5,
        76.6,
        76.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v": [
        77.2,
        76.4,
        76.7,
        76.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v": [
        77.1,
        76.3,
        76.6,
        76.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v": [
        77.2,
        76.3,
        76.6,
        76.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj": [
        77.3,
        76.4,
        76.7,
        76.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj": [
        77.3,
        76.4,
        76.7,
        76.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        77.2,
        76.3,
        76.6,
        76.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        77.2,
        76.4,
        76.6,
        76.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        77.2,
        76.4,
        76.7,
        76.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        77.2,
        76.5,
        76.7,
        76.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        77.2,
        76.5,
        76.7,
        76.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2": [
        77.2,
        76.6,
        76.8,
        76.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        77.2,
        76.5,
        76.8,
        76.9
      ],
      "up_blocks.0.resnets.2.conv1": [
        77.2,
        76.5,
        76.7,
        76.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q": [
        77.1,
        76.4,
        76.5,
        76.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v": [
        77.0,
        76.4,
        76.6,
        76.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2": [
        77.1,
        76.4,
        76.7,
        76.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        77.1,
        76.3,
        76.5,
        76.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        77.0,
        76.0,
        76.4,
        76.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        76.8,
        75.9,
        76.4,
        76.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        76.7,
        75.9,
        76.3,
        76.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        76.8,
        75.9,
        76.3,
        76.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2": [
        76.6,
        75.8,
        76.2,
        76.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2": [
        76.7,
        75.8,
        76.1,
        76.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        76.7,
        75.8,
        76.3,
        76.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0": [
        76.4,
        75.7,
        76.0,
        76.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        76.4,
        75.7,
        76.0,
        76.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        76.5,
        75.6,
        76.1,
        76.2
      ],
      "down_blocks.2.attentions.0.proj_in": [
        76.4,
        75.6,
        76.0,
        76.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        76.3,
        75.6,
        76.1,
        76.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0": [
        76.2,
        75.4,
        75.9,
        76.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        76.3,
        75.4,
        75.9,
        76.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2": [
        76.3,
        75.4,
        75.9,
        76.0
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        75.9,
        75.3,
        75.6,
        75.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj": [
        75.9,
        75.2,
        75.5,
        75.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        75.8,
        75.1,
        75.3,
        75.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        75.7,
        74.9,
        75.0,
        75.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        75.6,
        74.9,
        75.0,
        75.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2": [
        75.6,
        74.9,
        75.0,
        75.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        75.3,
        74.6,
        74.6,
        74.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0": [
        75.1,
        74.4,
        74.5,
        74.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        75.1,
        74.3,
        74.4,
        74.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        75.0,
        74.2,
        74.2,
        74.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        74.9,
        74.0,
        73.9,
        74.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        74.9,
        73.9,
        73.9,
        74.1
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        74.8,
        73.8,
        74.1,
        74.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        74.6,
        73.8,
        74.2,
        74.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        74.7,
        73.9,
        74.0,
        74.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        74.6,
        73.8,
        74.0,
        74.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        74.6,
        73.6,
        73.9,
        73.8
      ],
      "up_blocks.0.attentions.2.proj_in": [
        74.4,
        73.5,
        73.7,
        73.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        74.4,
        73.5,
        73.7,
        73.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        74.4,
        73.5,
        73.8,
        73.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        74.2,
        73.4,
        73.7,
        73.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        74.3,
        73.3,
        73.6,
        73.6
      ],
      "up_blocks.0.attentions.2.proj_out": [
        74.1,
        73.3,
        73.7,
        73.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        74.0,
        73.1,
        73.5,
        73.6
      ],
      "up_blocks.0.attentions.1.proj_out": [
        73.9,
        73.1,
        73.4,
        73.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        73.6,
        73.0,
        73.3,
        73.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        73.5,
        72.8,
        73.2,
        73.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        73.5,
        72.7,
        73.0,
        73.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        73.4,
        72.7,
        73.0,
        73.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        73.3,
        72.6,
        72.9,
        73.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        72.9,
        72.3,
        72.7,
        72.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        72.8,
        72.3,
        72.5,
        72.7
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        72.7,
        72.2,
        72.4,
        72.6
      ],
      "up_blocks.1.resnets.0.conv1": [
        72.6,
        72.1,
        72.4,
        72.4
      ],
      "up_blocks.0.attentions.1.proj_in": [
        72.2,
        71.7,
        71.9,
        72.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k": [
        72.1,
        71.5,
        71.8,
        71.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        71.8,
        71.3,
        71.5,
        71.8
      ],
      "down_blocks.1.resnets.1.conv2": [
        71.6,
        71.1,
        71.4,
        71.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        71.7,
        71.1,
        71.4,
        71.5
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        71.5,
        70.9,
        71.2,
        71.4
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        71.2,
        70.6,
        70.9,
        71.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        71.0,
        70.5,
        70.9,
        71.1
      ],
      "up_blocks.1.resnets.0.conv2": [
        70.9,
        70.3,
        70.7,
        70.8
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        70.6,
        69.9,
        70.4,
        70.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        70.4,
        69.8,
        70.2,
        70.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        70.3,
        69.7,
        70.0,
        70.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        70.2,
        69.6,
        70.0,
        70.2
      ],
      "down_blocks.1.resnets.0.conv2": [
        70.0,
        69.5,
        69.8,
        70.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        70.0,
        69.5,
        69.8,
        70.1
      ],
      "add_embedding.linear_1": [
        69.9,
        69.5,
        69.5,
        69.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        69.8,
        69.4,
        69.5,
        69.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q": [
        69.6,
        69.2,
        69.5,
        69.6
      ],
      "add_embedding.linear_2": [
        69.5,
        69.1,
        69.3,
        69.5
      ],
      "down_blocks.1.attentions.1.proj_out": [
        69.4,
        68.9,
        69.2,
        69.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        69.3,
        68.9,
        69.1,
        69.3
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        69.1,
        68.7,
        68.9,
        69.0
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        68.8,
        68.4,
        68.7,
        68.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        68.6,
        68.3,
        68.5,
        68.6
      ],
      "up_blocks.1.attentions.0.proj_out": [
        68.5,
        68.3,
        68.4,
        68.5
      ],
      "time_embedding.linear_1": [
        68.6,
        68.3,
        68.5,
        68.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        68.3,
        68.2,
        68.3,
        68.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        68.2,
        67.9,
        68.1,
        68.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        68.1,
        67.7,
        68.0,
        68.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        67.5,
        67.5,
        67.7,
        67.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        67.4,
        67.4,
        67.5,
        67.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        67.3,
        67.2,
        67.4,
        67.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        67.2,
        67.1,
        67.4,
        67.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        67.2,
        67.1,
        67.3,
        67.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        67.0,
        66.9,
        67.2,
        67.4
      ],
      "down_blocks.1.attentions.1.proj_in": [
        66.9,
        66.8,
        67.1,
        67.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        66.7,
        66.6,
        66.9,
        67.0
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        66.4,
        66.4,
        66.6,
        66.8
      ],
      "down_blocks.1.resnets.1.conv1": [
        66.3,
        66.4,
        66.6,
        66.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        66.2,
        66.3,
        66.4,
        66.6
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        65.7,
        65.8,
        65.9,
        66.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        65.4,
        65.6,
        65.5,
        65.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        65.4,
        65.6,
        65.6,
        65.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        65.2,
        65.4,
        65.4,
        65.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2": [
        65.1,
        65.4,
        65.3,
        65.5
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        64.8,
        65.0,
        65.2,
        65.4
      ],
      "up_blocks.1.attentions.1.proj_out": [
        64.8,
        65.0,
        65.1,
        65.4
      ],
      "up_blocks.1.attentions.0.proj_in": [
        64.7,
        64.9,
        64.9,
        65.2
      ],
      "up_blocks.1.resnets.2.conv2": [
        64.4,
        64.7,
        64.7,
        64.9
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        64.4,
        64.6,
        64.6,
        64.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        64.3,
        64.5,
        64.5,
        64.8
      ],
      "down_blocks.1.attentions.0.proj_out": [
        64.1,
        64.3,
        64.3,
        64.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        63.8,
        64.0,
        63.9,
        64.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        63.6,
        63.8,
        63.8,
        64.1
      ],
      "up_blocks.1.resnets.1.conv2": [
        63.7,
        63.8,
        63.9,
        64.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v": [
        63.8,
        63.8,
        63.9,
        64.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        63.6,
        63.6,
        63.7,
        64.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        63.7,
        63.7,
        63.9,
        64.2
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        63.7,
        63.7,
        63.8,
        64.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        63.5,
        63.4,
        63.5,
        63.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        63.4,
        63.3,
        63.4,
        63.8
      ],
      "down_blocks.1.downsamplers.0.conv": [
        62.8,
        62.7,
        62.8,
        63.2
      ],
      "up_blocks.1.attentions.2.proj_out": [
        62.7,
        62.6,
        62.7,
        63.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        62.6,
        62.5,
        62.6,
        63.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        62.5,
        62.4,
        62.6,
        63.0
      ],
      "down_blocks.1.resnets.0.conv1": [
        62.2,
        62.0,
        62.2,
        62.6
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        61.8,
        61.6,
        61.8,
        62.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        61.4,
        61.2,
        61.4,
        61.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        61.5,
        61.3,
        61.5,
        61.8
      ],
      "up_blocks.1.attentions.1.proj_in": [
        61.3,
        61.0,
        61.2,
        61.6
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        60.9,
        60.6,
        60.8,
        61.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        60.7,
        60.3,
        60.6,
        60.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        60.4,
        60.2,
        60.5,
        60.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        60.1,
        59.8,
        60.1,
        60.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        60.2,
        59.8,
        60.1,
        60.4
      ],
      "up_blocks.0.upsamplers.0.conv": [
        59.7,
        59.3,
        59.7,
        59.8
      ],
      "up_blocks.1.attentions.2.proj_in": [
        59.4,
        59.1,
        59.4,
        59.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k": [
        58.3,
        57.6,
        58.3,
        58.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        58.2,
        57.4,
        58.1,
        58.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        58.1,
        57.3,
        58.0,
        58.2
      ],
      "down_blocks.1.attentions.0.proj_in": [
        58.0,
        57.2,
        58.0,
        58.1
      ],
      "up_blocks.1.resnets.1.conv1": [
        57.8,
        57.0,
        57.8,
        57.9
      ],
      "down_blocks.0.resnets.0.conv1": [
        57.9,
        57.0,
        57.8,
        58.0
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        57.8,
        57.0,
        57.7,
        57.9
      ],
      "up_blocks.1.resnets.2.conv1": [
        57.6,
        56.8,
        57.5,
        57.7
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        57.4,
        56.6,
        57.3,
        57.5
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        57.2,
        56.5,
        57.1,
        57.3
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        57.3,
        56.5,
        57.2,
        57.4
      ],
      "down_blocks.0.downsamplers.0.conv": [
        57.0,
        56.2,
        56.9,
        57.1
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        56.5,
        55.8,
        56.4,
        56.6
      ],
      "time_embedding.linear_2": [
        56.2,
        55.6,
        56.1,
        56.3
      ],
      "down_blocks.0.resnets.1.conv1": [
        56.1,
        55.3,
        55.9,
        56.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        56.2,
        55.4,
        55.9,
        56.1
      ],
      "up_blocks.1.upsamplers.0.conv": [
        55.4,
        54.7,
        55.1,
        55.3
      ],
      "down_blocks.0.resnets.1.conv2": [
        55.3,
        54.7,
        55.1,
        55.2
      ],
      "down_blocks.0.resnets.0.conv2": [
        55.0,
        54.5,
        54.9,
        54.9
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        54.5,
        54.0,
        54.3,
        54.3
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        53.9,
        53.5,
        53.8,
        53.8
      ],
      "up_blocks.2.resnets.2.conv2": [
        53.9,
        53.4,
        53.7,
        53.7
      ],
      "up_blocks.2.resnets.0.conv2": [
        53.8,
        53.4,
        53.7,
        53.7
      ],
      "up_blocks.2.resnets.0.conv1": [
        53.6,
        53.2,
        53.6,
        53.5
      ],
      "up_blocks.2.resnets.2.conv1": [
        53.0,
        52.6,
        52.9,
        52.8
      ],
      "up_blocks.2.resnets.1.conv1": [
        52.3,
        52.0,
        52.3,
        52.2
      ],
      "up_blocks.2.resnets.1.conv2": [
        51.4,
        51.2,
        51.5,
        51.4
      ]
    },
    "6": {
      "metadata": {
        "nbits": 6,
        "out_ngroups": 1,
        "in_ngroups": 1,
        "cumulative": true,
        "candidates": [
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.6.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_v",
          "mid_block.resnets.0.time_emb_proj",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v",
          "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_k",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v",
          "up_blocks.0.resnets.0.time_emb_proj",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.2.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.1.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0",
          "mid_block.resnets.0.conv2",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k",
          "up_blocks.0.resnets.2.time_emb_proj",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.8.ff.net.2",
          "mid_block.resnets.1.time_emb_proj",
          "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_k",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2",
          "down_blocks.2.resnets.0.conv1",
          "down_blocks.2.resnets.0.time_emb_proj",
          "down_blocks.2.resnets.1.time_emb_proj",
          "down_blocks.2.resnets.1.conv2",
          "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.4.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k",
          "down_blocks.2.resnets.1.conv1",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.3.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_v",
          "mid_block.attentions.0.proj_out",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_q",
          "mid_block.resnets.1.conv2",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.7.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.9.ff.net.2",
          "mid_block.resnets.0.conv1",
          "mid_block.resnets.1.conv1",
          "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q",
          "up_blocks.0.resnets.0.conv1",
          "up_blocks.0.resnets.1.conv1",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_v",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj",
          "down_blocks.2.attentions.1.proj_in",
          "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2",
          "down_blocks.2.resnets.0.conv2",
          "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj",
          "up_blocks.0.resnets.0.conv2",
          "up_blocks.0.resnets.1.time_emb_proj",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.5.ff.net.2",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.2.attentions.1.proj_out",
          "up_blocks.0.attentions.0.proj_in",
          "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2",
          "up_blocks.0.resnets.2.conv1",
          "up_blocks.1.resnets.0.time_emb_proj",
          "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v",
          "mid_block.attentions.0.proj_in",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.attentions.0.proj_in",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v",
          "up_blocks.0.resnets.1.conv2",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v",
          "down_blocks.1.resnets.1.time_emb_proj",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.0.proj_out",
          "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0",
          "down_blocks.2.attentions.0.proj_out",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2",
          "up_blocks.0.resnets.2.conv2",
          "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2",
          "up_blocks.0.resnets.0.conv_shortcut",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.1.proj_in",
          "down_blocks.1.resnets.0.time_emb_proj",
          "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.2.proj_in",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.resnets.1.conv_shortcut",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.2.proj_out",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k",
          "up_blocks.1.resnets.0.conv1",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q",
          "down_blocks.2.resnets.0.conv_shortcut",
          "down_blocks.1.resnets.1.conv2",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q",
          "down_blocks.1.resnets.0.conv2",
          "up_blocks.0.attentions.1.proj_out",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k",
          "up_blocks.2.resnets.2.time_emb_proj",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2",
          "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2",
          "up_blocks.1.resnets.0.conv2",
          "up_blocks.1.resnets.1.time_emb_proj",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.resnets.2.time_emb_proj",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.1.attentions.1.proj_out",
          "down_blocks.1.downsamplers.0.conv",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v",
          "down_blocks.1.attentions.1.proj_in",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.0.proj_out",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2",
          "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "down_blocks.1.resnets.1.conv1",
          "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2",
          "down_blocks.1.resnets.0.conv1",
          "up_blocks.2.resnets.0.time_emb_proj",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v",
          "up_blocks.2.resnets.1.time_emb_proj",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "add_embedding.linear_2",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v",
          "up_blocks.1.attentions.1.proj_out",
          "up_blocks.1.resnets.1.conv2",
          "down_blocks.0.resnets.1.time_emb_proj",
          "add_embedding.linear_1",
          "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "time_embedding.linear_1",
          "up_blocks.1.attentions.0.proj_in",
          "down_blocks.1.attentions.0.proj_out",
          "up_blocks.1.resnets.2.conv2",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.0.upsamplers.0.conv",
          "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.1.resnets.1.conv1",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2",
          "up_blocks.0.resnets.2.conv_shortcut",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.0.resnets.0.time_emb_proj",
          "time_embedding.linear_2",
          "up_blocks.1.resnets.2.conv1",
          "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.1.attentions.1.proj_in",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.2.proj_out",
          "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.1.resnets.0.conv_shortcut",
          "down_blocks.1.attentions.0.proj_in",
          "up_blocks.1.attentions.2.proj_in",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.2.resnets.2.conv_shortcut",
          "down_blocks.0.resnets.0.conv1",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.resnets.2.conv_shortcut",
          "down_blocks.0.downsamplers.0.conv",
          "down_blocks.1.resnets.0.conv_shortcut",
          "up_blocks.1.resnets.1.conv_shortcut",
          "up_blocks.2.resnets.0.conv1",
          "down_blocks.0.resnets.1.conv2",
          "up_blocks.2.resnets.0.conv2",
          "up_blocks.1.upsamplers.0.conv",
          "down_blocks.0.resnets.1.conv1",
          "down_blocks.0.resnets.0.conv2",
          "up_blocks.2.resnets.1.conv_shortcut",
          "up_blocks.2.resnets.0.conv_shortcut",
          "up_blocks.2.resnets.2.conv2",
          "up_blocks.2.resnets.1.conv1",
          "up_blocks.2.resnets.2.conv1",
          "up_blocks.2.resnets.1.conv2"
        ],
        "sizes": [
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          13107200,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          409600,
          1310720,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          6553600,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1310720,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          6553600,
          2621440,
          409600,
          1310720,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          6553600,
          1638400,
          1638400,
          1638400,
          6553600,
          2621440,
          2621440,
          1638400,
          1638400,
          6553600,
          1638400,
          6553600,
          2621440,
          1638400,
          13107200,
          6553600,
          1638400,
          13107200,
          1638400,
          2621440,
          1638400,
          1638400,
          13107200,
          6553600,
          2621440,
          2621440,
          2621440,
          13107200,
          6553600,
          2621440,
          2621440,
          1638400,
          1638400,
          13107200,
          6553600,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          2621440,
          2621440,
          1310720,
          1310720,
          409600,
          2621440,
          1638400,
          2621440,
          6553600,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          13107200,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          14745600,
          2621440,
          2621440,
          6553600,
          1638400,
          6553600,
          1638400,
          13107200,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          6553600,
          1638400,
          2621440,
          6553600,
          1638400,
          409600,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          6553600,
          7372800,
          1638400,
          1638400,
          14745600,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          6553600,
          1638400,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1310720,
          409600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          409600,
          6553600,
          1638400,
          1638400,
          13107200,
          6553600,
          6553600,
          1638400,
          1638400,
          6553600,
          2621440,
          2621440,
          13107200,
          2621440,
          14745600,
          2621440,
          13107200,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1310720,
          409600,
          13107200,
          1638400,
          13107200,
          2621440,
          1638400,
          6553600,
          2621440,
          13107200,
          1638400,
          13107200,
          13107200,
          13107200,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          6553600,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          14745600,
          1638400,
          13107200,
          1638400,
          13107200,
          1638400,
          13107200,
          13107200,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1310720,
          409600,
          1310720,
          409600,
          1638400,
          1638400,
          1638400,
          6553600,
          13107200,
          6553600,
          14745600,
          14745600,
          13107200,
          1638400,
          1638400,
          6553600,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          13107200,
          2621440,
          1638400,
          2621440,
          2621440,
          1638400,
          29491200,
          29491200,
          409600,
          409600,
          1638400,
          1638400,
          2621440,
          13107200,
          1310720,
          1638400,
          1310720,
          409600,
          6553600,
          13107200,
          13107200,
          1638400,
          6553600,
          14745600,
          13107200,
          14745600,
          1638400,
          1310720,
          6553600,
          409600,
          409600,
          1310720,
          1638400,
          1638400,
          1638400,
          6553600,
          22118400,
          819200,
          6553600,
          1638400,
          1638400,
          409600,
          1310720,
          1310720,
          1638400,
          409600,
          1310720,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1310720,
          409600,
          13107200,
          13107200,
          6553600,
          6553600,
          1638400,
          14745600,
          6553600,
          1638400,
          6553600,
          6553600,
          1638400,
          409600,
          13107200,
          6553600,
          409600,
          13107200,
          1638400,
          819200,
          1638400,
          409600,
          409600,
          409600,
          1638400,
          13107200,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          6553600,
          1638400,
          1638400,
          409600,
          1310720,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          6553600,
          13107200,
          1638400,
          6553600,
          13107200,
          13107200,
          13107200,
          6553600,
          14745600,
          6553600,
          1638400,
          1638400,
          13107200,
          6553600,
          409600,
          13107200,
          6553600,
          6553600,
          13107200,
          1638400,
          1638400,
          6553600,
          1638400,
          13107200,
          1638400,
          6553600,
          1638400,
          409600,
          1638400,
          1638400,
          409600,
          1638400,
          6553600,
          3276800,
          1310720,
          1638400,
          1638400,
          1638400,
          13107200,
          13107200,
          6553600,
          13107200,
          6553600,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          6553600,
          1638400,
          409600,
          1638400,
          819200,
          6553600,
          409600,
          1638400,
          409600,
          409600,
          13107200,
          409600,
          1638400,
          409600,
          409600,
          3276800,
          409600,
          409600,
          409600,
          1638400,
          409600,
          1310720,
          11059200,
          409600,
          13107200,
          409600,
          1638400,
          409600,
          819200,
          3686400,
          409600,
          3686400,
          1638400,
          409600,
          409600,
          409600,
          1638400,
          1638400,
          3686400,
          819200,
          409600,
          409600,
          819200,
          409600,
          409600,
          3686400,
          409600,
          409600,
          3276800,
          1638400,
          409600,
          409600,
          1638400,
          3276800,
          3686400,
          3276800,
          1638400,
          1843200,
          409600,
          409600,
          409600,
          409600,
          1638400,
          3276800,
          409600,
          1638400,
          409600,
          409600,
          409600,
          409600,
          3686400,
          409600,
          3604480,
          1638400,
          409600,
          409600,
          409600,
          409600,
          409600,
          409600,
          3686400,
          1638400,
          1638400,
          14745600,
          3276800,
          7372800,
          1638400,
          2457600,
          3276800,
          409600,
          1638400,
          5529600,
          3276800,
          409600,
          409600,
          409600,
          409600,
          409600,
          3276800,
          1228800,
          409600,
          409600,
          3276800,
          204800,
          921600,
          3276800,
          614400,
          921600,
          204800,
          819200,
          2764800,
          921600,
          921600,
          3686400,
          921600,
          921600,
          204800,
          307200,
          921600,
          1843200,
          1843200,
          921600
        ]
      },
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k": [
        82.4,
        82.3,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0": [
        82.1,
        82.4,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.0,
        82.3,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k": [
        82.0,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q": [
        82.3,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k": [
        82.2,
        82.4,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k": [
        82.2,
        82.2,
        82.5,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        82.0,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k": [
        82.1,
        82.4,
        82.3,
        82.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        82.2,
        82.3,
        82.2,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        82.3,
        82.4,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.2,
        82.2,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k": [
        82.1,
        82.4,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0": [
        81.9,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.2,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        82.1,
        82.2,
        82.5,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.2": [
        82.0,
        82.2,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_q": [
        82.1,
        82.2,
        82.2,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_v": [
        82.0,
        82.2,
        82.3,
        82.0
      ],
      "mid_block.resnets.0.time_emb_proj": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q": [
        82.0,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k": [
        81.9,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v": [
        82.2,
        82.1,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        82.1,
        82.1,
        82.5,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k": [
        82.1,
        82.2,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        81.9,
        82.1,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v": [
        82.0,
        82.2,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        82.2,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k": [
        82.1,
        82.2,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2": [
        82.1,
        82.4,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q": [
        82.0,
        82.4,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k": [
        82.0,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2": [
        82.1,
        82.1,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k": [
        82.1,
        82.2,
        82.4,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q": [
        81.9,
        82.1,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q": [
        82.0,
        82.1,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q": [
        81.9,
        82.1,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q": [
        82.1,
        82.1,
        82.4,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0": [
        81.9,
        82.3,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q": [
        82.0,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k": [
        81.8,
        82.2,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q": [
        82.0,
        82.3,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k": [
        81.9,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k": [
        81.9,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q": [
        81.9,
        82.3,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k": [
        82.1,
        81.9,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_k": [
        82.0,
        82.2,
        82.3,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_q": [
        82.0,
        82.2,
        82.3,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_v": [
        81.9,
        82.2,
        82.4,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_q": [
        81.9,
        82.2,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_k": [
        81.8,
        82.1,
        82.4,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_k": [
        81.8,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        82.0,
        82.1,
        82.3,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_q": [
        82.0,
        82.1,
        82.3,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_q": [
        81.9,
        82.2,
        82.2,
        81.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        82.0,
        82.2,
        82.4,
        81.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        81.9,
        82.3,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q": [
        82.1,
        82.2,
        82.4,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.1,
        82.1,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.1,
        82.2,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k": [
        82.0,
        82.1,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k": [
        82.0,
        82.2,
        82.4,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k": [
        81.8,
        82.1,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q": [
        82.0,
        82.2,
        82.1,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k": [
        81.9,
        82.3,
        82.4,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q": [
        81.8,
        82.2,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        81.8,
        82.2,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q": [
        81.9,
        82.2,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v": [
        82.0,
        82.1,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k": [
        81.8,
        82.1,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        81.9,
        82.2,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.1,
        82.1,
        82.1,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v": [
        82.0,
        82.1,
        82.3,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        82.0,
        82.2,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k": [
        82.0,
        82.1,
        82.3,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        81.9,
        82.2,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2": [
        81.8,
        82.2,
        82.3,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k": [
        81.9,
        82.2,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v": [
        81.7,
        82.1,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        81.9,
        82.1,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        81.8,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        81.9,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k": [
        81.8,
        82.1,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q": [
        82.0,
        82.1,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v": [
        82.0,
        82.1,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        81.8,
        82.0,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q": [
        81.9,
        82.1,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v": [
        81.9,
        82.1,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k": [
        82.0,
        82.0,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v": [
        82.0,
        82.0,
        82.3,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v": [
        81.9,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        82.0,
        82.0,
        82.3,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v": [
        82.0,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q": [
        81.9,
        82.0,
        82.4,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k": [
        81.9,
        82.1,
        82.3,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q": [
        81.9,
        82.1,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q": [
        81.9,
        82.0,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.8,
        82.1,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v": [
        81.9,
        82.0,
        82.4,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        82.1,
        82.1,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q": [
        81.9,
        82.0,
        82.3,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        81.9,
        82.2,
        82.3,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q": [
        82.0,
        82.1,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k": [
        81.9,
        82.1,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2": [
        82.0,
        82.1,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k": [
        82.0,
        82.0,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v": [
        81.8,
        82.0,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q": [
        81.8,
        82.1,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v": [
        81.9,
        82.1,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k": [
        81.9,
        82.1,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        81.9,
        82.3,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k": [
        81.9,
        82.0,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        82.0,
        82.1,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        81.8,
        82.1,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q": [
        81.9,
        82.0,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q": [
        81.9,
        82.0,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k": [
        82.0,
        82.1,
        82.2,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v": [
        81.8,
        82.1,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k": [
        81.8,
        82.0,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k": [
        81.8,
        82.1,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        81.8,
        82.1,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q": [
        81.8,
        81.9,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k": [
        81.7,
        82.0,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        81.9,
        81.9,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q": [
        81.8,
        81.9,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k": [
        81.8,
        82.1,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k": [
        81.8,
        81.9,
        82.1,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k": [
        81.8,
        82.0,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k": [
        81.8,
        82.0,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0": [
        81.9,
        81.9,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q": [
        81.8,
        81.8,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0": [
        81.8,
        81.9,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q": [
        81.8,
        82.0,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v": [
        81.7,
        81.9,
        82.1,
        81.8
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        81.8,
        82.0,
        82.1,
        81.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k": [
        81.7,
        81.9,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        81.8,
        81.8,
        82.2,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        81.7,
        81.9,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        81.8,
        81.8,
        82.2,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_k": [
        81.7,
        81.9,
        82.2,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_v": [
        81.7,
        81.9,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_v": [
        81.7,
        81.9,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_q": [
        81.8,
        81.8,
        82.1,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        81.7,
        81.9,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_q": [
        81.8,
        81.9,
        82.1,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_v": [
        81.8,
        81.9,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        81.5,
        81.9,
        82.0,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        81.7,
        81.9,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        81.7,
        81.7,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_q": [
        81.7,
        81.7,
        82.0,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_v": [
        81.7,
        81.6,
        82.0,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_k": [
        81.8,
        81.9,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        81.7,
        81.8,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k": [
        81.7,
        81.9,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        81.8,
        81.9,
        82.1,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q": [
        81.9,
        81.8,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        81.7,
        81.9,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k": [
        81.8,
        81.8,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v": [
        81.8,
        81.8,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q": [
        81.7,
        81.9,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        81.7,
        82.0,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v": [
        81.7,
        81.8,
        82.2,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v": [
        81.6,
        81.8,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        81.8,
        81.8,
        82.0,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_k": [
        81.7,
        81.8,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        81.8,
        81.7,
        82.0,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.2": [
        81.7,
        81.9,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_k": [
        81.7,
        81.6,
        82.1,
        81.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        81.7,
        81.7,
        82.0,
        81.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        81.7,
        81.9,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        81.7,
        81.9,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        81.7,
        81.7,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.7,
        81.7,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        81.7,
        81.7,
        82.0,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q": [
        81.6,
        81.7,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v": [
        81.7,
        81.7,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        81.7,
        81.7,
        82.1,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q": [
        81.7,
        81.8,
        82.1,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k": [
        81.7,
        81.6,
        82.1,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        81.7,
        81.8,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k": [
        81.7,
        81.8,
        81.9,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q": [
        81.7,
        81.8,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k": [
        81.7,
        81.7,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q": [
        81.6,
        81.7,
        81.8,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q": [
        81.7,
        81.7,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v": [
        81.6,
        81.7,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k": [
        81.6,
        81.9,
        82.0,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q": [
        81.6,
        81.7,
        82.0,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        81.7,
        81.7,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q": [
        81.7,
        81.8,
        82.1,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        81.6,
        81.8,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2": [
        81.6,
        81.7,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k": [
        81.7,
        81.8,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v": [
        81.7,
        81.7,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        81.6,
        81.7,
        81.9,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2": [
        81.6,
        81.6,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k": [
        81.6,
        81.7,
        82.0,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v": [
        81.7,
        81.8,
        81.7,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q": [
        81.6,
        81.7,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        81.7,
        81.6,
        82.0,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2": [
        81.6,
        81.9,
        81.8,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        81.6,
        81.8,
        81.9,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2": [
        81.6,
        81.5,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v": [
        81.7,
        81.7,
        81.9,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        81.6,
        81.6,
        81.9,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        81.7,
        81.7,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2": [
        81.6,
        81.8,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v": [
        81.6,
        81.6,
        81.9,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        81.6,
        81.6,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        81.5,
        81.6,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        81.6,
        81.6,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        81.6,
        81.7,
        81.7,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q": [
        81.5,
        81.6,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        81.6,
        81.6,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2": [
        81.6,
        81.7,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k": [
        81.7,
        81.7,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v": [
        81.6,
        81.7,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k": [
        81.7,
        81.6,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        81.5,
        81.6,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2": [
        81.6,
        81.7,
        81.8,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k": [
        81.5,
        81.6,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v": [
        81.6,
        81.5,
        81.8,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q": [
        81.6,
        81.7,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q": [
        81.5,
        81.6,
        81.8,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        81.6,
        81.6,
        81.7,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2": [
        81.5,
        81.6,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v": [
        81.6,
        81.6,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k": [
        81.4,
        81.5,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        81.5,
        81.5,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q": [
        81.4,
        81.6,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k": [
        81.6,
        81.6,
        81.8,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v": [
        81.5,
        81.6,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v": [
        81.6,
        81.6,
        81.6,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q": [
        81.4,
        81.7,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k": [
        81.5,
        81.4,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q": [
        81.4,
        81.7,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k": [
        81.4,
        81.5,
        81.9,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v": [
        81.6,
        81.6,
        81.8,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q": [
        81.6,
        81.6,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k": [
        81.4,
        81.7,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q": [
        81.4,
        81.6,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v": [
        81.5,
        81.7,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q": [
        81.6,
        81.5,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k": [
        81.6,
        81.7,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q": [
        81.6,
        81.7,
        81.6,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0": [
        81.5,
        81.5,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q": [
        81.5,
        81.6,
        81.6,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v": [
        81.4,
        81.5,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v": [
        81.5,
        81.6,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q": [
        81.5,
        81.5,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k": [
        81.5,
        81.6,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v": [
        81.4,
        81.6,
        81.7,
        81.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        81.6,
        81.5,
        81.6,
        81.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        81.6,
        81.4,
        81.7,
        81.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q": [
        81.5,
        81.5,
        81.9,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.5,
        81.4,
        81.9,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.6,
        81.7,
        81.8,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_v": [
        81.5,
        81.6,
        81.8,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.2": [
        81.6,
        81.6,
        81.8,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_q": [
        81.4,
        81.6,
        81.8,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        81.6,
        81.5,
        81.7,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        81.6,
        81.6,
        81.9,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_q": [
        81.5,
        81.5,
        81.8,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_k": [
        81.6,
        81.6,
        81.8,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        81.5,
        81.7,
        81.7,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_k": [
        81.5,
        81.7,
        81.6,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_v": [
        81.5,
        81.6,
        81.9,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_k": [
        81.5,
        81.7,
        81.8,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_q": [
        81.4,
        81.6,
        81.9,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_v": [
        81.5,
        81.5,
        81.5,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_v": [
        81.5,
        81.6,
        81.9,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        81.7,
        81.6,
        81.9,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_k": [
        81.5,
        81.7,
        81.6,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        81.7,
        81.6,
        81.6,
        81.5
      ],
      "mid_block.resnets.0.conv2": [
        81.4,
        81.6,
        81.7,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.4,
        81.7,
        81.7,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k": [
        81.4,
        81.6,
        81.9,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2": [
        81.4,
        81.6,
        81.9,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v": [
        81.4,
        81.6,
        81.9,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        81.6,
        81.6,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        81.5,
        81.4,
        81.7,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        81.5,
        81.4,
        81.7,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k": [
        81.4,
        81.6,
        81.8,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        81.5,
        81.7,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        81.5,
        81.5,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v": [
        81.5,
        81.5,
        81.8,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v": [
        81.5,
        81.6,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        81.5,
        81.6,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v": [
        81.5,
        81.5,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        81.5,
        81.6,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k": [
        81.5,
        81.5,
        81.8,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q": [
        81.5,
        81.5,
        81.9,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        81.4,
        81.5,
        81.8,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v": [
        81.5,
        81.6,
        81.8,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0": [
        81.5,
        81.6,
        81.9,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k": [
        81.5,
        81.6,
        81.7,
        81.4
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        81.6,
        81.3,
        81.6,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        81.5,
        81.5,
        81.6,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_q": [
        81.6,
        81.6,
        81.6,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_v": [
        81.5,
        81.6,
        81.7,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.2": [
        81.5,
        81.5,
        81.5,
        81.5
      ],
      "mid_block.resnets.1.time_emb_proj": [
        81.5,
        81.6,
        81.6,
        81.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        81.5,
        81.6,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        81.5,
        81.5,
        81.6,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        81.5,
        81.6,
        81.7,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k": [
        81.5,
        81.5,
        81.8,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k": [
        81.6,
        81.5,
        81.6,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v": [
        81.6,
        81.6,
        81.6,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        81.5,
        81.5,
        81.6,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        81.5,
        81.6,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        81.5,
        81.6,
        81.7,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        81.6,
        81.5,
        81.7,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        81.5,
        81.6,
        81.7,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        81.5,
        81.6,
        81.7,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        81.4,
        81.6,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v": [
        81.5,
        81.5,
        81.7,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v": [
        81.5,
        81.5,
        81.6,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v": [
        81.5,
        81.4,
        81.6,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q": [
        81.6,
        81.5,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q": [
        81.5,
        81.6,
        81.7,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        81.5,
        81.4,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q": [
        81.5,
        81.5,
        81.7,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2": [
        81.6,
        81.6,
        81.6,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q": [
        81.5,
        81.6,
        81.6,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        81.5,
        81.5,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v": [
        81.5,
        81.5,
        81.8,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        81.6,
        81.5,
        81.6,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        81.4,
        81.4,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q": [
        81.6,
        81.6,
        81.6,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2": [
        81.5,
        81.5,
        81.7,
        81.3
      ],
      "down_blocks.2.resnets.0.conv1": [
        81.5,
        81.5,
        81.6,
        81.4
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        81.4,
        81.4,
        81.6,
        81.4
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        81.4,
        81.3,
        81.7,
        81.3
      ],
      "down_blocks.2.resnets.1.conv2": [
        81.4,
        81.2,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        81.4,
        81.2,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q": [
        81.3,
        81.4,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        81.4,
        81.3,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v": [
        81.4,
        81.3,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k": [
        81.3,
        81.5,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        81.3,
        81.3,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v": [
        81.3,
        81.6,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k": [
        81.4,
        81.3,
        81.8,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        81.3,
        81.5,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k": [
        81.4,
        81.3,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2": [
        81.4,
        81.3,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        81.4,
        81.3,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v": [
        81.4,
        81.3,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k": [
        81.4,
        81.3,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        81.4,
        81.3,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k": [
        81.4,
        81.3,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        81.4,
        81.5,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q": [
        81.4,
        81.2,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        81.4,
        81.3,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q": [
        81.4,
        81.3,
        81.6,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k": [
        81.4,
        81.4,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        81.5,
        81.3,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        81.3,
        81.2,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q": [
        81.4,
        81.3,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v": [
        81.3,
        81.4,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q": [
        81.4,
        81.3,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q": [
        81.4,
        81.4,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k": [
        81.3,
        81.3,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0": [
        81.4,
        81.3,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k": [
        81.4,
        81.3,
        81.6,
        81.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        81.3,
        81.4,
        81.6,
        81.3
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        81.4,
        81.3,
        81.5,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        81.4,
        81.3,
        81.6,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_q": [
        81.4,
        81.3,
        81.5,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_q": [
        81.4,
        81.4,
        81.7,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_q": [
        81.4,
        81.3,
        81.6,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_k": [
        81.3,
        81.3,
        81.8,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        81.3,
        81.3,
        81.6,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.2": [
        81.4,
        81.5,
        81.6,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        81.3,
        81.2,
        81.5,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        81.4,
        81.2,
        81.6,
        81.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        81.3,
        81.3,
        81.5,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2": [
        81.3,
        81.2,
        81.6,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q": [
        81.4,
        81.3,
        81.6,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v": [
        81.2,
        81.3,
        81.4,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        81.3,
        81.4,
        81.5,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2": [
        81.3,
        81.4,
        81.6,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2": [
        81.4,
        81.3,
        81.5,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k": [
        81.3,
        81.3,
        81.5,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        81.4,
        81.3,
        81.5,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2": [
        81.4,
        81.3,
        81.5,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k": [
        81.4,
        81.2,
        81.7,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v": [
        81.3,
        81.2,
        81.7,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        81.5,
        81.3,
        81.6,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k": [
        81.3,
        81.2,
        81.5,
        81.4
      ],
      "down_blocks.2.resnets.1.conv1": [
        81.4,
        81.4,
        81.4,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.5,
        81.3,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        81.3,
        81.3,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q": [
        81.4,
        81.2,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v": [
        81.3,
        81.3,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q": [
        81.2,
        81.1,
        81.5,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v": [
        81.3,
        81.1,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        81.3,
        81.3,
        81.5,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v": [
        81.4,
        81.4,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q": [
        81.3,
        81.3,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q": [
        81.4,
        81.2,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v": [
        81.4,
        81.0,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        81.3,
        81.1,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        81.4,
        81.1,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q": [
        81.4,
        81.2,
        81.4,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q": [
        81.2,
        81.3,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q": [
        81.4,
        81.2,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v": [
        81.3,
        81.2,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k": [
        81.4,
        81.1,
        81.6,
        81.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        81.4,
        81.2,
        81.6,
        81.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        81.3,
        81.3,
        81.5,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        81.4,
        81.2,
        81.5,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_q": [
        81.3,
        81.2,
        81.4,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        81.3,
        81.2,
        81.4,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_k": [
        81.3,
        81.3,
        81.4,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        81.5,
        81.3,
        81.5,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.2": [
        81.2,
        81.3,
        81.5,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_k": [
        81.4,
        81.0,
        81.4,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        81.3,
        81.2,
        81.4,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_q": [
        81.4,
        81.2,
        81.6,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        81.4,
        81.3,
        81.5,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        81.4,
        81.3,
        81.5,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        81.3,
        81.2,
        81.5,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_v": [
        81.3,
        81.3,
        81.5,
        81.2
      ],
      "mid_block.attentions.0.proj_out": [
        81.4,
        81.3,
        81.4,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v": [
        81.3,
        81.2,
        81.6,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v": [
        81.3,
        81.3,
        81.6,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v": [
        81.3,
        81.2,
        81.5,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q": [
        81.4,
        81.2,
        81.5,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        81.3,
        81.2,
        81.5,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        81.3,
        81.3,
        81.6,
        81.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        81.2,
        81.2,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2": [
        81.3,
        81.2,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v": [
        81.3,
        81.2,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q": [
        81.3,
        81.1,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k": [
        81.3,
        81.2,
        81.5,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q": [
        81.4,
        81.2,
        81.5,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        81.4,
        81.2,
        81.4,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_v": [
        81.2,
        81.2,
        81.4,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        81.2,
        81.2,
        81.6,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_q": [
        81.4,
        81.2,
        81.5,
        81.2
      ],
      "mid_block.resnets.1.conv2": [
        81.3,
        81.1,
        81.5,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v": [
        81.3,
        81.3,
        81.5,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        81.3,
        81.0,
        81.5,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v": [
        81.3,
        81.2,
        81.5,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        81.2,
        81.1,
        81.5,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        81.3,
        81.3,
        81.5,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        81.3,
        81.2,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        81.3,
        81.3,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        81.3,
        80.9,
        81.4,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v": [
        81.3,
        81.2,
        81.4,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k": [
        81.3,
        81.2,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v": [
        81.1,
        81.1,
        81.6,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        81.3,
        81.2,
        81.5,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_v": [
        81.4,
        81.2,
        81.5,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        81.2,
        81.2,
        81.5,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        81.3,
        81.1,
        81.5,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v": [
        81.3,
        81.2,
        81.5,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        81.2,
        81.2,
        81.4,
        81.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        81.3,
        81.2,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        81.2,
        81.1,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q": [
        81.3,
        81.2,
        81.4,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        81.3,
        81.2,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q": [
        81.2,
        81.2,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k": [
        81.3,
        81.2,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k": [
        81.2,
        81.1,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k": [
        81.1,
        81.1,
        81.6,
        81.3
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.3,
        81.2,
        81.5,
        81.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        81.3,
        81.1,
        81.4,
        81.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        81.3,
        81.3,
        81.4,
        81.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        81.2,
        81.1,
        81.4,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_k": [
        81.2,
        81.0,
        81.5,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_k": [
        81.2,
        81.1,
        81.3,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_v": [
        81.2,
        81.2,
        81.4,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.2": [
        81.2,
        81.1,
        81.4,
        81.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        81.1,
        81.0,
        81.5,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.2": [
        81.1,
        81.2,
        81.4,
        81.2
      ],
      "mid_block.resnets.0.conv1": [
        81.2,
        81.1,
        81.4,
        81.2
      ],
      "mid_block.resnets.1.conv1": [
        81.1,
        81.0,
        81.4,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        81.2,
        81.0,
        81.4,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v": [
        81.1,
        81.0,
        81.4,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v": [
        81.2,
        81.1,
        81.5,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2": [
        81.3,
        81.1,
        81.4,
        81.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        81.1,
        81.0,
        81.4,
        81.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2": [
        81.1,
        81.1,
        81.4,
        81.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q": [
        81.1,
        81.0,
        81.3,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        81.2,
        81.1,
        81.3,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v": [
        81.2,
        81.0,
        81.4,
        81.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v": [
        81.2,
        81.0,
        81.4,
        81.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        81.1,
        81.1,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        81.2,
        81.1,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k": [
        81.0,
        81.1,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k": [
        81.2,
        81.1,
        81.3,
        81.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k": [
        81.1,
        81.1,
        81.3,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v": [
        81.1,
        81.1,
        81.5,
        81.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q": [
        81.0,
        81.1,
        81.4,
        81.3
      ],
      "up_blocks.0.resnets.0.conv1": [
        81.0,
        81.1,
        81.4,
        81.3
      ],
      "up_blocks.0.resnets.1.conv1": [
        81.1,
        81.0,
        81.3,
        81.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        81.1,
        81.1,
        81.3,
        81.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        81.2,
        81.2,
        81.3,
        81.1
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_v": [
        81.2,
        81.1,
        81.3,
        81.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_v": [
        81.2,
        81.1,
        81.3,
        81.1
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_k": [
        81.2,
        81.1,
        81.4,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        81.1,
        81.2,
        81.3,
        81.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        81.3,
        81.1,
        81.3,
        81.3
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_v": [
        81.2,
        81.0,
        81.4,
        81.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        81.2,
        81.1,
        81.4,
        81.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.1,
        81.0,
        81.2,
        81.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2": [
        81.1,
        81.0,
        81.2,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        81.1,
        80.8,
        81.5,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        81.0,
        80.9,
        81.3,
        80.9
      ],
      "down_blocks.2.attentions.1.proj_in": [
        81.1,
        80.9,
        81.3,
        81.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2": [
        81.1,
        80.8,
        81.3,
        81.1
      ],
      "down_blocks.2.resnets.0.conv2": [
        81.1,
        80.9,
        81.3,
        80.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        81.1,
        80.8,
        81.3,
        81.0
      ],
      "up_blocks.0.resnets.0.conv2": [
        81.0,
        80.9,
        81.4,
        81.0
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        80.9,
        80.9,
        81.3,
        80.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        81.0,
        80.7,
        81.2,
        80.8
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.2": [
        81.1,
        80.9,
        81.3,
        80.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        81.0,
        80.9,
        81.2,
        80.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        81.0,
        81.1,
        81.2,
        80.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        81.0,
        80.9,
        81.2,
        80.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        80.9,
        80.9,
        81.3,
        80.9
      ],
      "down_blocks.2.attentions.1.proj_out": [
        81.1,
        81.0,
        81.2,
        80.8
      ],
      "up_blocks.0.attentions.0.proj_in": [
        81.2,
        80.8,
        81.2,
        80.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2": [
        81.0,
        81.0,
        81.2,
        80.8
      ],
      "up_blocks.0.resnets.2.conv1": [
        81.1,
        80.9,
        81.3,
        80.9
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        81.1,
        80.9,
        81.2,
        81.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        81.2,
        80.9,
        81.2,
        80.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v": [
        81.1,
        80.8,
        81.2,
        81.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v": [
        81.2,
        80.9,
        81.1,
        81.1
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.2,
        80.8,
        81.1,
        80.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        81.1,
        80.8,
        81.1,
        80.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v": [
        81.0,
        80.9,
        81.0,
        80.8
      ],
      "mid_block.attentions.0.proj_in": [
        81.1,
        80.8,
        81.1,
        80.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        81.0,
        80.7,
        81.0,
        80.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        80.9,
        80.9,
        80.9,
        80.5
      ],
      "down_blocks.2.attentions.0.proj_in": [
        80.9,
        80.8,
        80.9,
        80.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        80.9,
        80.8,
        80.9,
        80.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2": [
        81.0,
        80.6,
        80.9,
        80.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v": [
        80.9,
        80.8,
        81.1,
        80.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        81.0,
        80.9,
        81.0,
        80.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        81.0,
        80.7,
        81.0,
        80.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        80.9,
        80.6,
        80.9,
        80.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        81.0,
        81.0,
        80.9,
        80.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        80.9,
        80.9,
        81.0,
        80.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2": [
        80.8,
        80.9,
        81.0,
        80.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2": [
        81.0,
        80.9,
        80.9,
        80.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v": [
        80.8,
        80.9,
        81.1,
        80.7
      ],
      "up_blocks.0.resnets.1.conv2": [
        80.9,
        80.8,
        81.0,
        80.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        81.0,
        80.9,
        81.1,
        80.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v": [
        80.9,
        81.0,
        81.0,
        80.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2": [
        81.0,
        81.0,
        81.0,
        80.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2": [
        81.0,
        80.9,
        81.0,
        80.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0": [
        80.9,
        80.8,
        81.1,
        80.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        80.9,
        80.7,
        81.0,
        80.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj": [
        80.8,
        80.9,
        80.9,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2": [
        80.9,
        80.8,
        81.0,
        80.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        80.8,
        80.7,
        80.9,
        80.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        80.8,
        80.8,
        80.8,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v": [
        80.7,
        80.6,
        80.9,
        80.8
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        80.8,
        80.5,
        81.0,
        80.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0": [
        80.7,
        80.7,
        80.9,
        80.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        80.7,
        80.6,
        80.7,
        80.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        80.6,
        80.7,
        80.7,
        80.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        80.6,
        80.5,
        80.9,
        80.6
      ],
      "up_blocks.0.attentions.0.proj_out": [
        80.7,
        80.4,
        80.7,
        80.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        80.6,
        80.5,
        80.8,
        80.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        80.7,
        80.6,
        80.8,
        80.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        80.6,
        80.4,
        80.8,
        80.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        80.7,
        80.4,
        80.7,
        80.4
      ],
      "down_blocks.2.attentions.0.proj_out": [
        80.6,
        80.4,
        80.8,
        80.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v": [
        80.7,
        80.4,
        80.7,
        80.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        80.6,
        80.2,
        80.5,
        80.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        80.6,
        80.2,
        80.6,
        80.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2": [
        80.6,
        80.3,
        80.5,
        80.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0": [
        80.6,
        80.3,
        80.7,
        80.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0": [
        80.6,
        80.4,
        80.6,
        80.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        80.4,
        80.3,
        80.6,
        80.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        80.3,
        80.1,
        80.4,
        80.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v": [
        80.3,
        80.1,
        80.4,
        80.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        80.3,
        80.1,
        80.4,
        80.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        80.2,
        80.0,
        80.3,
        80.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v": [
        80.4,
        80.2,
        80.4,
        80.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        80.4,
        79.9,
        80.4,
        80.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2": [
        80.3,
        80.2,
        80.4,
        80.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        80.4,
        80.0,
        80.4,
        80.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0": [
        80.2,
        80.1,
        80.3,
        80.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2": [
        80.2,
        80.1,
        80.3,
        80.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj": [
        80.3,
        80.1,
        80.4,
        80.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj": [
        80.1,
        80.1,
        80.4,
        80.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj": [
        80.3,
        80.1,
        80.3,
        80.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2": [
        80.3,
        80.0,
        80.4,
        80.2
      ],
      "up_blocks.0.resnets.2.conv2": [
        80.3,
        80.2,
        80.3,
        80.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2": [
        80.3,
        80.1,
        80.4,
        80.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        80.2,
        80.1,
        80.4,
        80.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v": [
        80.2,
        80.0,
        80.5,
        80.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj": [
        80.2,
        80.1,
        80.4,
        80.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2": [
        80.2,
        80.1,
        80.5,
        80.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        80.3,
        79.9,
        80.4,
        80.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        80.2,
        80.1,
        80.2,
        80.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2": [
        80.1,
        80.0,
        80.4,
        80.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2": [
        80.2,
        80.0,
        80.2,
        80.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        80.1,
        80.1,
        80.2,
        80.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v": [
        80.1,
        79.9,
        80.3,
        80.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v": [
        80.2,
        79.9,
        80.3,
        80.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2": [
        80.2,
        79.9,
        80.1,
        80.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v": [
        80.1,
        79.9,
        80.2,
        80.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj": [
        80.2,
        79.8,
        80.2,
        79.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v": [
        80.1,
        79.9,
        80.2,
        80.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2": [
        80.2,
        80.0,
        80.1,
        80.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0": [
        80.1,
        80.0,
        80.3,
        79.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        80.1,
        79.9,
        80.0,
        80.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0": [
        80.1,
        80.0,
        80.2,
        79.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v": [
        80.1,
        79.9,
        80.3,
        80.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        80.0,
        79.9,
        80.1,
        79.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        80.0,
        79.8,
        80.3,
        79.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2": [
        80.1,
        79.9,
        80.1,
        79.7
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        80.0,
        79.9,
        80.2,
        79.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        79.9,
        79.9,
        80.0,
        79.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v": [
        80.1,
        79.9,
        80.0,
        79.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        80.1,
        79.8,
        80.0,
        79.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v": [
        80.0,
        79.6,
        80.0,
        79.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj": [
        79.8,
        79.5,
        79.9,
        79.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj": [
        79.9,
        79.6,
        80.0,
        80.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2": [
        80.0,
        79.7,
        80.1,
        79.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        80.0,
        79.8,
        80.1,
        79.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        80.0,
        79.6,
        80.1,
        79.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        79.9,
        79.7,
        80.0,
        80.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v": [
        79.9,
        79.7,
        80.0,
        79.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        80.0,
        79.6,
        79.9,
        79.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        80.0,
        79.7,
        80.2,
        79.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        79.8,
        79.6,
        79.9,
        79.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2": [
        80.0,
        79.6,
        79.8,
        79.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0": [
        80.0,
        79.7,
        79.9,
        79.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        79.9,
        79.6,
        79.8,
        79.5
      ],
      "up_blocks.0.attentions.1.proj_in": [
        79.9,
        79.5,
        79.9,
        79.6
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        79.9,
        79.6,
        79.8,
        79.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2": [
        79.8,
        79.5,
        79.8,
        79.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        79.8,
        79.4,
        79.9,
        79.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v": [
        79.9,
        79.6,
        79.9,
        79.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        79.8,
        79.4,
        79.7,
        79.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        79.7,
        79.5,
        79.7,
        79.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        79.6,
        79.6,
        79.6,
        79.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        79.7,
        79.6,
        79.5,
        79.4
      ],
      "up_blocks.0.attentions.2.proj_in": [
        79.6,
        79.5,
        79.5,
        79.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        79.8,
        79.5,
        79.4,
        79.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        79.6,
        79.4,
        79.5,
        79.5
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        79.6,
        79.5,
        79.4,
        79.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        79.4,
        79.3,
        79.5,
        79.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        79.5,
        79.3,
        79.6,
        79.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        79.4,
        79.3,
        79.4,
        79.4
      ],
      "up_blocks.0.attentions.2.proj_out": [
        79.2,
        79.3,
        79.5,
        79.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        79.2,
        79.2,
        79.5,
        79.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        79.1,
        79.0,
        79.4,
        79.1
      ],
      "up_blocks.1.resnets.0.conv1": [
        79.1,
        79.1,
        79.3,
        79.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        79.0,
        79.0,
        79.3,
        79.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        79.1,
        78.9,
        79.2,
        79.0
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        79.0,
        78.8,
        79.1,
        79.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        78.8,
        78.7,
        79.1,
        79.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        78.9,
        78.6,
        79.1,
        79.0
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        78.8,
        78.5,
        79.0,
        78.9
      ],
      "down_blocks.1.resnets.1.conv2": [
        78.9,
        78.3,
        79.0,
        78.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q": [
        78.7,
        78.4,
        78.8,
        78.8
      ],
      "down_blocks.1.resnets.0.conv2": [
        78.7,
        78.4,
        78.8,
        78.6
      ],
      "up_blocks.0.attentions.1.proj_out": [
        78.7,
        78.3,
        78.9,
        78.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        78.7,
        78.2,
        78.8,
        78.7
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        78.5,
        78.1,
        78.7,
        78.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        78.4,
        78.1,
        78.7,
        78.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        78.3,
        78.1,
        78.7,
        78.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        78.6,
        78.0,
        78.6,
        78.5
      ],
      "up_blocks.1.resnets.0.conv2": [
        78.5,
        78.0,
        78.4,
        78.4
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        78.4,
        78.1,
        78.3,
        78.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k": [
        78.3,
        77.7,
        78.2,
        78.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        78.3,
        77.8,
        78.3,
        78.1
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        78.0,
        78.0,
        78.3,
        78.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        78.0,
        77.6,
        78.2,
        78.1
      ],
      "down_blocks.1.attentions.1.proj_out": [
        77.9,
        77.6,
        78.1,
        77.9
      ],
      "down_blocks.1.downsamplers.0.conv": [
        77.8,
        77.6,
        78.1,
        77.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        77.7,
        77.6,
        78.0,
        77.8
      ],
      "down_blocks.1.attentions.1.proj_in": [
        77.8,
        77.5,
        77.7,
        77.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        77.5,
        77.1,
        77.5,
        77.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        77.4,
        77.1,
        77.3,
        77.5
      ],
      "up_blocks.1.attentions.0.proj_out": [
        77.4,
        76.9,
        77.5,
        77.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        77.2,
        76.9,
        77.1,
        77.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        77.1,
        76.9,
        76.9,
        76.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        77.1,
        76.7,
        76.9,
        77.1
      ],
      "down_blocks.1.resnets.1.conv1": [
        76.9,
        76.6,
        77.0,
        77.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        77.0,
        76.7,
        76.9,
        76.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        77.0,
        76.5,
        76.7,
        76.9
      ],
      "down_blocks.1.resnets.0.conv1": [
        76.7,
        76.6,
        76.8,
        76.9
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        77.0,
        76.6,
        76.9,
        76.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        76.8,
        76.5,
        76.6,
        76.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        76.5,
        76.6,
        76.5,
        76.7
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        76.4,
        76.4,
        76.3,
        76.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        76.5,
        76.2,
        76.2,
        76.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        76.3,
        76.1,
        76.2,
        76.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        76.3,
        76.2,
        76.0,
        76.4
      ],
      "add_embedding.linear_2": [
        76.4,
        76.2,
        76.0,
        76.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        76.4,
        76.2,
        75.9,
        76.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        76.2,
        76.0,
        75.8,
        76.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v": [
        76.0,
        75.7,
        75.7,
        76.0
      ],
      "up_blocks.1.attentions.1.proj_out": [
        75.9,
        75.8,
        75.6,
        75.9
      ],
      "up_blocks.1.resnets.1.conv2": [
        75.9,
        75.8,
        75.5,
        75.8
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        75.8,
        75.8,
        75.5,
        75.8
      ],
      "add_embedding.linear_1": [
        75.7,
        75.7,
        75.4,
        75.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2": [
        75.7,
        75.8,
        75.3,
        75.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        75.5,
        75.4,
        75.3,
        75.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        75.7,
        75.5,
        75.2,
        75.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        75.6,
        75.5,
        75.1,
        75.3
      ],
      "time_embedding.linear_1": [
        75.5,
        75.5,
        75.1,
        75.3
      ],
      "up_blocks.1.attentions.0.proj_in": [
        75.5,
        75.3,
        75.0,
        75.1
      ],
      "down_blocks.1.attentions.0.proj_out": [
        75.4,
        75.1,
        74.9,
        75.1
      ],
      "up_blocks.1.resnets.2.conv2": [
        75.2,
        75.0,
        74.7,
        74.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        75.3,
        74.8,
        74.8,
        74.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        75.2,
        74.8,
        74.7,
        74.7
      ],
      "up_blocks.0.upsamplers.0.conv": [
        74.9,
        74.5,
        74.3,
        74.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        74.8,
        74.6,
        74.2,
        74.4
      ],
      "up_blocks.1.resnets.1.conv1": [
        74.7,
        74.4,
        74.3,
        74.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        74.6,
        74.2,
        74.2,
        74.3
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        74.4,
        74.1,
        74.1,
        74.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        74.4,
        74.1,
        73.9,
        74.1
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        74.4,
        74.0,
        74.0,
        74.0
      ],
      "time_embedding.linear_2": [
        74.2,
        73.9,
        74.1,
        74.2
      ],
      "up_blocks.1.resnets.2.conv1": [
        74.1,
        73.8,
        73.8,
        73.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        74.0,
        73.6,
        73.8,
        73.7
      ],
      "up_blocks.1.attentions.1.proj_in": [
        73.8,
        73.5,
        73.6,
        73.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        73.6,
        73.3,
        73.5,
        73.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        73.8,
        73.5,
        73.6,
        73.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        73.7,
        73.3,
        73.4,
        73.4
      ],
      "up_blocks.1.attentions.2.proj_out": [
        73.4,
        73.1,
        73.0,
        73.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        73.3,
        72.9,
        72.8,
        72.8
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        73.2,
        72.8,
        72.8,
        72.7
      ],
      "down_blocks.1.attentions.0.proj_in": [
        72.9,
        72.5,
        72.5,
        72.6
      ],
      "up_blocks.1.attentions.2.proj_in": [
        72.9,
        72.5,
        72.4,
        72.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        72.7,
        72.1,
        72.3,
        72.4
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        72.5,
        71.9,
        72.2,
        72.3
      ],
      "down_blocks.0.resnets.0.conv1": [
        72.3,
        72.0,
        72.1,
        72.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        72.2,
        71.9,
        71.9,
        72.2
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        72.0,
        71.6,
        71.7,
        71.9
      ],
      "down_blocks.0.downsamplers.0.conv": [
        71.8,
        71.5,
        71.7,
        71.8
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        71.7,
        71.3,
        71.4,
        71.4
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        71.3,
        70.9,
        70.9,
        71.2
      ],
      "up_blocks.2.resnets.0.conv1": [
        70.9,
        70.6,
        70.6,
        70.7
      ],
      "down_blocks.0.resnets.1.conv2": [
        70.6,
        70.2,
        70.5,
        70.3
      ],
      "up_blocks.2.resnets.0.conv2": [
        70.2,
        69.8,
        70.1,
        70.1
      ],
      "up_blocks.1.upsamplers.0.conv": [
        69.6,
        69.3,
        69.5,
        69.6
      ],
      "down_blocks.0.resnets.1.conv1": [
        69.0,
        68.6,
        68.8,
        68.9
      ],
      "down_blocks.0.resnets.0.conv2": [
        68.8,
        68.4,
        68.7,
        68.6
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        68.0,
        67.7,
        68.0,
        67.9
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        67.9,
        67.5,
        67.8,
        67.6
      ],
      "up_blocks.2.resnets.2.conv2": [
        67.6,
        67.2,
        67.5,
        67.4
      ],
      "up_blocks.2.resnets.1.conv1": [
        66.8,
        66.4,
        66.6,
        66.6
      ],
      "up_blocks.2.resnets.2.conv1": [
        66.1,
        65.6,
        65.9,
        65.9
      ],
      "up_blocks.2.resnets.1.conv2": [
        65.3,
        65.0,
        65.2,
        65.1
      ]
    },
    "8": {
      "metadata": {
        "nbits": 8,
        "out_ngroups": 1,
        "in_ngroups": 1,
        "cumulative": true,
        "candidates": [
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0",
          "down_blocks.2.attentions.1.proj_out",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k",
          "down_blocks.2.resnets.1.time_emb_proj",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v",
          "down_blocks.2.resnets.1.conv2",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q",
          "up_blocks.0.resnets.1.time_emb_proj",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_k",
          "down_blocks.1.attentions.0.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.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj",
          "down_blocks.2.resnets.0.conv1",
          "down_blocks.2.resnets.1.conv1",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k",
          "up_blocks.0.resnets.0.time_emb_proj",
          "up_blocks.0.resnets.0.conv2",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.1.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.4.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.5.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.6.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.9.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q",
          "down_blocks.2.attentions.1.proj_in",
          "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_v",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2",
          "down_blocks.2.resnets.0.time_emb_proj",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v",
          "mid_block.attentions.0.proj_in",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.2.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.2.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.3.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.5.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_k",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0",
          "mid_block.resnets.0.conv2",
          "mid_block.resnets.1.conv1",
          "mid_block.resnets.1.time_emb_proj",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.4.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_k",
          "mid_block.resnets.0.conv1",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
          "down_blocks.1.resnets.1.time_emb_proj",
          "down_blocks.2.attentions.0.proj_in",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0",
          "up_blocks.0.attentions.0.proj_in",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k",
          "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2",
          "up_blocks.0.resnets.1.conv2",
          "up_blocks.0.resnets.2.time_emb_proj",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.0.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "mid_block.attentions.0.transformer_blocks.1.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_v",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_v",
          "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.8.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.9.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj",
          "mid_block.resnets.0.time_emb_proj",
          "mid_block.resnets.1.conv2",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v",
          "down_blocks.2.resnets.0.conv2",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v",
          "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0",
          "mid_block.attentions.0.transformer_blocks.7.ff.net.2",
          "mid_block.attentions.0.transformer_blocks.8.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.9.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0",
          "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2",
          "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj",
          "up_blocks.0.attentions.0.proj_out",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k",
          "up_blocks.0.resnets.0.conv1",
          "mid_block.attentions.0.transformer_blocks.2.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.3.attn2.to_q",
          "mid_block.attentions.0.transformer_blocks.7.attn1.to_q",
          "mid_block.attentions.0.transformer_blocks.8.attn1.to_v",
          "mid_block.attentions.0.proj_out",
          "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v",
          "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2",
          "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj",
          "up_blocks.1.resnets.0.time_emb_proj",
          "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "mid_block.attentions.0.transformer_blocks.1.attn2.to_q",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v",
          "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2",
          "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj",
          "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0",
          "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k",
          "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0",
          "up_blocks.0.resnets.1.conv1",
          "mid_block.attentions.0.transformer_blocks.4.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.5.attn2.to_k",
          "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k",
          "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q",
          "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k",
          "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v",
          "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0",
          "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v",
          "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0",
          "up_blocks.0.resnets.2.conv1",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q",
          "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k",
          "up_blocks.0.attentions.1.proj_in",
          "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0",
          "up_blocks.0.resnets.2.conv2",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k",
          "down_blocks.2.attentions.0.proj_out",
          "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q",
          "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj",
          "up_blocks.0.resnets.0.conv_shortcut",
          "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj",
          "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q",
          "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2",
          "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2",
          "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k",
          "up_blocks.1.resnets.0.conv1",
          "down_blocks.1.downsamplers.0.conv",
          "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v",
          "up_blocks.0.attentions.1.proj_out",
          "up_blocks.0.resnets.1.conv_shortcut",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k",
          "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
          "down_blocks.1.resnets.1.conv1",
          "up_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.resnets.1.conv2",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "down_blocks.1.resnets.0.time_emb_proj",
          "up_blocks.0.attentions.2.proj_in",
          "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2",
          "down_blocks.2.resnets.0.conv_shortcut",
          "up_blocks.1.resnets.1.time_emb_proj",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q",
          "down_blocks.1.attentions.1.proj_in",
          "down_blocks.1.resnets.0.conv2",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2",
          "up_blocks.0.attentions.2.proj_out",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q",
          "up_blocks.1.resnets.0.conv2",
          "up_blocks.2.resnets.2.time_emb_proj",
          "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "down_blocks.1.attentions.1.proj_out",
          "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.resnets.2.time_emb_proj",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2",
          "up_blocks.1.attentions.0.proj_out",
          "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v",
          "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.1.resnets.1.conv2",
          "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v",
          "up_blocks.2.resnets.2.conv_shortcut",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v",
          "down_blocks.1.resnets.0.conv1",
          "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
          "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
          "down_blocks.1.attentions.0.proj_out",
          "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0",
          "up_blocks.1.attentions.1.proj_out",
          "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2",
          "up_blocks.1.resnets.1.conv1",
          "up_blocks.0.resnets.2.conv_shortcut",
          "down_blocks.1.attentions.0.proj_in",
          "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2",
          "up_blocks.2.resnets.1.time_emb_proj",
          "down_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.1.attentions.2.transformer_blocks.1.ff.net.0.proj",
          "up_blocks.2.resnets.0.time_emb_proj",
          "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.0.proj_in",
          "up_blocks.1.resnets.2.conv2",
          "time_embedding.linear_1",
          "up_blocks.1.resnets.2.conv1",
          "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
          "add_embedding.linear_1",
          "up_blocks.0.upsamplers.0.conv",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0",
          "add_embedding.linear_2",
          "up_blocks.1.attentions.1.proj_in",
          "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj",
          "up_blocks.1.resnets.0.conv_shortcut",
          "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj",
          "down_blocks.0.downsamplers.0.conv",
          "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0",
          "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v",
          "up_blocks.1.attentions.2.proj_out",
          "down_blocks.1.resnets.0.conv_shortcut",
          "up_blocks.1.attentions.2.proj_in",
          "down_blocks.0.resnets.1.time_emb_proj",
          "up_blocks.1.resnets.1.conv_shortcut",
          "up_blocks.2.resnets.2.conv2",
          "down_blocks.0.resnets.0.time_emb_proj",
          "time_embedding.linear_2",
          "up_blocks.1.resnets.2.conv_shortcut",
          "up_blocks.1.upsamplers.0.conv",
          "down_blocks.0.resnets.0.conv1",
          "up_blocks.2.resnets.0.conv2",
          "up_blocks.2.resnets.1.conv2",
          "up_blocks.2.resnets.1.conv_shortcut",
          "down_blocks.0.resnets.1.conv2",
          "up_blocks.2.resnets.0.conv1",
          "up_blocks.2.resnets.0.conv_shortcut",
          "down_blocks.0.resnets.0.conv2",
          "up_blocks.2.resnets.1.conv1",
          "up_blocks.2.resnets.2.conv1",
          "down_blocks.0.resnets.1.conv1"
        ],
        "sizes": [
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          2621440,
          6553600,
          6553600,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          2621440,
          1638400,
          6553600,
          13107200,
          2621440,
          1638400,
          409600,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          409600,
          2621440,
          1638400,
          13107200,
          2621440,
          2621440,
          13107200,
          6553600,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          13107200,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          13107200,
          2621440,
          2621440,
          14745600,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          6553600,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          13107200,
          1638400,
          1638400,
          13107200,
          2621440,
          409600,
          1638400,
          1638400,
          2621440,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          6553600,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          13107200,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          13107200,
          7372800,
          14745600,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          2621440,
          13107200,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          2621440,
          2621440,
          1638400,
          2621440,
          1638400,
          6553600,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          14745600,
          1310720,
          409600,
          1638400,
          1638400,
          2621440,
          2621440,
          6553600,
          1638400,
          1638400,
          2621440,
          13107200,
          1638400,
          13107200,
          1638400,
          2621440,
          6553600,
          6553600,
          6553600,
          1638400,
          1638400,
          2621440,
          1638400,
          6553600,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          409600,
          1310720,
          409600,
          409600,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          2621440,
          13107200,
          1638400,
          6553600,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          13107200,
          1638400,
          2621440,
          6553600,
          13107200,
          6553600,
          1638400,
          2621440,
          6553600,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          13107200,
          1638400,
          2621440,
          2621440,
          13107200,
          6553600,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          2621440,
          13107200,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          6553600,
          2621440,
          1638400,
          6553600,
          1638400,
          2621440,
          1638400,
          2621440,
          13107200,
          2621440,
          1638400,
          1310720,
          409600,
          1310720,
          1310720,
          1310720,
          1310720,
          1310720,
          1310720,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          6553600,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          13107200,
          1638400,
          1638400,
          2621440,
          1638400,
          2621440,
          1638400,
          14745600,
          14745600,
          1638400,
          1310720,
          2621440,
          2621440,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          409600,
          409600,
          1310720,
          1638400,
          1638400,
          14745600,
          1310720,
          409600,
          819200,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          13107200,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          6553600,
          1638400,
          1638400,
          1638400,
          6553600,
          13107200,
          1638400,
          6553600,
          1638400,
          2621440,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          2621440,
          1638400,
          1638400,
          1638400,
          6553600,
          6553600,
          1638400,
          2621440,
          1638400,
          1638400,
          6553600,
          14745600,
          1638400,
          409600,
          1310720,
          409600,
          409600,
          2621440,
          13107200,
          1638400,
          2621440,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          6553600,
          2621440,
          13107200,
          1638400,
          14745600,
          409600,
          409600,
          1310720,
          409600,
          1638400,
          13107200,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          14745600,
          1638400,
          13107200,
          6553600,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          2621440,
          13107200,
          1638400,
          1638400,
          2621440,
          1638400,
          1638400,
          1638400,
          13107200,
          2621440,
          13107200,
          409600,
          409600,
          409600,
          1310720,
          1310720,
          409600,
          1638400,
          1638400,
          1638400,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          2621440,
          29491200,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          1638400,
          6553600,
          6553600,
          13107200,
          1638400,
          6553600,
          1638400,
          1638400,
          13107200,
          1638400,
          1638400,
          13107200,
          819200,
          1638400,
          1638400,
          409600,
          409600,
          1310720,
          6553600,
          13107200,
          1638400,
          13107200,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          2621440,
          1638400,
          1638400,
          1638400,
          6553600,
          6553600,
          1638400,
          1638400,
          29491200,
          2621440,
          2621440,
          13107200,
          6553600,
          1310720,
          409600,
          1638400,
          1638400,
          13107200,
          6553600,
          1638400,
          1638400,
          1638400,
          1638400,
          13107200,
          1638400,
          6553600,
          13107200,
          1638400,
          13107200,
          13107200,
          1638400,
          13107200,
          1638400,
          13107200,
          1638400,
          1638400,
          22118400,
          409600,
          409600,
          409600,
          1638400,
          1638400,
          1638400,
          14745600,
          1310720,
          409600,
          1310720,
          1638400,
          1638400,
          6553600,
          1638400,
          13107200,
          3276800,
          13107200,
          13107200,
          409600,
          409600,
          409600,
          6553600,
          13107200,
          6553600,
          1638400,
          1638400,
          6553600,
          409600,
          409600,
          13107200,
          1638400,
          409600,
          11059200,
          3686400,
          13107200,
          409600,
          409600,
          409600,
          1638400,
          3276800,
          409600,
          409600,
          3686400,
          409600,
          409600,
          3686400,
          409600,
          409600,
          819200,
          1638400,
          1638400,
          819200,
          819200,
          409600,
          409600,
          3686400,
          1638400,
          1638400,
          1638400,
          409600,
          409600,
          3686400,
          409600,
          3276800,
          409600,
          409600,
          3276800,
          819200,
          409600,
          3276800,
          3276800,
          1638400,
          409600,
          1638400,
          409600,
          1638400,
          3686400,
          409600,
          204800,
          409600,
          1843200,
          1638400,
          409600,
          1638400,
          409600,
          409600,
          409600,
          409600,
          1638400,
          7372800,
          2457600,
          409600,
          3276800,
          1638400,
          409600,
          3276800,
          409600,
          3276800,
          409600,
          409600,
          409600,
          409600,
          3686400,
          409600,
          5529600,
          3276800,
          3604480,
          14745600,
          409600,
          1638400,
          409600,
          3276800,
          1228800,
          3276800,
          921600,
          409600,
          409600,
          409600,
          204800,
          409600,
          409600,
          819200,
          921600,
          409600,
          1638400,
          614400,
          3686400,
          921600,
          921600,
          921600,
          204800,
          921600,
          2764800,
          307200,
          921600,
          1843200,
          1843200,
          921600
        ]
      },
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q": [
        82.0,
        82.4,
        82.6,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        82.2,
        82.4,
        82.3,
        82.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q": [
        82.3,
        82.4,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q": [
        82.2,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k": [
        82.1,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k": [
        82.1,
        82.4,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q": [
        82.2,
        82.2,
        82.5,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        82.3,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v": [
        82.4,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k": [
        82.2,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_k": [
        82.0,
        82.2,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_q": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.2,
        82.4,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v": [
        82.2,
        82.4,
        82.4,
        82.3
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q": [
        82.3,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        82.0,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k": [
        82.1,
        82.4,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v": [
        82.3,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2": [
        82.3,
        82.3,
        82.5,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2": [
        82.3,
        82.3,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        82.2,
        82.3,
        82.5,
        82.3
      ],
      "down_blocks.2.attentions.1.proj_out": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v": [
        82.1,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        82.2,
        82.4,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v": [
        82.2,
        82.4,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.0,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v": [
        82.1,
        82.3,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q": [
        82.2,
        82.3,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q": [
        82.1,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k": [
        82.1,
        82.2,
        82.5,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        82.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_k": [
        82.2,
        82.3,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        82.1,
        82.3,
        82.5,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_k": [
        82.2,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        82.1,
        82.3,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        82.2,
        82.2,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_k": [
        82.2,
        82.1,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_v": [
        82.1,
        82.4,
        82.5,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_v": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2": [
        82.0,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        82.1,
        82.1,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        82.1,
        82.1,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k": [
        82.2,
        82.1,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v": [
        82.2,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.resnets.1.time_emb_proj": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q": [
        81.9,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q": [
        82.1,
        82.1,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v": [
        82.1,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v": [
        82.0,
        82.3,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q": [
        82.2,
        82.2,
        82.5,
        82.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_v": [
        82.0,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k": [
        81.9,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2": [
        82.1,
        82.3,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        81.9,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        82.0,
        82.2,
        82.4,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": [
        82.0,
        82.2,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": [
        82.0,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        82.1,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k": [
        82.2,
        82.1,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v": [
        82.1,
        82.2,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q": [
        82.1,
        82.1,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k": [
        82.0,
        82.1,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        82.1,
        82.2,
        82.4,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        82.2,
        82.3,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k": [
        82.1,
        82.2,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v": [
        82.1,
        82.2,
        82.2,
        82.1
      ],
      "down_blocks.2.resnets.1.conv2": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        82.2,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k": [
        82.1,
        82.1,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k": [
        82.2,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v": [
        82.1,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k": [
        81.9,
        82.1,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q": [
        82.1,
        82.1,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2": [
        82.0,
        82.3,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        82.0,
        82.3,
        82.2,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k": [
        82.1,
        82.3,
        82.2,
        82.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q": [
        82.2,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q": [
        82.1,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k": [
        82.0,
        82.1,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v": [
        82.2,
        82.3,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "up_blocks.0.resnets.1.time_emb_proj": [
        82.1,
        82.1,
        82.3,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": [
        82.2,
        82.2,
        82.2,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.3,
        82.2,
        82.2,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_v": [
        81.9,
        82.2,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_q": [
        81.9,
        82.2,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_k": [
        81.9,
        82.3,
        82.3,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        82.1,
        82.4,
        82.5,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_q": [
        81.9,
        82.2,
        82.4,
        82.2
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_q": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        82.0,
        82.2,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.0,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        82.0,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": [
        82.0,
        82.2,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": [
        82.0,
        82.1,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": [
        82.1,
        82.1,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q": [
        82.0,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2": [
        82.0,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q": [
        81.9,
        82.3,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q": [
        82.1,
        82.2,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2": [
        82.0,
        82.3,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v": [
        82.1,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        82.1,
        82.3,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q": [
        82.2,
        82.3,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k": [
        81.8,
        82.3,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q": [
        82.2,
        82.3,
        82.4,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        82.2,
        82.1,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k": [
        82.2,
        82.3,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v": [
        82.0,
        82.1,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2": [
        82.0,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": [
        82.1,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        82.3,
        82.1,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k": [
        82.1,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q": [
        82.1,
        82.1,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        82.2,
        82.1,
        82.4,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        82.2,
        82.2,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        82.2,
        82.3,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q": [
        82.1,
        82.3,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        82.1,
        82.3,
        82.1,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        82.1,
        82.1,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2": [
        81.8,
        82.3,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v": [
        82.0,
        82.3,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q": [
        82.1,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        82.0,
        82.1,
        82.3,
        82.2
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        82.0,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v": [
        82.0,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q": [
        82.0,
        82.2,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v": [
        82.1,
        82.3,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        82.0,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v": [
        82.2,
        82.1,
        82.3,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q": [
        82.2,
        82.2,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k": [
        82.1,
        82.2,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        82.1,
        82.2,
        82.4,
        82.1
      ],
      "down_blocks.2.resnets.0.conv1": [
        82.0,
        82.2,
        82.3,
        82.1
      ],
      "down_blocks.2.resnets.1.conv1": [
        82.1,
        82.1,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v": [
        82.2,
        82.2,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v": [
        82.1,
        82.3,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        82.0,
        82.2,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q": [
        82.0,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v": [
        82.1,
        82.0,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k": [
        82.1,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k": [
        82.0,
        82.1,
        82.2,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k": [
        82.0,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k": [
        82.2,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v": [
        81.9,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        82.0,
        82.2,
        82.4,
        82.2
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2": [
        81.8,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": [
        81.8,
        82.3,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q": [
        82.1,
        82.1,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v": [
        82.1,
        82.3,
        82.4,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q": [
        82.1,
        82.0,
        82.3,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k": [
        82.1,
        82.2,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v": [
        82.1,
        82.1,
        82.4,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k": [
        81.9,
        82.2,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k": [
        82.0,
        82.1,
        82.4,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v": [
        82.0,
        82.2,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k": [
        82.1,
        82.1,
        82.2,
        82.1
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k": [
        82.0,
        82.2,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        82.1,
        82.1,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2": [
        81.8,
        82.1,
        82.4,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q": [
        81.9,
        82.3,
        82.4,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0": [
        82.0,
        82.1,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v": [
        81.9,
        82.2,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0": [
        81.9,
        82.2,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        82.0,
        82.2,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0": [
        82.0,
        82.1,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0": [
        82.0,
        82.1,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k": [
        81.9,
        82.1,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q": [
        82.0,
        82.1,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k": [
        81.7,
        82.1,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q": [
        81.8,
        82.1,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k": [
        81.9,
        82.2,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k": [
        82.1,
        82.0,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "up_blocks.0.resnets.0.time_emb_proj": [
        81.8,
        82.0,
        82.1,
        81.9
      ],
      "up_blocks.0.resnets.0.conv2": [
        81.9,
        82.0,
        82.2,
        82.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.7,
        82.0,
        82.3,
        81.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        82.0,
        82.1,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": [
        82.0,
        82.0,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": [
        81.8,
        82.1,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.9,
        82.1,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_k": [
        81.8,
        82.2,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.2": [
        81.8,
        81.9,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_k": [
        81.8,
        82.1,
        82.1,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_v": [
        81.9,
        82.1,
        82.3,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_k": [
        81.9,
        82.0,
        82.1,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        81.9,
        82.1,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_v": [
        81.8,
        82.1,
        82.2,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        81.9,
        81.9,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_q": [
        81.9,
        82.0,
        82.2,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_v": [
        81.8,
        82.1,
        82.2,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.4.ff.net.2": [
        82.0,
        82.0,
        82.2,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.5.ff.net.2": [
        82.0,
        82.1,
        82.1,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.6.ff.net.2": [
        82.1,
        82.0,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_q": [
        81.8,
        82.0,
        82.2,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_k": [
        82.0,
        82.1,
        82.2,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        81.7,
        82.0,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.2": [
        82.1,
        82.0,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k": [
        81.9,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0": [
        82.0,
        82.0,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v": [
        82.0,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v": [
        81.9,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q": [
        81.9,
        82.0,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": [
        82.0,
        82.1,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q": [
        81.7,
        82.0,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q": [
        81.9,
        82.1,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": [
        81.8,
        82.1,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q": [
        82.0,
        82.1,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v": [
        81.8,
        82.1,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v": [
        81.8,
        82.0,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v": [
        81.8,
        82.1,
        82.2,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q": [
        81.9,
        82.0,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q": [
        82.0,
        82.1,
        82.3,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.9,
        82.0,
        82.1,
        81.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        81.9,
        82.0,
        82.1,
        82.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        81.9,
        82.0,
        82.2,
        82.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        81.8,
        82.0,
        82.1,
        81.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        81.8,
        81.9,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        81.9,
        82.2,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": [
        81.9,
        81.9,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": [
        82.0,
        82.1,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        82.0,
        82.1,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.0,
        82.1,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k": [
        81.9,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v": [
        81.9,
        82.0,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q": [
        81.9,
        82.2,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        81.9,
        82.0,
        82.1,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k": [
        81.9,
        81.9,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v": [
        81.9,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0": [
        82.0,
        82.1,
        81.9,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2": [
        82.0,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q": [
        82.0,
        82.1,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k": [
        81.9,
        82.0,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        82.0,
        82.0,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0": [
        81.9,
        81.9,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2": [
        82.0,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q": [
        82.0,
        82.0,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        81.9,
        82.0,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k": [
        81.9,
        82.1,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        82.0,
        82.0,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q": [
        82.0,
        82.0,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v": [
        81.9,
        82.0,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        81.9,
        82.1,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        82.1,
        82.1,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q": [
        81.8,
        82.1,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.proj_in": [
        81.9,
        82.0,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": [
        82.0,
        82.0,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": [
        82.0,
        82.0,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": [
        82.0,
        82.1,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k": [
        82.0,
        82.0,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v": [
        81.9,
        82.0,
        82.2,
        82.1
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        82.0,
        82.0,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q": [
        81.9,
        82.0,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v": [
        82.0,
        82.0,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        82.0,
        82.0,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k": [
        81.9,
        82.1,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v": [
        82.0,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q": [
        82.0,
        82.0,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k": [
        82.0,
        82.0,
        82.1,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        81.9,
        82.1,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q": [
        81.9,
        82.0,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q": [
        82.2,
        82.0,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k": [
        81.8,
        82.0,
        82.3,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k": [
        82.0,
        82.1,
        82.3,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        82.1,
        82.0,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v": [
        82.0,
        82.0,
        82.2,
        82.0
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2": [
        82.0,
        82.0,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        81.9,
        81.9,
        82.2,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2": [
        82.0,
        82.0,
        82.2,
        81.9
      ],
      "down_blocks.2.resnets.0.time_emb_proj": [
        81.9,
        81.9,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.8,
        82.0,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": [
        82.0,
        82.0,
        82.2,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        81.8,
        82.0,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        81.8,
        82.0,
        82.3,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        81.9,
        81.9,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k": [
        81.8,
        81.9,
        82.0,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q": [
        82.0,
        82.1,
        82.0,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k": [
        82.2,
        82.0,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v": [
        82.0,
        82.0,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj": [
        82.0,
        82.1,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2": [
        81.9,
        82.0,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k": [
        81.8,
        82.1,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        81.9,
        82.0,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj": [
        82.0,
        82.0,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v": [
        82.0,
        82.1,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        81.9,
        82.1,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k": [
        81.9,
        82.0,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v": [
        82.0,
        81.9,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        82.0,
        82.0,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k": [
        82.0,
        82.0,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v": [
        81.9,
        81.9,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        81.9,
        81.9,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k": [
        81.9,
        82.0,
        82.3,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v": [
        81.9,
        82.0,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0": [
        82.1,
        82.1,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": [
        81.8,
        82.0,
        82.3,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v": [
        82.0,
        82.0,
        82.2,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q": [
        81.9,
        82.1,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k": [
        81.9,
        81.9,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0": [
        81.9,
        82.0,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2": [
        81.9,
        82.0,
        82.2,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v": [
        82.0,
        82.0,
        82.0,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k": [
        81.8,
        82.2,
        82.2,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v": [
        82.0,
        82.0,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0": [
        82.0,
        82.0,
        82.2,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2": [
        81.9,
        82.0,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q": [
        81.7,
        81.9,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0": [
        81.9,
        82.0,
        82.0,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k": [
        81.9,
        82.0,
        81.9,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q": [
        81.9,
        82.0,
        82.0,
        82.0
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v": [
        81.8,
        82.0,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k": [
        82.0,
        82.0,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k": [
        81.7,
        81.9,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q": [
        81.9,
        82.0,
        82.0,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q": [
        81.9,
        82.0,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k": [
        81.8,
        81.8,
        82.0,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2": [
        81.8,
        81.9,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k": [
        82.1,
        82.0,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0": [
        81.6,
        82.0,
        82.0,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2": [
        82.0,
        81.9,
        82.1,
        82.0
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0": [
        81.8,
        82.1,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v": [
        81.8,
        81.9,
        81.9,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0": [
        82.1,
        81.7,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k": [
        81.8,
        82.0,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj": [
        81.9,
        82.0,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k": [
        81.9,
        82.0,
        82.1,
        81.9
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0": [
        81.9,
        82.0,
        82.1,
        81.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        81.8,
        82.0,
        82.0,
        81.8
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        82.0,
        82.0,
        82.0,
        81.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        81.9,
        82.0,
        82.1,
        81.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        81.7,
        82.0,
        82.2,
        81.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        81.9,
        82.1,
        82.1,
        81.7
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": [
        81.8,
        82.1,
        82.2,
        81.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        81.9,
        82.0,
        82.2,
        81.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        82.0,
        82.0,
        82.1,
        81.9
      ],
      "mid_block.attentions.0.proj_in": [
        81.8,
        82.0,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": [
        81.9,
        81.9,
        82.1,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_q": [
        81.9,
        82.0,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        81.9,
        81.9,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        82.0,
        81.8,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_q": [
        82.0,
        82.0,
        82.1,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.2.ff.net.2": [
        82.0,
        81.9,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_k": [
        82.0,
        82.0,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        81.8,
        82.0,
        82.1,
        82.0
      ],
      "mid_block.attentions.0.transformer_blocks.3.ff.net.2": [
        81.9,
        81.9,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        81.9,
        82.1,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_q": [
        81.8,
        81.8,
        82.2,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_v": [
        81.9,
        82.0,
        81.9,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_v": [
        81.9,
        81.9,
        81.9,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        81.8,
        82.0,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_v": [
        81.9,
        81.9,
        82.1,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_k": [
        82.0,
        82.0,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj": [
        81.9,
        81.9,
        82.0,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_k": [
        82.0,
        81.9,
        82.1,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0": [
        81.8,
        81.9,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_v": [
        81.8,
        82.0,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_q": [
        81.7,
        82.0,
        82.0,
        81.9
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_v": [
        81.9,
        81.8,
        82.0,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0": [
        82.0,
        82.1,
        82.0,
        82.0
      ],
      "mid_block.resnets.0.conv2": [
        81.9,
        82.0,
        82.0,
        81.9
      ],
      "mid_block.resnets.1.conv1": [
        81.9,
        82.0,
        82.0,
        81.9
      ],
      "mid_block.resnets.1.time_emb_proj": [
        81.9,
        82.1,
        82.1,
        81.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": [
        81.9,
        82.0,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k": [
        81.9,
        81.9,
        82.1,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v": [
        81.9,
        82.0,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2": [
        81.9,
        82.0,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v": [
        81.9,
        82.0,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v": [
        82.0,
        81.9,
        82.0,
        81.9
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        81.9,
        81.9,
        82.1,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q": [
        81.9,
        81.9,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2": [
        81.8,
        81.9,
        82.0,
        82.0
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q": [
        81.9,
        81.9,
        82.1,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k": [
        81.9,
        82.0,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0": [
        82.0,
        81.9,
        82.1,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k": [
        82.0,
        82.1,
        82.0,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        81.9,
        81.9,
        82.0,
        81.9
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v": [
        81.8,
        81.8,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k": [
        81.9,
        81.9,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0": [
        81.9,
        82.0,
        82.0,
        81.7
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        81.7,
        82.0,
        81.9,
        81.9
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        81.9,
        82.0,
        81.9,
        81.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v": [
        81.7,
        81.9,
        82.1,
        81.8
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_v": [
        81.8,
        81.8,
        82.0,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_k": [
        81.8,
        81.9,
        81.9,
        81.8
      ],
      "mid_block.resnets.0.conv1": [
        81.8,
        81.8,
        81.9,
        81.8
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": [
        81.9,
        81.8,
        81.9,
        81.7
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        81.9,
        81.8,
        82.0,
        81.8
      ],
      "down_blocks.1.resnets.1.time_emb_proj": [
        81.8,
        81.9,
        81.8,
        81.8
      ],
      "down_blocks.2.attentions.0.proj_in": [
        82.0,
        81.8,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        81.8,
        81.8,
        81.9,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v": [
        81.9,
        81.7,
        82.1,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0": [
        81.8,
        81.8,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q": [
        81.6,
        81.8,
        81.8,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2": [
        81.8,
        81.6,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k": [
        81.8,
        81.9,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q": [
        82.0,
        81.8,
        82.0,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj": [
        81.7,
        81.8,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q": [
        81.8,
        81.6,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k": [
        82.0,
        81.9,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q": [
        81.7,
        81.8,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k": [
        81.7,
        81.7,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v": [
        81.9,
        81.7,
        81.9,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        81.6,
        81.8,
        82.0,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q": [
        81.8,
        81.8,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k": [
        81.7,
        81.8,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0": [
        81.7,
        81.7,
        81.8,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        81.9,
        81.9,
        81.9,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v": [
        81.9,
        81.8,
        81.9,
        81.8
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q": [
        81.9,
        81.9,
        82.0,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v": [
        81.8,
        81.9,
        81.8,
        81.7
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        81.8,
        81.8,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.0.proj_in": [
        81.9,
        81.7,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": [
        81.8,
        81.5,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": [
        81.9,
        81.8,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k": [
        81.7,
        81.7,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2": [
        81.7,
        81.9,
        81.9,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q": [
        81.8,
        81.8,
        81.8,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v": [
        81.7,
        81.9,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        81.9,
        81.8,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2": [
        81.9,
        81.7,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj": [
        81.8,
        81.8,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q": [
        81.7,
        81.8,
        82.0,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2": [
        81.8,
        81.7,
        81.8,
        81.8
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v": [
        81.7,
        81.7,
        82.0,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k": [
        81.9,
        81.9,
        81.9,
        81.8
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        81.7,
        81.8,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2": [
        81.8,
        81.3,
        81.8,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0": [
        81.8,
        81.6,
        81.7,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0": [
        81.8,
        81.8,
        81.7,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0": [
        81.8,
        81.8,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q": [
        81.8,
        81.8,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v": [
        81.8,
        81.8,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v": [
        81.9,
        81.7,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0": [
        81.8,
        81.7,
        81.8,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k": [
        81.8,
        81.8,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q": [
        81.9,
        81.7,
        81.8,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k": [
        82.0,
        81.8,
        81.8,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        81.6,
        81.7,
        81.8,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2": [
        81.8,
        81.9,
        81.6,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v": [
        81.8,
        81.8,
        81.9,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q": [
        81.7,
        81.9,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v": [
        81.7,
        81.7,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q": [
        81.7,
        81.7,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2": [
        81.7,
        81.6,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2": [
        81.7,
        81.6,
        81.7,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v": [
        81.7,
        81.7,
        81.9,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v": [
        81.8,
        81.8,
        81.6,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0": [
        81.7,
        81.7,
        81.8,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q": [
        81.6,
        81.7,
        81.9,
        81.7
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2": [
        81.8,
        81.7,
        81.8,
        81.6
      ],
      "up_blocks.0.resnets.1.conv2": [
        81.6,
        81.7,
        81.8,
        81.6
      ],
      "up_blocks.0.resnets.2.time_emb_proj": [
        81.7,
        81.7,
        81.6,
        81.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": [
        81.6,
        81.5,
        81.7,
        81.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": [
        81.7,
        81.6,
        81.6,
        81.7
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": [
        81.6,
        81.5,
        81.7,
        81.6
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q": [
        81.6,
        81.5,
        81.7,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": [
        81.7,
        81.6,
        81.7,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        81.7,
        81.6,
        81.7,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_v": [
        81.8,
        81.7,
        81.7,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_v": [
        81.8,
        81.6,
        81.7,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_q": [
        81.7,
        81.7,
        81.7,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0": [
        81.7,
        81.7,
        81.8,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_k": [
        81.7,
        81.7,
        81.7,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_v": [
        81.7,
        81.7,
        81.7,
        81.7
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        81.7,
        81.7,
        81.7,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0": [
        81.8,
        81.6,
        81.8,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.2": [
        81.6,
        81.7,
        81.7,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_k": [
        81.5,
        81.7,
        81.8,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        81.7,
        81.7,
        81.7,
        81.7
      ],
      "mid_block.resnets.0.time_emb_proj": [
        81.7,
        81.7,
        81.6,
        81.5
      ],
      "mid_block.resnets.1.conv2": [
        81.7,
        81.7,
        81.8,
        81.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.8,
        81.7,
        81.7,
        81.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": [
        81.7,
        81.6,
        81.7,
        81.5
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": [
        81.7,
        81.6,
        81.7,
        81.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": [
        81.7,
        81.6,
        81.8,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": [
        81.7,
        81.6,
        81.7,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj": [
        81.7,
        81.7,
        81.8,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0": [
        81.6,
        81.7,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0": [
        81.7,
        81.7,
        81.7,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k": [
        81.7,
        81.7,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v": [
        81.6,
        81.7,
        81.7,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v": [
        81.6,
        81.6,
        81.7,
        81.5
      ],
      "down_blocks.2.resnets.0.conv2": [
        81.8,
        81.6,
        81.6,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": [
        81.8,
        81.6,
        81.6,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        81.7,
        81.7,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2": [
        81.7,
        81.7,
        81.7,
        81.7
      ],
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0": [
        81.7,
        81.5,
        81.8,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q": [
        81.8,
        81.7,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v": [
        81.7,
        81.6,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        81.8,
        81.7,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q": [
        81.7,
        81.7,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": [
        81.7,
        81.6,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": [
        81.8,
        81.6,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": [
        81.8,
        81.7,
        81.8,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k": [
        81.6,
        81.7,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj": [
        81.6,
        81.5,
        81.8,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0": [
        81.7,
        81.6,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v": [
        81.5,
        81.4,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k": [
        81.7,
        81.6,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k": [
        81.5,
        81.5,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q": [
        81.7,
        81.5,
        81.8,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q": [
        81.7,
        81.4,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj": [
        81.6,
        81.6,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v": [
        81.7,
        81.7,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj": [
        81.7,
        81.5,
        81.6,
        81.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": [
        81.6,
        81.6,
        81.7,
        81.4
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": [
        81.7,
        81.5,
        81.6,
        81.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        81.7,
        81.6,
        81.7,
        81.5
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": [
        81.6,
        81.6,
        81.6,
        81.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k": [
        81.7,
        81.5,
        81.8,
        81.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        81.6,
        81.6,
        81.7,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        81.7,
        81.5,
        81.6,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_q": [
        81.5,
        81.5,
        81.6,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0": [
        81.6,
        81.6,
        81.6,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.7.ff.net.2": [
        81.6,
        81.4,
        81.6,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_q": [
        81.6,
        81.5,
        81.6,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_q": [
        81.6,
        81.4,
        81.5,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q": [
        81.6,
        81.5,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0": [
        81.5,
        81.6,
        81.6,
        81.5
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0": [
        81.5,
        81.5,
        81.6,
        81.7
      ],
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2": [
        81.7,
        81.5,
        81.6,
        81.6
      ],
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        81.7,
        81.6,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.0.proj_out": [
        81.6,
        81.4,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q": [
        81.6,
        81.5,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q": [
        81.7,
        81.4,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q": [
        81.7,
        81.5,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0": [
        81.6,
        81.5,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k": [
        81.5,
        81.5,
        81.6,
        81.5
      ],
      "up_blocks.0.resnets.0.conv1": [
        81.7,
        81.5,
        81.6,
        81.6
      ],
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_q": [
        81.5,
        81.5,
        81.6,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_q": [
        81.6,
        81.4,
        81.6,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_q": [
        81.7,
        81.5,
        81.6,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_v": [
        81.6,
        81.6,
        81.7,
        81.5
      ],
      "mid_block.attentions.0.proj_out": [
        81.7,
        81.5,
        81.6,
        81.6
      ],
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v": [
        81.7,
        81.5,
        81.7,
        81.5
      ],
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2": [
        81.7,
        81.5,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2": [
        81.5,
        81.6,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        81.6,
        81.5,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q": [
        81.7,
        81.6,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2": [
        81.7,
        81.5,
        81.5,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v": [
        81.7,
        81.6,
        81.7,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v": [
        81.6,
        81.7,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        81.6,
        81.6,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v": [
        81.5,
        81.6,
        81.6,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q": [
        81.5,
        81.6,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj": [
        81.6,
        81.6,
        81.6,
        81.5
      ],
      "up_blocks.1.resnets.0.time_emb_proj": [
        81.5,
        81.5,
        81.5,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        81.6,
        81.6,
        81.7,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_q": [
        81.5,
        81.6,
        81.6,
        81.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": [
        81.6,
        81.6,
        81.5,
        81.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        81.5,
        81.6,
        81.6,
        81.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": [
        81.4,
        81.6,
        81.7,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": [
        81.6,
        81.6,
        81.7,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        81.5,
        81.6,
        81.6,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q": [
        81.5,
        81.6,
        81.5,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj": [
        81.7,
        81.6,
        81.6,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0": [
        81.5,
        81.6,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q": [
        81.5,
        81.5,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        81.6,
        81.6,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0": [
        81.6,
        81.6,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        81.5,
        81.4,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k": [
        81.5,
        81.5,
        81.7,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        81.5,
        81.4,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v": [
        81.6,
        81.5,
        81.5,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0": [
        81.5,
        81.5,
        81.5,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2": [
        81.5,
        81.5,
        81.6,
        81.6
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2": [
        81.5,
        81.4,
        81.7,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v": [
        81.4,
        81.5,
        81.5,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0": [
        81.5,
        81.6,
        81.5,
        81.5
      ],
      "up_blocks.0.resnets.1.conv1": [
        81.5,
        81.5,
        81.5,
        81.5
      ],
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_k": [
        81.4,
        81.4,
        81.5,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_k": [
        81.6,
        81.6,
        81.6,
        81.4
      ],
      "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj": [
        81.5,
        81.5,
        81.4,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2": [
        81.4,
        81.5,
        81.6,
        81.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": [
        81.5,
        81.5,
        81.5,
        81.4
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        81.4,
        81.4,
        81.5,
        81.4
      ],
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q": [
        81.4,
        81.5,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": [
        81.4,
        81.5,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        81.4,
        81.4,
        81.4,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": [
        81.3,
        81.4,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        81.2,
        81.5,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v": [
        81.4,
        81.5,
        81.5,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0": [
        81.4,
        81.4,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0": [
        81.4,
        81.3,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj": [
        81.3,
        81.7,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v": [
        81.4,
        81.5,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2": [
        81.5,
        81.4,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj": [
        81.4,
        81.5,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0": [
        81.5,
        81.5,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj": [
        81.5,
        81.5,
        81.4,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj": [
        81.5,
        81.5,
        81.6,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0": [
        81.5,
        81.4,
        81.5,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj": [
        81.5,
        81.5,
        81.5,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v": [
        81.4,
        81.5,
        81.5,
        81.4
      ],
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj": [
        81.5,
        81.5,
        81.6,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v": [
        81.4,
        81.6,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0": [
        81.5,
        81.5,
        81.7,
        81.5
      ],
      "up_blocks.0.resnets.2.conv1": [
        81.3,
        81.5,
        81.7,
        81.4
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        81.4,
        81.5,
        81.5,
        81.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": [
        81.3,
        81.4,
        81.5,
        81.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": [
        81.3,
        81.5,
        81.7,
        81.4
      ],
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k": [
        81.3,
        81.5,
        81.6,
        81.5
      ],
      "up_blocks.0.attentions.1.proj_in": [
        81.3,
        81.4,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0": [
        81.4,
        81.6,
        81.6,
        81.3
      ],
      "up_blocks.0.resnets.2.conv2": [
        81.4,
        81.4,
        81.6,
        81.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": [
        81.3,
        81.4,
        81.7,
        81.4
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        81.5,
        81.5,
        81.5,
        81.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": [
        81.3,
        81.4,
        81.7,
        81.3
      ],
      "down_blocks.2.attentions.0.proj_out": [
        81.4,
        81.4,
        81.6,
        81.3
      ],
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q": [
        81.5,
        81.4,
        81.7,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2": [
        81.4,
        81.4,
        81.6,
        81.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q": [
        81.4,
        81.5,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj": [
        81.4,
        81.2,
        81.5,
        81.2
      ],
      "up_blocks.0.resnets.0.conv_shortcut": [
        81.4,
        81.4,
        81.4,
        81.4
      ],
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj": [
        81.3,
        81.4,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj": [
        81.2,
        81.3,
        81.3,
        81.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": [
        81.4,
        81.2,
        81.5,
        81.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": [
        81.3,
        81.2,
        81.4,
        81.3
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        81.2,
        81.2,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2": [
        81.4,
        81.2,
        81.4,
        81.2
      ],
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj": [
        81.1,
        81.3,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2": [
        81.2,
        81.2,
        81.5,
        81.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0": [
        81.3,
        81.1,
        81.4,
        81.2
      ],
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0": [
        81.4,
        81.3,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2": [
        81.2,
        81.3,
        81.5,
        81.3
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": [
        81.2,
        81.2,
        81.4,
        81.3
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        81.3,
        81.3,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        81.3,
        81.4,
        81.5,
        81.3
      ],
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v": [
        81.3,
        81.4,
        81.3,
        81.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": [
        81.2,
        81.4,
        81.4,
        81.2
      ],
      "up_blocks.1.resnets.0.conv1": [
        81.2,
        81.3,
        81.4,
        81.3
      ],
      "down_blocks.1.downsamplers.0.conv": [
        81.1,
        81.3,
        81.4,
        81.3
      ],
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj": [
        81.0,
        81.2,
        81.5,
        81.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k": [
        81.1,
        81.1,
        81.4,
        81.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": [
        81.2,
        81.3,
        81.5,
        81.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        81.2,
        81.2,
        81.3,
        81.2
      ],
      "up_blocks.0.attentions.1.proj_out": [
        81.2,
        81.2,
        81.4,
        81.2
      ],
      "up_blocks.0.resnets.1.conv_shortcut": [
        81.2,
        81.3,
        81.4,
        81.3
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": [
        81.1,
        81.2,
        81.5,
        81.2
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        81.1,
        81.3,
        81.3,
        81.2
      ],
      "down_blocks.1.resnets.1.conv1": [
        81.2,
        81.2,
        81.5,
        81.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": [
        81.2,
        81.3,
        81.5,
        81.0
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        81.1,
        81.2,
        81.3,
        81.2
      ],
      "down_blocks.1.resnets.1.conv2": [
        81.1,
        81.1,
        81.4,
        81.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": [
        81.1,
        81.3,
        81.3,
        81.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        81.0,
        81.2,
        81.5,
        81.0
      ],
      "down_blocks.1.resnets.0.time_emb_proj": [
        81.0,
        81.2,
        81.3,
        81.1
      ],
      "up_blocks.0.attentions.2.proj_in": [
        81.0,
        81.3,
        81.3,
        81.1
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        81.0,
        81.1,
        81.2,
        81.2
      ],
      "down_blocks.2.resnets.0.conv_shortcut": [
        80.9,
        81.2,
        81.2,
        81.1
      ],
      "up_blocks.1.resnets.1.time_emb_proj": [
        81.1,
        81.2,
        81.1,
        81.0
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": [
        81.0,
        81.2,
        81.2,
        81.1
      ],
      "down_blocks.1.attentions.1.proj_in": [
        80.9,
        81.1,
        81.1,
        81.0
      ],
      "down_blocks.1.resnets.0.conv2": [
        80.8,
        81.1,
        81.3,
        80.9
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        80.9,
        81.1,
        81.3,
        80.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        80.8,
        81.0,
        81.2,
        80.9
      ],
      "up_blocks.0.attentions.2.proj_out": [
        80.9,
        81.0,
        81.1,
        81.0
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": [
        80.9,
        81.1,
        81.2,
        80.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q": [
        80.7,
        81.0,
        81.2,
        81.0
      ],
      "up_blocks.1.resnets.0.conv2": [
        80.8,
        81.0,
        81.1,
        80.8
      ],
      "up_blocks.2.resnets.2.time_emb_proj": [
        80.6,
        80.8,
        81.1,
        80.8
      ],
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        80.6,
        80.8,
        81.0,
        80.7
      ],
      "down_blocks.1.attentions.1.proj_out": [
        80.6,
        80.7,
        81.0,
        80.7
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        80.6,
        80.6,
        80.9,
        80.6
      ],
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        80.6,
        80.6,
        80.9,
        80.7
      ],
      "up_blocks.1.resnets.2.time_emb_proj": [
        80.6,
        80.6,
        80.9,
        80.6
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        80.7,
        80.6,
        80.9,
        80.6
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        80.7,
        80.5,
        80.8,
        80.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        80.7,
        80.5,
        80.8,
        80.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": [
        80.5,
        80.6,
        80.9,
        80.5
      ],
      "up_blocks.1.attentions.0.proj_out": [
        80.6,
        80.5,
        80.8,
        80.5
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": [
        80.5,
        80.4,
        80.8,
        80.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        80.7,
        80.5,
        80.7,
        80.5
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        80.5,
        80.5,
        80.7,
        80.5
      ],
      "up_blocks.1.resnets.1.conv2": [
        80.6,
        80.4,
        80.6,
        80.5
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": [
        80.5,
        80.5,
        80.6,
        80.5
      ],
      "up_blocks.2.resnets.2.conv_shortcut": [
        80.4,
        80.5,
        80.5,
        80.3
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": [
        80.5,
        80.6,
        80.6,
        80.2
      ],
      "down_blocks.1.resnets.0.conv1": [
        80.5,
        80.5,
        80.7,
        80.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": [
        80.3,
        80.3,
        80.6,
        80.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": [
        80.3,
        80.4,
        80.4,
        80.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": [
        80.4,
        80.4,
        80.3,
        80.2
      ],
      "down_blocks.1.attentions.0.proj_out": [
        80.2,
        80.5,
        80.4,
        80.2
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": [
        80.2,
        80.4,
        80.5,
        80.2
      ],
      "up_blocks.1.attentions.1.proj_out": [
        80.3,
        80.4,
        80.3,
        80.2
      ],
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": [
        80.1,
        80.3,
        80.4,
        80.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": [
        80.2,
        80.3,
        80.3,
        80.1
      ],
      "up_blocks.1.resnets.1.conv1": [
        80.1,
        80.2,
        80.2,
        80.2
      ],
      "up_blocks.0.resnets.2.conv_shortcut": [
        80.1,
        80.1,
        80.1,
        80.2
      ],
      "down_blocks.1.attentions.0.proj_in": [
        80.1,
        80.2,
        80.3,
        80.1
      ],
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": [
        80.1,
        80.2,
        80.2,
        79.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2": [
        79.9,
        80.1,
        80.3,
        79.8
      ],
      "up_blocks.2.resnets.1.time_emb_proj": [
        79.8,
        80.0,
        80.2,
        79.9
      ],
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": [
        79.8,
        79.9,
        80.1,
        79.9
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": [
        79.9,
        79.9,
        80.1,
        79.9
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj": [
        79.8,
        79.7,
        80.0,
        79.7
      ],
      "up_blocks.2.resnets.0.time_emb_proj": [
        79.8,
        79.8,
        79.8,
        79.8
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": [
        79.8,
        79.8,
        79.8,
        79.8
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": [
        79.9,
        79.8,
        80.0,
        79.8
      ],
      "up_blocks.1.attentions.0.proj_in": [
        79.7,
        79.8,
        80.0,
        80.0
      ],
      "up_blocks.1.resnets.2.conv2": [
        79.6,
        79.9,
        80.0,
        79.7
      ],
      "time_embedding.linear_1": [
        79.7,
        79.6,
        79.8,
        79.6
      ],
      "up_blocks.1.resnets.2.conv1": [
        79.7,
        79.5,
        79.8,
        79.6
      ],
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": [
        79.7,
        79.4,
        79.8,
        79.5
      ],
      "add_embedding.linear_1": [
        79.7,
        79.4,
        79.7,
        79.5
      ],
      "up_blocks.0.upsamplers.0.conv": [
        79.5,
        79.3,
        79.6,
        79.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0": [
        79.6,
        79.3,
        79.6,
        79.4
      ],
      "add_embedding.linear_2": [
        79.6,
        79.4,
        79.6,
        79.4
      ],
      "up_blocks.1.attentions.1.proj_in": [
        79.6,
        79.3,
        79.5,
        79.4
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": [
        79.4,
        79.4,
        79.5,
        79.3
      ],
      "up_blocks.1.resnets.0.conv_shortcut": [
        79.4,
        79.1,
        79.2,
        79.2
      ],
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": [
        79.4,
        79.0,
        79.4,
        79.0
      ],
      "down_blocks.0.downsamplers.0.conv": [
        79.3,
        78.9,
        79.5,
        79.1
      ],
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": [
        79.3,
        78.8,
        79.4,
        79.2
      ],
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v": [
        79.2,
        79.0,
        79.3,
        79.1
      ],
      "up_blocks.1.attentions.2.proj_out": [
        79.3,
        79.0,
        79.1,
        79.3
      ],
      "down_blocks.1.resnets.0.conv_shortcut": [
        79.2,
        78.8,
        79.0,
        78.9
      ],
      "up_blocks.1.attentions.2.proj_in": [
        79.3,
        78.8,
        79.1,
        79.0
      ],
      "down_blocks.0.resnets.1.time_emb_proj": [
        79.0,
        78.8,
        78.9,
        78.8
      ],
      "up_blocks.1.resnets.1.conv_shortcut": [
        78.9,
        78.5,
        78.7,
        78.5
      ],
      "up_blocks.2.resnets.2.conv2": [
        78.4,
        78.0,
        78.0,
        78.1
      ],
      "down_blocks.0.resnets.0.time_emb_proj": [
        78.1,
        78.0,
        78.2,
        78.1
      ],
      "time_embedding.linear_2": [
        78.1,
        77.9,
        78.5,
        78.6
      ],
      "up_blocks.1.resnets.2.conv_shortcut": [
        77.9,
        77.8,
        78.4,
        78.3
      ],
      "up_blocks.1.upsamplers.0.conv": [
        77.7,
        77.5,
        78.2,
        78.1
      ],
      "down_blocks.0.resnets.0.conv1": [
        77.6,
        77.4,
        78.1,
        78.2
      ],
      "up_blocks.2.resnets.0.conv2": [
        77.4,
        77.3,
        77.7,
        78.0
      ],
      "up_blocks.2.resnets.1.conv2": [
        77.4,
        77.3,
        77.8,
        77.8
      ],
      "up_blocks.2.resnets.1.conv_shortcut": [
        77.2,
        77.1,
        77.6,
        77.6
      ],
      "down_blocks.0.resnets.1.conv2": [
        77.3,
        77.2,
        77.6,
        77.8
      ],
      "up_blocks.2.resnets.0.conv1": [
        77.3,
        77.0,
        77.3,
        77.5
      ],
      "up_blocks.2.resnets.0.conv_shortcut": [
        77.1,
        77.1,
        77.0,
        77.0
      ],
      "down_blocks.0.resnets.0.conv2": [
        77.0,
        77.0,
        77.0,
        77.0
      ],
      "up_blocks.2.resnets.1.conv1": [
        76.7,
        76.5,
        76.7,
        76.4
      ],
      "up_blocks.2.resnets.2.conv1": [
        76.2,
        76.1,
        76.1,
        76.0
      ],
      "down_blocks.0.resnets.1.conv1": [
        75.8,
        75.5,
        75.9,
        75.9
      ]
    }
  },
  "model_version": "stabilityai/stable-diffusion-xl-base-1.0",
  "baselines": {
    "original": 82.2,
    "linear_8bit": 66.025,
    "recipe_6.55_bit_mixedpalette": 79.9,
    "recipe_4.50_bit_mixedpalette": 75.8,
    "recipe_3.41_bit_mixedpalette": 71.7,
    "recipe_3.05_bit_mixedpalette": 69.2,
    "recipe_2.81_bit_mixedpalette": 66.8,
    "recipe_2.31_bit_mixedpalette": 64.3
  },
  "recipes": {
    "recipe_6.55_bit_mixedpalette": {
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k": 4,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q": 4,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v": 4,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k": 4,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q": 4,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q": 4,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": 2,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0": 4,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0": 4,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k": 4,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0": 6,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v": 6,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0": 6,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v": 6,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q": 4,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k": 6,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k": 4,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k": 6,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v": 6,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q": 6,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q": 4,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q": 6,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v": 6,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k": 6,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q": 6,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k": 4,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q": 4,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q": 4,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k": 6,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q": 6,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q": 4,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0": 6,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k": 4,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q": 4,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k": 4,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v": 6,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v": 6,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v": 6,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k": 6,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_q": 4,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 4,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2": 6,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0": 6,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_q": 6,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k": 6,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0": 6,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v": 8,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q": 4,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k": 6,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k": 6,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k": 4,
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2": 6,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_q": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v": 6,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0": 6,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0": 6,
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2": 6,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0": 6,
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2": 6,
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2": 4,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k": 4,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0": 6,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q": 4,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj": 6,
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2": 6,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0": 6,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_v": 6,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0": 6,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q": 4,
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2": 6,
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2": 6,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v": 6,
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2": 6,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_v": 6,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v": 6,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q": 4,
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2": 6,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k": 4,
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2": 6,
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2": 6,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v": 6,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0": 6,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k": 6,
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2": 4,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v": 6,
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2": 6,
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj": 6,
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj": 6,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0": 6,
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj": 6,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0": 6,
      "down_blocks.2.resnets.1.time_emb_proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q": 6,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q": 6,
      "mid_block.attentions.0.transformer_blocks.7.ff.net.2": 4,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v": 6,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0": 6,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k": 6,
      "mid_block.attentions.0.transformer_blocks.8.ff.net.2": 6,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v": 6,
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj": 6,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0": 6,
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj": 6,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0": 6,
      "mid_block.resnets.1.time_emb_proj": 6,
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2": 6,
      "mid_block.attentions.0.transformer_blocks.6.ff.net.2": 4,
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj": 6,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q": 6,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_v": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q": 4,
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj": 6,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k": 6,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0": 6,
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj": 6,
      "mid_block.attentions.0.transformer_blocks.5.ff.net.2": 8,
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj": 6,
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2": 4,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k": 6,
      "mid_block.attentions.0.transformer_blocks.2.ff.net.2": 6,
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0": 8,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0": 6,
      "mid_block.attentions.0.transformer_blocks.4.ff.net.2": 6,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_k": 6,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v": 6,
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj": 6,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q": 6,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v": 6,
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k": 6,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q": 6,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v": 6,
      "up_blocks.0.resnets.0.time_emb_proj": 6,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0": 6,
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2": 8,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k": 6,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0": 8,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k": 6,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0": 6,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k": 6,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q": 8,
      "mid_block.attentions.0.transformer_blocks.9.ff.net.2": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v": 6,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0": 6,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k": 6,
      "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q": 6,
      "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj": 6,
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2": 6,
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj": 6,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q": 6,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0": 6,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v": 6,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_k": 6,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0": 6,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_v": 6,
      "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj": 6,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v": 6,
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2": 8,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k": 6,
      "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj": 6,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_q": 6,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v": 6,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0": 6,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_q": 6,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q": 6,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0": 6,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k": 6,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q": 6,
      "mid_block.attentions.0.transformer_blocks.3.ff.net.2": 6,
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2": 6,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k": 6,
      "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj": 6,
      "mid_block.resnets.0.time_emb_proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k": 6,
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2": 6,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_k": 6,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q": 6,
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj": 6,
      "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj": 6,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v": 8,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k": 6,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0": 6,
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2": 8,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k": 6,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q": 6,
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q": 6,
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj": 6,
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj": 6,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_v": 8,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k": 6,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k": 6,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q": 6,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q": 6,
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2": 6,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v": 8,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj": 8,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k": 6,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q": 6,
      "mid_block.attentions.0.transformer_blocks.1.ff.net.2": 6,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_v": 6,
      "mid_block.resnets.1.conv1": 8,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v": 6,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0": 6,
      "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k": 8,
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2": 6,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q": 6,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k": 6,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q": 6,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 8,
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2": 6,
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj": 8,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k": 6,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v": 6,
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj": 6,
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj": 8,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q": 6,
      "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj": 6,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v": 8,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v": 6,
      "down_blocks.2.attentions.1.proj_in": 8,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_k": 6,
      "mid_block.resnets.1.conv2": 6,
      "down_blocks.2.resnets.1.conv1": 6,
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v": 6,
      "up_blocks.0.resnets.0.conv1": 8,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v": 6,
      "down_blocks.2.resnets.1.conv2": 6,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_q": 6,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0": 8,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0": 8,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2": 8,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v": 8,
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2": 8,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_v": 8,
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj": 8,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0": 6,
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2": 8,
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2": 8,
      "up_blocks.0.resnets.1.time_emb_proj": 8,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2": 8,
      "mid_block.resnets.0.conv1": 6,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0": 6,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0": 8,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v": 8,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v": 8,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v": 8,
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2": 8,
      "down_blocks.2.resnets.0.time_emb_proj": 6,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v": 8,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj": 6,
      "mid_block.resnets.0.conv2": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.0.resnets.1.conv2": 8,
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2": 8,
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2": 8,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v": 8,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v": 8,
      "down_blocks.2.attentions.1.proj_out": 8,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0": 8,
      "up_blocks.0.resnets.2.time_emb_proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v": 8,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0": 8,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0": 8,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0": 8,
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj": 8,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0": 8,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v": 8,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0": 8,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v": 8,
      "mid_block.attentions.0.proj_out": 6,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_q": 6,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v": 6,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v": 8,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0": 8,
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj": 8,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0": 8,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v": 8,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0": 8,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_k": 6,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0": 8,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v": 8,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0": 8,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0": 8,
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj": 8,
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2": 8,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v": 8,
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj": 8,
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2": 8,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.0.resnets.1.conv1": 8,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v": 8,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v": 8,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v": 8,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": 6,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_v": 8,
      "up_blocks.0.resnets.0.conv2": 8,
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj": 8,
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj": 8,
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2": 8,
      "up_blocks.0.resnets.2.conv2": 8,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0": 8,
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj": 8,
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2": 8,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 8,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v": 8,
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2": 8,
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj": 8,
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2": 8,
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 8,
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj": 8,
      "up_blocks.0.attentions.0.proj_out": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2": 8,
      "up_blocks.0.resnets.2.conv1": 8,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0": 8,
      "down_blocks.2.resnets.0.conv2": 8,
      "up_blocks.0.resnets.0.conv_shortcut": 8,
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v": 8,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj": 8,
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj": 8,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": 8,
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj": 8,
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj": 8,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": 6,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj": 8,
      "down_blocks.2.attentions.0.proj_in": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "up_blocks.0.attentions.0.proj_in": 8,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": 8,
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj": 8,
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj": 8,
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj": 8,
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.2.attentions.0.proj_out": 8,
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2": 8,
      "mid_block.attentions.0.proj_in": 8,
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": 8,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": 8,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": 8,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": 8,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k": 4,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0": 8,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": 6,
      "up_blocks.0.attentions.2.proj_in": 16,
      "down_blocks.2.resnets.0.conv1": 6,
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj": 8,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": 8,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 8,
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": 8,
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj": 8,
      "up_blocks.1.resnets.0.time_emb_proj": 8,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": 8,
      "up_blocks.0.attentions.2.proj_out": 16,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q": 6,
      "add_embedding.linear_2": 16,
      "down_blocks.1.resnets.1.time_emb_proj": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 8,
      "up_blocks.0.resnets.1.conv_shortcut": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 8,
      "up_blocks.0.attentions.1.proj_in": 8,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 8,
      "add_embedding.linear_1": 16,
      "up_blocks.1.resnets.0.conv1": 8,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": 8,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": 8,
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": 16,
      "up_blocks.1.resnets.0.conv2": 16,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": 16,
      "up_blocks.0.attentions.1.proj_out": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 16,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 8,
      "down_blocks.1.resnets.0.time_emb_proj": 16,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 16,
      "down_blocks.1.resnets.1.conv2": 16,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 16,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": 8,
      "up_blocks.1.attentions.0.proj_out": 16,
      "down_blocks.1.attentions.1.proj_out": 16,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": 8,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": 8,
      "down_blocks.1.resnets.1.conv1": 8,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": 8,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": 16,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k": 16,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 16,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": 8,
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": 16,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": 8,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": 16,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": 16,
      "down_blocks.2.resnets.0.conv_shortcut": 16,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 16,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 8,
      "down_blocks.1.resnets.0.conv2": 16,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 8,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q": 16,
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": 16,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": 8,
      "down_blocks.1.attentions.1.proj_in": 16,
      "up_blocks.1.resnets.1.time_emb_proj": 16,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 16,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 16,
      "up_blocks.0.resnets.2.conv_shortcut": 16,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": 16,
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": 16,
      "up_blocks.1.attentions.0.proj_in": 16,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 16,
      "down_blocks.1.downsamplers.0.conv": 8,
      "up_blocks.1.resnets.2.time_emb_proj": 16,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 16,
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2": 16,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": 16,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 16,
      "up_blocks.1.attentions.1.proj_out": 16,
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": 16,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": 16,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 16,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0": 16,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": 16,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": 16,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 16,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v": 16,
      "up_blocks.0.upsamplers.0.conv": 16,
      "down_blocks.1.attentions.0.proj_out": 16,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 16,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": 16,
      "up_blocks.1.resnets.1.conv2": 16,
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": 16,
      "up_blocks.1.resnets.2.conv2": 16,
      "up_blocks.1.resnets.1.conv1": 16,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": 8,
      "up_blocks.1.attentions.2.proj_out": 16,
      "up_blocks.1.attentions.1.proj_in": 16,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 16,
      "down_blocks.1.resnets.0.conv1": 16,
      "up_blocks.1.attentions.2.proj_in": 16,
      "up_blocks.1.resnets.2.conv1": 16,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": 16,
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": 16,
      "down_blocks.0.resnets.1.time_emb_proj": 16,
      "up_blocks.2.resnets.0.time_emb_proj": 16,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": 16,
      "down_blocks.1.attentions.0.proj_in": 16,
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj": 16,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 16,
      "up_blocks.2.resnets.1.time_emb_proj": 16,
      "up_blocks.2.resnets.2.time_emb_proj": 16,
      "down_blocks.0.resnets.0.time_emb_proj": 16,
      "up_blocks.1.resnets.0.conv_shortcut": 16,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k": 6,
      "down_blocks.0.resnets.1.conv2": 16,
      "down_blocks.0.resnets.0.conv1": 16,
      "up_blocks.1.resnets.1.conv_shortcut": 16,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.resnets.1.conv1": 16,
      "down_blocks.1.resnets.0.conv_shortcut": 16,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.0.resnets.0.conv2": 16,
      "down_blocks.0.downsamplers.0.conv": 16,
      "up_blocks.2.resnets.0.conv1": 16,
      "up_blocks.2.resnets.0.conv2": 16,
      "up_blocks.1.resnets.2.conv_shortcut": 16,
      "up_blocks.2.resnets.1.conv2": 16,
      "time_embedding.linear_2": 16,
      "up_blocks.2.resnets.1.conv1": 16,
      "up_blocks.1.upsamplers.0.conv": 16,
      "time_embedding.linear_1": 16,
      "up_blocks.2.resnets.0.conv_shortcut": 16,
      "up_blocks.2.resnets.2.conv1": 16,
      "up_blocks.2.resnets.1.conv_shortcut": 16,
      "up_blocks.2.resnets.2.conv2": 16,
      "up_blocks.2.resnets.2.conv_shortcut": 16
    },
    "recipe_4.50_bit_mixedpalette": {
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 2,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v": 2,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v": 4,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 4,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2": 4,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0": 6,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0": 4,
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2": 4,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2": 4,
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2": 4,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k": 4,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0": 6,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q": 4,
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2": 4,
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2": 4,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q": 4,
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k": 4,
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2": 4,
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2": 4,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k": 4,
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v": 4,
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2": 4,
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj": 4,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0": 4,
      "down_blocks.2.resnets.1.time_emb_proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.7.ff.net.2": 4,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.8.ff.net.2": 4,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v": 4,
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj": 4,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj": 4,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0": 4,
      "mid_block.resnets.1.time_emb_proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2": 4,
      "mid_block.attentions.0.transformer_blocks.6.ff.net.2": 4,
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q": 4,
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj": 4,
      "mid_block.attentions.0.transformer_blocks.5.ff.net.2": 4,
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj": 4,
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2": 4,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.2.ff.net.2": 4,
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.4.ff.net.2": 4,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v": 4,
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v": 4,
      "up_blocks.0.resnets.0.time_emb_proj": 4,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2": 6,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.9.ff.net.2": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0": 6,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj": 4,
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2": 4,
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q": 4,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0": 4,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj": 4,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2": 6,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj": 4,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.3.ff.net.2": 4,
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2": 6,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj": 4,
      "mid_block.resnets.0.time_emb_proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2": 6,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q": 4,
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj": 4,
      "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2": 4,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj": 6,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj": 4,
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj": 4,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2": 4,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.1.ff.net.2": 4,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_v": 4,
      "mid_block.resnets.1.conv1": 6,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2": 4,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2": 6,
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj": 6,
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v": 4,
      "down_blocks.2.attentions.1.proj_in": 4,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_k": 4,
      "mid_block.resnets.1.conv2": 6,
      "down_blocks.2.resnets.1.conv1": 6,
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v": 4,
      "up_blocks.0.resnets.0.conv1": 6,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v": 4,
      "down_blocks.2.resnets.1.conv2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0": 6,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2": 6,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2": 6,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj": 6,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0": 6,
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2": 6,
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2": 6,
      "up_blocks.0.resnets.1.time_emb_proj": 6,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2": 6,
      "mid_block.resnets.0.conv1": 6,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v": 6,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2": 6,
      "down_blocks.2.resnets.0.time_emb_proj": 6,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v": 6,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj": 6,
      "mid_block.resnets.0.conv2": 6,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.0.resnets.1.conv2": 6,
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2": 6,
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2": 6,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v": 6,
      "down_blocks.2.attentions.1.proj_out": 6,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0": 6,
      "up_blocks.0.resnets.2.time_emb_proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v": 6,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0": 6,
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj": 6,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0": 6,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v": 6,
      "mid_block.attentions.0.proj_out": 4,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v": 6,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0": 6,
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v": 6,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0": 6,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_k": 6,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0": 6,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0": 6,
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj": 6,
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2": 6,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v": 6,
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2": 6,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k": 2,
      "up_blocks.0.resnets.1.conv1": 6,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v": 6,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v": 6,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": 6,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_v": 4,
      "up_blocks.0.resnets.0.conv2": 6,
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj": 6,
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2": 6,
      "up_blocks.0.resnets.2.conv2": 6,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0": 6,
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj": 6,
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2": 6,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2": 6,
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj": 6,
      "up_blocks.0.attentions.0.proj_out": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2": 6,
      "up_blocks.0.resnets.2.conv1": 6,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0": 6,
      "down_blocks.2.resnets.0.conv2": 6,
      "up_blocks.0.resnets.0.conv_shortcut": 6,
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": 6,
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj": 6,
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj": 6,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj": 6,
      "down_blocks.2.attentions.0.proj_in": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.0.attentions.0.proj_in": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": 4,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": 6,
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj": 6,
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.2.attentions.0.proj_out": 6,
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2": 6,
      "mid_block.attentions.0.proj_in": 6,
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": 6,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k": 4,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": 4,
      "up_blocks.0.attentions.2.proj_in": 6,
      "down_blocks.2.resnets.0.conv1": 6,
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": 6,
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj": 6,
      "up_blocks.1.resnets.0.time_emb_proj": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": 6,
      "up_blocks.0.attentions.2.proj_out": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q": 6,
      "add_embedding.linear_2": 8,
      "down_blocks.1.resnets.1.time_emb_proj": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.0.resnets.1.conv_shortcut": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.0.attentions.1.proj_in": 6,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "add_embedding.linear_1": 8,
      "up_blocks.1.resnets.0.conv1": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": 8,
      "up_blocks.1.resnets.0.conv2": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": 8,
      "up_blocks.0.attentions.1.proj_out": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 6,
      "down_blocks.1.resnets.0.time_emb_proj": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.1.resnets.1.conv2": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.1.attentions.0.proj_out": 8,
      "down_blocks.1.attentions.1.proj_out": 8,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": 8,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": 6,
      "down_blocks.1.resnets.1.conv1": 8,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": 8,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": 8,
      "down_blocks.2.resnets.0.conv_shortcut": 8,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.1.resnets.0.conv2": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q": 8,
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": 8,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": 6,
      "down_blocks.1.attentions.1.proj_in": 8,
      "up_blocks.1.resnets.1.time_emb_proj": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 8,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.0.resnets.2.conv_shortcut": 8,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": 8,
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": 8,
      "up_blocks.1.attentions.0.proj_in": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.1.downsamplers.0.conv": 8,
      "up_blocks.1.resnets.2.time_emb_proj": 8,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2": 8,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.1.attentions.1.proj_out": 8,
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0": 8,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v": 8,
      "up_blocks.0.upsamplers.0.conv": 8,
      "down_blocks.1.attentions.0.proj_out": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.1.resnets.1.conv2": 8,
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": 8,
      "up_blocks.1.resnets.2.conv2": 8,
      "up_blocks.1.resnets.1.conv1": 8,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": 6,
      "up_blocks.1.attentions.2.proj_out": 8,
      "up_blocks.1.attentions.1.proj_in": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.resnets.0.conv1": 8,
      "up_blocks.1.attentions.2.proj_in": 8,
      "up_blocks.1.resnets.2.conv1": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": 8,
      "down_blocks.0.resnets.1.time_emb_proj": 8,
      "up_blocks.2.resnets.0.time_emb_proj": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.1.attentions.0.proj_in": 8,
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.2.resnets.1.time_emb_proj": 8,
      "up_blocks.2.resnets.2.time_emb_proj": 8,
      "down_blocks.0.resnets.0.time_emb_proj": 16,
      "up_blocks.1.resnets.0.conv_shortcut": 8,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k": 6,
      "down_blocks.0.resnets.1.conv2": 16,
      "down_blocks.0.resnets.0.conv1": 16,
      "up_blocks.1.resnets.1.conv_shortcut": 16,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.resnets.1.conv1": 16,
      "down_blocks.1.resnets.0.conv_shortcut": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.0.resnets.0.conv2": 16,
      "down_blocks.0.downsamplers.0.conv": 8,
      "up_blocks.2.resnets.0.conv1": 16,
      "up_blocks.2.resnets.0.conv2": 16,
      "up_blocks.1.resnets.2.conv_shortcut": 16,
      "up_blocks.2.resnets.1.conv2": 16,
      "time_embedding.linear_2": 16,
      "up_blocks.2.resnets.1.conv1": 16,
      "up_blocks.1.upsamplers.0.conv": 16,
      "time_embedding.linear_1": 8,
      "up_blocks.2.resnets.0.conv_shortcut": 16,
      "up_blocks.2.resnets.2.conv1": 16,
      "up_blocks.2.resnets.1.conv_shortcut": 16,
      "up_blocks.2.resnets.2.conv2": 16,
      "up_blocks.2.resnets.2.conv_shortcut": 8
    },
    "recipe_3.41_bit_mixedpalette": {
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v": 1,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k": 1,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q": 1,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v": 2,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v": 2,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0": 2,
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2": 2,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2": 2,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2": 2,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2": 2,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v": 2,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v": 2,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0": 4,
      "down_blocks.2.resnets.1.time_emb_proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.7.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.8.ff.net.2": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0": 4,
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj": 2,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0": 2,
      "mid_block.resnets.1.time_emb_proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2": 2,
      "mid_block.attentions.0.transformer_blocks.6.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj": 2,
      "mid_block.attentions.0.transformer_blocks.5.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj": 2,
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.2.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2": 2,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.4.ff.net.2": 4,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj": 2,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v": 2,
      "up_blocks.0.resnets.0.time_emb_proj": 4,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2": 4,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.9.ff.net.2": 2,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj": 4,
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2": 4,
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj": 2,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q": 4,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0": 4,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj": 4,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2": 2,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj": 4,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.3.ff.net.2": 4,
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj": 4,
      "mid_block.resnets.0.time_emb_proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2": 4,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q": 4,
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj": 4,
      "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2": 4,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k": 4,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj": 4,
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj": 4,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2": 4,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.1.ff.net.2": 4,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_v": 4,
      "mid_block.resnets.1.conv1": 4,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2": 4,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2": 4,
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj": 4,
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v": 4,
      "down_blocks.2.attentions.1.proj_in": 4,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_k": 4,
      "mid_block.resnets.1.conv2": 4,
      "down_blocks.2.resnets.1.conv1": 4,
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v": 4,
      "up_blocks.0.resnets.0.conv1": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v": 4,
      "down_blocks.2.resnets.1.conv2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2": 4,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2": 4,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj": 4,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 4,
      "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2": 4,
      "up_blocks.0.resnets.1.time_emb_proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2": 4,
      "mid_block.resnets.0.conv1": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2": 4,
      "down_blocks.2.resnets.0.time_emb_proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj": 4,
      "mid_block.resnets.0.conv2": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.0.resnets.1.conv2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v": 4,
      "down_blocks.2.attentions.1.proj_out": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0": 4,
      "up_blocks.0.resnets.2.time_emb_proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0": 6,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v": 4,
      "mid_block.attentions.0.proj_out": 4,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2": 6,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k": 2,
      "up_blocks.0.resnets.1.conv1": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_v": 4,
      "up_blocks.0.resnets.0.conv2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2": 4,
      "up_blocks.0.resnets.2.conv2": 4,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2": 4,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj": 4,
      "up_blocks.0.attentions.0.proj_out": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2": 4,
      "up_blocks.0.resnets.2.conv1": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0": 6,
      "down_blocks.2.resnets.0.conv2": 4,
      "up_blocks.0.resnets.0.conv_shortcut": 4,
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj": 6,
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj": 6,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj": 6,
      "down_blocks.2.attentions.0.proj_in": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.0.attentions.0.proj_in": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": 4,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj": 6,
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": 4,
      "down_blocks.2.attentions.0.proj_out": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2": 4,
      "mid_block.attentions.0.proj_in": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k": 4,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": 4,
      "up_blocks.0.attentions.2.proj_in": 6,
      "down_blocks.2.resnets.0.conv1": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": 6,
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj": 6,
      "up_blocks.1.resnets.0.time_emb_proj": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": 6,
      "up_blocks.0.attentions.2.proj_out": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q": 4,
      "add_embedding.linear_2": 8,
      "down_blocks.1.resnets.1.time_emb_proj": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.0.resnets.1.conv_shortcut": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.0.attentions.1.proj_in": 6,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k": 2,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "add_embedding.linear_1": 8,
      "up_blocks.1.resnets.0.conv1": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.1.resnets.0.conv2": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.0.attentions.1.proj_out": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 6,
      "down_blocks.1.resnets.0.time_emb_proj": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "down_blocks.1.resnets.1.conv2": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.1.attentions.0.proj_out": 6,
      "down_blocks.1.attentions.1.proj_out": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": 6,
      "down_blocks.1.resnets.1.conv1": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": 8,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": 6,
      "down_blocks.2.resnets.0.conv_shortcut": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.1.resnets.0.conv2": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": 6,
      "down_blocks.1.attentions.1.proj_in": 6,
      "up_blocks.1.resnets.1.time_emb_proj": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.0.resnets.2.conv_shortcut": 8,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": 8,
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.1.attentions.0.proj_in": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.1.downsamplers.0.conv": 6,
      "up_blocks.1.resnets.2.time_emb_proj": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2": 8,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.1.attentions.1.proj_out": 8,
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 8,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0": 8,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": 4,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 8,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v": 8,
      "up_blocks.0.upsamplers.0.conv": 8,
      "down_blocks.1.attentions.0.proj_out": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": 8,
      "up_blocks.1.resnets.1.conv2": 8,
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": 8,
      "up_blocks.1.resnets.2.conv2": 8,
      "up_blocks.1.resnets.1.conv1": 8,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": 6,
      "up_blocks.1.attentions.2.proj_out": 8,
      "up_blocks.1.attentions.1.proj_in": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.resnets.0.conv1": 6,
      "up_blocks.1.attentions.2.proj_in": 8,
      "up_blocks.1.resnets.2.conv1": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": 8,
      "down_blocks.0.resnets.1.time_emb_proj": 8,
      "up_blocks.2.resnets.0.time_emb_proj": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.1.attentions.0.proj_in": 8,
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.2.resnets.1.time_emb_proj": 8,
      "up_blocks.2.resnets.2.time_emb_proj": 6,
      "down_blocks.0.resnets.0.time_emb_proj": 8,
      "up_blocks.1.resnets.0.conv_shortcut": 8,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k": 6,
      "down_blocks.0.resnets.1.conv2": 8,
      "down_blocks.0.resnets.0.conv1": 8,
      "up_blocks.1.resnets.1.conv_shortcut": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 6,
      "down_blocks.0.resnets.1.conv1": 16,
      "down_blocks.1.resnets.0.conv_shortcut": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.0.resnets.0.conv2": 8,
      "down_blocks.0.downsamplers.0.conv": 8,
      "up_blocks.2.resnets.0.conv1": 8,
      "up_blocks.2.resnets.0.conv2": 8,
      "up_blocks.1.resnets.2.conv_shortcut": 8,
      "up_blocks.2.resnets.1.conv2": 8,
      "time_embedding.linear_2": 8,
      "up_blocks.2.resnets.1.conv1": 8,
      "up_blocks.1.upsamplers.0.conv": 8,
      "time_embedding.linear_1": 8,
      "up_blocks.2.resnets.0.conv_shortcut": 8,
      "up_blocks.2.resnets.2.conv1": 16,
      "up_blocks.2.resnets.1.conv_shortcut": 8,
      "up_blocks.2.resnets.2.conv2": 8,
      "up_blocks.2.resnets.2.conv_shortcut": 8
    },
    "recipe_3.05_bit_mixedpalette": {
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v": 1,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k": 1,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q": 1,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q": 1,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q": 1,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0": 1,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k": 1,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q": 1,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0": 1,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v": 1,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v": 2,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0": 2,
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2": 2,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2": 2,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2": 2,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2": 2,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v": 2,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v": 2,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0": 2,
      "down_blocks.2.resnets.1.time_emb_proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.7.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.8.ff.net.2": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj": 2,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0": 2,
      "mid_block.resnets.1.time_emb_proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2": 2,
      "mid_block.attentions.0.transformer_blocks.6.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj": 2,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj": 2,
      "mid_block.attentions.0.transformer_blocks.5.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj": 2,
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.2.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2": 2,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.4.ff.net.2": 2,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj": 2,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v": 2,
      "up_blocks.0.resnets.0.time_emb_proj": 2,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2": 2,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.9.ff.net.2": 2,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": 2,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj": 2,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj": 2,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2": 2,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj": 2,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.3.ff.net.2": 2,
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj": 2,
      "mid_block.resnets.0.time_emb_proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2": 4,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj": 4,
      "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj": 2,
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj": 4,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2": 4,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.1.ff.net.2": 2,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_v": 2,
      "mid_block.resnets.1.conv1": 4,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 2,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2": 4,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2": 4,
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj": 4,
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v": 2,
      "down_blocks.2.attentions.1.proj_in": 4,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_k": 4,
      "mid_block.resnets.1.conv2": 4,
      "down_blocks.2.resnets.1.conv1": 4,
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v": 4,
      "up_blocks.0.resnets.0.conv1": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v": 2,
      "down_blocks.2.resnets.1.conv2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2": 4,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2": 4,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj": 4,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 4,
      "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2": 4,
      "up_blocks.0.resnets.1.time_emb_proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2": 4,
      "mid_block.resnets.0.conv1": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2": 4,
      "down_blocks.2.resnets.0.time_emb_proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj": 4,
      "mid_block.resnets.0.conv2": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.0.resnets.1.conv2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v": 4,
      "down_blocks.2.attentions.1.proj_out": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0": 4,
      "up_blocks.0.resnets.2.time_emb_proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v": 4,
      "mid_block.attentions.0.proj_out": 4,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k": 2,
      "up_blocks.0.resnets.1.conv1": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_v": 4,
      "up_blocks.0.resnets.0.conv2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2": 4,
      "up_blocks.0.resnets.2.conv2": 4,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2": 4,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj": 4,
      "up_blocks.0.attentions.0.proj_out": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2": 4,
      "up_blocks.0.resnets.2.conv1": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0": 4,
      "down_blocks.2.resnets.0.conv2": 4,
      "up_blocks.0.resnets.0.conv_shortcut": 4,
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj": 4,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj": 4,
      "down_blocks.2.attentions.0.proj_in": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.0.attentions.0.proj_in": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": 4,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": 4,
      "down_blocks.2.attentions.0.proj_out": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2": 4,
      "mid_block.attentions.0.proj_in": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k": 4,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0": 4,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": 4,
      "up_blocks.0.attentions.2.proj_in": 6,
      "down_blocks.2.resnets.0.conv1": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj": 4,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj": 6,
      "up_blocks.1.resnets.0.time_emb_proj": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": 6,
      "up_blocks.0.attentions.2.proj_out": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q": 4,
      "add_embedding.linear_2": 6,
      "down_blocks.1.resnets.1.time_emb_proj": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.0.resnets.1.conv_shortcut": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.0.attentions.1.proj_in": 6,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k": 2,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 4,
      "add_embedding.linear_1": 6,
      "up_blocks.1.resnets.0.conv1": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.1.resnets.0.conv2": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.0.attentions.1.proj_out": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "down_blocks.1.resnets.0.time_emb_proj": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "down_blocks.1.resnets.1.conv2": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": 6,
      "up_blocks.1.attentions.0.proj_out": 6,
      "down_blocks.1.attentions.1.proj_out": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": 6,
      "down_blocks.1.resnets.1.conv1": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": 6,
      "down_blocks.2.resnets.0.conv_shortcut": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.1.resnets.0.conv2": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": 6,
      "down_blocks.1.attentions.1.proj_in": 6,
      "up_blocks.1.resnets.1.time_emb_proj": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.0.resnets.2.conv_shortcut": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.1.attentions.0.proj_in": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.1.downsamplers.0.conv": 6,
      "up_blocks.1.resnets.2.time_emb_proj": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.1.attentions.1.proj_out": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0": 8,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": 4,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v": 6,
      "up_blocks.0.upsamplers.0.conv": 6,
      "down_blocks.1.attentions.0.proj_out": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.1.resnets.1.conv2": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": 8,
      "up_blocks.1.resnets.2.conv2": 6,
      "up_blocks.1.resnets.1.conv1": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": 6,
      "up_blocks.1.attentions.2.proj_out": 8,
      "up_blocks.1.attentions.1.proj_in": 8,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.resnets.0.conv1": 6,
      "up_blocks.1.attentions.2.proj_in": 8,
      "up_blocks.1.resnets.2.conv1": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": 8,
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": 8,
      "down_blocks.0.resnets.1.time_emb_proj": 6,
      "up_blocks.2.resnets.0.time_emb_proj": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": 8,
      "down_blocks.1.attentions.0.proj_in": 8,
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 8,
      "up_blocks.2.resnets.1.time_emb_proj": 6,
      "up_blocks.2.resnets.2.time_emb_proj": 6,
      "down_blocks.0.resnets.0.time_emb_proj": 8,
      "up_blocks.1.resnets.0.conv_shortcut": 8,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k": 6,
      "down_blocks.0.resnets.1.conv2": 8,
      "down_blocks.0.resnets.0.conv1": 8,
      "up_blocks.1.resnets.1.conv_shortcut": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 4,
      "down_blocks.0.resnets.1.conv1": 8,
      "down_blocks.1.resnets.0.conv_shortcut": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.0.resnets.0.conv2": 8,
      "down_blocks.0.downsamplers.0.conv": 8,
      "up_blocks.2.resnets.0.conv1": 8,
      "up_blocks.2.resnets.0.conv2": 8,
      "up_blocks.1.resnets.2.conv_shortcut": 8,
      "up_blocks.2.resnets.1.conv2": 8,
      "time_embedding.linear_2": 8,
      "up_blocks.2.resnets.1.conv1": 8,
      "up_blocks.1.upsamplers.0.conv": 8,
      "time_embedding.linear_1": 6,
      "up_blocks.2.resnets.0.conv_shortcut": 8,
      "up_blocks.2.resnets.2.conv1": 8,
      "up_blocks.2.resnets.1.conv_shortcut": 8,
      "up_blocks.2.resnets.2.conv2": 8,
      "up_blocks.2.resnets.2.conv_shortcut": 8
    },
    "recipe_2.81_bit_mixedpalette": {
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v": 1,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k": 1,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q": 1,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q": 1,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q": 1,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0": 1,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k": 1,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q": 1,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0": 1,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v": 1,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v": 1,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0": 1,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0": 1,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k": 1,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_k": 1,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0": 1,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0": 1,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k": 1,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0": 1,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 1,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q": 1,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2": 2,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2": 2,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2": 2,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v": 2,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v": 2,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k": 2,
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0": 2,
      "down_blocks.2.resnets.1.time_emb_proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.7.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.8.ff.net.2": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj": 2,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0": 2,
      "mid_block.resnets.1.time_emb_proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2": 2,
      "mid_block.attentions.0.transformer_blocks.6.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj": 2,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0": 2,
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj": 2,
      "mid_block.attentions.0.transformer_blocks.5.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj": 2,
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.2.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2": 2,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.4.ff.net.2": 2,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj": 2,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v": 2,
      "up_blocks.0.resnets.0.time_emb_proj": 2,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2": 2,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.9.ff.net.2": 2,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": 2,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj": 2,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj": 2,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2": 2,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj": 2,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.3.ff.net.2": 2,
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2": 2,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj": 2,
      "mid_block.resnets.0.time_emb_proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2": 2,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj": 2,
      "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj": 2,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj": 2,
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj": 2,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2": 2,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 2,
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.1.ff.net.2": 2,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_v": 2,
      "mid_block.resnets.1.conv1": 4,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj": 2,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 2,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2": 2,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q": 2,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2": 2,
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj": 4,
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q": 4,
      "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v": 2,
      "down_blocks.2.attentions.1.proj_in": 4,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_k": 4,
      "mid_block.resnets.1.conv2": 2,
      "down_blocks.2.resnets.1.conv1": 4,
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v": 2,
      "up_blocks.0.resnets.0.conv1": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v": 2,
      "down_blocks.2.resnets.1.conv2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2": 4,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2": 4,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_v": 4,
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj": 4,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2": 4,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 4,
      "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj": 2,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 2,
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2": 4,
      "up_blocks.0.resnets.1.time_emb_proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2": 4,
      "mid_block.resnets.0.conv1": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2": 4,
      "down_blocks.2.resnets.0.time_emb_proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v": 4,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj": 4,
      "mid_block.resnets.0.conv2": 4,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.0.resnets.1.conv2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v": 4,
      "down_blocks.2.attentions.1.proj_out": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0": 4,
      "up_blocks.0.resnets.2.time_emb_proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v": 4,
      "mid_block.attentions.0.proj_out": 4,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_k": 4,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k": 2,
      "up_blocks.0.resnets.1.conv1": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_v": 4,
      "up_blocks.0.resnets.0.conv2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2": 4,
      "up_blocks.0.resnets.2.conv2": 4,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2": 4,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj": 4,
      "up_blocks.0.attentions.0.proj_out": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2": 4,
      "up_blocks.0.resnets.2.conv1": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0": 4,
      "down_blocks.2.resnets.0.conv2": 4,
      "up_blocks.0.resnets.0.conv_shortcut": 4,
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj": 4,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj": 4,
      "down_blocks.2.attentions.0.proj_in": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.0.attentions.0.proj_in": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": 2,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": 4,
      "down_blocks.2.attentions.0.proj_out": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2": 4,
      "mid_block.attentions.0.proj_in": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k": 4,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0": 4,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": 4,
      "up_blocks.0.attentions.2.proj_in": 4,
      "down_blocks.2.resnets.0.conv1": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj": 4,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj": 4,
      "up_blocks.1.resnets.0.time_emb_proj": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.proj_out": 4,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q": 4,
      "add_embedding.linear_2": 6,
      "down_blocks.1.resnets.1.time_emb_proj": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.0.resnets.1.conv_shortcut": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.0.attentions.1.proj_in": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k": 2,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 4,
      "add_embedding.linear_1": 6,
      "up_blocks.1.resnets.0.conv1": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.1.resnets.0.conv2": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.0.attentions.1.proj_out": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "down_blocks.1.resnets.0.time_emb_proj": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "down_blocks.1.resnets.1.conv2": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.1.attentions.0.proj_out": 6,
      "down_blocks.1.attentions.1.proj_out": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": 4,
      "down_blocks.1.resnets.1.conv1": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": 6,
      "down_blocks.2.resnets.0.conv_shortcut": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.1.resnets.0.conv2": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q": 6,
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": 4,
      "down_blocks.1.attentions.1.proj_in": 6,
      "up_blocks.1.resnets.1.time_emb_proj": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.0.resnets.2.conv_shortcut": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.1.attentions.0.proj_in": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.1.downsamplers.0.conv": 6,
      "up_blocks.1.resnets.2.time_emb_proj": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.1.attentions.1.proj_out": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": 4,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v": 6,
      "up_blocks.0.upsamplers.0.conv": 6,
      "down_blocks.1.attentions.0.proj_out": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.1.resnets.1.conv2": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": 6,
      "up_blocks.1.resnets.2.conv2": 6,
      "up_blocks.1.resnets.1.conv1": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": 6,
      "up_blocks.1.attentions.2.proj_out": 6,
      "up_blocks.1.attentions.1.proj_in": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.1.resnets.0.conv1": 6,
      "up_blocks.1.attentions.2.proj_in": 6,
      "up_blocks.1.resnets.2.conv1": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": 6,
      "down_blocks.0.resnets.1.time_emb_proj": 6,
      "up_blocks.2.resnets.0.time_emb_proj": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": 6,
      "down_blocks.1.attentions.0.proj_in": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.2.resnets.1.time_emb_proj": 6,
      "up_blocks.2.resnets.2.time_emb_proj": 6,
      "down_blocks.0.resnets.0.time_emb_proj": 6,
      "up_blocks.1.resnets.0.conv_shortcut": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k": 6,
      "down_blocks.0.resnets.1.conv2": 8,
      "down_blocks.0.resnets.0.conv1": 6,
      "up_blocks.1.resnets.1.conv_shortcut": 8,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 4,
      "down_blocks.0.resnets.1.conv1": 8,
      "down_blocks.1.resnets.0.conv_shortcut": 8,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": 6,
      "down_blocks.0.resnets.0.conv2": 8,
      "down_blocks.0.downsamplers.0.conv": 8,
      "up_blocks.2.resnets.0.conv1": 8,
      "up_blocks.2.resnets.0.conv2": 8,
      "up_blocks.1.resnets.2.conv_shortcut": 6,
      "up_blocks.2.resnets.1.conv2": 8,
      "time_embedding.linear_2": 6,
      "up_blocks.2.resnets.1.conv1": 8,
      "up_blocks.1.upsamplers.0.conv": 8,
      "time_embedding.linear_1": 6,
      "up_blocks.2.resnets.0.conv_shortcut": 8,
      "up_blocks.2.resnets.2.conv1": 8,
      "up_blocks.2.resnets.1.conv_shortcut": 8,
      "up_blocks.2.resnets.2.conv2": 8,
      "up_blocks.2.resnets.2.conv_shortcut": 6
    },
    "recipe_2.31_bit_mixedpalette": {
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.3.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.4.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.2.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.1.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v": 1,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k": 1,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q": 1,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q": 1,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q": 1,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0": 1,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k": 1,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q": 1,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0": 1,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v": 1,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v": 1,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0": 1,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0": 1,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k": 1,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_k": 1,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0": 1,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0": 1,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k": 1,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0": 1,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": 1,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q": 1,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.8.attn1.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2": 1,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.6.attn1.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2": 1,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k": 1,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2": 1,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v": 1,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v": 1,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2": 1,
      "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj": 1,
      "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj": 1,
      "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v": 1,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj": 1,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0": 1,
      "down_blocks.2.resnets.1.time_emb_proj": 1,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.7.ff.net.2": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0": 1,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.8.ff.net.2": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0": 1,
      "mid_block.resnets.1.time_emb_proj": 1,
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2": 1,
      "mid_block.attentions.0.transformer_blocks.6.ff.net.2": 1,
      "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj": 1,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.9.attn1.to_v": 1,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q": 1,
      "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj": 1,
      "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v": 1,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0": 1,
      "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj": 1,
      "mid_block.attentions.0.transformer_blocks.5.ff.net.2": 1,
      "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj": 1,
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.2.ff.net.2": 1,
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2": 1,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0": 1,
      "mid_block.attentions.0.transformer_blocks.4.ff.net.2": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_k": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj": 1,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v": 1,
      "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj": 1,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v": 1,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q": 1,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v": 1,
      "up_blocks.0.resnets.0.time_emb_proj": 1,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0": 1,
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2": 1,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0": 1,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k": 1,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k": 1,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q": 1,
      "mid_block.attentions.0.transformer_blocks.9.ff.net.2": 1,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": 1,
      "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v": 1,
      "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0": 1,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k": 1,
      "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj": 1,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": 2,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj": 2,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj": 2,
      "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2": 2,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj": 2,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": 2,
      "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.3.ff.net.2": 2,
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2": 2,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj": 2,
      "mid_block.resnets.0.time_emb_proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2": 2,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj": 2,
      "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2": 2,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k": 2,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj": 2,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj": 2,
      "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj": 2,
      "mid_block.attentions.0.transformer_blocks.2.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2": 2,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v": 2,
      "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": 2,
      "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.1.ff.net.2": 2,
      "mid_block.attentions.0.transformer_blocks.3.attn1.to_v": 2,
      "mid_block.resnets.1.conv1": 2,
      "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0": 2,
      "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj": 2,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": 2,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2": 2,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.2": 2,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q": 2,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q": 2,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2": 2,
      "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k": 2,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj": 2,
      "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q": 2,
      "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj": 2,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": 2,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v": 2,
      "down_blocks.2.attentions.1.proj_in": 2,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_k": 2,
      "mid_block.resnets.1.conv2": 2,
      "down_blocks.2.resnets.1.conv1": 2,
      "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v": 2,
      "up_blocks.0.resnets.0.conv1": 2,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v": 2,
      "down_blocks.2.resnets.1.conv2": 2,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0": 2,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 2,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2": 2,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_v": 2,
      "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj": 2,
      "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2": 2,
      "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": 2,
      "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj": 2,
      "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": 2,
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2": 2,
      "up_blocks.0.resnets.1.time_emb_proj": 4,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": 2,
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2": 2,
      "mid_block.resnets.0.conv1": 2,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0": 2,
      "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0": 2,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v": 2,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2": 2,
      "down_blocks.2.resnets.0.time_emb_proj": 2,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": 2,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v": 2,
      "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": 2,
      "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj": 2,
      "mid_block.resnets.0.conv2": 2,
      "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": 2,
      "up_blocks.0.resnets.1.conv2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2": 2,
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v": 2,
      "down_blocks.2.attentions.1.proj_out": 2,
      "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0": 2,
      "up_blocks.0.resnets.2.time_emb_proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v": 4,
      "mid_block.attentions.0.proj_out": 4,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_q": 2,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v": 2,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": 2,
      "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_k": 2,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": 2,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k": 2,
      "up_blocks.0.resnets.1.conv1": 2,
      "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v": 4,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v": 2,
      "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": 4,
      "mid_block.attentions.0.transformer_blocks.1.attn1.to_v": 4,
      "up_blocks.0.resnets.0.conv2": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2": 4,
      "up_blocks.0.resnets.2.conv2": 4,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0": 4,
      "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2": 4,
      "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2": 4,
      "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj": 4,
      "up_blocks.0.attentions.0.proj_out": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 2,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2": 4,
      "up_blocks.0.resnets.2.conv1": 4,
      "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0": 4,
      "down_blocks.2.resnets.0.conv2": 4,
      "up_blocks.0.resnets.0.conv_shortcut": 4,
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": 4,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": 4,
      "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj": 4,
      "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k": 2,
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": 2,
      "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj": 4,
      "down_blocks.2.attentions.0.proj_in": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.0.attentions.0.proj_in": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": 2,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj": 4,
      "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": 4,
      "down_blocks.2.attentions.0.proj_out": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2": 4,
      "mid_block.attentions.0.proj_in": 4,
      "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k": 4,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0": 4,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": 4,
      "up_blocks.0.attentions.2.proj_in": 4,
      "down_blocks.2.resnets.0.conv1": 4,
      "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj": 4,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": 6,
      "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": 4,
      "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": 4,
      "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj": 4,
      "up_blocks.1.resnets.0.time_emb_proj": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": 4,
      "up_blocks.0.attentions.2.proj_out": 4,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q": 4,
      "add_embedding.linear_2": 6,
      "down_blocks.1.resnets.1.time_emb_proj": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.0.resnets.1.conv_shortcut": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.0.attentions.1.proj_in": 4,
      "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k": 2,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 4,
      "add_embedding.linear_1": 4,
      "up_blocks.1.resnets.0.conv1": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": 4,
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": 4,
      "up_blocks.1.resnets.0.conv2": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": 4,
      "up_blocks.0.attentions.1.proj_out": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": 4,
      "down_blocks.1.resnets.0.time_emb_proj": 4,
      "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 4,
      "down_blocks.1.resnets.1.conv2": 4,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": 4,
      "up_blocks.1.attentions.0.proj_out": 6,
      "down_blocks.1.attentions.1.proj_out": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": 4,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": 4,
      "down_blocks.1.resnets.1.conv1": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": 6,
      "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": 4,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": 6,
      "down_blocks.2.resnets.0.conv_shortcut": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": 6,
      "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.1.resnets.0.conv2": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q": 4,
      "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": 4,
      "down_blocks.1.attentions.1.proj_in": 6,
      "up_blocks.1.resnets.1.time_emb_proj": 4,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": 6,
      "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.0.resnets.2.conv_shortcut": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.1.attentions.0.proj_in": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.1.downsamplers.0.conv": 6,
      "up_blocks.1.resnets.2.time_emb_proj": 4,
      "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.1.attentions.1.proj_out": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": 4,
      "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": 4,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v": 6,
      "up_blocks.0.upsamplers.0.conv": 6,
      "down_blocks.1.attentions.0.proj_out": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": 6,
      "up_blocks.1.resnets.1.conv2": 6,
      "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": 6,
      "up_blocks.1.resnets.2.conv2": 6,
      "up_blocks.1.resnets.1.conv1": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": 6,
      "up_blocks.1.attentions.2.proj_out": 6,
      "up_blocks.1.attentions.1.proj_in": 6,
      "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.1.resnets.0.conv1": 6,
      "up_blocks.1.attentions.2.proj_in": 6,
      "up_blocks.1.resnets.2.conv1": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": 6,
      "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": 6,
      "down_blocks.0.resnets.1.time_emb_proj": 6,
      "up_blocks.2.resnets.0.time_emb_proj": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": 6,
      "down_blocks.1.attentions.0.proj_in": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": 6,
      "up_blocks.2.resnets.1.time_emb_proj": 6,
      "up_blocks.2.resnets.2.time_emb_proj": 6,
      "down_blocks.0.resnets.0.time_emb_proj": 6,
      "up_blocks.1.resnets.0.conv_shortcut": 6,
      "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k": 6,
      "down_blocks.0.resnets.1.conv2": 6,
      "down_blocks.0.resnets.0.conv1": 6,
      "up_blocks.1.resnets.1.conv_shortcut": 6,
      "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": 4,
      "down_blocks.0.resnets.1.conv1": 8,
      "down_blocks.1.resnets.0.conv_shortcut": 6,
      "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": 4,
      "down_blocks.0.resnets.0.conv2": 8,
      "down_blocks.0.downsamplers.0.conv": 6,
      "up_blocks.2.resnets.0.conv1": 6,
      "up_blocks.2.resnets.0.conv2": 6,
      "up_blocks.1.resnets.2.conv_shortcut": 6,
      "up_blocks.2.resnets.1.conv2": 8,
      "time_embedding.linear_2": 6,
      "up_blocks.2.resnets.1.conv1": 8,
      "up_blocks.1.upsamplers.0.conv": 6,
      "time_embedding.linear_1": 6,
      "up_blocks.2.resnets.0.conv_shortcut": 8,
      "up_blocks.2.resnets.2.conv1": 8,
      "up_blocks.2.resnets.1.conv_shortcut": 8,
      "up_blocks.2.resnets.2.conv2": 8,
      "up_blocks.2.resnets.2.conv_shortcut": 6
    }
  }
}