diff --git "a/fp8_hooks_maxabs.json" "b/fp8_hooks_maxabs.json" --- "a/fp8_hooks_maxabs.json" +++ "b/fp8_hooks_maxabs.json" @@ -1,1555 +1,18871 @@ - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - Page not found · GitHub - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
- - - -
- Skip to content - - - - - - - - - - - - - - - - - - - -
- - - - - - - - - - - - - - -
- -
- - - - - - - - -
- - - - - -
- - - - - - - - - - - -
-
- - - -
-
- -
-
- 404 “This is not the web page you are looking for” - - - - - - - - - - - - -
-
- -
-
- -
- - -
-
-
- Contact Support — - GitHub Status — - @githubstatus -
-
- -
- -
- - - - - - - - - - - - - - - - - - - - - - -
- -
-
- - - +{ + "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 + ] + ] + } + } + } +} \ No newline at end of file