fp8_hooks_maxabs.json

{
    "GlobalRank": null,
    "LocalRank": null,
    "Mode": "DynamicRange",
    "Nodes": {
        "conv_in": {
            "inputs": [
                [
                    [
                        4.78125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.4765625
                    ]
                ]
            }
        },
        "time_embedding.linear_1": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.171875
                    ]
                ]
            }
        },
        "time_embedding.linear_2": {
            "inputs": [
                [
                    [
                        3.671875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1318359375
                    ]
                ]
            }
        },
        "add_embedding.linear_1": {
            "inputs": [
                [
                    [
                        7.40625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        1.390625
                    ]
                ]
            }
        },
        "add_embedding.linear_2": {
            "inputs": [
                [
                    [
                        4.78125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.484375
                    ]
                ]
            }
        },
        "down_blocks.0.resnets.0.conv1": {
            "inputs": [
                [
                    [
                        9.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.57421875
                    ]
                ]
            }
        },
        "down_blocks.0.resnets.0.time_emb_proj": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.57421875
                    ]
                ]
            }
        },
        "down_blocks.0.resnets.0.conv2": {
            "inputs": [
                [
                    [
                        7.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.486328125
                    ]
                ]
            }
        },
        "down_blocks.0.resnets.1.conv1": {
            "inputs": [
                [
                    [
                        9.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.65234375
                    ]
                ]
            }
        },
        "down_blocks.0.resnets.1.time_emb_proj": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.294921875
                    ]
                ]
            }
        },
        "down_blocks.0.resnets.1.conv2": {
            "inputs": [
                [
                    [
                        8.4375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.41796875
                    ]
                ]
            }
        },
        "down_blocks.0.downsamplers.0.conv": {
            "inputs": [
                [
                    [
                        6.9375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.234375
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.proj_in": {
            "inputs": [
                [
                    [
                        7.9375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.177734375
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": {
            "inputs": [
                [
                    [
                        9.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2265625
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": {
            "inputs": [
                [
                    [
                        9.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.19921875
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": {
            "inputs": [
                [
                    [
                        9.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1484375
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.59375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2314453125
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.0.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        9.1875
                    ]
                ],
                [
                    [
                        9.5625
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.0.transformer_blocks.0.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        5.75
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.0.transformer_blocks.0.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        314.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": {
            "inputs": [
                [
                    [
                        11.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1611328125
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.287109375
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1279296875
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        4.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.10302734375
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.0.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        107.5
                    ]
                ],
                [
                    [
                        5.6875
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.0.transformer_blocks.0.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        20.125
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.0.transformer_blocks.0.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        856.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        7.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2255859375
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": {
            "inputs": [
                [
                    [
                        32.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.3125
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": {
            "inputs": [
                [
                    [
                        10.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1865234375
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": {
            "inputs": [
                [
                    [
                        10.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1826171875
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": {
            "inputs": [
                [
                    [
                        10.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1826171875
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.3359375
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.1.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.8125
                    ]
                ],
                [
                    [
                        9.5
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.0.transformer_blocks.1.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.98828125
                    ]
                ],
                [
                    [
                        6.21875
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.0.transformer_blocks.1.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        376.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.98828125
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": {
            "inputs": [
                [
                    [
                        14.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1494140625
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.33203125
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.154296875
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        13.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2412109375
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.1.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        97.0
                    ]
                ],
                [
                    [
                        4.65625
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.0.transformer_blocks.1.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        17.375
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.0.transformer_blocks.1.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        2576.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        9.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.240234375
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": {
            "inputs": [
                [
                    [
                        70.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.291015625
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.0.proj_out": {
            "inputs": [
                [
                    [
                        20.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1083984375
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.proj_in": {
            "inputs": [
                [
                    [
                        11.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1748046875
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": {
            "inputs": [
                [
                    [
                        7.40625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.193359375
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": {
            "inputs": [
                [
                    [
                        7.40625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1923828125
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": {
            "inputs": [
                [
                    [
                        7.40625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1494140625
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.15625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.248046875
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.0.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.96875
                    ]
                ],
                [
                    [
                        7.90625
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.1.transformer_blocks.0.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        6.40625
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.1.transformer_blocks.0.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        199.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": {
            "inputs": [
                [
                    [
                        12.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1337890625
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1533203125
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0986328125
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        4.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1884765625
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.0.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        135.0
                    ]
                ],
                [
                    [
                        4.125
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.1.transformer_blocks.0.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.953125
                    ]
                ],
                [
                    [
                        15.125
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.1.transformer_blocks.0.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1864.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.953125
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        6.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1953125
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": {
            "inputs": [
                [
                    [
                        19.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2578125
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": {
            "inputs": [
                [
                    [
                        9.1875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.193359375
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": {
            "inputs": [
                [
                    [
                        9.1875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.201171875
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": {
            "inputs": [
                [
                    [
                        9.1875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1591796875
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.71875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.35546875
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.1.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        8.875
                    ]
                ],
                [
                    [
                        8.75
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.1.transformer_blocks.1.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        6.03125
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.1.transformer_blocks.1.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        173.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": {
            "inputs": [
                [
                    [
                        13.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1201171875
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.15234375
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.09716796875
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        8.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1943359375
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.1.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        152.0
                    ]
                ],
                [
                    [
                        4.46875
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.1.transformer_blocks.1.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        19.625
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.1.transformer_blocks.1.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1792.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        5.84375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.201171875
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": {
            "inputs": [
                [
                    [
                        33.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.22265625
                    ]
                ]
            }
        },
        "down_blocks.1.attentions.1.proj_out": {
            "inputs": [
                [
                    [
                        22.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.14453125
                    ]
                ]
            }
        },
        "down_blocks.1.resnets.0.conv1": {
            "inputs": [
                [
                    [
                        15.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.8203125
                    ]
                ]
            }
        },
        "down_blocks.1.resnets.0.time_emb_proj": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.625
                    ]
                ]
            }
        },
        "down_blocks.1.resnets.0.conv2": {
            "inputs": [
                [
                    [
                        7.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.271484375
                    ]
                ]
            }
        },
        "down_blocks.1.resnets.0.conv_shortcut": {
            "inputs": [
                [
                    [
                        19.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.251953125
                    ]
                ]
            }
        },
        "down_blocks.1.resnets.1.conv1": {
            "inputs": [
                [
                    [
                        7.46875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        1.1953125
                    ]
                ]
            }
        },
        "down_blocks.1.resnets.1.time_emb_proj": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.326171875
                    ]
                ]
            }
        },
        "down_blocks.1.resnets.1.conv2": {
            "inputs": [
                [
                    [
                        7.84375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.322265625
                    ]
                ]
            }
        },
        "down_blocks.1.downsamplers.0.conv": {
            "inputs": [
                [
                    [
                        27.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.25390625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.proj_in": {
            "inputs": [
                [
                    [
                        9.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.3359375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": {
            "inputs": [
                [
                    [
                        3.359375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1455078125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": {
            "inputs": [
                [
                    [
                        3.359375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.146484375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": {
            "inputs": [
                [
                    [
                        3.359375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.14453125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.1875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.15625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.0.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        8.1875
                    ]
                ],
                [
                    [
                        8.1875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.0.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        6.8125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.0.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        282.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": {
            "inputs": [
                [
                    [
                        9.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.09521484375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2333984375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.119140625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        12.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.08203125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.0.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        104.0
                    ]
                ],
                [
                    [
                        7.71875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.0.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        21.125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.0.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1904.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        3.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2158203125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": {
            "inputs": [
                [
                    [
                        23.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.171875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_q": {
            "inputs": [
                [
                    [
                        4.96875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.134765625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_k": {
            "inputs": [
                [
                    [
                        4.96875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1298828125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_v": {
            "inputs": [
                [
                    [
                        4.96875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1279296875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.1.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1689453125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.1.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        5.96875
                    ]
                ],
                [
                    [
                        6.28125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.1.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.98828125
                    ]
                ],
                [
                    [
                        8.6875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.1.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        155.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.98828125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_q": {
            "inputs": [
                [
                    [
                        11.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.10400390625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2353515625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1298828125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.1.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        5.96875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.05322265625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.1.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        95.0
                    ]
                ],
                [
                    [
                        5.40625
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.1.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        21.75
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.1.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1192.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        2.84375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.15625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.1.ff.net.2": {
            "inputs": [
                [
                    [
                        18.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1748046875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_q": {
            "inputs": [
                [
                    [
                        7.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.158203125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_k": {
            "inputs": [
                [
                    [
                        7.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1357421875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_v": {
            "inputs": [
                [
                    [
                        7.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.142578125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.2.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        5.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1533203125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.2.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.375
                    ]
                ],
                [
                    [
                        7.65625
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.2.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        6.625
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.2.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        194.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_q": {
            "inputs": [
                [
                    [
                        14.1875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.09912109375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1630859375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12353515625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.2.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        4.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0439453125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.2.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        126.0
                    ]
                ],
                [
                    [
                        4.78125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.2.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        20.5
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.2.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        980.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        3.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2119140625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.2.ff.net.2": {
            "inputs": [
                [
                    [
                        22.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.158203125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_q": {
            "inputs": [
                [
                    [
                        8.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1435546875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_k": {
            "inputs": [
                [
                    [
                        8.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.14453125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_v": {
            "inputs": [
                [
                    [
                        8.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.15625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.3.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.53125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1513671875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.3.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.5625
                    ]
                ],
                [
                    [
                        7.71875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.3.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        7.96875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.3.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        188.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_q": {
            "inputs": [
                [
                    [
                        14.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.10888671875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.169921875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.11767578125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.3.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        4.53125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.049072265625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.3.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        122.0
                    ]
                ],
                [
                    [
                        6.28125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.3.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        25.875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.3.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        2528.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        3.671875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.193359375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.3.ff.net.2": {
            "inputs": [
                [
                    [
                        26.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.13671875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_q": {
            "inputs": [
                [
                    [
                        6.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.123046875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_k": {
            "inputs": [
                [
                    [
                        6.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1279296875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_v": {
            "inputs": [
                [
                    [
                        6.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.119140625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.4.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1337890625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.4.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.0
                    ]
                ],
                [
                    [
                        7.15625
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.4.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        6.96875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.4.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        185.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_q": {
            "inputs": [
                [
                    [
                        16.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0810546875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1357421875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1044921875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.4.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        3.796875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.04638671875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.4.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        110.0
                    ]
                ],
                [
                    [
                        4.5625
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.4.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        23.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.4.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1448.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        3.53125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.15625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.4.ff.net.2": {
            "inputs": [
                [
                    [
                        26.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1455078125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_q": {
            "inputs": [
                [
                    [
                        7.28125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.130859375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_k": {
            "inputs": [
                [
                    [
                        7.28125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.126953125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_v": {
            "inputs": [
                [
                    [
                        7.28125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1259765625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.5.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12451171875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.5.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.65625
                    ]
                ],
                [
                    [
                        7.3125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.5.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        5.6875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.5.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        172.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_q": {
            "inputs": [
                [
                    [
                        17.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.08349609375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.09716796875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.09228515625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.5.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        3.34375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.040771484375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.5.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        65.0
                    ]
                ],
                [
                    [
                        4.09375
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.5.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        26.25
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.5.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1104.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        3.890625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1533203125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.5.ff.net.2": {
            "inputs": [
                [
                    [
                        22.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1494140625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_q": {
            "inputs": [
                [
                    [
                        8.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1240234375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_k": {
            "inputs": [
                [
                    [
                        8.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_v": {
            "inputs": [
                [
                    [
                        8.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1220703125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.6.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1357421875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.6.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.21875
                    ]
                ],
                [
                    [
                        6.5625
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.6.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.98828125
                    ]
                ],
                [
                    [
                        5.3125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.6.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        149.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.98828125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_q": {
            "inputs": [
                [
                    [
                        20.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.080078125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.06884765625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.049560546875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.6.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        1.21875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0260009765625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.6.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        43.25
                    ]
                ],
                [
                    [
                        3.640625
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.6.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.98046875
                    ]
                ],
                [
                    [
                        20.875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.6.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        940.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.98046875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        3.59375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.18359375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.6.ff.net.2": {
            "inputs": [
                [
                    [
                        22.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.14453125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_q": {
            "inputs": [
                [
                    [
                        8.9375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_k": {
            "inputs": [
                [
                    [
                        8.9375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.11962890625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_v": {
            "inputs": [
                [
                    [
                        8.9375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.11767578125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.7.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        3.71875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.11328125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.7.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        5.9375
                    ]
                ],
                [
                    [
                        5.75
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.7.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.984375
                    ]
                ],
                [
                    [
                        5.03125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.7.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        145.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.984375
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_q": {
            "inputs": [
                [
                    [
                        18.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0751953125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.08740234375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0693359375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.7.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        3.296875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.039794921875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.7.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        42.75
                    ]
                ],
                [
                    [
                        3.734375
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.7.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.953125
                    ]
                ],
                [
                    [
                        24.125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.7.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        988.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.953125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        3.734375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.146484375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.7.ff.net.2": {
            "inputs": [
                [
                    [
                        28.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.154296875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_q": {
            "inputs": [
                [
                    [
                        8.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1171875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_k": {
            "inputs": [
                [
                    [
                        8.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.11865234375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_v": {
            "inputs": [
                [
                    [
                        8.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.11083984375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.8.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.78125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.169921875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.8.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        5.8125
                    ]
                ],
                [
                    [
                        6.0625
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.8.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.984375
                    ]
                ],
                [
                    [
                        5.71875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.8.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        139.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.984375
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_q": {
            "inputs": [
                [
                    [
                        19.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.068359375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.083984375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.059326171875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.8.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        1.515625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.039794921875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.8.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        49.75
                    ]
                ],
                [
                    [
                        3.921875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.8.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.984375
                    ]
                ],
                [
                    [
                        24.25
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.8.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1368.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.984375
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        3.90625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1455078125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.8.ff.net.2": {
            "inputs": [
                [
                    [
                        24.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1796875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_q": {
            "inputs": [
                [
                    [
                        8.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.130859375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_k": {
            "inputs": [
                [
                    [
                        8.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.126953125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_v": {
            "inputs": [
                [
                    [
                        8.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1240234375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.9.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.65625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1435546875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.9.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.0625
                    ]
                ],
                [
                    [
                        6.53125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.9.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.984375
                    ]
                ],
                [
                    [
                        5.5625
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.9.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        129.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.984375
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_q": {
            "inputs": [
                [
                    [
                        16.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.07275390625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.087890625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.08984375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.9.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        6.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.134765625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.9.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        63.5
                    ]
                ],
                [
                    [
                        3.71875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.9.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        32.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.9.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1312.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        5.15625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1484375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.transformer_blocks.9.ff.net.2": {
            "inputs": [
                [
                    [
                        30.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.17578125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.0.proj_out": {
            "inputs": [
                [
                    [
                        44.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.11962890625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.proj_in": {
            "inputs": [
                [
                    [
                        8.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.490234375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": {
            "inputs": [
                [
                    [
                        3.59375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.140625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": {
            "inputs": [
                [
                    [
                        3.59375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.142578125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": {
            "inputs": [
                [
                    [
                        3.59375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12255859375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2294921875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.0.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.90625
                    ]
                ],
                [
                    [
                        8.4375
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.0.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        5.4375
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.0.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        230.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": {
            "inputs": [
                [
                    [
                        12.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.10498046875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2060546875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2021484375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        5.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1494140625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.0.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        131.0
                    ]
                ],
                [
                    [
                        6.9375
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.0.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        23.375
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.0.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1608.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        5.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.177734375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": {
            "inputs": [
                [
                    [
                        34.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.251953125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_q": {
            "inputs": [
                [
                    [
                        5.09375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.138671875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_k": {
            "inputs": [
                [
                    [
                        5.09375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1494140625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_v": {
            "inputs": [
                [
                    [
                        5.09375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12890625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.1.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1787109375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.1.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.3125
                    ]
                ],
                [
                    [
                        6.4375
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.1.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.984375
                    ]
                ],
                [
                    [
                        6.15625
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.1.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        166.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.984375
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_q": {
            "inputs": [
                [
                    [
                        16.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.15234375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2392578125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1474609375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.1.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        6.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.07568359375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.1.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        117.0
                    ]
                ],
                [
                    [
                        7.71875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.1.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        19.75
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.1.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1528.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        3.453125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1787109375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.1.ff.net.2": {
            "inputs": [
                [
                    [
                        30.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.16015625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_q": {
            "inputs": [
                [
                    [
                        6.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.146484375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_k": {
            "inputs": [
                [
                    [
                        6.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.146484375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_v": {
            "inputs": [
                [
                    [
                        6.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1435546875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.2.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.96875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1728515625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.2.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.5625
                    ]
                ],
                [
                    [
                        7.21875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.2.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.98828125
                    ]
                ],
                [
                    [
                        6.8125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.2.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        157.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.98828125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_q": {
            "inputs": [
                [
                    [
                        17.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.134765625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.224609375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1494140625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.2.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        17.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.07568359375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.2.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        134.0
                    ]
                ],
                [
                    [
                        9.5
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.2.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        19.875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.2.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1400.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        3.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.162109375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.2.ff.net.2": {
            "inputs": [
                [
                    [
                        46.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.142578125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_q": {
            "inputs": [
                [
                    [
                        6.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1298828125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_k": {
            "inputs": [
                [
                    [
                        6.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1298828125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_v": {
            "inputs": [
                [
                    [
                        6.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1337890625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.3.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.1875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1435546875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.3.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.40625
                    ]
                ],
                [
                    [
                        6.53125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.3.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.98046875
                    ]
                ],
                [
                    [
                        5.875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.3.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        157.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.98046875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_q": {
            "inputs": [
                [
                    [
                        17.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1328125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1806640625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.142578125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.3.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        13.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.07373046875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.3.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        142.0
                    ]
                ],
                [
                    [
                        8.0625
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.3.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        20.125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.3.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1624.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.03125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1494140625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.3.ff.net.2": {
            "inputs": [
                [
                    [
                        54.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1640625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_q": {
            "inputs": [
                [
                    [
                        6.65625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1298828125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_k": {
            "inputs": [
                [
                    [
                        6.65625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1318359375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_v": {
            "inputs": [
                [
                    [
                        6.65625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.4.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.34375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1513671875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.4.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.15625
                    ]
                ],
                [
                    [
                        5.71875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.4.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.96875
                    ]
                ],
                [
                    [
                        4.84375
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.4.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        152.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.96875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_q": {
            "inputs": [
                [
                    [
                        16.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1201171875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.212890625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.142578125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.4.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        18.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.064453125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.4.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        139.0
                    ]
                ],
                [
                    [
                        6.40625
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.4.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        24.875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.4.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        2304.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.1875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1923828125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.4.ff.net.2": {
            "inputs": [
                [
                    [
                        87.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.166015625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_q": {
            "inputs": [
                [
                    [
                        8.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1298828125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_k": {
            "inputs": [
                [
                    [
                        8.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1201171875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_v": {
            "inputs": [
                [
                    [
                        8.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12451171875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.5.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.09375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1611328125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.5.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        5.78125
                    ]
                ],
                [
                    [
                        6.03125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.5.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        5.5625
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.5.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        176.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_q": {
            "inputs": [
                [
                    [
                        14.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.126953125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.21484375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.13671875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.5.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        12.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.06396484375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.5.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        168.0
                    ]
                ],
                [
                    [
                        6.71875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.5.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        26.875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.5.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1872.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.84375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1640625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.5.ff.net.2": {
            "inputs": [
                [
                    [
                        92.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.158203125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_q": {
            "inputs": [
                [
                    [
                        8.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.138671875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_k": {
            "inputs": [
                [
                    [
                        8.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.123046875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_v": {
            "inputs": [
                [
                    [
                        8.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1201171875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.6.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1494140625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.6.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.3125
                    ]
                ],
                [
                    [
                        6.3125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.6.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        5.3125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.6.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        166.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_q": {
            "inputs": [
                [
                    [
                        12.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.09033203125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.169921875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.11328125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.6.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        28.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0576171875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.6.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        156.0
                    ]
                ],
                [
                    [
                        4.625
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.6.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        30.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.6.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        2096.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        5.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1572265625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.6.ff.net.2": {
            "inputs": [
                [
                    [
                        111.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1484375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_q": {
            "inputs": [
                [
                    [
                        7.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.130859375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_k": {
            "inputs": [
                [
                    [
                        7.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.138671875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_v": {
            "inputs": [
                [
                    [
                        7.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12890625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.7.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        5.1875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12255859375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.7.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        5.875
                    ]
                ],
                [
                    [
                        6.65625
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.7.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9609375
                    ]
                ],
                [
                    [
                        6.6875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.7.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        139.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9609375
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_q": {
            "inputs": [
                [
                    [
                        9.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.09375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1396484375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.123046875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.7.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        20.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.06298828125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.7.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        142.0
                    ]
                ],
                [
                    [
                        5.25
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.7.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        34.25
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.7.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        2368.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        5.84375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1591796875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.7.ff.net.2": {
            "inputs": [
                [
                    [
                        107.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.15234375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_q": {
            "inputs": [
                [
                    [
                        7.46875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.13671875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_k": {
            "inputs": [
                [
                    [
                        7.46875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1328125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_v": {
            "inputs": [
                [
                    [
                        7.46875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1416015625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.8.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        7.09375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12890625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.8.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.34375
                    ]
                ],
                [
                    [
                        7.875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.8.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.953125
                    ]
                ],
                [
                    [
                        7.40625
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.8.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        164.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.953125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_q": {
            "inputs": [
                [
                    [
                        8.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0986328125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1630859375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.126953125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.8.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        17.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0595703125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.8.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        151.0
                    ]
                ],
                [
                    [
                        5.625
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.8.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        30.125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.8.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        4160.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        6.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.16796875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.8.ff.net.2": {
            "inputs": [
                [
                    [
                        98.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1533203125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_q": {
            "inputs": [
                [
                    [
                        7.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.130859375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_k": {
            "inputs": [
                [
                    [
                        7.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1298828125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_v": {
            "inputs": [
                [
                    [
                        7.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1328125
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.9.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.28125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1240234375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.9.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.625
                    ]
                ],
                [
                    [
                        7.46875
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.9.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        6.625
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.9.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        223.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_q": {
            "inputs": [
                [
                    [
                        7.90625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.08349609375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.11962890625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.08984375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.9.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        16.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.055419921875
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.9.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        119.0
                    ]
                ],
                [
                    [
                        5.78125
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.9.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        39.5
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.9.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        3680.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        6.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.177734375
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.transformer_blocks.9.ff.net.2": {
            "inputs": [
                [
                    [
                        55.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.16015625
                    ]
                ]
            }
        },
        "down_blocks.2.attentions.1.proj_out": {
            "inputs": [
                [
                    [
                        28.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.07666015625
                    ]
                ]
            }
        },
        "down_blocks.2.resnets.0.conv1": {
            "inputs": [
                [
                    [
                        8.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.58203125
                    ]
                ]
            }
        },
        "down_blocks.2.resnets.0.time_emb_proj": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.65625
                    ]
                ]
            }
        },
        "down_blocks.2.resnets.0.conv2": {
            "inputs": [
                [
                    [
                        7.4375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.29296875
                    ]
                ]
            }
        },
        "down_blocks.2.resnets.0.conv_shortcut": {
            "inputs": [
                [
                    [
                        84.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.265625
                    ]
                ]
            }
        },
        "down_blocks.2.resnets.1.conv1": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.494140625
                    ]
                ]
            }
        },
        "down_blocks.2.resnets.1.time_emb_proj": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.66796875
                    ]
                ]
            }
        },
        "down_blocks.2.resnets.1.conv2": {
            "inputs": [
                [
                    [
                        9.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.3984375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.proj_in": {
            "inputs": [
                [
                    [
                        9.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.341796875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": {
            "inputs": [
                [
                    [
                        4.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.158203125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": {
            "inputs": [
                [
                    [
                        4.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1572265625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": {
            "inputs": [
                [
                    [
                        4.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.34375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2001953125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.0.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        8.5625
                    ]
                ],
                [
                    [
                        9.6875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.0.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        6.9375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.0.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        372.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": {
            "inputs": [
                [
                    [
                        11.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.09521484375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1708984375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2109375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        5.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.052734375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.0.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        85.0
                    ]
                ],
                [
                    [
                        4.75
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.0.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        33.25
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.0.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        732.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        3.015625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.173828125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": {
            "inputs": [
                [
                    [
                        31.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.392578125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_q": {
            "inputs": [
                [
                    [
                        6.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.158203125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_k": {
            "inputs": [
                [
                    [
                        6.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1728515625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_v": {
            "inputs": [
                [
                    [
                        6.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1552734375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.1.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1806640625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.1.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        8.25
                    ]
                ],
                [
                    [
                        10.875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.1.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.98828125
                    ]
                ],
                [
                    [
                        6.0625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.1.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        312.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.98828125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_q": {
            "inputs": [
                [
                    [
                        13.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12890625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1708984375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.13671875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.1.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        5.40625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.05908203125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.1.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        88.0
                    ]
                ],
                [
                    [
                        6.84375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.1.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        20.125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.1.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        864.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        3.15625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1953125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.1.ff.net.2": {
            "inputs": [
                [
                    [
                        22.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.265625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_q": {
            "inputs": [
                [
                    [
                        7.21875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1640625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_k": {
            "inputs": [
                [
                    [
                        7.21875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1650390625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_v": {
            "inputs": [
                [
                    [
                        7.21875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.146484375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.2.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.228515625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.2.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.28125
                    ]
                ],
                [
                    [
                        8.25
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.2.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        6.84375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.2.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        236.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_q": {
            "inputs": [
                [
                    [
                        16.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1396484375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.29296875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.51953125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.2.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        5.53125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.06640625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.2.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        108.0
                    ]
                ],
                [
                    [
                        7.125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.2.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        20.25
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.2.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        840.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        3.640625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.193359375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.2.ff.net.2": {
            "inputs": [
                [
                    [
                        34.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2392578125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_q": {
            "inputs": [
                [
                    [
                        6.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1552734375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_k": {
            "inputs": [
                [
                    [
                        6.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1611328125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_v": {
            "inputs": [
                [
                    [
                        6.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.146484375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.3.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.09375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.220703125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.3.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        8.375
                    ]
                ],
                [
                    [
                        9.5
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.3.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.98828125
                    ]
                ],
                [
                    [
                        7.6875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.3.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        231.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.98828125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_q": {
            "inputs": [
                [
                    [
                        18.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1953125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1533203125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.3.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        5.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0703125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.3.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        127.0
                    ]
                ],
                [
                    [
                        6.6875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.3.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        18.5
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.3.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        924.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        3.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1728515625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.3.ff.net.2": {
            "inputs": [
                [
                    [
                        44.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.197265625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_q": {
            "inputs": [
                [
                    [
                        7.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1591796875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_k": {
            "inputs": [
                [
                    [
                        7.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.146484375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_v": {
            "inputs": [
                [
                    [
                        7.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1513671875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.4.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        5.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.318359375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.4.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.71875
                    ]
                ],
                [
                    [
                        7.78125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.4.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        7.15625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.4.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        190.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_q": {
            "inputs": [
                [
                    [
                        19.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12060546875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1689453125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1484375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.4.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        5.21875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.072265625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.4.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        106.5
                    ]
                ],
                [
                    [
                        4.96875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.4.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        20.125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.4.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        888.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.34375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1845703125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.4.ff.net.2": {
            "inputs": [
                [
                    [
                        54.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2275390625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_q": {
            "inputs": [
                [
                    [
                        7.84375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.14453125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_k": {
            "inputs": [
                [
                    [
                        7.84375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.154296875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_v": {
            "inputs": [
                [
                    [
                        7.84375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.138671875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.5.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        7.09375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.294921875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.5.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.625
                    ]
                ],
                [
                    [
                        7.59375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.5.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        7.625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.5.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        204.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_q": {
            "inputs": [
                [
                    [
                        20.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1162109375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.17578125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1279296875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.5.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        4.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.06787109375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.5.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        96.5
                    ]
                ],
                [
                    [
                        4.625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.5.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        22.5
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.5.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1168.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.40625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.5.ff.net.2": {
            "inputs": [
                [
                    [
                        53.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.197265625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_q": {
            "inputs": [
                [
                    [
                        8.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1396484375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_k": {
            "inputs": [
                [
                    [
                        8.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.158203125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_v": {
            "inputs": [
                [
                    [
                        8.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.146484375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.6.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.40625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.353515625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.6.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.0
                    ]
                ],
                [
                    [
                        7.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.6.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        7.40625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.6.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        198.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_q": {
            "inputs": [
                [
                    [
                        20.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.11572265625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.177734375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.123046875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.6.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        3.703125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.061767578125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.6.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        98.0
                    ]
                ],
                [
                    [
                        3.96875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.6.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        25.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.6.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        2144.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.71875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1689453125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.6.ff.net.2": {
            "inputs": [
                [
                    [
                        56.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.216796875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_q": {
            "inputs": [
                [
                    [
                        9.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1572265625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_k": {
            "inputs": [
                [
                    [
                        9.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1630859375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_v": {
            "inputs": [
                [
                    [
                        9.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.138671875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.7.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        5.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.30859375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.7.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.25
                    ]
                ],
                [
                    [
                        7.5625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.7.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        6.65625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.7.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        171.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_q": {
            "inputs": [
                [
                    [
                        20.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.09375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.158203125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1220703125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.7.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        13.4375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.052734375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.7.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        93.5
                    ]
                ],
                [
                    [
                        4.46875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.7.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        30.625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.7.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1784.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.236328125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.7.ff.net.2": {
            "inputs": [
                [
                    [
                        51.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.337890625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_q": {
            "inputs": [
                [
                    [
                        9.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1416015625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_k": {
            "inputs": [
                [
                    [
                        9.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.146484375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_v": {
            "inputs": [
                [
                    [
                        9.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.134765625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.8.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.78125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.3359375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.8.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.59375
                    ]
                ],
                [
                    [
                        7.15625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.8.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        8.8125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.8.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        143.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_q": {
            "inputs": [
                [
                    [
                        19.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0859375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12158203125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0947265625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.8.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        7.71875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.09130859375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.8.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        138.0
                    ]
                ],
                [
                    [
                        5.46875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.8.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        39.25
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.8.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        5216.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.29296875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.8.ff.net.2": {
            "inputs": [
                [
                    [
                        50.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.33984375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_q": {
            "inputs": [
                [
                    [
                        10.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.16015625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_k": {
            "inputs": [
                [
                    [
                        10.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1494140625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_v": {
            "inputs": [
                [
                    [
                        10.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1435546875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.9.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.302734375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.9.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.71875
                    ]
                ],
                [
                    [
                        7.125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.9.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        7.1875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.9.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        151.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_q": {
            "inputs": [
                [
                    [
                        23.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12890625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.11181640625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0810546875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.9.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        5.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.080078125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.9.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        91.0
                    ]
                ],
                [
                    [
                        4.78125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.9.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        31.125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.9.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        3392.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2021484375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.transformer_blocks.9.ff.net.2": {
            "inputs": [
                [
                    [
                        46.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.474609375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.0.proj_out": {
            "inputs": [
                [
                    [
                        162.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.109375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.proj_in": {
            "inputs": [
                [
                    [
                        8.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.28515625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": {
            "inputs": [
                [
                    [
                        5.15625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1533203125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": {
            "inputs": [
                [
                    [
                        5.15625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1611328125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": {
            "inputs": [
                [
                    [
                        5.15625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.162109375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.31640625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.0.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.125
                    ]
                ],
                [
                    [
                        9.6875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.0.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.98828125
                    ]
                ],
                [
                    [
                        6.6875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.0.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        228.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.98828125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": {
            "inputs": [
                [
                    [
                        12.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.09521484375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1640625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1328125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        2.90625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.15234375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.0.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        79.5
                    ]
                ],
                [
                    [
                        4.59375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.0.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        20.125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.0.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        648.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        5.59375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.220703125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": {
            "inputs": [
                [
                    [
                        22.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.259765625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_q": {
            "inputs": [
                [
                    [
                        6.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1494140625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_k": {
            "inputs": [
                [
                    [
                        6.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1572265625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_v": {
            "inputs": [
                [
                    [
                        6.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1455078125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.1.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.71875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1845703125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.1.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.5625
                    ]
                ],
                [
                    [
                        8.4375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.1.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9765625
                    ]
                ],
                [
                    [
                        6.75
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.1.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        241.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9765625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_q": {
            "inputs": [
                [
                    [
                        13.9375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1591796875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1982421875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.158203125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.1.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        4.84375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.06884765625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.1.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        65.0
                    ]
                ],
                [
                    [
                        6.75
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.1.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        25.125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.1.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        684.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        7.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2890625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.1.ff.net.2": {
            "inputs": [
                [
                    [
                        23.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.248046875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_q": {
            "inputs": [
                [
                    [
                        8.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.154296875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_k": {
            "inputs": [
                [
                    [
                        8.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.15234375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_v": {
            "inputs": [
                [
                    [
                        8.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.138671875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.2.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.185546875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.2.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.03125
                    ]
                ],
                [
                    [
                        7.75
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.2.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.98046875
                    ]
                ],
                [
                    [
                        7.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.2.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        184.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.98046875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_q": {
            "inputs": [
                [
                    [
                        15.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1630859375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.203125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.181640625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.2.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        5.53125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0703125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.2.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        126.0
                    ]
                ],
                [
                    [
                        6.34375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.2.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        24.375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.2.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        904.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        6.71875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2236328125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.2.ff.net.2": {
            "inputs": [
                [
                    [
                        45.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.20703125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_q": {
            "inputs": [
                [
                    [
                        8.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1669921875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_k": {
            "inputs": [
                [
                    [
                        8.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1474609375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_v": {
            "inputs": [
                [
                    [
                        8.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.16796875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.3.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.90625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2216796875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.3.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.9375
                    ]
                ],
                [
                    [
                        9.0625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.3.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        7.3125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.3.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        182.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_q": {
            "inputs": [
                [
                    [
                        15.9375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.140625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.201171875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1689453125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.3.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        7.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.06787109375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.3.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        154.0
                    ]
                ],
                [
                    [
                        6.65625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.3.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        18.25
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.3.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1360.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        6.84375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.275390625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.3.ff.net.2": {
            "inputs": [
                [
                    [
                        86.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2431640625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_q": {
            "inputs": [
                [
                    [
                        8.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1494140625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_k": {
            "inputs": [
                [
                    [
                        8.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1572265625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_v": {
            "inputs": [
                [
                    [
                        8.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.146484375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.4.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2353515625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.4.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.3125
                    ]
                ],
                [
                    [
                        8.375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.4.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        7.625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.4.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        173.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_q": {
            "inputs": [
                [
                    [
                        17.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1259765625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.193359375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.15625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.4.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        5.78125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.06396484375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.4.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        118.5
                    ]
                ],
                [
                    [
                        5.40625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.4.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        19.375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.4.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1072.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        6.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2099609375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.4.ff.net.2": {
            "inputs": [
                [
                    [
                        39.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1982421875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_q": {
            "inputs": [
                [
                    [
                        8.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1455078125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_k": {
            "inputs": [
                [
                    [
                        8.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1474609375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_v": {
            "inputs": [
                [
                    [
                        8.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.15234375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.5.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        5.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2490234375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.5.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.3125
                    ]
                ],
                [
                    [
                        7.3125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.5.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9765625
                    ]
                ],
                [
                    [
                        7.65625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.5.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        194.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9765625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_q": {
            "inputs": [
                [
                    [
                        17.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1669921875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.15625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.5.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        7.1875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0732421875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.5.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        117.5
                    ]
                ],
                [
                    [
                        4.6875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.5.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        23.5
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.5.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1816.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        6.84375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.26171875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.5.ff.net.2": {
            "inputs": [
                [
                    [
                        43.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1923828125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_q": {
            "inputs": [
                [
                    [
                        10.4375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.162109375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_k": {
            "inputs": [
                [
                    [
                        10.4375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.150390625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_v": {
            "inputs": [
                [
                    [
                        10.4375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.158203125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.6.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        7.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.345703125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.6.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.09375
                    ]
                ],
                [
                    [
                        8.5625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.6.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        8.5625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.6.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        204.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_q": {
            "inputs": [
                [
                    [
                        17.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1220703125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1337890625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.142578125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.6.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        4.09375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0634765625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.6.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        199.0
                    ]
                ],
                [
                    [
                        5.40625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.6.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        22.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.6.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        2320.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        6.71875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.19140625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.6.ff.net.2": {
            "inputs": [
                [
                    [
                        53.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.17578125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_q": {
            "inputs": [
                [
                    [
                        9.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.15234375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_k": {
            "inputs": [
                [
                    [
                        9.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.162109375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_v": {
            "inputs": [
                [
                    [
                        9.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1484375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.7.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.267578125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.7.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.875
                    ]
                ],
                [
                    [
                        8.1875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.7.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        7.6875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.7.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        211.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_q": {
            "inputs": [
                [
                    [
                        16.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1005859375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1494140625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1279296875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.7.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        5.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.059326171875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.7.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        144.0
                    ]
                ],
                [
                    [
                        5.28125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.7.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        25.75
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.7.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        2640.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        6.59375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2275390625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.7.ff.net.2": {
            "inputs": [
                [
                    [
                        56.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.19140625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_q": {
            "inputs": [
                [
                    [
                        10.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1484375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_k": {
            "inputs": [
                [
                    [
                        10.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1611328125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_v": {
            "inputs": [
                [
                    [
                        10.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1611328125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.8.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        7.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2099609375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.8.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.9375
                    ]
                ],
                [
                    [
                        7.875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.8.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        9.125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.8.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        201.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_q": {
            "inputs": [
                [
                    [
                        16.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.09423828125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.11572265625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.8.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        6.03125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.11474609375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.8.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        123.5
                    ]
                ],
                [
                    [
                        4.5
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.8.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        28.25
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.8.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        2944.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        6.09375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.228515625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.8.ff.net.2": {
            "inputs": [
                [
                    [
                        55.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.27734375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_q": {
            "inputs": [
                [
                    [
                        11.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.146484375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_k": {
            "inputs": [
                [
                    [
                        11.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1572265625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_v": {
            "inputs": [
                [
                    [
                        11.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.15625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.9.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        5.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.25
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.9.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.625
                    ]
                ],
                [
                    [
                        7.125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.9.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        9.125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.9.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        136.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_q": {
            "inputs": [
                [
                    [
                        17.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1220703125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1494140625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.103515625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.9.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        5.15625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.076171875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.9.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        97.5
                    ]
                ],
                [
                    [
                        4.78125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.9.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        34.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.9.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        2336.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        5.09375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.transformer_blocks.9.ff.net.2": {
            "inputs": [
                [
                    [
                        43.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.37890625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.1.proj_out": {
            "inputs": [
                [
                    [
                        120.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.11865234375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.proj_in": {
            "inputs": [
                [
                    [
                        12.4375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.251953125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_q": {
            "inputs": [
                [
                    [
                        4.15625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1630859375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_k": {
            "inputs": [
                [
                    [
                        4.15625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1982421875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_v": {
            "inputs": [
                [
                    [
                        4.15625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.130859375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.0.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.5390625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.0.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.59375
                    ]
                ],
                [
                    [
                        8.875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.0.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        5.0625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.0.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        208.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_q": {
            "inputs": [
                [
                    [
                        10.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.10009765625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.341796875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.259765625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.0.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        2.421875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2392578125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.0.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        116.5
                    ]
                ],
                [
                    [
                        4.75
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.0.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        31.125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.0.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1080.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        3.84375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.19140625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.0.ff.net.2": {
            "inputs": [
                [
                    [
                        16.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.365234375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_q": {
            "inputs": [
                [
                    [
                        4.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.158203125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_k": {
            "inputs": [
                [
                    [
                        4.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.171875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_v": {
            "inputs": [
                [
                    [
                        4.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.126953125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.1.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.84375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.26171875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.1.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.34375
                    ]
                ],
                [
                    [
                        8.4375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.1.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        5.90625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.1.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        227.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_q": {
            "inputs": [
                [
                    [
                        10.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1689453125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.55078125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.640625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.1.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        2.71875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.07373046875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.1.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        80.5
                    ]
                ],
                [
                    [
                        6.34375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.1.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        42.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.1.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1056.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.185546875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.1.ff.net.2": {
            "inputs": [
                [
                    [
                        23.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.244140625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_q": {
            "inputs": [
                [
                    [
                        5.28125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1611328125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_k": {
            "inputs": [
                [
                    [
                        5.28125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1591796875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_v": {
            "inputs": [
                [
                    [
                        5.28125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1328125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.2.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.263671875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.2.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.84375
                    ]
                ],
                [
                    [
                        8.9375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.2.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        6.78125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.2.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        227.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_q": {
            "inputs": [
                [
                    [
                        11.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.158203125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.5
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.412109375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.2.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        3.703125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.08056640625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.2.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        98.0
                    ]
                ],
                [
                    [
                        4.9375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.2.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        49.5
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.2.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        676.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.173828125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.2.ff.net.2": {
            "inputs": [
                [
                    [
                        35.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2392578125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_q": {
            "inputs": [
                [
                    [
                        5.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1650390625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_k": {
            "inputs": [
                [
                    [
                        5.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1796875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_v": {
            "inputs": [
                [
                    [
                        5.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12890625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.3.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.34765625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.3.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.625
                    ]
                ],
                [
                    [
                        8.25
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.3.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        5.90625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.3.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        189.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_q": {
            "inputs": [
                [
                    [
                        12.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1513671875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.515625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.4921875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.3.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        3.203125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.06494140625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.3.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        105.5
                    ]
                ],
                [
                    [
                        4.75
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.3.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        19.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.3.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        924.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.78125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1982421875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.3.ff.net.2": {
            "inputs": [
                [
                    [
                        37.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2578125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_q": {
            "inputs": [
                [
                    [
                        6.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.162109375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_k": {
            "inputs": [
                [
                    [
                        6.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.169921875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_v": {
            "inputs": [
                [
                    [
                        6.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.158203125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.4.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        5.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.494140625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.4.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.09375
                    ]
                ],
                [
                    [
                        7.59375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.4.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        6.1875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.4.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        163.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_q": {
            "inputs": [
                [
                    [
                        13.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.10791015625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1728515625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.18359375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.4.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        3.109375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.07958984375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.4.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        76.0
                    ]
                ],
                [
                    [
                        4.71875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.4.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        17.25
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.4.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        648.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.84375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.19921875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.4.ff.net.2": {
            "inputs": [
                [
                    [
                        52.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.28125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_q": {
            "inputs": [
                [
                    [
                        6.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1650390625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_k": {
            "inputs": [
                [
                    [
                        6.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.15625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_v": {
            "inputs": [
                [
                    [
                        6.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1298828125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.5.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        5.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.337890625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.5.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.71875
                    ]
                ],
                [
                    [
                        7.6875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.5.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.98828125
                    ]
                ],
                [
                    [
                        5.6875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.5.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        187.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.98828125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_q": {
            "inputs": [
                [
                    [
                        14.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0947265625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.134765625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.10205078125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.5.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        3.234375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.05810546875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.5.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        79.0
                    ]
                ],
                [
                    [
                        4.21875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.5.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        25.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.5.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        828.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.78125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1826171875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.5.ff.net.2": {
            "inputs": [
                [
                    [
                        28.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.19140625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_q": {
            "inputs": [
                [
                    [
                        6.40625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1826171875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_k": {
            "inputs": [
                [
                    [
                        6.40625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.18359375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_v": {
            "inputs": [
                [
                    [
                        6.40625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1376953125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.6.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.30859375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.6.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.3125
                    ]
                ],
                [
                    [
                        6.625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.6.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        5.625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.6.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        154.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_q": {
            "inputs": [
                [
                    [
                        14.1875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0810546875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.11865234375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.08154296875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.6.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        2.453125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.04638671875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.6.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        60.25
                    ]
                ],
                [
                    [
                        4.125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.6.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        34.25
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.6.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1480.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.84375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.18359375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.6.ff.net.2": {
            "inputs": [
                [
                    [
                        50.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.220703125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_q": {
            "inputs": [
                [
                    [
                        7.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.16796875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_k": {
            "inputs": [
                [
                    [
                        7.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.19140625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_v": {
            "inputs": [
                [
                    [
                        7.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1513671875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.7.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.03125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.32421875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.7.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.09375
                    ]
                ],
                [
                    [
                        8.25
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.7.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        9.0625
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.7.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        177.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_q": {
            "inputs": [
                [
                    [
                        14.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0869140625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1318359375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.08642578125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.7.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        2.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0419921875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.7.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        61.75
                    ]
                ],
                [
                    [
                        3.59375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.7.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        20.5
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.7.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1536.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        5.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1669921875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.7.ff.net.2": {
            "inputs": [
                [
                    [
                        43.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.38671875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_q": {
            "inputs": [
                [
                    [
                        8.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1806640625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_k": {
            "inputs": [
                [
                    [
                        8.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.208984375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_v": {
            "inputs": [
                [
                    [
                        8.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1337890625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.8.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.65625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2119140625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.8.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.90625
                    ]
                ],
                [
                    [
                        8.4375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.8.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.98828125
                    ]
                ],
                [
                    [
                        10.6875
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.8.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        211.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.98828125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_q": {
            "inputs": [
                [
                    [
                        17.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.08447265625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12890625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0751953125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.8.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        4.28125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.043701171875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.8.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        70.5
                    ]
                ],
                [
                    [
                        3.8125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.8.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.984375
                    ]
                ],
                [
                    [
                        38.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.8.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1952.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.984375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        5.4375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1669921875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.8.ff.net.2": {
            "inputs": [
                [
                    [
                        42.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.298828125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_q": {
            "inputs": [
                [
                    [
                        7.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1640625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_k": {
            "inputs": [
                [
                    [
                        7.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.251953125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_v": {
            "inputs": [
                [
                    [
                        7.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.138671875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.9.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.78125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2470703125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.9.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.96875
                    ]
                ],
                [
                    [
                        7.125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.9.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        5.78125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.9.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        188.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_q": {
            "inputs": [
                [
                    [
                        19.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1787109375
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.134765625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.07421875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.9.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        4.40625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.07763671875
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.9.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        80.0
                    ]
                ],
                [
                    [
                        4.28125
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.9.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        46.0
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.9.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        3088.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        5.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1845703125
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.transformer_blocks.9.ff.net.2": {
            "inputs": [
                [
                    [
                        43.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.5625
                    ]
                ]
            }
        },
        "up_blocks.0.attentions.2.proj_out": {
            "inputs": [
                [
                    [
                        174.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12109375
                    ]
                ]
            }
        },
        "up_blocks.0.resnets.0.conv1": {
            "inputs": [
                [
                    [
                        10.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.435546875
                    ]
                ]
            }
        },
        "up_blocks.0.resnets.0.time_emb_proj": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.62109375
                    ]
                ]
            }
        },
        "up_blocks.0.resnets.0.conv2": {
            "inputs": [
                [
                    [
                        13.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.384765625
                    ]
                ]
            }
        },
        "up_blocks.0.resnets.0.conv_shortcut": {
            "inputs": [
                [
                    [
                        59.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.09228515625
                    ]
                ]
            }
        },
        "up_blocks.0.resnets.1.conv1": {
            "inputs": [
                [
                    [
                        11.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.53515625
                    ]
                ]
            }
        },
        "up_blocks.0.resnets.1.time_emb_proj": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.76171875
                    ]
                ]
            }
        },
        "up_blocks.0.resnets.1.conv2": {
            "inputs": [
                [
                    [
                        11.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.50390625
                    ]
                ]
            }
        },
        "up_blocks.0.resnets.1.conv_shortcut": {
            "inputs": [
                [
                    [
                        94.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.162109375
                    ]
                ]
            }
        },
        "up_blocks.0.resnets.2.conv1": {
            "inputs": [
                [
                    [
                        9.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.91015625
                    ]
                ]
            }
        },
        "up_blocks.0.resnets.2.time_emb_proj": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.82421875
                    ]
                ]
            }
        },
        "up_blocks.0.resnets.2.conv2": {
            "inputs": [
                [
                    [
                        8.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.26953125
                    ]
                ]
            }
        },
        "up_blocks.0.resnets.2.conv_shortcut": {
            "inputs": [
                [
                    [
                        94.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1396484375
                    ]
                ]
            }
        },
        "up_blocks.0.upsamplers.0.conv": {
            "inputs": [
                [
                    [
                        67.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.400390625
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.proj_in": {
            "inputs": [
                [
                    [
                        6.9375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.26171875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": {
            "inputs": [
                [
                    [
                        7.96875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.20703125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": {
            "inputs": [
                [
                    [
                        7.96875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.197265625
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": {
            "inputs": [
                [
                    [
                        7.96875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1923828125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        5.78125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.34765625
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.0.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        9.125
                    ]
                ],
                [
                    [
                        10.125
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.0.transformer_blocks.0.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.98046875
                    ]
                ],
                [
                    [
                        7.25
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.0.transformer_blocks.0.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        268.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.98046875
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": {
            "inputs": [
                [
                    [
                        14.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1376953125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2236328125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0751953125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        3.59375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1650390625
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.0.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        91.5
                    ]
                ],
                [
                    [
                        4.1875
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.0.transformer_blocks.0.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        22.875
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.0.transformer_blocks.0.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        836.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        7.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1982421875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": {
            "inputs": [
                [
                    [
                        29.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.380859375
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_q": {
            "inputs": [
                [
                    [
                        12.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.201171875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_k": {
            "inputs": [
                [
                    [
                        12.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2578125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_v": {
            "inputs": [
                [
                    [
                        12.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1953125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.1.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        5.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.380859375
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.1.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        9.875
                    ]
                ],
                [
                    [
                        11.5625
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.0.transformer_blocks.1.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        7.25
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.0.transformer_blocks.1.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        412.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_q": {
            "inputs": [
                [
                    [
                        16.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1357421875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.244140625
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.10498046875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.1.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        17.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.314453125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.1.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        159.0
                    ]
                ],
                [
                    [
                        4.59375
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.0.transformer_blocks.1.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        27.5
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.0.transformer_blocks.1.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1336.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        9.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1962890625
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.transformer_blocks.1.ff.net.2": {
            "inputs": [
                [
                    [
                        33.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.201171875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.0.proj_out": {
            "inputs": [
                [
                    [
                        34.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.23828125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.proj_in": {
            "inputs": [
                [
                    [
                        7.1875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.205078125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": {
            "inputs": [
                [
                    [
                        8.4375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.23046875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": {
            "inputs": [
                [
                    [
                        8.4375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.228515625
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": {
            "inputs": [
                [
                    [
                        8.4375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.173828125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.90625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.328125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.0.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        9.125
                    ]
                ],
                [
                    [
                        9.3125
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.1.transformer_blocks.0.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        5.625
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.1.transformer_blocks.0.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        272.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": {
            "inputs": [
                [
                    [
                        13.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1357421875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1982421875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0751953125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        9.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.0.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        108.0
                    ]
                ],
                [
                    [
                        4.9375
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.1.transformer_blocks.0.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        21.625
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.1.transformer_blocks.0.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        984.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        7.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.251953125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": {
            "inputs": [
                [
                    [
                        23.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2578125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_q": {
            "inputs": [
                [
                    [
                        9.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2099609375
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_k": {
            "inputs": [
                [
                    [
                        9.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2138671875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_v": {
            "inputs": [
                [
                    [
                        9.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2099609375
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.1.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.28125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.326171875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.1.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        9.5
                    ]
                ],
                [
                    [
                        10.5
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.1.transformer_blocks.1.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        6.625
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.1.transformer_blocks.1.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        508.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_q": {
            "inputs": [
                [
                    [
                        13.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1328125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.22265625
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12451171875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.1.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        10.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.291015625
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.1.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        141.0
                    ]
                ],
                [
                    [
                        4.65625
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.1.transformer_blocks.1.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        22.875
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.1.transformer_blocks.1.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1376.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        7.90625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.248046875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.transformer_blocks.1.ff.net.2": {
            "inputs": [
                [
                    [
                        68.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2421875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.1.proj_out": {
            "inputs": [
                [
                    [
                        21.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.16796875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.proj_in": {
            "inputs": [
                [
                    [
                        9.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2255859375
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": {
            "inputs": [
                [
                    [
                        10.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.23046875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": {
            "inputs": [
                [
                    [
                        10.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2314453125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": {
            "inputs": [
                [
                    [
                        10.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1630859375
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.78125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.298828125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.0.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.875
                    ]
                ],
                [
                    [
                        9.5625
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.2.transformer_blocks.0.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        6.90625
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.2.transformer_blocks.0.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        243.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": {
            "inputs": [
                [
                    [
                        14.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1318359375
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.255859375
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12353515625
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        14.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.203125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.0.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        170.0
                    ]
                ],
                [
                    [
                        3.4375
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.2.transformer_blocks.0.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        24.375
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.2.transformer_blocks.0.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        968.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        8.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.396484375
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": {
            "inputs": [
                [
                    [
                        15.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.318359375
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_q": {
            "inputs": [
                [
                    [
                        9.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2236328125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_k": {
            "inputs": [
                [
                    [
                        9.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.263671875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_v": {
            "inputs": [
                [
                    [
                        9.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1748046875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.1.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        5.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.427734375
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.1.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        8.125
                    ]
                ],
                [
                    [
                        10.375
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.2.transformer_blocks.1.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.98046875
                    ]
                ],
                [
                    [
                        6.53125
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.2.transformer_blocks.1.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        314.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.98046875
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_q": {
            "inputs": [
                [
                    [
                        14.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1298828125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.29296875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1591796875
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.1.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        16.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.298828125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.1.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        117.0
                    ]
                ],
                [
                    [
                        3.671875
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.2.transformer_blocks.1.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        37.5
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.2.transformer_blocks.1.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1448.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        7.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.23828125
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.transformer_blocks.1.ff.net.2": {
            "inputs": [
                [
                    [
                        29.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.30859375
                    ]
                ]
            }
        },
        "up_blocks.1.attentions.2.proj_out": {
            "inputs": [
                [
                    [
                        22.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1591796875
                    ]
                ]
            }
        },
        "up_blocks.1.resnets.0.conv1": {
            "inputs": [
                [
                    [
                        12.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        1.0078125
                    ]
                ]
            }
        },
        "up_blocks.1.resnets.0.time_emb_proj": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1552734375
                    ]
                ]
            }
        },
        "up_blocks.1.resnets.0.conv2": {
            "inputs": [
                [
                    [
                        11.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.59375
                    ]
                ]
            }
        },
        "up_blocks.1.resnets.0.conv_shortcut": {
            "inputs": [
                [
                    [
                        146.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2236328125
                    ]
                ]
            }
        },
        "up_blocks.1.resnets.1.conv1": {
            "inputs": [
                [
                    [
                        7.59375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        1.015625
                    ]
                ]
            }
        },
        "up_blocks.1.resnets.1.time_emb_proj": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.470703125
                    ]
                ]
            }
        },
        "up_blocks.1.resnets.1.conv2": {
            "inputs": [
                [
                    [
                        7.90625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.5546875
                    ]
                ]
            }
        },
        "up_blocks.1.resnets.1.conv_shortcut": {
            "inputs": [
                [
                    [
                        42.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.24609375
                    ]
                ]
            }
        },
        "up_blocks.1.resnets.2.conv1": {
            "inputs": [
                [
                    [
                        8.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.64453125
                    ]
                ]
            }
        },
        "up_blocks.1.resnets.2.time_emb_proj": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.7578125
                    ]
                ]
            }
        },
        "up_blocks.1.resnets.2.conv2": {
            "inputs": [
                [
                    [
                        8.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.263671875
                    ]
                ]
            }
        },
        "up_blocks.1.resnets.2.conv_shortcut": {
            "inputs": [
                [
                    [
                        31.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1259765625
                    ]
                ]
            }
        },
        "up_blocks.1.upsamplers.0.conv": {
            "inputs": [
                [
                    [
                        20.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1923828125
                    ]
                ]
            }
        },
        "up_blocks.2.resnets.0.conv1": {
            "inputs": [
                [
                    [
                        9.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.87890625
                    ]
                ]
            }
        },
        "up_blocks.2.resnets.0.time_emb_proj": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.296875
                    ]
                ]
            }
        },
        "up_blocks.2.resnets.0.conv2": {
            "inputs": [
                [
                    [
                        12.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.90625
                    ]
                ]
            }
        },
        "up_blocks.2.resnets.0.conv_shortcut": {
            "inputs": [
                [
                    [
                        54.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.3515625
                    ]
                ]
            }
        },
        "up_blocks.2.resnets.1.conv1": {
            "inputs": [
                [
                    [
                        9.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.703125
                    ]
                ]
            }
        },
        "up_blocks.2.resnets.1.time_emb_proj": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1572265625
                    ]
                ]
            }
        },
        "up_blocks.2.resnets.1.conv2": {
            "inputs": [
                [
                    [
                        12.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.66015625
                    ]
                ]
            }
        },
        "up_blocks.2.resnets.1.conv_shortcut": {
            "inputs": [
                [
                    [
                        25.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.2314453125
                    ]
                ]
            }
        },
        "up_blocks.2.resnets.2.conv1": {
            "inputs": [
                [
                    [
                        21.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.78515625
                    ]
                ]
            }
        },
        "up_blocks.2.resnets.2.time_emb_proj": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1845703125
                    ]
                ]
            }
        },
        "up_blocks.2.resnets.2.conv2": {
            "inputs": [
                [
                    [
                        9.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.9140625
                    ]
                ]
            }
        },
        "up_blocks.2.resnets.2.conv_shortcut": {
            "inputs": [
                [
                    [
                        13.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.25
                    ]
                ]
            }
        },
        "mid_block.attentions.0.proj_in": {
            "inputs": [
                [
                    [
                        8.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.298828125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.0.attn1.to_q": {
            "inputs": [
                [
                    [
                        3.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.140625
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.0.attn1.to_k": {
            "inputs": [
                [
                    [
                        3.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.14453125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.0.attn1.to_v": {
            "inputs": [
                [
                    [
                        3.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1328125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        5.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1923828125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.0.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        8.0625
                    ]
                ],
                [
                    [
                        8.375
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.0.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        6.4375
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.0.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        274.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.0.attn2.to_q": {
            "inputs": [
                [
                    [
                        11.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.07470703125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.0.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12109375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.0.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1064453125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        3.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.046875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.0.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        83.0
                    ]
                ],
                [
                    [
                        4.59375
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.0.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        22.625
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.0.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1216.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        2.640625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1806640625
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.0.ff.net.2": {
            "inputs": [
                [
                    [
                        20.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1630859375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.1.attn1.to_q": {
            "inputs": [
                [
                    [
                        6.53125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.15625
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.1.attn1.to_k": {
            "inputs": [
                [
                    [
                        6.53125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1533203125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.1.attn1.to_v": {
            "inputs": [
                [
                    [
                        6.53125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1435546875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.1.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1494140625
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.1.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ],
                [
                    [
                        9.6875
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.1.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        7.59375
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.1.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        215.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.1.attn2.to_q": {
            "inputs": [
                [
                    [
                        12.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.07958984375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.1.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1259765625
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.1.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1064453125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.1.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        4.78125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.042236328125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.1.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        83.5
                    ]
                ],
                [
                    [
                        5.21875
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.1.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        23.75
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.1.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1120.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.1.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        3.484375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.16015625
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.1.ff.net.2": {
            "inputs": [
                [
                    [
                        22.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.142578125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.2.attn1.to_q": {
            "inputs": [
                [
                    [
                        8.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.140625
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.2.attn1.to_k": {
            "inputs": [
                [
                    [
                        8.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1767578125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.2.attn1.to_v": {
            "inputs": [
                [
                    [
                        8.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.15234375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.2.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.09375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1435546875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.2.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.90625
                    ]
                ],
                [
                    [
                        9.0
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.2.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.984375
                    ]
                ],
                [
                    [
                        7.03125
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.2.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        233.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.984375
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.2.attn2.to_q": {
            "inputs": [
                [
                    [
                        13.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.083984375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.2.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1220703125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.2.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.10546875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.2.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        5.59375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0439453125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.2.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        72.0
                    ]
                ],
                [
                    [
                        4.03125
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.2.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        24.75
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.2.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        864.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.2.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        3.546875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1611328125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.2.ff.net.2": {
            "inputs": [
                [
                    [
                        28.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.14453125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.3.attn1.to_q": {
            "inputs": [
                [
                    [
                        8.9375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1376953125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.3.attn1.to_k": {
            "inputs": [
                [
                    [
                        8.9375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1513671875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.3.attn1.to_v": {
            "inputs": [
                [
                    [
                        8.9375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1435546875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.3.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1337890625
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.3.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        8.4375
                    ]
                ],
                [
                    [
                        8.875
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.3.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        7.75
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.3.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        219.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.3.attn2.to_q": {
            "inputs": [
                [
                    [
                        12.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.06298828125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.3.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.10791015625
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.3.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.07958984375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.3.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        13.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.03515625
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.3.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        62.0
                    ]
                ],
                [
                    [
                        3.921875
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.3.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        24.25
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.3.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1184.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.3.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.34375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.154296875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.3.ff.net.2": {
            "inputs": [
                [
                    [
                        29.875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.138671875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.4.attn1.to_q": {
            "inputs": [
                [
                    [
                        8.1875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1416015625
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.4.attn1.to_k": {
            "inputs": [
                [
                    [
                        8.1875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1455078125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.4.attn1.to_v": {
            "inputs": [
                [
                    [
                        8.1875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.15234375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.4.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        7.4375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1708984375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.4.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        8.375
                    ]
                ],
                [
                    [
                        9.125
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.4.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        8.4375
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.4.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        212.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.4.attn2.to_q": {
            "inputs": [
                [
                    [
                        11.8125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0576171875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.4.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.09619140625
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.4.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0693359375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.4.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        2.359375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0322265625
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.4.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        60.5
                    ]
                ],
                [
                    [
                        3.640625
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.4.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        32.0
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.4.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1080.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.4.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        3.96875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.158203125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.4.ff.net.2": {
            "inputs": [
                [
                    [
                        37.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.146484375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.5.attn1.to_q": {
            "inputs": [
                [
                    [
                        7.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.126953125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.5.attn1.to_k": {
            "inputs": [
                [
                    [
                        7.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1484375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.5.attn1.to_v": {
            "inputs": [
                [
                    [
                        7.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.138671875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.5.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        6.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.150390625
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.5.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        6.96875
                    ]
                ],
                [
                    [
                        8.4375
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.5.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        7.375
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.5.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        189.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.5.attn2.to_q": {
            "inputs": [
                [
                    [
                        9.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0556640625
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.5.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.08642578125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.5.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.06591796875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.5.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        2.546875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.033935546875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.5.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        47.5
                    ]
                ],
                [
                    [
                        3.46875
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.5.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        27.25
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.5.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1072.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.5.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.34375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.16796875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.5.ff.net.2": {
            "inputs": [
                [
                    [
                        34.25
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1552734375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.6.attn1.to_q": {
            "inputs": [
                [
                    [
                        6.40625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1357421875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.6.attn1.to_k": {
            "inputs": [
                [
                    [
                        6.40625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1318359375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.6.attn1.to_v": {
            "inputs": [
                [
                    [
                        6.40625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.123046875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.6.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1240234375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.6.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.0
                    ]
                ],
                [
                    [
                        8.125
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.6.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        7.03125
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.6.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        175.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.6.attn2.to_q": {
            "inputs": [
                [
                    [
                        8.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.059814453125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.6.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.08349609375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.6.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.055908203125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.6.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        6.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.030517578125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.6.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        61.75
                    ]
                ],
                [
                    [
                        3.90625
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.6.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.984375
                    ]
                ],
                [
                    [
                        31.5
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.6.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1528.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.984375
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.6.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.46875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1435546875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.6.ff.net.2": {
            "inputs": [
                [
                    [
                        44.5
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1396484375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.7.attn1.to_q": {
            "inputs": [
                [
                    [
                        6.15625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.12890625
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.7.attn1.to_k": {
            "inputs": [
                [
                    [
                        6.15625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.126953125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.7.attn1.to_v": {
            "inputs": [
                [
                    [
                        6.15625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1416015625
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.7.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        7.21875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1298828125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.7.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.03125
                    ]
                ],
                [
                    [
                        7.6875
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.7.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        8.5
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.7.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        153.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.7.attn2.to_q": {
            "inputs": [
                [
                    [
                        6.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.054443359375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.7.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.07958984375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.7.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.05029296875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.7.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        10.375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.031982421875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.7.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        57.25
                    ]
                ],
                [
                    [
                        3.40625
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.7.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        30.25
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.7.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        1080.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.7.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1455078125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.7.ff.net.2": {
            "inputs": [
                [
                    [
                        42.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.14453125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.8.attn1.to_q": {
            "inputs": [
                [
                    [
                        5.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.134765625
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.8.attn1.to_k": {
            "inputs": [
                [
                    [
                        5.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1318359375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.8.attn1.to_v": {
            "inputs": [
                [
                    [
                        5.625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1435546875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.8.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        8.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1484375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.8.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.84375
                    ]
                ],
                [
                    [
                        8.625
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.8.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.99609375
                    ]
                ],
                [
                    [
                        11.375
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.8.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        179.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.99609375
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.8.attn2.to_q": {
            "inputs": [
                [
                    [
                        5.96875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.05517578125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.8.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0830078125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.8.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.052734375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.8.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        1.7109375
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.0296630859375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.8.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        61.0
                    ]
                ],
                [
                    [
                        3.59375
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.8.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.9921875
                    ]
                ],
                [
                    [
                        26.75
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.8.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        2016.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.9921875
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.8.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.90625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.17578125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.8.ff.net.2": {
            "inputs": [
                [
                    [
                        56.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1435546875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.9.attn1.to_q": {
            "inputs": [
                [
                    [
                        4.96875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1298828125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.9.attn1.to_k": {
            "inputs": [
                [
                    [
                        4.96875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1240234375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.9.attn1.to_v": {
            "inputs": [
                [
                    [
                        4.96875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1396484375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.9.attn1.to_out.0": {
            "inputs": [
                [
                    [
                        4.90625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.14453125
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.9.attn1.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        7.15625
                    ]
                ],
                [
                    [
                        7.71875
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.9.attn1.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        1.0
                    ]
                ],
                [
                    [
                        8.25
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.9.attn1.attention_module.softmax": {
            "inputs": [
                [
                    [
                        153.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        1.0
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.9.attn2.to_q": {
            "inputs": [
                [
                    [
                        5.0625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.046630859375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.9.attn2.to_k": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.08740234375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.9.attn2.to_v": {
            "inputs": [
                [
                    [
                        852.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.04248046875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.9.attn2.to_out.0": {
            "inputs": [
                [
                    [
                        1.0078125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.026123046875
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.9.attn2.attention_module.bmm1": {
            "inputs": [
                [
                    [
                        63.0
                    ]
                ],
                [
                    [
                        3.671875
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.9.attn2.attention_module.bmm2": {
            "inputs": [
                [
                    [
                        0.90234375
                    ]
                ],
                [
                    [
                        28.625
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.9.attn2.attention_module.softmax": {
            "inputs": [
                [
                    [
                        2224.0
                    ]
                ]
            ],
            "outputs": [
                [
                    [
                        0.90234375
                    ]
                ]
            ]
        },
        "mid_block.attentions.0.transformer_blocks.9.ff.net.0.proj": {
            "inputs": [
                [
                    [
                        4.40625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1552734375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.transformer_blocks.9.ff.net.2": {
            "inputs": [
                [
                    [
                        51.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.146484375
                    ]
                ]
            }
        },
        "mid_block.attentions.0.proj_out": {
            "inputs": [
                [
                    [
                        26.0
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.1396484375
                    ]
                ]
            }
        },
        "mid_block.resnets.0.conv1": {
            "inputs": [
                [
                    [
                        9.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.41015625
                    ]
                ]
            }
        },
        "mid_block.resnets.0.time_emb_proj": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.53515625
                    ]
                ]
            }
        },
        "mid_block.resnets.0.conv2": {
            "inputs": [
                [
                    [
                        11.125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.71484375
                    ]
                ]
            }
        },
        "mid_block.resnets.1.conv1": {
            "inputs": [
                [
                    [
                        8.5625
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.482421875
                    ]
                ]
            }
        },
        "mid_block.resnets.1.time_emb_proj": {
            "inputs": [
                [
                    [
                        7.6875
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.640625
                    ]
                ]
            }
        },
        "mid_block.resnets.1.conv2": {
            "inputs": [
                [
                    [
                        13.3125
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.61328125
                    ]
                ]
            }
        },
        "conv_out": {
            "inputs": [
                [
                    [
                        11.75
                    ]
                ]
            ],
            "params": {
                "weight": [
                    [
                        0.21875
                    ]
                ]
            }
        }
    }
}