diff --git "a/openvino_model.xml" "b/openvino_model.xml"
--- "a/openvino_model.xml"
+++ "b/openvino_model.xml"
@@ -1,40 +1,112 @@
 <?xml version="1.0"?>
-<net name="Model6" version="11">
+<net name="Model9" version="11">
 	<layers>
-		<layer id="0" name="sample" type="Parameter" version="opset1">
-			<data shape="?,3,?,?" element_type="f32" />
+		<layer id="0" name="latent_sample" type="Parameter" version="opset1">
+			<data shape="?,4,?,?" element_type="f32" />
 			<output>
-				<port id="0" precision="FP32" names="sample">
+				<port id="0" precision="FP32" names="latent_sample">
 					<dim>-1</dim>
-					<dim>3</dim>
+					<dim>4</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="1" name="self.encoder.conv_in.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="128, 3, 3, 3" offset="0" size="13824" />
+		<layer id="1" name="self.post_quant_conv.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="4, 4, 1, 1" offset="0" size="64" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.conv_in.weight">
-					<dim>128</dim>
-					<dim>3</dim>
+				<port id="0" precision="FP32" names="self.post_quant_conv.weight">
+					<dim>4</dim>
+					<dim>4</dim>
+					<dim>1</dim>
+					<dim>1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="2" name="__module.post_quant_conv/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="0, 0" pads_end="0, 0" auto_pad="explicit" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>4</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>4</dim>
+					<dim>4</dim>
+					<dim>1</dim>
+					<dim>1</dim>
+				</port>
+			</input>
+			<output>
+				<port id="2" precision="FP32">
+					<dim>-1</dim>
+					<dim>4</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="3" name="__module.post_quant_conv/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 4, 1, 1" offset="64" size="16" />
+			<output>
+				<port id="0" precision="FP32">
+					<dim>1</dim>
+					<dim>4</dim>
+					<dim>1</dim>
+					<dim>1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="4" name="__module.post_quant_conv/aten::_convolution/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>4</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>1</dim>
+					<dim>4</dim>
+					<dim>1</dim>
+					<dim>1</dim>
+				</port>
+			</input>
+			<output>
+				<port id="2" precision="FP32" names="17,input.1">
+					<dim>-1</dim>
+					<dim>4</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="5" name="self.decoder.conv_in.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 4, 3, 3" offset="80" size="73728" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.conv_in.weight">
+					<dim>512</dim>
+					<dim>4</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="2" name="__module.encoder.conv_in/aten::_convolution/Convolution" type="Convolution" version="opset1">
+		<layer id="6" name="__module.decoder.conv_in/aten::_convolution/Convolution" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>3</dim>
+					<dim>4</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>128</dim>
-					<dim>3</dim>
+					<dim>512</dim>
+					<dim>4</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
@@ -42,130 +114,130 @@
 			<output>
 				<port id="2" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="3" name="__module.encoder.conv_in/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 128, 1, 1" offset="13824" size="512" />
+		<layer id="7" name="__module.decoder.conv_in/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="73808" size="2048" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="4" name="__module.encoder.conv_in/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="8" name="__module.decoder.conv_in/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
 					<dim>1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="92,input.1">
+				<port id="2" precision="FP32" names="57,input.3">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="5" name="self.encoder.down_blocks.0.resnets.0.norm1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="128" offset="14336" size="512" />
+		<layer id="9" name="self.decoder.mid_block.resnets.0.norm1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="75856" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.0.resnets.0.norm1.weight">
-					<dim>128</dim>
+				<port id="0" precision="FP32" names="self.decoder.mid_block.resnets.0.norm1.weight">
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="6" name="self.encoder.down_blocks.0.resnets.0.norm1.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="128" offset="14848" size="512" />
+		<layer id="10" name="self.decoder.mid_block.resnets.0.norm1.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="77904" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.0.resnets.0.norm1.bias">
-					<dim>128</dim>
+				<port id="0" precision="FP32" names="self.decoder.mid_block.resnets.0.norm1.bias">
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="7" name="__module.encoder.down_blocks.0.resnets.0.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+		<layer id="11" name="__module.decoder.mid_block.resnets.0.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
 			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>128</dim>
+					<dim>512</dim>
 				</port>
 				<port id="2" precision="FP32">
-					<dim>128</dim>
+					<dim>512</dim>
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="106,input.3">
+				<port id="3" precision="FP32" names="74,input.5">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="8" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish" type="Swish" version="opset4">
+		<layer id="12" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="107,input.5">
+				<port id="1" precision="FP32" names="75,input.7">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="9" name="self.encoder.down_blocks.0.resnets.0.conv1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="128, 128, 3, 3" offset="15360" size="589824" />
+		<layer id="13" name="self.decoder.mid_block.resnets.0.conv1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512, 3, 3" offset="79952" size="9437184" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.0.resnets.0.conv1.weight">
-					<dim>128</dim>
-					<dim>128</dim>
+				<port id="0" precision="FP32" names="self.decoder.mid_block.resnets.0.conv1.weight">
+					<dim>512</dim>
+					<dim>512</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="10" name="__module.encoder.down_blocks.0.resnets.0.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
+		<layer id="14" name="__module.decoder.mid_block.resnets.0.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>128</dim>
-					<dim>128</dim>
+					<dim>512</dim>
+					<dim>512</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
@@ -173,130 +245,130 @@
 			<output>
 				<port id="2" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="11" name="__module.encoder.down_blocks.0.resnets.0.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 128, 1, 1" offset="605184" size="512" />
+		<layer id="15" name="__module.decoder.mid_block.resnets.0.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="9517136" size="2048" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="12" name="__module.encoder.down_blocks.0.resnets.0.conv1/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="16" name="__module.decoder.mid_block.resnets.0.conv1/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
 					<dim>1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="114,input.7">
+				<port id="2" precision="FP32" names="82,input.9">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="13" name="self.encoder.down_blocks.0.resnets.0.norm2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="128" offset="605696" size="512" />
+		<layer id="17" name="self.decoder.mid_block.resnets.0.norm2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="9519184" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.0.resnets.0.norm2.weight">
-					<dim>128</dim>
+				<port id="0" precision="FP32" names="self.decoder.mid_block.resnets.0.norm2.weight">
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="14" name="self.encoder.down_blocks.0.resnets.0.norm2.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="128" offset="606208" size="512" />
+		<layer id="18" name="self.decoder.mid_block.resnets.0.norm2.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="9521232" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.0.resnets.0.norm2.bias">
-					<dim>128</dim>
+				<port id="0" precision="FP32" names="self.decoder.mid_block.resnets.0.norm2.bias">
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="15" name="__module.encoder.down_blocks.0.resnets.0.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+		<layer id="19" name="__module.decoder.mid_block.resnets.0.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
 			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>128</dim>
+					<dim>512</dim>
 				</port>
 				<port id="2" precision="FP32">
-					<dim>128</dim>
+					<dim>512</dim>
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="117,input.9">
+				<port id="3" precision="FP32" names="85,input.11">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="16" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_1" type="Swish" version="opset4">
+		<layer id="20" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_1" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="118,input.11">
+				<port id="1" precision="FP32" names="86,input.13">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="17" name="self.encoder.down_blocks.0.resnets.0.conv2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="128, 128, 3, 3" offset="606720" size="589824" />
+		<layer id="21" name="self.decoder.mid_block.resnets.0.conv2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512, 3, 3" offset="9523280" size="9437184" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.0.resnets.0.conv2.weight">
-					<dim>128</dim>
-					<dim>128</dim>
+				<port id="0" precision="FP32" names="self.decoder.mid_block.resnets.0.conv2.weight">
+					<dim>512</dim>
+					<dim>512</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="18" name="__module.encoder.down_blocks.0.resnets.0.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
+		<layer id="22" name="__module.decoder.mid_block.resnets.0.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>128</dim>
-					<dim>128</dim>
+					<dim>512</dim>
+					<dim>512</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
@@ -304,717 +376,1101 @@
 			<output>
 				<port id="2" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="19" name="__module.encoder.down_blocks.0.resnets.0.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 128, 1, 1" offset="1196544" size="512" />
+		<layer id="23" name="__module.decoder.mid_block.resnets.0.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="18960464" size="2048" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="20" name="__module.encoder.down_blocks.0.resnets.0.conv2/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="24" name="__module.decoder.mid_block.resnets.0.conv2/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
 					<dim>1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="126,hidden_states.1">
+				<port id="2" precision="FP32" names="94,hidden_states.1">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="21" name="__module.encoder.down_blocks.0.resnets.0/aten::add/Add" type="Add" version="opset1">
+		<layer id="25" name="__module.decoder.mid_block.resnets.0/aten::add/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="127,128,input.15">
+				<port id="2" precision="FP32" names="95,96,hidden_states.3">
 					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="22" name="self.encoder.down_blocks.0.resnets.1.norm1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="128" offset="1197056" size="512" />
-			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.0.resnets.1.norm1.weight">
-					<dim>128</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="23" name="self.encoder.down_blocks.0.resnets.1.norm1.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="128" offset="1197568" size="512" />
+		<layer id="26" name="Constant_243393" type="Const" version="opset1">
+			<data element_type="i64" shape="3" offset="18962512" size="24" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.0.resnets.1.norm1.bias">
-					<dim>128</dim>
+				<port id="0" precision="I64">
+					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="24" name="__module.encoder.down_blocks.0.resnets.1.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
-			<data num_groups="32" epsilon="9.9999999747524271e-07" />
+		<layer id="27" name="__module.decoder.mid_block.attentions.0/aten::view/Reshape" type="Reshape" version="opset1">
+			<data special_zero="true" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
-				</port>
-				<port id="1" precision="FP32">
-					<dim>128</dim>
-				</port>
-				<port id="2" precision="FP32">
-					<dim>128</dim>
-				</port>
-			</input>
-			<output>
-				<port id="3" precision="FP32" names="136,input.17">
-					<dim>-1</dim>
-					<dim>128</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
-			</output>
-		</layer>
-		<layer id="25" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_2" type="Swish" version="opset4">
-			<input>
-				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>128</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
+				<port id="1" precision="I64">
+					<dim>3</dim>
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="137,input.19">
-					<dim>-1</dim>
-					<dim>128</dim>
+				<port id="2" precision="FP32" names="112">
 					<dim>-1</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="26" name="self.encoder.down_blocks.0.resnets.1.conv1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="128, 128, 3, 3" offset="1198080" size="589824" />
+		<layer id="28" name="__module.decoder.mid_block.attentions.0/aten::transpose/ScatterElementsUpdate" type="Const" version="opset1">
+			<data element_type="i32" shape="3" offset="18962536" size="12" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.0.resnets.1.conv1.weight">
-					<dim>128</dim>
-					<dim>128</dim>
-					<dim>3</dim>
+				<port id="0" precision="I32">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="27" name="__module.encoder.down_blocks.0.resnets.1.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
-			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
+		<layer id="29" name="__module.decoder.mid_block.attentions.0/aten::transpose/Transpose" type="Transpose" version="opset1">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
-					<dim>-1</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 				</port>
-				<port id="1" precision="FP32">
-					<dim>128</dim>
-					<dim>128</dim>
-					<dim>3</dim>
+				<port id="1" precision="I32">
 					<dim>3</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32">
-					<dim>-1</dim>
-					<dim>128</dim>
+				<port id="2" precision="FP32" names="113,hidden_states.5">
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="28" name="__module.encoder.down_blocks.0.resnets.1.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 128, 1, 1" offset="1787904" size="512" />
+		<layer id="30" name="__module.decoder.mid_block.attentions.0/aten::transpose/ScatterElementsUpdate_1" type="Const" version="opset1">
+			<data element_type="i32" shape="3" offset="18962536" size="12" />
 			<output>
-				<port id="0" precision="FP32">
-					<dim>1</dim>
-					<dim>128</dim>
-					<dim>1</dim>
-					<dim>1</dim>
+				<port id="0" precision="I32">
+					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="29" name="__module.encoder.down_blocks.0.resnets.1.conv1/aten::_convolution/Add" type="Add" version="opset1">
-			<data auto_broadcast="numpy" />
+		<layer id="31" name="__module.decoder.mid_block.attentions.0/aten::transpose/Transpose_1" type="Transpose" version="opset1">
 			<input>
 				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>512</dim>
 				</port>
-				<port id="1" precision="FP32">
-					<dim>1</dim>
-					<dim>128</dim>
-					<dim>1</dim>
-					<dim>1</dim>
+				<port id="1" precision="I32">
+					<dim>3</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="144,input.21">
-					<dim>-1</dim>
-					<dim>128</dim>
+				<port id="2" precision="FP32" names="115,input.17">
 					<dim>-1</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="30" name="self.encoder.down_blocks.0.resnets.1.norm2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="128" offset="1788416" size="512" />
+		<layer id="32" name="self.decoder.mid_block.attentions.0.group_norm.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="18962548" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.0.resnets.1.norm2.weight">
-					<dim>128</dim>
+				<port id="0" precision="FP32" names="self.decoder.mid_block.attentions.0.group_norm.weight">
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="31" name="self.encoder.down_blocks.0.resnets.1.norm2.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="128" offset="1788928" size="512" />
+		<layer id="33" name="self.decoder.mid_block.attentions.0.group_norm.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="18964596" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.0.resnets.1.norm2.bias">
-					<dim>128</dim>
+				<port id="0" precision="FP32" names="self.decoder.mid_block.attentions.0.group_norm.bias">
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="32" name="__module.encoder.down_blocks.0.resnets.1.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+		<layer id="34" name="__module.decoder.mid_block.attentions.0.group_norm/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
 			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
-					<dim>-1</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>128</dim>
+					<dim>512</dim>
 				</port>
 				<port id="2" precision="FP32">
-					<dim>128</dim>
+					<dim>512</dim>
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="147,input.23">
-					<dim>-1</dim>
-					<dim>128</dim>
+				<port id="3" precision="FP32" names="118">
 					<dim>-1</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="33" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_3" type="Swish" version="opset4">
-			<input>
-				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>128</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
-				</port>
-			</input>
-			<output>
-				<port id="1" precision="FP32" names="148,input.25">
-					<dim>-1</dim>
-					<dim>128</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="34" name="self.encoder.down_blocks.0.resnets.1.conv2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="128, 128, 3, 3" offset="1789440" size="589824" />
+		<layer id="35" name="self.decoder.mid_block.attentions.0.to_q.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512" offset="18966644" size="1048576" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.0.resnets.1.conv2.weight">
-					<dim>128</dim>
-					<dim>128</dim>
-					<dim>3</dim>
-					<dim>3</dim>
+				<port id="0" precision="FP32" names="self.decoder.mid_block.attentions.0.to_q.weight">
+					<dim>512</dim>
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="35" name="__module.encoder.down_blocks.0.resnets.1.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
-			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
+		<layer id="36" name="__module.decoder.mid_block.attentions.0.to_q/aten::linear/MatMul" type="MatMul" version="opset1">
+			<data transpose_a="true" transpose_b="true" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
-					<dim>-1</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>128</dim>
-					<dim>128</dim>
-					<dim>3</dim>
-					<dim>3</dim>
+					<dim>512</dim>
+					<dim>512</dim>
 				</port>
 			</input>
 			<output>
 				<port id="2" precision="FP32">
-					<dim>-1</dim>
-					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="36" name="__module.encoder.down_blocks.0.resnets.1.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 128, 1, 1" offset="2379264" size="512" />
+		<layer id="37" name="Constant_243274" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 1, 512" offset="20015220" size="2048" />
 			<output>
 				<port id="0" precision="FP32">
-					<dim>1</dim>
-					<dim>128</dim>
 					<dim>1</dim>
 					<dim>1</dim>
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="37" name="__module.encoder.down_blocks.0.resnets.1.conv2/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="38" name="__module.decoder.mid_block.attentions.0.to_q/aten::linear/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>512</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>1</dim>
-					<dim>128</dim>
 					<dim>1</dim>
 					<dim>1</dim>
+					<dim>512</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="156,hidden_states.3">
-					<dim>-1</dim>
-					<dim>128</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="38" name="__module.encoder.down_blocks.0.resnets.1/aten::add/Add" type="Add" version="opset1">
-			<data auto_broadcast="numpy" />
-			<input>
-				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>128</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
-				</port>
-				<port id="1" precision="FP32">
-					<dim>-1</dim>
-					<dim>128</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
-				</port>
-			</input>
-			<output>
-				<port id="2" precision="FP32" names="157,158,hidden_states.5">
-					<dim>-1</dim>
-					<dim>128</dim>
+				<port id="2" precision="FP32" names="122,query.1">
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="39" name="self.encoder.down_blocks.0.downsamplers.0.conv.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="128, 128, 3, 3" offset="2379776" size="589824" />
+		<layer id="39" name="Constant_243394" type="Const" version="opset1">
+			<data element_type="i64" shape="4" offset="20017268" size="32" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.0.downsamplers.0.conv.weight">
-					<dim>128</dim>
-					<dim>128</dim>
-					<dim>3</dim>
-					<dim>3</dim>
+				<port id="0" precision="I64">
+					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="40" name="__module.encoder.down_blocks.0.downsamplers.0.conv/aten::_convolution/Convolution" type="Convolution" version="opset1">
-			<data strides="2, 2" dilations="1, 1" pads_begin="0, 0" pads_end="1, 1" auto_pad="explicit" />
+		<layer id="40" name="__module.decoder.mid_block.attentions.0/aten::view/Reshape_1" type="Reshape" version="opset1">
+			<data special_zero="true" />
 			<input>
 				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>512</dim>
 				</port>
-				<port id="1" precision="FP32">
-					<dim>128</dim>
-					<dim>128</dim>
-					<dim>3</dim>
-					<dim>3</dim>
+				<port id="1" precision="I64">
+					<dim>4</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32">
-					<dim>-1</dim>
-					<dim>128</dim>
+				<port id="2" precision="FP32" names="136">
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>1</dim>
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="41" name="__module.encoder.down_blocks.0.downsamplers.0.conv/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 128, 1, 1" offset="2969600" size="512" />
+		<layer id="41" name="Constant_243140" type="Const" version="opset1">
+			<data element_type="i64" shape="4" offset="20017300" size="32" />
 			<output>
-				<port id="0" precision="FP32">
-					<dim>1</dim>
-					<dim>128</dim>
-					<dim>1</dim>
-					<dim>1</dim>
+				<port id="0" precision="I64">
+					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="42" name="__module.encoder.down_blocks.0.downsamplers.0.conv/aten::_convolution/Add" type="Add" version="opset1">
-			<data auto_broadcast="numpy" />
+		<layer id="42" name="__module.decoder.mid_block.attentions.0/aten::transpose/Transpose_3" type="Reshape" version="opset1">
+			<data special_zero="true" />
 			<input>
 				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-				</port>
-				<port id="1" precision="FP32">
-					<dim>1</dim>
-					<dim>128</dim>
-					<dim>1</dim>
 					<dim>1</dim>
+					<dim>512</dim>
+				</port>
+				<port id="1" precision="I64">
+					<dim>4</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="168,input.29">
-					<dim>-1</dim>
-					<dim>128</dim>
+				<port id="2" precision="FP32" names="137,query">
 					<dim>-1</dim>
+					<dim>1</dim>
 					<dim>-1</dim>
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="43" name="self.encoder.down_blocks.1.resnets.0.conv_shortcut.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="256, 128, 1, 1" offset="2970112" size="131072" />
+		<layer id="43" name="self.decoder.mid_block.attentions.0.to_k.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512" offset="20017332" size="1048576" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.1.resnets.0.conv_shortcut.weight">
-					<dim>256</dim>
-					<dim>128</dim>
-					<dim>1</dim>
-					<dim>1</dim>
+				<port id="0" precision="FP32" names="self.decoder.mid_block.attentions.0.to_k.weight">
+					<dim>512</dim>
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="44" name="__module.encoder.down_blocks.1.resnets.0.conv_shortcut/aten::_convolution/Convolution" type="Convolution" version="opset1">
-			<data strides="1, 1" dilations="1, 1" pads_begin="0, 0" pads_end="0, 0" auto_pad="explicit" />
+		<layer id="44" name="__module.decoder.mid_block.attentions.0.to_k/aten::linear/MatMul" type="MatMul" version="opset1">
+			<data transpose_a="true" transpose_b="true" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
-					<dim>-1</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>256</dim>
-					<dim>128</dim>
-					<dim>1</dim>
-					<dim>1</dim>
+					<dim>512</dim>
+					<dim>512</dim>
 				</port>
 			</input>
 			<output>
 				<port id="2" precision="FP32">
-					<dim>-1</dim>
-					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="45" name="__module.encoder.down_blocks.1.resnets.0.conv_shortcut/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 256, 1, 1" offset="3101184" size="1024" />
+		<layer id="45" name="Constant_243275" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 1, 512" offset="21065908" size="2048" />
 			<output>
 				<port id="0" precision="FP32">
-					<dim>1</dim>
-					<dim>256</dim>
 					<dim>1</dim>
 					<dim>1</dim>
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="46" name="__module.encoder.down_blocks.1.resnets.0.conv_shortcut/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="46" name="__module.decoder.mid_block.attentions.0.to_k/aten::linear/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>512</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>1</dim>
-					<dim>256</dim>
 					<dim>1</dim>
 					<dim>1</dim>
+					<dim>512</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="210,input_tensor.1">
-					<dim>-1</dim>
-					<dim>256</dim>
+				<port id="2" precision="FP32" names="125,key.1">
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="47" name="self.encoder.down_blocks.1.resnets.0.norm1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="128" offset="3102208" size="512" />
-			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.1.resnets.0.norm1.weight">
-					<dim>128</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="48" name="self.encoder.down_blocks.1.resnets.0.norm1.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="128" offset="3102720" size="512" />
+		<layer id="47" name="Constant_243395" type="Const" version="opset1">
+			<data element_type="i64" shape="4" offset="20017268" size="32" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.1.resnets.0.norm1.bias">
-					<dim>128</dim>
+				<port id="0" precision="I64">
+					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="49" name="__module.encoder.down_blocks.1.resnets.0.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
-			<data num_groups="32" epsilon="9.9999999747524271e-07" />
+		<layer id="48" name="__module.decoder.mid_block.attentions.0/aten::view/Reshape_2" type="Reshape" version="opset1">
+			<data special_zero="true" />
 			<input>
 				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>512</dim>
 				</port>
-				<port id="1" precision="FP32">
-					<dim>128</dim>
-				</port>
-				<port id="2" precision="FP32">
-					<dim>128</dim>
+				<port id="1" precision="I64">
+					<dim>4</dim>
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="183,input.31">
-					<dim>-1</dim>
-					<dim>128</dim>
+				<port id="2" precision="FP32" names="139">
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>1</dim>
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="49" name="Constant_243142" type="Const" version="opset1">
+			<data element_type="i64" shape="4" offset="20017300" size="32" />
+			<output>
+				<port id="0" precision="I64">
+					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="50" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_4" type="Swish" version="opset4">
+		<layer id="50" name="__module.decoder.mid_block.attentions.0/aten::transpose/Transpose_4" type="Reshape" version="opset1">
+			<data special_zero="true" />
 			<input>
 				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>1</dim>
+					<dim>512</dim>
+				</port>
+				<port id="1" precision="I64">
+					<dim>4</dim>
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="184,input.33">
-					<dim>-1</dim>
-					<dim>128</dim>
+				<port id="2" precision="FP32" names="140,key">
 					<dim>-1</dim>
+					<dim>1</dim>
 					<dim>-1</dim>
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="51" name="self.encoder.down_blocks.1.resnets.0.conv1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="256, 128, 3, 3" offset="3103232" size="1179648" />
+		<layer id="51" name="self.decoder.mid_block.attentions.0.to_v.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512" offset="21067956" size="1048576" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.1.resnets.0.conv1.weight">
-					<dim>256</dim>
-					<dim>128</dim>
-					<dim>3</dim>
-					<dim>3</dim>
+				<port id="0" precision="FP32" names="self.decoder.mid_block.attentions.0.to_v.weight">
+					<dim>512</dim>
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="52" name="__module.encoder.down_blocks.1.resnets.0.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
-			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
+		<layer id="52" name="__module.decoder.mid_block.attentions.0.to_v/aten::linear/MatMul" type="MatMul" version="opset1">
+			<data transpose_a="true" transpose_b="true" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>128</dim>
-					<dim>-1</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>256</dim>
-					<dim>128</dim>
-					<dim>3</dim>
-					<dim>3</dim>
+					<dim>512</dim>
+					<dim>512</dim>
 				</port>
 			</input>
 			<output>
 				<port id="2" precision="FP32">
-					<dim>-1</dim>
-					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="53" name="__module.encoder.down_blocks.1.resnets.0.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 256, 1, 1" offset="4282880" size="1024" />
+		<layer id="53" name="Constant_243276" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 1, 512" offset="22116532" size="2048" />
 			<output>
 				<port id="0" precision="FP32">
-					<dim>1</dim>
-					<dim>256</dim>
 					<dim>1</dim>
 					<dim>1</dim>
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="54" name="__module.encoder.down_blocks.1.resnets.0.conv1/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="54" name="__module.decoder.mid_block.attentions.0.to_v/aten::linear/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>512</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>1</dim>
-					<dim>256</dim>
 					<dim>1</dim>
 					<dim>1</dim>
+					<dim>512</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="191,input.35">
-					<dim>-1</dim>
-					<dim>256</dim>
+				<port id="2" precision="FP32" names="128,value.1">
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="55" name="self.encoder.down_blocks.1.resnets.0.norm2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="256" offset="4283904" size="1024" />
-			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.1.resnets.0.norm2.weight">
-					<dim>256</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="56" name="self.encoder.down_blocks.1.resnets.0.norm2.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="256" offset="4284928" size="1024" />
+		<layer id="55" name="Constant_243396" type="Const" version="opset1">
+			<data element_type="i64" shape="4" offset="20017268" size="32" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.1.resnets.0.norm2.bias">
-					<dim>256</dim>
+				<port id="0" precision="I64">
+					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="57" name="__module.encoder.down_blocks.1.resnets.0.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
-			<data num_groups="32" epsilon="9.9999999747524271e-07" />
+		<layer id="56" name="__module.decoder.mid_block.attentions.0/aten::view/Reshape_3" type="Reshape" version="opset1">
+			<data special_zero="true" />
 			<input>
 				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>512</dim>
 				</port>
-				<port id="1" precision="FP32">
-					<dim>256</dim>
-				</port>
-				<port id="2" precision="FP32">
-					<dim>256</dim>
+				<port id="1" precision="I64">
+					<dim>4</dim>
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="194,input.37">
-					<dim>-1</dim>
-					<dim>256</dim>
+				<port id="2" precision="FP32" names="142">
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>1</dim>
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="57" name="Constant_243146" type="Const" version="opset1">
+			<data element_type="i64" shape="4" offset="20017300" size="32" />
+			<output>
+				<port id="0" precision="I64">
+					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="58" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_5" type="Swish" version="opset4">
+		<layer id="58" name="__module.decoder.mid_block.attentions.0/aten::transpose/Transpose_5" type="Reshape" version="opset1">
+			<data special_zero="true" />
 			<input>
 				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
+					<dim>1</dim>
+					<dim>512</dim>
+				</port>
+				<port id="1" precision="I64">
+					<dim>4</dim>
+				</port>
+			</input>
+			<output>
+				<port id="2" precision="FP32" names="143,value">
+					<dim>-1</dim>
+					<dim>1</dim>
+					<dim>-1</dim>
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="59" name="__module.decoder.mid_block.attentions.0/aten::scaled_dot_product_attention/ScaledDotProductAttention" type="ScaledDotProductAttention" version="opset13">
+			<data causal="false" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>1</dim>
+					<dim>-1</dim>
+					<dim>512</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>-1</dim>
+					<dim>1</dim>
+					<dim>-1</dim>
+					<dim>512</dim>
+				</port>
+				<port id="2" precision="FP32">
+					<dim>-1</dim>
+					<dim>1</dim>
+					<dim>-1</dim>
+					<dim>512</dim>
+				</port>
+			</input>
+			<output>
+				<port id="3" precision="FP32" names="144,hidden_states.7">
+					<dim>-1</dim>
+					<dim>1</dim>
+					<dim>-1</dim>
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="60" name="Constant_243148" type="Const" version="opset1">
+			<data element_type="i64" shape="4" offset="22118580" size="32" />
+			<output>
+				<port id="0" precision="I64">
+					<dim>4</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="61" name="__module.decoder.mid_block.attentions.0/aten::transpose/Transpose_6" type="Reshape" version="opset1">
+			<data special_zero="true" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>1</dim>
+					<dim>-1</dim>
+					<dim>512</dim>
+				</port>
+				<port id="1" precision="I64">
+					<dim>4</dim>
+				</port>
+			</input>
+			<output>
+				<port id="2" precision="FP32" names="145">
+					<dim>-1</dim>
+					<dim>-1</dim>
+					<dim>1</dim>
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="62" name="Constant_243397" type="Const" version="opset1">
+			<data element_type="i64" shape="3" offset="22118612" size="24" />
+			<output>
+				<port id="0" precision="I64">
+					<dim>3</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="63" name="__module.decoder.mid_block.attentions.0/aten::reshape/Reshape" type="Reshape" version="opset1">
+			<data special_zero="true" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>-1</dim>
+					<dim>1</dim>
+					<dim>512</dim>
+				</port>
+				<port id="1" precision="I64">
+					<dim>3</dim>
+				</port>
+			</input>
+			<output>
+				<port id="2" precision="FP32" names="149,150,hidden_states.9,input.21">
+					<dim>-1</dim>
+					<dim>-1</dim>
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="64" name="self.decoder.mid_block.attentions.0.to_out.0.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512" offset="22118636" size="1048576" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.mid_block.attentions.0.to_out.0.weight">
+					<dim>512</dim>
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="65" name="__module.decoder.mid_block.attentions.0.to_out.0/aten::linear/MatMul" type="MatMul" version="opset1">
+			<data transpose_a="false" transpose_b="true" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>-1</dim>
+					<dim>512</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>512</dim>
+					<dim>512</dim>
+				</port>
+			</input>
+			<output>
+				<port id="2" precision="FP32">
+					<dim>-1</dim>
+					<dim>-1</dim>
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="66" name="Constant_243277" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 1, 512" offset="23167212" size="2048" />
+			<output>
+				<port id="0" precision="FP32">
+					<dim>1</dim>
+					<dim>1</dim>
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="67" name="__module.decoder.mid_block.attentions.0.to_out.0/aten::linear/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>-1</dim>
+					<dim>512</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>1</dim>
+					<dim>1</dim>
+					<dim>512</dim>
+				</port>
+			</input>
+			<output>
+				<port id="2" precision="FP32" names="153,input.23">
+					<dim>-1</dim>
+					<dim>-1</dim>
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="68" name="__module.decoder.mid_block.attentions.0/aten::transpose/ScatterElementsUpdate_7" type="Const" version="opset1">
+			<data element_type="i32" shape="3" offset="18962536" size="12" />
+			<output>
+				<port id="0" precision="I32">
+					<dim>3</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="69" name="__module.decoder.mid_block.attentions.0/aten::transpose/Transpose_7" type="Transpose" version="opset1">
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>-1</dim>
+					<dim>512</dim>
+				</port>
+				<port id="1" precision="I32">
+					<dim>3</dim>
+				</port>
+			</input>
+			<output>
+				<port id="2" precision="FP32" names="155">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="70" name="ShapeOf_243372" type="ShapeOf" version="opset3">
+			<data output_type="i64" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</input>
+			<output>
+				<port id="1" precision="I64">
+					<dim>4</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="71" name="Constant_243373" type="Const" version="opset1">
+			<data element_type="i64" shape="1" offset="23169260" size="8" />
+			<output>
+				<port id="0" precision="I64">
+					<dim>1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="72" name="Constant_243374" type="Const" version="opset1">
+			<data element_type="i64" shape="" offset="23169260" size="8" />
+			<output>
+				<port id="0" precision="I64" />
+			</output>
+		</layer>
+		<layer id="73" name="Gather_243375" type="Gather" version="opset8">
+			<data batch_dims="0" />
+			<input>
+				<port id="0" precision="I64">
+					<dim>4</dim>
+				</port>
+				<port id="1" precision="I64">
+					<dim>1</dim>
+				</port>
+				<port id="2" precision="I64" />
+			</input>
+			<output>
+				<port id="3" precision="I64" names="103,114">
+					<dim>1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="74" name="__module.decoder.mid_block.attentions.0/prim::ListConstruct/Reshape_0" type="Const" version="opset1">
+			<data element_type="i64" shape="1" offset="23169268" size="8" />
+			<output>
+				<port id="0" precision="I64">
+					<dim>1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="75" name="Constant_243500" type="Const" version="opset1">
+			<data element_type="i64" shape="2" offset="23169276" size="16" />
+			<output>
+				<port id="0" precision="I64">
+					<dim>2</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="76" name="Constant_243501" type="Const" version="opset1">
+			<data element_type="i64" shape="" offset="23169260" size="8" />
+			<output>
+				<port id="0" precision="I64" />
+			</output>
+		</layer>
+		<layer id="77" name="Gather_243502" type="Gather" version="opset8">
+			<data batch_dims="0" />
+			<input>
+				<port id="0" precision="I64">
+					<dim>4</dim>
+				</port>
+				<port id="1" precision="I64">
+					<dim>2</dim>
+				</port>
+				<port id="2" precision="I64" />
+			</input>
+			<output>
+				<port id="3" precision="I64">
+					<dim>2</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="78" name="__module.decoder.mid_block.attentions.0/prim::ListConstruct/Concat_5" type="Concat" version="opset1">
+			<data axis="0" />
+			<input>
+				<port id="0" precision="I64">
+					<dim>1</dim>
+				</port>
+				<port id="1" precision="I64">
+					<dim>1</dim>
+				</port>
+				<port id="2" precision="I64">
+					<dim>2</dim>
+				</port>
+			</input>
+			<output>
+				<port id="3" precision="I64" names="156">
+					<dim>4</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="79" name="__module.decoder.mid_block.attentions.0/aten::reshape/Reshape_1" type="Reshape" version="opset1">
+			<data special_zero="false" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+				</port>
+				<port id="1" precision="I64">
+					<dim>4</dim>
+				</port>
+			</input>
+			<output>
+				<port id="2" precision="FP32" names="157,hidden_states.13">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="80" name="__module.decoder.mid_block.attentions.0/aten::add/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</input>
+			<output>
+				<port id="2" precision="FP32" names="158,159,hidden_states.15,input.25">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="81" name="self.decoder.mid_block.resnets.1.norm1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="23169292" size="2048" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.mid_block.resnets.1.norm1.weight">
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="82" name="self.decoder.mid_block.resnets.1.norm1.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="23171340" size="2048" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.mid_block.resnets.1.norm1.bias">
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="83" name="__module.decoder.mid_block.resnets.1.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+			<data num_groups="32" epsilon="9.9999999747524271e-07" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>512</dim>
+				</port>
+				<port id="2" precision="FP32">
+					<dim>512</dim>
+				</port>
+			</input>
+			<output>
+				<port id="3" precision="FP32" names="168,input.27">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="84" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_2" type="Swish" version="opset4">
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</input>
+			<output>
+				<port id="1" precision="FP32" names="169,input.29">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="85" name="self.decoder.mid_block.resnets.1.conv1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512, 3, 3" offset="23173388" size="9437184" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.mid_block.resnets.1.conv1.weight">
+					<dim>512</dim>
+					<dim>512</dim>
+					<dim>3</dim>
+					<dim>3</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="86" name="__module.decoder.mid_block.resnets.1.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>512</dim>
+					<dim>512</dim>
+					<dim>3</dim>
+					<dim>3</dim>
+				</port>
+			</input>
+			<output>
+				<port id="2" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="87" name="__module.decoder.mid_block.resnets.1.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="32610572" size="2048" />
+			<output>
+				<port id="0" precision="FP32">
+					<dim>1</dim>
+					<dim>512</dim>
+					<dim>1</dim>
+					<dim>1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="88" name="__module.decoder.mid_block.resnets.1.conv1/aten::_convolution/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>1</dim>
+					<dim>512</dim>
+					<dim>1</dim>
+					<dim>1</dim>
+				</port>
+			</input>
+			<output>
+				<port id="2" precision="FP32" names="176,input.31">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="89" name="self.decoder.mid_block.resnets.1.norm2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="32612620" size="2048" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.mid_block.resnets.1.norm2.weight">
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="90" name="self.decoder.mid_block.resnets.1.norm2.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="32614668" size="2048" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.mid_block.resnets.1.norm2.bias">
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="91" name="__module.decoder.mid_block.resnets.1.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+			<data num_groups="32" epsilon="9.9999999747524271e-07" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>512</dim>
+				</port>
+				<port id="2" precision="FP32">
+					<dim>512</dim>
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="195,input.39">
+				<port id="3" precision="FP32" names="179,input.33">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="59" name="self.encoder.down_blocks.1.resnets.0.conv2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="256, 256, 3, 3" offset="4285952" size="2359296" />
+		<layer id="92" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_3" type="Swish" version="opset4">
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</input>
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.1.resnets.0.conv2.weight">
-					<dim>256</dim>
-					<dim>256</dim>
+				<port id="1" precision="FP32" names="180,input.35">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="93" name="self.decoder.mid_block.resnets.1.conv2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512, 3, 3" offset="32616716" size="9437184" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.mid_block.resnets.1.conv2.weight">
+					<dim>512</dim>
+					<dim>512</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="60" name="__module.encoder.down_blocks.1.resnets.0.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
+		<layer id="94" name="__module.decoder.mid_block.resnets.1.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>256</dim>
-					<dim>256</dim>
+					<dim>512</dim>
+					<dim>512</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
@@ -1022,155 +1478,155 @@
 			<output>
 				<port id="2" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="61" name="__module.encoder.down_blocks.1.resnets.0.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 256, 1, 1" offset="6645248" size="1024" />
+		<layer id="95" name="__module.decoder.mid_block.resnets.1.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="42053900" size="2048" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="62" name="__module.encoder.down_blocks.1.resnets.0.conv2/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="96" name="__module.decoder.mid_block.resnets.1.conv2/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
 					<dim>1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="203,hidden_states.9">
+				<port id="2" precision="FP32" names="188,hidden_states.17">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="63" name="__module.encoder.down_blocks.1.resnets.0/aten::add/Add" type="Add" version="opset1">
+		<layer id="97" name="__module.decoder.mid_block.resnets.1/aten::add/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="211,212,input.43">
+				<port id="2" precision="FP32" names="189,190,191,input.39,sample">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="64" name="self.encoder.down_blocks.1.resnets.1.norm1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="256" offset="6646272" size="1024" />
+		<layer id="98" name="self.decoder.up_blocks.0.resnets.0.norm1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="42055948" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.1.resnets.1.norm1.weight">
-					<dim>256</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.resnets.0.norm1.weight">
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="65" name="self.encoder.down_blocks.1.resnets.1.norm1.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="256" offset="6647296" size="1024" />
+		<layer id="99" name="self.decoder.up_blocks.0.resnets.0.norm1.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="42057996" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.1.resnets.1.norm1.bias">
-					<dim>256</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.resnets.0.norm1.bias">
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="66" name="__module.encoder.down_blocks.1.resnets.1.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+		<layer id="100" name="__module.decoder.up_blocks.0.resnets.0.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
 			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>256</dim>
+					<dim>512</dim>
 				</port>
 				<port id="2" precision="FP32">
-					<dim>256</dim>
+					<dim>512</dim>
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="220,input.45">
+				<port id="3" precision="FP32" names="207,input.41">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="67" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_6" type="Swish" version="opset4">
+		<layer id="101" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_4" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="221,input.47">
+				<port id="1" precision="FP32" names="208,input.43">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="68" name="self.encoder.down_blocks.1.resnets.1.conv1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="256, 256, 3, 3" offset="6648320" size="2359296" />
+		<layer id="102" name="self.decoder.up_blocks.0.resnets.0.conv1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512, 3, 3" offset="42060044" size="9437184" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.1.resnets.1.conv1.weight">
-					<dim>256</dim>
-					<dim>256</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.resnets.0.conv1.weight">
+					<dim>512</dim>
+					<dim>512</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="69" name="__module.encoder.down_blocks.1.resnets.1.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
+		<layer id="103" name="__module.decoder.up_blocks.0.resnets.0.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>256</dim>
-					<dim>256</dim>
+					<dim>512</dim>
+					<dim>512</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
@@ -1178,130 +1634,286 @@
 			<output>
 				<port id="2" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="70" name="__module.encoder.down_blocks.1.resnets.1.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 256, 1, 1" offset="9007616" size="1024" />
+		<layer id="104" name="__module.decoder.up_blocks.0.resnets.0.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="51497228" size="2048" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="71" name="__module.encoder.down_blocks.1.resnets.1.conv1/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="105" name="__module.decoder.up_blocks.0.resnets.0.conv1/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
 					<dim>1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="228,input.49">
+				<port id="2" precision="FP32" names="215,input.45">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="72" name="self.encoder.down_blocks.1.resnets.1.norm2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="256" offset="9008640" size="1024" />
+		<layer id="106" name="self.decoder.up_blocks.0.resnets.0.norm2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="51499276" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.1.resnets.1.norm2.weight">
-					<dim>256</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.resnets.0.norm2.weight">
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="73" name="self.encoder.down_blocks.1.resnets.1.norm2.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="256" offset="9009664" size="1024" />
+		<layer id="107" name="self.decoder.up_blocks.0.resnets.0.norm2.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="51501324" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.1.resnets.1.norm2.bias">
-					<dim>256</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.resnets.0.norm2.bias">
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="74" name="__module.encoder.down_blocks.1.resnets.1.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+		<layer id="108" name="__module.decoder.up_blocks.0.resnets.0.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
 			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>256</dim>
+					<dim>512</dim>
 				</port>
 				<port id="2" precision="FP32">
-					<dim>256</dim>
+					<dim>512</dim>
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="231,input.51">
+				<port id="3" precision="FP32" names="218,input.47">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="75" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_7" type="Swish" version="opset4">
+		<layer id="109" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_5" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="232,input.53">
+				<port id="1" precision="FP32" names="219,input.49">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="76" name="self.encoder.down_blocks.1.resnets.1.conv2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="256, 256, 3, 3" offset="9010688" size="2359296" />
+		<layer id="110" name="self.decoder.up_blocks.0.resnets.0.conv2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512, 3, 3" offset="51503372" size="9437184" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.1.resnets.1.conv2.weight">
-					<dim>256</dim>
-					<dim>256</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.resnets.0.conv2.weight">
+					<dim>512</dim>
+					<dim>512</dim>
+					<dim>3</dim>
+					<dim>3</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="111" name="__module.decoder.up_blocks.0.resnets.0.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>512</dim>
+					<dim>512</dim>
+					<dim>3</dim>
+					<dim>3</dim>
+				</port>
+			</input>
+			<output>
+				<port id="2" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="112" name="__module.decoder.up_blocks.0.resnets.0.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="60940556" size="2048" />
+			<output>
+				<port id="0" precision="FP32">
+					<dim>1</dim>
+					<dim>512</dim>
+					<dim>1</dim>
+					<dim>1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="113" name="__module.decoder.up_blocks.0.resnets.0.conv2/aten::_convolution/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>1</dim>
+					<dim>512</dim>
+					<dim>1</dim>
+					<dim>1</dim>
+				</port>
+			</input>
+			<output>
+				<port id="2" precision="FP32" names="227,hidden_states.19">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="114" name="__module.decoder.up_blocks.0.resnets.0/aten::add/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</input>
+			<output>
+				<port id="2" precision="FP32" names="228,229,input.53">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="115" name="self.decoder.up_blocks.0.resnets.1.norm1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="60942604" size="2048" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.resnets.1.norm1.weight">
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="116" name="self.decoder.up_blocks.0.resnets.1.norm1.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="60944652" size="2048" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.resnets.1.norm1.bias">
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="117" name="__module.decoder.up_blocks.0.resnets.1.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+			<data num_groups="32" epsilon="9.9999999747524271e-07" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>512</dim>
+				</port>
+				<port id="2" precision="FP32">
+					<dim>512</dim>
+				</port>
+			</input>
+			<output>
+				<port id="3" precision="FP32" names="237,input.55">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="118" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_6" type="Swish" version="opset4">
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</input>
+			<output>
+				<port id="1" precision="FP32" names="238,input.57">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="119" name="self.decoder.up_blocks.0.resnets.1.conv1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512, 3, 3" offset="60946700" size="9437184" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.resnets.1.conv1.weight">
+					<dim>512</dim>
+					<dim>512</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="77" name="__module.encoder.down_blocks.1.resnets.1.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
+		<layer id="120" name="__module.decoder.up_blocks.0.resnets.1.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>256</dim>
-					<dim>256</dim>
+					<dim>512</dim>
+					<dim>512</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
@@ -1309,174 +1921,172 @@
 			<output>
 				<port id="2" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="78" name="__module.encoder.down_blocks.1.resnets.1.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 256, 1, 1" offset="11369984" size="1024" />
+		<layer id="121" name="__module.decoder.up_blocks.0.resnets.1.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="70383884" size="2048" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="79" name="__module.encoder.down_blocks.1.resnets.1.conv2/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="122" name="__module.decoder.up_blocks.0.resnets.1.conv1/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
 					<dim>1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="240,hidden_states.11">
+				<port id="2" precision="FP32" names="245,input.59">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="80" name="__module.encoder.down_blocks.1.resnets.1/aten::add/Add" type="Add" version="opset1">
-			<data auto_broadcast="numpy" />
+		<layer id="123" name="self.decoder.up_blocks.0.resnets.1.norm2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="70385932" size="2048" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.resnets.1.norm2.weight">
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="124" name="self.decoder.up_blocks.0.resnets.1.norm2.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="70387980" size="2048" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.resnets.1.norm2.bias">
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="125" name="__module.decoder.up_blocks.0.resnets.1.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>-1</dim>
-					<dim>256</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
+					<dim>512</dim>
+				</port>
+				<port id="2" precision="FP32">
+					<dim>512</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="241,242,hidden_states.13">
+				<port id="3" precision="FP32" names="248,input.61">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="81" name="self.encoder.down_blocks.1.downsamplers.0.conv.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="256, 256, 3, 3" offset="11371008" size="2359296" />
-			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.1.downsamplers.0.conv.weight">
-					<dim>256</dim>
-					<dim>256</dim>
-					<dim>3</dim>
-					<dim>3</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="82" name="__module.encoder.down_blocks.1.downsamplers.0.conv/aten::_convolution/Convolution" type="Convolution" version="opset1">
-			<data strides="2, 2" dilations="1, 1" pads_begin="0, 0" pads_end="1, 1" auto_pad="explicit" />
+		<layer id="126" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_7" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
-				<port id="1" precision="FP32">
-					<dim>256</dim>
-					<dim>256</dim>
-					<dim>3</dim>
-					<dim>3</dim>
-				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32">
+				<port id="1" precision="FP32" names="249,input.63">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="83" name="__module.encoder.down_blocks.1.downsamplers.0.conv/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 256, 1, 1" offset="13730304" size="1024" />
+		<layer id="127" name="self.decoder.up_blocks.0.resnets.1.conv2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512, 3, 3" offset="70390028" size="9437184" />
 			<output>
-				<port id="0" precision="FP32">
-					<dim>1</dim>
-					<dim>256</dim>
-					<dim>1</dim>
-					<dim>1</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.resnets.1.conv2.weight">
+					<dim>512</dim>
+					<dim>512</dim>
+					<dim>3</dim>
+					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="84" name="__module.encoder.down_blocks.1.downsamplers.0.conv/aten::_convolution/Add" type="Add" version="opset1">
-			<data auto_broadcast="numpy" />
+		<layer id="128" name="__module.decoder.up_blocks.0.resnets.1.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>1</dim>
-					<dim>256</dim>
-					<dim>1</dim>
-					<dim>1</dim>
+					<dim>512</dim>
+					<dim>512</dim>
+					<dim>3</dim>
+					<dim>3</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="252,input.57">
+				<port id="2" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="85" name="self.encoder.down_blocks.2.resnets.0.conv_shortcut.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512, 256, 1, 1" offset="13731328" size="524288" />
+		<layer id="129" name="__module.decoder.up_blocks.0.resnets.1.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="79827212" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.2.resnets.0.conv_shortcut.weight">
+				<port id="0" precision="FP32">
+					<dim>1</dim>
 					<dim>512</dim>
-					<dim>256</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="86" name="__module.encoder.down_blocks.2.resnets.0.conv_shortcut/aten::_convolution/Convolution" type="Convolution" version="opset1">
-			<data strides="1, 1" dilations="1, 1" pads_begin="0, 0" pads_end="0, 0" auto_pad="explicit" />
+		<layer id="130" name="__module.decoder.up_blocks.0.resnets.1.conv2/aten::_convolution/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
+					<dim>1</dim>
 					<dim>512</dim>
-					<dim>256</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32">
+				<port id="2" precision="FP32" names="257,hidden_states.21">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -1484,18 +2094,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="87" name="__module.encoder.down_blocks.2.resnets.0.conv_shortcut/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 512, 1, 1" offset="14255616" size="2048" />
-			<output>
-				<port id="0" precision="FP32">
-					<dim>1</dim>
-					<dim>512</dim>
-					<dim>1</dim>
-					<dim>1</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="88" name="__module.encoder.down_blocks.2.resnets.0.conv_shortcut/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="131" name="__module.decoder.up_blocks.0.resnets.1/aten::add/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
@@ -1505,14 +2104,14 @@
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>1</dim>
+					<dim>-1</dim>
 					<dim>512</dim>
-					<dim>1</dim>
-					<dim>1</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="294,input_tensor">
+				<port id="2" precision="FP32" names="258,259,input.67">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -1520,88 +2119,88 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="89" name="self.encoder.down_blocks.2.resnets.0.norm1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="256" offset="14257664" size="1024" />
+		<layer id="132" name="self.decoder.up_blocks.0.resnets.2.norm1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="79829260" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.2.resnets.0.norm1.weight">
-					<dim>256</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.resnets.2.norm1.weight">
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="90" name="self.encoder.down_blocks.2.resnets.0.norm1.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="256" offset="14258688" size="1024" />
+		<layer id="133" name="self.decoder.up_blocks.0.resnets.2.norm1.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="79831308" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.2.resnets.0.norm1.bias">
-					<dim>256</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.resnets.2.norm1.bias">
+					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="91" name="__module.encoder.down_blocks.2.resnets.0.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+		<layer id="134" name="__module.decoder.up_blocks.0.resnets.2.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
 			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>256</dim>
+					<dim>512</dim>
 				</port>
 				<port id="2" precision="FP32">
-					<dim>256</dim>
+					<dim>512</dim>
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="267,input.59">
+				<port id="3" precision="FP32" names="267,input.69">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="92" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_8" type="Swish" version="opset4">
+		<layer id="135" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_8" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="268,input.61">
+				<port id="1" precision="FP32" names="268,input.71">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="93" name="self.encoder.down_blocks.2.resnets.0.conv1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512, 256, 3, 3" offset="14259712" size="4718592" />
+		<layer id="136" name="self.decoder.up_blocks.0.resnets.2.conv1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512, 3, 3" offset="79833356" size="9437184" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.2.resnets.0.conv1.weight">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.resnets.2.conv1.weight">
+					<dim>512</dim>
 					<dim>512</dim>
-					<dim>256</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="94" name="__module.encoder.down_blocks.2.resnets.0.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
+		<layer id="137" name="__module.decoder.up_blocks.0.resnets.2.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
 					<dim>512</dim>
-					<dim>256</dim>
+					<dim>512</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
@@ -1615,8 +2214,8 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="95" name="__module.encoder.down_blocks.2.resnets.0.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 512, 1, 1" offset="18978304" size="2048" />
+		<layer id="138" name="__module.decoder.up_blocks.0.resnets.2.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="89270540" size="2048" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
@@ -1626,7 +2225,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="96" name="__module.encoder.down_blocks.2.resnets.0.conv1/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="139" name="__module.decoder.up_blocks.0.resnets.2.conv1/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
@@ -1643,7 +2242,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="275,input.63">
+				<port id="2" precision="FP32" names="275,input.73">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -1651,23 +2250,23 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="97" name="self.encoder.down_blocks.2.resnets.0.norm2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="18980352" size="2048" />
+		<layer id="140" name="self.decoder.up_blocks.0.resnets.2.norm2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="89272588" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.2.resnets.0.norm2.weight">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.resnets.2.norm2.weight">
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="98" name="self.encoder.down_blocks.2.resnets.0.norm2.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="18982400" size="2048" />
+		<layer id="141" name="self.decoder.up_blocks.0.resnets.2.norm2.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="89274636" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.2.resnets.0.norm2.bias">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.resnets.2.norm2.bias">
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="99" name="__module.encoder.down_blocks.2.resnets.0.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+		<layer id="142" name="__module.decoder.up_blocks.0.resnets.2.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
 			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
@@ -1684,7 +2283,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="278,input.65">
+				<port id="3" precision="FP32" names="278,input.75">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -1692,7 +2291,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="100" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_9" type="Swish" version="opset4">
+		<layer id="143" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_9" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
@@ -1702,7 +2301,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="279,input.67">
+				<port id="1" precision="FP32" names="279,input.77">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -1710,10 +2309,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="101" name="self.encoder.down_blocks.2.resnets.0.conv2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512, 512, 3, 3" offset="18984448" size="9437184" />
+		<layer id="144" name="self.decoder.up_blocks.0.resnets.2.conv2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512, 3, 3" offset="89276684" size="9437184" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.2.resnets.0.conv2.weight">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.resnets.2.conv2.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -1721,7 +2320,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="102" name="__module.encoder.down_blocks.2.resnets.0.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
+		<layer id="145" name="__module.decoder.up_blocks.0.resnets.2.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
@@ -1746,8 +2345,8 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="103" name="__module.encoder.down_blocks.2.resnets.0.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 512, 1, 1" offset="28421632" size="2048" />
+		<layer id="146" name="__module.decoder.up_blocks.0.resnets.2.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="98713868" size="2048" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
@@ -1757,7 +2356,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="104" name="__module.encoder.down_blocks.2.resnets.0.conv2/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="147" name="__module.decoder.up_blocks.0.resnets.2.conv2/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
@@ -1774,7 +2373,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="287,hidden_states.17">
+				<port id="2" precision="FP32" names="287,hidden_states.23">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -1782,7 +2381,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="105" name="__module.encoder.down_blocks.2.resnets.0/aten::add/Add" type="Add" version="opset1">
+		<layer id="148" name="__module.decoder.up_blocks.0.resnets.2/aten::add/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
@@ -1799,7 +2398,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="295,296,input.71">
+				<port id="2" precision="FP32" names="288,289,hidden_states.25">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -1807,24 +2406,24 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="106" name="self.encoder.down_blocks.2.resnets.1.norm1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="28423680" size="2048" />
+		<layer id="149" name="__module.decoder.up_blocks.0.upsamplers.0/aten::upsample_nearest2d/Multiply" type="Const" version="opset1">
+			<data element_type="f32" shape="2" offset="98715916" size="8" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.2.resnets.1.norm1.weight">
-					<dim>512</dim>
+				<port id="0" precision="FP32">
+					<dim>2</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="107" name="self.encoder.down_blocks.2.resnets.1.norm1.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="28425728" size="2048" />
+		<layer id="150" name="Constant_234855" type="Const" version="opset1">
+			<data element_type="i32" shape="2" offset="98715924" size="8" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.2.resnets.1.norm1.bias">
-					<dim>512</dim>
+				<port id="0" precision="I32">
+					<dim>2</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="108" name="__module.encoder.down_blocks.2.resnets.1.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
-			<data num_groups="32" epsilon="9.9999999747524271e-07" />
+		<layer id="151" name="__module.decoder.up_blocks.0.upsamplers.0/aten::upsample_nearest2d/Interpolate" type="Interpolate" version="opset11">
+			<data mode="nearest" shape_calculation_mode="scales" coordinate_transformation_mode="asymmetric" nearest_mode="floor" antialias="false" pads_begin="0, 0, 0, 0" pads_end="0, 0, 0, 0" cube_coeff="-0.75" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
@@ -1833,32 +2432,14 @@
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>512</dim>
-				</port>
-				<port id="2" precision="FP32">
-					<dim>512</dim>
-				</port>
-			</input>
-			<output>
-				<port id="3" precision="FP32" names="304,input.73">
-					<dim>-1</dim>
-					<dim>512</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
+					<dim>2</dim>
 				</port>
-			</output>
-		</layer>
-		<layer id="109" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_10" type="Swish" version="opset4">
-			<input>
-				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>512</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
+				<port id="2" precision="I32">
+					<dim>2</dim>
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="305,input.75">
+				<port id="3" precision="FP32" names="292,input.81">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -1866,10 +2447,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="110" name="self.encoder.down_blocks.2.resnets.1.conv1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512, 512, 3, 3" offset="28427776" size="9437184" />
+		<layer id="152" name="self.decoder.up_blocks.0.upsamplers.0.conv.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512, 3, 3" offset="98715932" size="9437184" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.2.resnets.1.conv1.weight">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.0.upsamplers.0.conv.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -1877,7 +2458,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="111" name="__module.encoder.down_blocks.2.resnets.1.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
+		<layer id="153" name="__module.decoder.up_blocks.0.upsamplers.0.conv/aten::_convolution/Convolution" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
@@ -1902,8 +2483,8 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="112" name="__module.encoder.down_blocks.2.resnets.1.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 512, 1, 1" offset="37864960" size="2048" />
+		<layer id="154" name="__module.decoder.up_blocks.0.upsamplers.0.conv/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="108153116" size="2048" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
@@ -1913,7 +2494,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="113" name="__module.encoder.down_blocks.2.resnets.1.conv1/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="155" name="__module.decoder.up_blocks.0.upsamplers.0.conv/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
@@ -1930,7 +2511,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="312,input.77">
+				<port id="2" precision="FP32" names="299,input.83">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -1938,23 +2519,23 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="114" name="self.encoder.down_blocks.2.resnets.1.norm2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="37867008" size="2048" />
+		<layer id="156" name="self.decoder.up_blocks.1.resnets.0.norm1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="108155164" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.2.resnets.1.norm2.weight">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.resnets.0.norm1.weight">
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="115" name="self.encoder.down_blocks.2.resnets.1.norm2.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="37869056" size="2048" />
+		<layer id="157" name="self.decoder.up_blocks.1.resnets.0.norm1.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="108157212" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.2.resnets.1.norm2.bias">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.resnets.0.norm1.bias">
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="116" name="__module.encoder.down_blocks.2.resnets.1.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+		<layer id="158" name="__module.decoder.up_blocks.1.resnets.0.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
 			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
@@ -1971,7 +2552,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="315,input.79">
+				<port id="3" precision="FP32" names="315,input.85">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -1979,7 +2560,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="117" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_11" type="Swish" version="opset4">
+		<layer id="159" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_10" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
@@ -1989,7 +2570,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="316,input.81">
+				<port id="1" precision="FP32" names="316,input.87">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -1997,10 +2578,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="118" name="self.encoder.down_blocks.2.resnets.1.conv2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512, 512, 3, 3" offset="37871104" size="9437184" />
+		<layer id="160" name="self.decoder.up_blocks.1.resnets.0.conv1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512, 3, 3" offset="108159260" size="9437184" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.2.resnets.1.conv2.weight">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.resnets.0.conv1.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -2008,7 +2589,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="119" name="__module.encoder.down_blocks.2.resnets.1.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
+		<layer id="161" name="__module.decoder.up_blocks.1.resnets.0.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2033,8 +2614,8 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="120" name="__module.encoder.down_blocks.2.resnets.1.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 512, 1, 1" offset="47308288" size="2048" />
+		<layer id="162" name="__module.decoder.up_blocks.1.resnets.0.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="117596444" size="2048" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
@@ -2044,7 +2625,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="121" name="__module.encoder.down_blocks.2.resnets.1.conv2/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="163" name="__module.decoder.up_blocks.1.resnets.0.conv1/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2061,7 +2642,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="324,hidden_states.19">
+				<port id="2" precision="FP32" names="323,input.89">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2069,8 +2650,24 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="122" name="__module.encoder.down_blocks.2.resnets.1/aten::add/Add" type="Add" version="opset1">
-			<data auto_broadcast="numpy" />
+		<layer id="164" name="self.decoder.up_blocks.1.resnets.0.norm2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="117598492" size="2048" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.resnets.0.norm2.weight">
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="165" name="self.decoder.up_blocks.1.resnets.0.norm2.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="117600540" size="2048" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.resnets.0.norm2.bias">
+					<dim>512</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="166" name="__module.decoder.up_blocks.1.resnets.0.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
@@ -2079,6 +2676,24 @@
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
+					<dim>512</dim>
+				</port>
+				<port id="2" precision="FP32">
+					<dim>512</dim>
+				</port>
+			</input>
+			<output>
+				<port id="3" precision="FP32" names="326,input.91">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="167" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_11" type="Swish" version="opset4">
+			<input>
+				<port id="0" precision="FP32">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2086,7 +2701,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="325,326,hidden_states.21">
+				<port id="1" precision="FP32" names="327,input.93">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2094,10 +2709,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="123" name="self.encoder.down_blocks.2.downsamplers.0.conv.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512, 512, 3, 3" offset="47310336" size="9437184" />
+		<layer id="168" name="self.decoder.up_blocks.1.resnets.0.conv2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512, 3, 3" offset="117602588" size="9437184" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.2.downsamplers.0.conv.weight">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.resnets.0.conv2.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -2105,8 +2720,8 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="124" name="__module.encoder.down_blocks.2.downsamplers.0.conv/aten::_convolution/Convolution" type="Convolution" version="opset1">
-			<data strides="2, 2" dilations="1, 1" pads_begin="0, 0" pads_end="1, 1" auto_pad="explicit" />
+		<layer id="169" name="__module.decoder.up_blocks.1.resnets.0.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
@@ -2130,8 +2745,8 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="125" name="__module.encoder.down_blocks.2.downsamplers.0.conv/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 512, 1, 1" offset="56747520" size="2048" />
+		<layer id="170" name="__module.decoder.up_blocks.1.resnets.0.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="127039772" size="2048" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
@@ -2141,7 +2756,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="126" name="__module.encoder.down_blocks.2.downsamplers.0.conv/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="171" name="__module.decoder.up_blocks.1.resnets.0.conv2/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2158,7 +2773,32 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="336,input.85">
+				<port id="2" precision="FP32" names="335,hidden_states.27">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="172" name="__module.decoder.up_blocks.1.resnets.0/aten::add/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</input>
+			<output>
+				<port id="2" precision="FP32" names="336,337,input.97">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2166,23 +2806,23 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="127" name="self.encoder.down_blocks.3.resnets.0.norm1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="56749568" size="2048" />
+		<layer id="173" name="self.decoder.up_blocks.1.resnets.1.norm1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="127041820" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.3.resnets.0.norm1.weight">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.resnets.1.norm1.weight">
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="128" name="self.encoder.down_blocks.3.resnets.0.norm1.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="56751616" size="2048" />
+		<layer id="174" name="self.decoder.up_blocks.1.resnets.1.norm1.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="127043868" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.3.resnets.0.norm1.bias">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.resnets.1.norm1.bias">
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="129" name="__module.encoder.down_blocks.3.resnets.0.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+		<layer id="175" name="__module.decoder.up_blocks.1.resnets.1.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
 			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2199,7 +2839,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="348,input.87">
+				<port id="3" precision="FP32" names="345,input.99">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2207,7 +2847,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="130" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_12" type="Swish" version="opset4">
+		<layer id="176" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_12" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
@@ -2217,7 +2857,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="349,input.89">
+				<port id="1" precision="FP32" names="346,input.101">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2225,10 +2865,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="131" name="self.encoder.down_blocks.3.resnets.0.conv1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512, 512, 3, 3" offset="56753664" size="9437184" />
+		<layer id="177" name="self.decoder.up_blocks.1.resnets.1.conv1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512, 3, 3" offset="127045916" size="9437184" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.3.resnets.0.conv1.weight">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.resnets.1.conv1.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -2236,7 +2876,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="132" name="__module.encoder.down_blocks.3.resnets.0.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
+		<layer id="178" name="__module.decoder.up_blocks.1.resnets.1.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2261,8 +2901,8 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="133" name="__module.encoder.down_blocks.3.resnets.0.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 512, 1, 1" offset="66190848" size="2048" />
+		<layer id="179" name="__module.decoder.up_blocks.1.resnets.1.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="136483100" size="2048" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
@@ -2272,7 +2912,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="134" name="__module.encoder.down_blocks.3.resnets.0.conv1/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="180" name="__module.decoder.up_blocks.1.resnets.1.conv1/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2289,7 +2929,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="356,input.91">
+				<port id="2" precision="FP32" names="353,input.103">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2297,23 +2937,23 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="135" name="self.encoder.down_blocks.3.resnets.0.norm2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="66192896" size="2048" />
+		<layer id="181" name="self.decoder.up_blocks.1.resnets.1.norm2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="136485148" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.3.resnets.0.norm2.weight">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.resnets.1.norm2.weight">
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="136" name="self.encoder.down_blocks.3.resnets.0.norm2.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="66194944" size="2048" />
+		<layer id="182" name="self.decoder.up_blocks.1.resnets.1.norm2.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="136487196" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.3.resnets.0.norm2.bias">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.resnets.1.norm2.bias">
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="137" name="__module.encoder.down_blocks.3.resnets.0.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+		<layer id="183" name="__module.decoder.up_blocks.1.resnets.1.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
 			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2330,7 +2970,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="359,input.93">
+				<port id="3" precision="FP32" names="356,input.105">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2338,7 +2978,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="138" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_13" type="Swish" version="opset4">
+		<layer id="184" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_13" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
@@ -2348,7 +2988,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="360,input.95">
+				<port id="1" precision="FP32" names="357,input.107">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2356,10 +2996,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="139" name="self.encoder.down_blocks.3.resnets.0.conv2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512, 512, 3, 3" offset="66196992" size="9437184" />
+		<layer id="185" name="self.decoder.up_blocks.1.resnets.1.conv2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512, 3, 3" offset="136489244" size="9437184" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.3.resnets.0.conv2.weight">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.resnets.1.conv2.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -2367,7 +3007,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="140" name="__module.encoder.down_blocks.3.resnets.0.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
+		<layer id="186" name="__module.decoder.up_blocks.1.resnets.1.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2392,8 +3032,8 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="141" name="__module.encoder.down_blocks.3.resnets.0.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 512, 1, 1" offset="75634176" size="2048" />
+		<layer id="187" name="__module.decoder.up_blocks.1.resnets.1.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="145926428" size="2048" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
@@ -2403,7 +3043,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="142" name="__module.encoder.down_blocks.3.resnets.0.conv2/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="188" name="__module.decoder.up_blocks.1.resnets.1.conv2/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2420,7 +3060,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="368,hidden_states.25">
+				<port id="2" precision="FP32" names="365,hidden_states.29">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2428,7 +3068,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="143" name="__module.encoder.down_blocks.3.resnets.0/aten::add/Add" type="Add" version="opset1">
+		<layer id="189" name="__module.decoder.up_blocks.1.resnets.1/aten::add/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2445,7 +3085,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="369,370,input.99">
+				<port id="2" precision="FP32" names="366,367,input.111">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2453,23 +3093,23 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="144" name="self.encoder.down_blocks.3.resnets.1.norm1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="75636224" size="2048" />
+		<layer id="190" name="self.decoder.up_blocks.1.resnets.2.norm1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="145928476" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.3.resnets.1.norm1.weight">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.resnets.2.norm1.weight">
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="145" name="self.encoder.down_blocks.3.resnets.1.norm1.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="75638272" size="2048" />
+		<layer id="191" name="self.decoder.up_blocks.1.resnets.2.norm1.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="145930524" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.3.resnets.1.norm1.bias">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.resnets.2.norm1.bias">
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="146" name="__module.encoder.down_blocks.3.resnets.1.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+		<layer id="192" name="__module.decoder.up_blocks.1.resnets.2.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
 			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2486,7 +3126,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="378,input.101">
+				<port id="3" precision="FP32" names="375,input.113">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2494,7 +3134,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="147" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_14" type="Swish" version="opset4">
+		<layer id="193" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_14" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
@@ -2504,7 +3144,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="379,input.103">
+				<port id="1" precision="FP32" names="376,input.115">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2512,10 +3152,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="148" name="self.encoder.down_blocks.3.resnets.1.conv1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512, 512, 3, 3" offset="75640320" size="9437184" />
+		<layer id="194" name="self.decoder.up_blocks.1.resnets.2.conv1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512, 3, 3" offset="145932572" size="9437184" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.3.resnets.1.conv1.weight">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.resnets.2.conv1.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -2523,7 +3163,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="149" name="__module.encoder.down_blocks.3.resnets.1.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
+		<layer id="195" name="__module.decoder.up_blocks.1.resnets.2.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2548,8 +3188,8 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="150" name="__module.encoder.down_blocks.3.resnets.1.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 512, 1, 1" offset="85077504" size="2048" />
+		<layer id="196" name="__module.decoder.up_blocks.1.resnets.2.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="155369756" size="2048" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
@@ -2559,7 +3199,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="151" name="__module.encoder.down_blocks.3.resnets.1.conv1/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="197" name="__module.decoder.up_blocks.1.resnets.2.conv1/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2576,7 +3216,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="386,input.105">
+				<port id="2" precision="FP32" names="383,input.117">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2584,23 +3224,23 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="152" name="self.encoder.down_blocks.3.resnets.1.norm2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="85079552" size="2048" />
+		<layer id="198" name="self.decoder.up_blocks.1.resnets.2.norm2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="155371804" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.3.resnets.1.norm2.weight">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.resnets.2.norm2.weight">
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="153" name="self.encoder.down_blocks.3.resnets.1.norm2.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="85081600" size="2048" />
+		<layer id="199" name="self.decoder.up_blocks.1.resnets.2.norm2.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="155373852" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.3.resnets.1.norm2.bias">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.resnets.2.norm2.bias">
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="154" name="__module.encoder.down_blocks.3.resnets.1.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+		<layer id="200" name="__module.decoder.up_blocks.1.resnets.2.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
 			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2617,7 +3257,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="389,input.107">
+				<port id="3" precision="FP32" names="386,input.119">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2625,7 +3265,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="155" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_15" type="Swish" version="opset4">
+		<layer id="201" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_15" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
@@ -2635,7 +3275,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="390,input.109">
+				<port id="1" precision="FP32" names="387,input.121">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2643,10 +3283,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="156" name="self.encoder.down_blocks.3.resnets.1.conv2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512, 512, 3, 3" offset="85083648" size="9437184" />
+		<layer id="202" name="self.decoder.up_blocks.1.resnets.2.conv2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512, 3, 3" offset="155375900" size="9437184" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.down_blocks.3.resnets.1.conv2.weight">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.resnets.2.conv2.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -2654,7 +3294,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="157" name="__module.encoder.down_blocks.3.resnets.1.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
+		<layer id="203" name="__module.decoder.up_blocks.1.resnets.2.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2679,8 +3319,8 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="158" name="__module.encoder.down_blocks.3.resnets.1.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 512, 1, 1" offset="94520832" size="2048" />
+		<layer id="204" name="__module.decoder.up_blocks.1.resnets.2.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="164813084" size="2048" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
@@ -2690,7 +3330,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="159" name="__module.encoder.down_blocks.3.resnets.1.conv2/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="205" name="__module.decoder.up_blocks.1.resnets.2.conv2/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2707,7 +3347,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="398,hidden_states.27">
+				<port id="2" precision="FP32" names="395,hidden_states.31">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2715,7 +3355,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="160" name="__module.encoder.down_blocks.3.resnets.1/aten::add/Add" type="Add" version="opset1">
+		<layer id="206" name="__module.decoder.up_blocks.1.resnets.2/aten::add/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2732,7 +3372,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="399,400,input.113">
+				<port id="2" precision="FP32" names="396,397,hidden_states.33">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2740,24 +3380,60 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="161" name="self.encoder.mid_block.resnets.0.norm1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="94522880" size="2048" />
+		<layer id="207" name="__module.decoder.up_blocks.1.upsamplers.0/aten::upsample_nearest2d/Multiply" type="Const" version="opset1">
+			<data element_type="f32" shape="2" offset="98715916" size="8" />
+			<output>
+				<port id="0" precision="FP32">
+					<dim>2</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="208" name="Constant_235287" type="Const" version="opset1">
+			<data element_type="i32" shape="2" offset="98715924" size="8" />
+			<output>
+				<port id="0" precision="I32">
+					<dim>2</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="209" name="__module.decoder.up_blocks.1.upsamplers.0/aten::upsample_nearest2d/Interpolate" type="Interpolate" version="opset11">
+			<data mode="nearest" shape_calculation_mode="scales" coordinate_transformation_mode="asymmetric" nearest_mode="floor" antialias="false" pads_begin="0, 0, 0, 0" pads_end="0, 0, 0, 0" cube_coeff="-0.75" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>2</dim>
+				</port>
+				<port id="2" precision="I32">
+					<dim>2</dim>
+				</port>
+			</input>
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.mid_block.resnets.0.norm1.weight">
+				<port id="3" precision="FP32" names="400,input.125">
+					<dim>-1</dim>
 					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="162" name="self.encoder.mid_block.resnets.0.norm1.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="94524928" size="2048" />
+		<layer id="210" name="self.decoder.up_blocks.1.upsamplers.0.conv.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512, 512, 3, 3" offset="164815132" size="9437184" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.mid_block.resnets.0.norm1.bias">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.1.upsamplers.0.conv.weight">
 					<dim>512</dim>
+					<dim>512</dim>
+					<dim>3</dim>
+					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="163" name="__module.encoder.mid_block.resnets.0.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
-			<data num_groups="32" epsilon="9.9999999747524271e-07" />
+		<layer id="211" name="__module.decoder.up_blocks.1.upsamplers.0.conv/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
@@ -2767,21 +3443,33 @@
 				</port>
 				<port id="1" precision="FP32">
 					<dim>512</dim>
+					<dim>512</dim>
+					<dim>3</dim>
+					<dim>3</dim>
 				</port>
+			</input>
+			<output>
 				<port id="2" precision="FP32">
+					<dim>-1</dim>
 					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
-			</input>
+			</output>
+		</layer>
+		<layer id="212" name="__module.decoder.up_blocks.1.upsamplers.0.conv/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 512, 1, 1" offset="174252316" size="2048" />
 			<output>
-				<port id="3" precision="FP32" names="414,input.115">
-					<dim>-1</dim>
+				<port id="0" precision="FP32">
+					<dim>1</dim>
 					<dim>512</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
+					<dim>1</dim>
+					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="164" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_16" type="Swish" version="opset4">
+		<layer id="213" name="__module.decoder.up_blocks.1.upsamplers.0.conv/aten::_convolution/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
@@ -2789,9 +3477,15 @@
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
+				<port id="1" precision="FP32">
+					<dim>1</dim>
+					<dim>512</dim>
+					<dim>1</dim>
+					<dim>1</dim>
+				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="415,input.117">
+				<port id="2" precision="FP32" names="407,input.127">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2799,19 +3493,19 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="165" name="self.encoder.mid_block.resnets.0.conv1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512, 512, 3, 3" offset="94526976" size="9437184" />
+		<layer id="214" name="self.decoder.up_blocks.2.resnets.0.conv_shortcut.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="256, 512, 1, 1" offset="174254364" size="524288" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.mid_block.resnets.0.conv1.weight">
-					<dim>512</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.0.conv_shortcut.weight">
+					<dim>256</dim>
 					<dim>512</dim>
-					<dim>3</dim>
-					<dim>3</dim>
+					<dim>1</dim>
+					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="166" name="__module.encoder.mid_block.resnets.0.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
-			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
+		<layer id="215" name="__module.decoder.up_blocks.2.resnets.0.conv_shortcut/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="0, 0" pads_end="0, 0" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
@@ -2820,74 +3514,74 @@
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
+					<dim>256</dim>
 					<dim>512</dim>
-					<dim>512</dim>
-					<dim>3</dim>
-					<dim>3</dim>
+					<dim>1</dim>
+					<dim>1</dim>
 				</port>
 			</input>
 			<output>
 				<port id="2" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="167" name="__module.encoder.mid_block.resnets.0.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 512, 1, 1" offset="103964160" size="2048" />
+		<layer id="216" name="__module.decoder.up_blocks.2.resnets.0.conv_shortcut/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 256, 1, 1" offset="174778652" size="1024" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="168" name="__module.encoder.mid_block.resnets.0.conv1/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="217" name="__module.decoder.up_blocks.2.resnets.0.conv_shortcut/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
 					<dim>1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="422,input.119">
+				<port id="2" precision="FP32" names="451,input_tensor.1">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="169" name="self.encoder.mid_block.resnets.0.norm2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="103966208" size="2048" />
+		<layer id="218" name="self.decoder.up_blocks.2.resnets.0.norm1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="174779676" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.mid_block.resnets.0.norm2.weight">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.0.norm1.weight">
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="170" name="self.encoder.mid_block.resnets.0.norm2.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="103968256" size="2048" />
+		<layer id="219" name="self.decoder.up_blocks.2.resnets.0.norm1.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="512" offset="174781724" size="2048" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.mid_block.resnets.0.norm2.bias">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.0.norm1.bias">
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="171" name="__module.encoder.mid_block.resnets.0.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+		<layer id="220" name="__module.decoder.up_blocks.2.resnets.0.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
 			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2904,7 +3598,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="425,input.121">
+				<port id="3" precision="FP32" names="424,input.129">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2912,7 +3606,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="172" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_17" type="Swish" version="opset4">
+		<layer id="221" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_16" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
@@ -2922,7 +3616,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="426,input.123">
+				<port id="1" precision="FP32" names="425,input.131">
 					<dim>-1</dim>
 					<dim>512</dim>
 					<dim>-1</dim>
@@ -2930,18 +3624,18 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="173" name="self.encoder.mid_block.resnets.0.conv2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512, 512, 3, 3" offset="103970304" size="9437184" />
+		<layer id="222" name="self.decoder.up_blocks.2.resnets.0.conv1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="256, 512, 3, 3" offset="174783772" size="4718592" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.mid_block.resnets.0.conv2.weight">
-					<dim>512</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.0.conv1.weight">
+					<dim>256</dim>
 					<dim>512</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="174" name="__module.encoder.mid_block.resnets.0.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
+		<layer id="223" name="__module.decoder.up_blocks.2.resnets.0.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
@@ -2951,7 +3645,7 @@
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>512</dim>
 					<dim>3</dim>
 					<dim>3</dim>
@@ -2960,1101 +3654,1045 @@
 			<output>
 				<port id="2" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="175" name="__module.encoder.mid_block.resnets.0.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 512, 1, 1" offset="113407488" size="2048" />
+		<layer id="224" name="__module.decoder.up_blocks.2.resnets.0.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 256, 1, 1" offset="179502364" size="1024" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="176" name="__module.encoder.mid_block.resnets.0.conv2/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="225" name="__module.decoder.up_blocks.2.resnets.0.conv1/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
 					<dim>1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="434,hidden_states.29">
+				<port id="2" precision="FP32" names="432,input.133">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="177" name="__module.encoder.mid_block.resnets.0/aten::add/Add" type="Add" version="opset1">
-			<data auto_broadcast="numpy" />
-			<input>
-				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>512</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
-				</port>
-				<port id="1" precision="FP32">
-					<dim>-1</dim>
-					<dim>512</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
-				</port>
-			</input>
+		<layer id="226" name="self.decoder.up_blocks.2.resnets.0.norm2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="256" offset="179503388" size="1024" />
 			<output>
-				<port id="2" precision="FP32" names="435,436,hidden_states.31">
-					<dim>-1</dim>
-					<dim>512</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.0.norm2.weight">
+					<dim>256</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="178" name="Constant_232994" type="Const" version="opset1">
-			<data element_type="i64" shape="3" offset="113409536" size="24" />
+		<layer id="227" name="self.decoder.up_blocks.2.resnets.0.norm2.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="256" offset="179504412" size="1024" />
 			<output>
-				<port id="0" precision="I64">
-					<dim>3</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.0.norm2.bias">
+					<dim>256</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="179" name="__module.encoder.mid_block.attentions.0/aten::view/Reshape" type="Reshape" version="opset1">
-			<data special_zero="true" />
+		<layer id="228" name="__module.decoder.up_blocks.2.resnets.0.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
-				<port id="1" precision="I64">
-					<dim>3</dim>
+				<port id="1" precision="FP32">
+					<dim>256</dim>
+				</port>
+				<port id="2" precision="FP32">
+					<dim>256</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="452">
+				<port id="3" precision="FP32" names="435,input.135">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="180" name="__module.encoder.mid_block.attentions.0/aten::transpose/ScatterElementsUpdate" type="Const" version="opset1">
-			<data element_type="i32" shape="3" offset="113409560" size="12" />
-			<output>
-				<port id="0" precision="I32">
-					<dim>3</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="181" name="__module.encoder.mid_block.attentions.0/aten::transpose/Transpose" type="Transpose" version="opset1">
+		<layer id="229" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_17" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
+					<dim>-1</dim>
 					<dim>-1</dim>
-				</port>
-				<port id="1" precision="I32">
-					<dim>3</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="453,hidden_states.33">
+				<port id="1" precision="FP32" names="436,input.137">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="182" name="__module.encoder.mid_block.attentions.0/aten::transpose/ScatterElementsUpdate_1" type="Const" version="opset1">
-			<data element_type="i32" shape="3" offset="113409560" size="12" />
+		<layer id="230" name="self.decoder.up_blocks.2.resnets.0.conv2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="256, 256, 3, 3" offset="179505436" size="2359296" />
 			<output>
-				<port id="0" precision="I32">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.0.conv2.weight">
+					<dim>256</dim>
+					<dim>256</dim>
+					<dim>3</dim>
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="183" name="__module.encoder.mid_block.attentions.0/aten::transpose/Transpose_1" type="Transpose" version="opset1">
+		<layer id="231" name="__module.decoder.up_blocks.2.resnets.0.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 				</port>
-				<port id="1" precision="I32">
+				<port id="1" precision="FP32">
+					<dim>256</dim>
+					<dim>256</dim>
+					<dim>3</dim>
 					<dim>3</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="455,input.127">
+				<port id="2" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="184" name="self.encoder.mid_block.attentions.0.group_norm.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="113409572" size="2048" />
-			<output>
-				<port id="0" precision="FP32" names="self.encoder.mid_block.attentions.0.group_norm.weight">
-					<dim>512</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="185" name="self.encoder.mid_block.attentions.0.group_norm.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="113411620" size="2048" />
+		<layer id="232" name="__module.decoder.up_blocks.2.resnets.0.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 256, 1, 1" offset="181864732" size="1024" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.mid_block.attentions.0.group_norm.bias">
-					<dim>512</dim>
+				<port id="0" precision="FP32">
+					<dim>1</dim>
+					<dim>256</dim>
+					<dim>1</dim>
+					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="186" name="__module.encoder.mid_block.attentions.0.group_norm/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
-			<data num_groups="32" epsilon="9.9999999747524271e-07" />
+		<layer id="233" name="__module.decoder.up_blocks.2.resnets.0.conv2/aten::_convolution/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
+					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>512</dim>
-				</port>
-				<port id="2" precision="FP32">
-					<dim>512</dim>
+					<dim>1</dim>
+					<dim>256</dim>
+					<dim>1</dim>
+					<dim>1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="458">
+				<port id="2" precision="FP32" names="444,hidden_states.35">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="187" name="self.encoder.mid_block.attentions.0.to_q.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512, 512" offset="113413668" size="1048576" />
-			<output>
-				<port id="0" precision="FP32" names="self.encoder.mid_block.attentions.0.to_q.weight">
-					<dim>512</dim>
-					<dim>512</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="188" name="__module.encoder.mid_block.attentions.0.to_q/aten::linear/MatMul" type="MatMul" version="opset1">
-			<data transpose_a="true" transpose_b="true" />
+		<layer id="234" name="__module.decoder.up_blocks.2.resnets.0/aten::add/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
+					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>512</dim>
-					<dim>512</dim>
+					<dim>-1</dim>
+					<dim>256</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32">
+				<port id="2" precision="FP32" names="452,453,input.141">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="189" name="Constant_232767" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 1, 512" offset="114462244" size="2048" />
-			<output>
-				<port id="0" precision="FP32">
-					<dim>1</dim>
-					<dim>1</dim>
-					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="190" name="__module.encoder.mid_block.attentions.0.to_q/aten::linear/Add" type="Add" version="opset1">
-			<data auto_broadcast="numpy" />
-			<input>
-				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>-1</dim>
-					<dim>512</dim>
-				</port>
-				<port id="1" precision="FP32">
-					<dim>1</dim>
-					<dim>1</dim>
-					<dim>512</dim>
-				</port>
-			</input>
+		<layer id="235" name="self.decoder.up_blocks.2.resnets.1.norm1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="256" offset="181865756" size="1024" />
 			<output>
-				<port id="2" precision="FP32" names="462,query.1">
-					<dim>-1</dim>
-					<dim>-1</dim>
-					<dim>512</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.1.norm1.weight">
+					<dim>256</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="191" name="Constant_232995" type="Const" version="opset1">
-			<data element_type="i64" shape="4" offset="114464292" size="32" />
+		<layer id="236" name="self.decoder.up_blocks.2.resnets.1.norm1.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="256" offset="181866780" size="1024" />
 			<output>
-				<port id="0" precision="I64">
-					<dim>4</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.1.norm1.bias">
+					<dim>256</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="192" name="__module.encoder.mid_block.attentions.0/aten::view/Reshape_1" type="Reshape" version="opset1">
-			<data special_zero="true" />
+		<layer id="237" name="__module.decoder.up_blocks.2.resnets.1.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 				</port>
-				<port id="1" precision="I64">
-					<dim>4</dim>
+				<port id="1" precision="FP32">
+					<dim>256</dim>
+				</port>
+				<port id="2" precision="FP32">
+					<dim>256</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="476">
+				<port id="3" precision="FP32" names="461,input.143">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>1</dim>
-					<dim>512</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="193" name="Constant_232668" type="Const" version="opset1">
-			<data element_type="i64" shape="4" offset="114464324" size="32" />
-			<output>
-				<port id="0" precision="I64">
-					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="194" name="__module.encoder.mid_block.attentions.0/aten::transpose/Transpose_3" type="Reshape" version="opset1">
-			<data special_zero="true" />
+		<layer id="238" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_18" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>1</dim>
-					<dim>512</dim>
-				</port>
-				<port id="1" precision="I64">
-					<dim>4</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="477,query">
+				<port id="1" precision="FP32" names="462,input.145">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
-					<dim>1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="195" name="self.encoder.mid_block.attentions.0.to_k.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512, 512" offset="114464356" size="1048576" />
+		<layer id="239" name="self.decoder.up_blocks.2.resnets.1.conv1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="256, 256, 3, 3" offset="181867804" size="2359296" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.mid_block.attentions.0.to_k.weight">
-					<dim>512</dim>
-					<dim>512</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.1.conv1.weight">
+					<dim>256</dim>
+					<dim>256</dim>
+					<dim>3</dim>
+					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="196" name="__module.encoder.mid_block.attentions.0.to_k/aten::linear/MatMul" type="MatMul" version="opset1">
-			<data transpose_a="true" transpose_b="true" />
+		<layer id="240" name="__module.decoder.up_blocks.2.resnets.1.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
+					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>512</dim>
-					<dim>512</dim>
+					<dim>256</dim>
+					<dim>256</dim>
+					<dim>3</dim>
+					<dim>3</dim>
 				</port>
 			</input>
 			<output>
 				<port id="2" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="197" name="Constant_232768" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 1, 512" offset="115512932" size="2048" />
+		<layer id="241" name="__module.decoder.up_blocks.2.resnets.1.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 256, 1, 1" offset="184227100" size="1024" />
 			<output>
 				<port id="0" precision="FP32">
+					<dim>1</dim>
+					<dim>256</dim>
 					<dim>1</dim>
 					<dim>1</dim>
-					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="198" name="__module.encoder.mid_block.attentions.0.to_k/aten::linear/Add" type="Add" version="opset1">
+		<layer id="242" name="__module.decoder.up_blocks.2.resnets.1.conv1/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 				</port>
 				<port id="1" precision="FP32">
+					<dim>1</dim>
+					<dim>256</dim>
 					<dim>1</dim>
 					<dim>1</dim>
-					<dim>512</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="465,key.1">
+				<port id="2" precision="FP32" names="469,input.147">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="199" name="Constant_232996" type="Const" version="opset1">
-			<data element_type="i64" shape="4" offset="114464292" size="32" />
+		<layer id="243" name="self.decoder.up_blocks.2.resnets.1.norm2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="256" offset="184228124" size="1024" />
 			<output>
-				<port id="0" precision="I64">
-					<dim>4</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.1.norm2.weight">
+					<dim>256</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="200" name="__module.encoder.mid_block.attentions.0/aten::view/Reshape_2" type="Reshape" version="opset1">
-			<data special_zero="true" />
+		<layer id="244" name="self.decoder.up_blocks.2.resnets.1.norm2.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="256" offset="184229148" size="1024" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.1.norm2.bias">
+					<dim>256</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="245" name="__module.decoder.up_blocks.2.resnets.1.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 				</port>
-				<port id="1" precision="I64">
-					<dim>4</dim>
+				<port id="1" precision="FP32">
+					<dim>256</dim>
+				</port>
+				<port id="2" precision="FP32">
+					<dim>256</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="479">
+				<port id="3" precision="FP32" names="472,input.149">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>1</dim>
-					<dim>512</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="201" name="Constant_232670" type="Const" version="opset1">
-			<data element_type="i64" shape="4" offset="114464324" size="32" />
-			<output>
-				<port id="0" precision="I64">
-					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="202" name="__module.encoder.mid_block.attentions.0/aten::transpose/Transpose_4" type="Reshape" version="opset1">
-			<data special_zero="true" />
+		<layer id="246" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_19" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>1</dim>
-					<dim>512</dim>
-				</port>
-				<port id="1" precision="I64">
-					<dim>4</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="480,key">
+				<port id="1" precision="FP32" names="473,input.151">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
-					<dim>1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="203" name="self.encoder.mid_block.attentions.0.to_v.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512, 512" offset="115514980" size="1048576" />
+		<layer id="247" name="self.decoder.up_blocks.2.resnets.1.conv2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="256, 256, 3, 3" offset="184230172" size="2359296" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.mid_block.attentions.0.to_v.weight">
-					<dim>512</dim>
-					<dim>512</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.1.conv2.weight">
+					<dim>256</dim>
+					<dim>256</dim>
+					<dim>3</dim>
+					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="204" name="__module.encoder.mid_block.attentions.0.to_v/aten::linear/MatMul" type="MatMul" version="opset1">
-			<data transpose_a="true" transpose_b="true" />
+		<layer id="248" name="__module.decoder.up_blocks.2.resnets.1.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
+					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>512</dim>
-					<dim>512</dim>
+					<dim>256</dim>
+					<dim>256</dim>
+					<dim>3</dim>
+					<dim>3</dim>
 				</port>
 			</input>
 			<output>
 				<port id="2" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="205" name="Constant_232769" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 1, 512" offset="116563556" size="2048" />
+		<layer id="249" name="__module.decoder.up_blocks.2.resnets.1.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 256, 1, 1" offset="186589468" size="1024" />
 			<output>
 				<port id="0" precision="FP32">
+					<dim>1</dim>
+					<dim>256</dim>
 					<dim>1</dim>
 					<dim>1</dim>
-					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="206" name="__module.encoder.mid_block.attentions.0.to_v/aten::linear/Add" type="Add" version="opset1">
+		<layer id="250" name="__module.decoder.up_blocks.2.resnets.1.conv2/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 				</port>
 				<port id="1" precision="FP32">
+					<dim>1</dim>
+					<dim>256</dim>
 					<dim>1</dim>
 					<dim>1</dim>
-					<dim>512</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="468,value.1">
+				<port id="2" precision="FP32" names="481,hidden_states.37">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="207" name="Constant_232997" type="Const" version="opset1">
-			<data element_type="i64" shape="4" offset="114464292" size="32" />
-			<output>
-				<port id="0" precision="I64">
-					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="208" name="__module.encoder.mid_block.attentions.0/aten::view/Reshape_3" type="Reshape" version="opset1">
-			<data special_zero="true" />
+		<layer id="251" name="__module.decoder.up_blocks.2.resnets.1/aten::add/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 				</port>
-				<port id="1" precision="I64">
-					<dim>4</dim>
+				<port id="1" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="482">
+				<port id="2" precision="FP32" names="482,483,input.155">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>1</dim>
-					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="209" name="Constant_232674" type="Const" version="opset1">
-			<data element_type="i64" shape="4" offset="114464324" size="32" />
+		<layer id="252" name="self.decoder.up_blocks.2.resnets.2.norm1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="256" offset="186590492" size="1024" />
 			<output>
-				<port id="0" precision="I64">
-					<dim>4</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.2.norm1.weight">
+					<dim>256</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="210" name="__module.encoder.mid_block.attentions.0/aten::transpose/Transpose_5" type="Reshape" version="opset1">
-			<data special_zero="true" />
-			<input>
-				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>-1</dim>
-					<dim>1</dim>
-					<dim>512</dim>
-				</port>
-				<port id="1" precision="I64">
-					<dim>4</dim>
-				</port>
-			</input>
+		<layer id="253" name="self.decoder.up_blocks.2.resnets.2.norm1.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="256" offset="186591516" size="1024" />
 			<output>
-				<port id="2" precision="FP32" names="483,value">
-					<dim>-1</dim>
-					<dim>1</dim>
-					<dim>-1</dim>
-					<dim>512</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.2.norm1.bias">
+					<dim>256</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="211" name="__module.encoder.mid_block.attentions.0/aten::scaled_dot_product_attention/ScaledDotProductAttention" type="ScaledDotProductAttention" version="opset13">
-			<data causal="false" />
+		<layer id="254" name="__module.decoder.up_blocks.2.resnets.2.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>1</dim>
+					<dim>256</dim>
+					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>-1</dim>
-					<dim>1</dim>
-					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 				</port>
 				<port id="2" precision="FP32">
-					<dim>-1</dim>
-					<dim>1</dim>
-					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="484,hidden_states.35">
+				<port id="3" precision="FP32" names="491,input.157">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
-					<dim>1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="212" name="Constant_232676" type="Const" version="opset1">
-			<data element_type="i64" shape="4" offset="116565604" size="32" />
-			<output>
-				<port id="0" precision="I64">
-					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="213" name="__module.encoder.mid_block.attentions.0/aten::transpose/Transpose_6" type="Reshape" version="opset1">
-			<data special_zero="true" />
+		<layer id="255" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_20" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>1</dim>
+					<dim>256</dim>
+					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
-				</port>
-				<port id="1" precision="I64">
-					<dim>4</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="485">
+				<port id="1" precision="FP32" names="492,input.159">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>1</dim>
-					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="214" name="Constant_232998" type="Const" version="opset1">
-			<data element_type="i64" shape="3" offset="116565636" size="24" />
+		<layer id="256" name="self.decoder.up_blocks.2.resnets.2.conv1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="256, 256, 3, 3" offset="186592540" size="2359296" />
 			<output>
-				<port id="0" precision="I64">
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.2.conv1.weight">
+					<dim>256</dim>
+					<dim>256</dim>
 					<dim>3</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="215" name="__module.encoder.mid_block.attentions.0/aten::reshape/Reshape" type="Reshape" version="opset1">
-			<data special_zero="true" />
-			<input>
-				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>-1</dim>
-					<dim>1</dim>
-					<dim>512</dim>
-				</port>
-				<port id="1" precision="I64">
 					<dim>3</dim>
 				</port>
-			</input>
-			<output>
-				<port id="2" precision="FP32" names="489,490,hidden_states.37,input.131">
-					<dim>-1</dim>
-					<dim>-1</dim>
-					<dim>512</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="216" name="self.encoder.mid_block.attentions.0.to_out.0.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512, 512" offset="116565660" size="1048576" />
-			<output>
-				<port id="0" precision="FP32" names="self.encoder.mid_block.attentions.0.to_out.0.weight">
-					<dim>512</dim>
-					<dim>512</dim>
-				</port>
 			</output>
 		</layer>
-		<layer id="217" name="__module.encoder.mid_block.attentions.0.to_out.0/aten::linear/MatMul" type="MatMul" version="opset1">
-			<data transpose_a="false" transpose_b="true" />
+		<layer id="257" name="__module.decoder.up_blocks.2.resnets.2.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>512</dim>
-					<dim>512</dim>
+					<dim>256</dim>
+					<dim>256</dim>
+					<dim>3</dim>
+					<dim>3</dim>
 				</port>
 			</input>
 			<output>
 				<port id="2" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="218" name="Constant_232770" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 1, 512" offset="117614236" size="2048" />
+		<layer id="258" name="__module.decoder.up_blocks.2.resnets.2.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 256, 1, 1" offset="188951836" size="1024" />
 			<output>
 				<port id="0" precision="FP32">
+					<dim>1</dim>
+					<dim>256</dim>
 					<dim>1</dim>
 					<dim>1</dim>
-					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="219" name="__module.encoder.mid_block.attentions.0.to_out.0/aten::linear/Add" type="Add" version="opset1">
+		<layer id="259" name="__module.decoder.up_blocks.2.resnets.2.conv1/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 				</port>
 				<port id="1" precision="FP32">
+					<dim>1</dim>
+					<dim>256</dim>
 					<dim>1</dim>
 					<dim>1</dim>
-					<dim>512</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="493,input.133">
+				<port id="2" precision="FP32" names="499,input.161">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
-					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="220" name="__module.encoder.mid_block.attentions.0/aten::transpose/ScatterElementsUpdate_7" type="Const" version="opset1">
-			<data element_type="i32" shape="3" offset="113409560" size="12" />
+		<layer id="260" name="self.decoder.up_blocks.2.resnets.2.norm2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="256" offset="188952860" size="1024" />
 			<output>
-				<port id="0" precision="I32">
-					<dim>3</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.2.norm2.weight">
+					<dim>256</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="221" name="__module.encoder.mid_block.attentions.0/aten::transpose/Transpose_7" type="Transpose" version="opset1">
-			<input>
-				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>-1</dim>
-					<dim>512</dim>
-				</port>
-				<port id="1" precision="I32">
-					<dim>3</dim>
-				</port>
-			</input>
+		<layer id="261" name="self.decoder.up_blocks.2.resnets.2.norm2.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="256" offset="188953884" size="1024" />
 			<output>
-				<port id="2" precision="FP32" names="495">
-					<dim>-1</dim>
-					<dim>512</dim>
-					<dim>-1</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.2.norm2.bias">
+					<dim>256</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="222" name="ShapeOf_232952" type="ShapeOf" version="opset3">
-			<data output_type="i64" />
+		<layer id="262" name="__module.decoder.up_blocks.2.resnets.2.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
-			</input>
-			<output>
-				<port id="1" precision="I64">
-					<dim>4</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="223" name="Constant_232953" type="Const" version="opset1">
-			<data element_type="i64" shape="1" offset="117616284" size="8" />
-			<output>
-				<port id="0" precision="I64">
-					<dim>1</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="224" name="Constant_232954" type="Const" version="opset1">
-			<data element_type="i64" shape="" offset="117616284" size="8" />
-			<output>
-				<port id="0" precision="I64" />
-			</output>
-		</layer>
-		<layer id="225" name="Gather_232955" type="Gather" version="opset8">
-			<data batch_dims="0" />
-			<input>
-				<port id="0" precision="I64">
-					<dim>4</dim>
+				<port id="1" precision="FP32">
+					<dim>256</dim>
 				</port>
-				<port id="1" precision="I64">
-					<dim>1</dim>
+				<port id="2" precision="FP32">
+					<dim>256</dim>
 				</port>
-				<port id="2" precision="I64" />
 			</input>
 			<output>
-				<port id="3" precision="I64" names="443,454">
-					<dim>1</dim>
+				<port id="3" precision="FP32" names="502,input.163">
+					<dim>-1</dim>
+					<dim>256</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="226" name="__module.encoder.mid_block.attentions.0/prim::ListConstruct/Reshape_0" type="Const" version="opset1">
-			<data element_type="i64" shape="1" offset="117616292" size="8" />
-			<output>
-				<port id="0" precision="I64">
-					<dim>1</dim>
+		<layer id="263" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_21" type="Swish" version="opset4">
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
-			</output>
-		</layer>
-		<layer id="227" name="Constant_233110" type="Const" version="opset1">
-			<data element_type="i64" shape="2" offset="117616300" size="16" />
+			</input>
 			<output>
-				<port id="0" precision="I64">
-					<dim>2</dim>
+				<port id="1" precision="FP32" names="503,input.165">
+					<dim>-1</dim>
+					<dim>256</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="228" name="Constant_233111" type="Const" version="opset1">
-			<data element_type="i64" shape="" offset="117616284" size="8" />
+		<layer id="264" name="self.decoder.up_blocks.2.resnets.2.conv2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="256, 256, 3, 3" offset="188954908" size="2359296" />
 			<output>
-				<port id="0" precision="I64" />
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.resnets.2.conv2.weight">
+					<dim>256</dim>
+					<dim>256</dim>
+					<dim>3</dim>
+					<dim>3</dim>
+				</port>
 			</output>
 		</layer>
-		<layer id="229" name="Gather_233112" type="Gather" version="opset8">
-			<data batch_dims="0" />
+		<layer id="265" name="__module.decoder.up_blocks.2.resnets.2.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
-				<port id="0" precision="I64">
-					<dim>4</dim>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
-				<port id="1" precision="I64">
-					<dim>2</dim>
+				<port id="1" precision="FP32">
+					<dim>256</dim>
+					<dim>256</dim>
+					<dim>3</dim>
+					<dim>3</dim>
 				</port>
-				<port id="2" precision="I64" />
 			</input>
 			<output>
-				<port id="3" precision="I64">
-					<dim>2</dim>
+				<port id="2" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="230" name="__module.encoder.mid_block.attentions.0/prim::ListConstruct/Concat_5" type="Concat" version="opset1">
-			<data axis="0" />
-			<input>
-				<port id="0" precision="I64">
+		<layer id="266" name="__module.decoder.up_blocks.2.resnets.2.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 256, 1, 1" offset="191314204" size="1024" />
+			<output>
+				<port id="0" precision="FP32">
+					<dim>1</dim>
+					<dim>256</dim>
 					<dim>1</dim>
-				</port>
-				<port id="1" precision="I64">
 					<dim>1</dim>
-				</port>
-				<port id="2" precision="I64">
-					<dim>2</dim>
-				</port>
-			</input>
-			<output>
-				<port id="3" precision="I64" names="496">
-					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="231" name="__module.encoder.mid_block.attentions.0/aten::reshape/Reshape_1" type="Reshape" version="opset1">
-			<data special_zero="false" />
+		<layer id="267" name="__module.decoder.up_blocks.2.resnets.2.conv2/aten::_convolution/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
+					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
-				<port id="1" precision="I64">
-					<dim>4</dim>
+				<port id="1" precision="FP32">
+					<dim>1</dim>
+					<dim>256</dim>
+					<dim>1</dim>
+					<dim>1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="497,hidden_states.41">
+				<port id="2" precision="FP32" names="511,hidden_states.39">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="232" name="__module.encoder.mid_block.attentions.0/aten::add/Add" type="Add" version="opset1">
+		<layer id="268" name="__module.decoder.up_blocks.2.resnets.2/aten::add/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="498,499,hidden_states.43,input.135">
+				<port id="2" precision="FP32" names="512,513,hidden_states.41">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="233" name="self.encoder.mid_block.resnets.1.norm1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="117616316" size="2048" />
+		<layer id="269" name="__module.decoder.up_blocks.2.upsamplers.0/aten::upsample_nearest2d/Multiply" type="Const" version="opset1">
+			<data element_type="f32" shape="2" offset="98715916" size="8" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.mid_block.resnets.1.norm1.weight">
-					<dim>512</dim>
+				<port id="0" precision="FP32">
+					<dim>2</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="234" name="self.encoder.mid_block.resnets.1.norm1.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="117618364" size="2048" />
+		<layer id="270" name="Constant_235768" type="Const" version="opset1">
+			<data element_type="i32" shape="2" offset="98715924" size="8" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.mid_block.resnets.1.norm1.bias">
-					<dim>512</dim>
+				<port id="0" precision="I32">
+					<dim>2</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="235" name="__module.encoder.mid_block.resnets.1.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
-			<data num_groups="32" epsilon="9.9999999747524271e-07" />
+		<layer id="271" name="__module.decoder.up_blocks.2.upsamplers.0/aten::upsample_nearest2d/Interpolate" type="Interpolate" version="opset11">
+			<data mode="nearest" shape_calculation_mode="scales" coordinate_transformation_mode="asymmetric" nearest_mode="floor" antialias="false" pads_begin="0, 0, 0, 0" pads_end="0, 0, 0, 0" cube_coeff="-0.75" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>512</dim>
+					<dim>2</dim>
 				</port>
-				<port id="2" precision="FP32">
-					<dim>512</dim>
+				<port id="2" precision="I32">
+					<dim>2</dim>
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="507,input.137">
+				<port id="3" precision="FP32" names="516,input.169">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="236" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_18" type="Swish" version="opset4">
+		<layer id="272" name="self.decoder.up_blocks.2.upsamplers.0.conv.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="256, 256, 3, 3" offset="191315228" size="2359296" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.2.upsamplers.0.conv.weight">
+					<dim>256</dim>
+					<dim>256</dim>
+					<dim>3</dim>
+					<dim>3</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="273" name="__module.decoder.up_blocks.2.upsamplers.0.conv/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>256</dim>
+					<dim>256</dim>
+					<dim>3</dim>
+					<dim>3</dim>
+				</port>
+			</input>
+			<output>
+				<port id="2" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="274" name="__module.decoder.up_blocks.2.upsamplers.0.conv/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 256, 1, 1" offset="193674524" size="1024" />
+			<output>
+				<port id="0" precision="FP32">
+					<dim>1</dim>
+					<dim>256</dim>
+					<dim>1</dim>
+					<dim>1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="275" name="__module.decoder.up_blocks.2.upsamplers.0.conv/aten::_convolution/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
+				<port id="1" precision="FP32">
+					<dim>1</dim>
+					<dim>256</dim>
+					<dim>1</dim>
+					<dim>1</dim>
+				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="508,input.139">
+				<port id="2" precision="FP32" names="523,input.171">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="237" name="self.encoder.mid_block.resnets.1.conv1.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512, 512, 3, 3" offset="117620412" size="9437184" />
+		<layer id="276" name="self.decoder.up_blocks.3.resnets.0.conv_shortcut.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="128, 256, 1, 1" offset="193675548" size="131072" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.mid_block.resnets.1.conv1.weight">
-					<dim>512</dim>
-					<dim>512</dim>
-					<dim>3</dim>
-					<dim>3</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.0.conv_shortcut.weight">
+					<dim>128</dim>
+					<dim>256</dim>
+					<dim>1</dim>
+					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="238" name="__module.encoder.mid_block.resnets.1.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
-			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
+		<layer id="277" name="__module.decoder.up_blocks.3.resnets.0.conv_shortcut/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="0, 0" pads_end="0, 0" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>512</dim>
-					<dim>512</dim>
-					<dim>3</dim>
-					<dim>3</dim>
+					<dim>128</dim>
+					<dim>256</dim>
+					<dim>1</dim>
+					<dim>1</dim>
 				</port>
 			</input>
 			<output>
 				<port id="2" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="239" name="__module.encoder.mid_block.resnets.1.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 512, 1, 1" offset="127057596" size="2048" />
+		<layer id="278" name="__module.decoder.up_blocks.3.resnets.0.conv_shortcut/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 128, 1, 1" offset="193806620" size="512" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
-					<dim>512</dim>
+					<dim>128</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="240" name="__module.encoder.mid_block.resnets.1.conv1/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="279" name="__module.decoder.up_blocks.3.resnets.0.conv_shortcut/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
 					<dim>1</dim>
-					<dim>512</dim>
+					<dim>128</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="515,input.141">
+				<port id="2" precision="FP32" names="565,input_tensor">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="241" name="self.encoder.mid_block.resnets.1.norm2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="127059644" size="2048" />
+		<layer id="280" name="self.decoder.up_blocks.3.resnets.0.norm1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="256" offset="193807132" size="1024" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.mid_block.resnets.1.norm2.weight">
-					<dim>512</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.0.norm1.weight">
+					<dim>256</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="242" name="self.encoder.mid_block.resnets.1.norm2.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="127061692" size="2048" />
+		<layer id="281" name="self.decoder.up_blocks.3.resnets.0.norm1.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="256" offset="193808156" size="1024" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.mid_block.resnets.1.norm2.bias">
-					<dim>512</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.0.norm1.bias">
+					<dim>256</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="243" name="__module.encoder.mid_block.resnets.1.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+		<layer id="282" name="__module.decoder.up_blocks.3.resnets.0.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
 			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>512</dim>
+					<dim>256</dim>
 				</port>
 				<port id="2" precision="FP32">
-					<dim>512</dim>
+					<dim>256</dim>
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="518,input.143">
+				<port id="3" precision="FP32" names="538,input.173">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="244" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_19" type="Swish" version="opset4">
+		<layer id="283" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_22" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="519,input.145">
+				<port id="1" precision="FP32" names="539,input.175">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="245" name="self.encoder.mid_block.resnets.1.conv2.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512, 512, 3, 3" offset="127063740" size="9437184" />
+		<layer id="284" name="self.decoder.up_blocks.3.resnets.0.conv1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="128, 256, 3, 3" offset="193809180" size="1179648" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.mid_block.resnets.1.conv2.weight">
-					<dim>512</dim>
-					<dim>512</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.0.conv1.weight">
+					<dim>128</dim>
+					<dim>256</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="246" name="__module.encoder.mid_block.resnets.1.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
+		<layer id="285" name="__module.decoder.up_blocks.3.resnets.0.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>256</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>512</dim>
-					<dim>512</dim>
+					<dim>128</dim>
+					<dim>256</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
@@ -4062,155 +4700,130 @@
 			<output>
 				<port id="2" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="247" name="__module.encoder.mid_block.resnets.1.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 512, 1, 1" offset="136500924" size="2048" />
+		<layer id="286" name="__module.decoder.up_blocks.3.resnets.0.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 128, 1, 1" offset="194988828" size="512" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
-					<dim>512</dim>
-					<dim>1</dim>
-					<dim>1</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="248" name="__module.encoder.mid_block.resnets.1.conv2/aten::_convolution/Add" type="Add" version="opset1">
-			<data auto_broadcast="numpy" />
-			<input>
-				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>512</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
-				</port>
-				<port id="1" precision="FP32">
-					<dim>1</dim>
-					<dim>512</dim>
+					<dim>128</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
-			</input>
-			<output>
-				<port id="2" precision="FP32" names="527,hidden_states">
-					<dim>-1</dim>
-					<dim>512</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
-				</port>
 			</output>
 		</layer>
-		<layer id="249" name="__module.encoder.mid_block.resnets.1/aten::add/Add" type="Add" version="opset1">
+		<layer id="287" name="__module.decoder.up_blocks.3.resnets.0.conv1/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
-				</port>
-				<port id="1" precision="FP32">
-					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
+				<port id="1" precision="FP32">
+					<dim>1</dim>
+					<dim>128</dim>
+					<dim>1</dim>
+					<dim>1</dim>
+				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="528,529,input.149">
+				<port id="2" precision="FP32" names="546,input.177">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="250" name="self.encoder.conv_norm_out.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="136502972" size="2048" />
+		<layer id="288" name="self.decoder.up_blocks.3.resnets.0.norm2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="128" offset="194989340" size="512" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.conv_norm_out.weight">
-					<dim>512</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.0.norm2.weight">
+					<dim>128</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="251" name="self.encoder.conv_norm_out.bias" type="Const" version="opset1">
-			<data element_type="f32" shape="512" offset="136505020" size="2048" />
+		<layer id="289" name="self.decoder.up_blocks.3.resnets.0.norm2.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="128" offset="194989852" size="512" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.conv_norm_out.bias">
-					<dim>512</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.0.norm2.bias">
+					<dim>128</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="252" name="__module.encoder.conv_norm_out/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+		<layer id="290" name="__module.decoder.up_blocks.3.resnets.0.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
 			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>512</dim>
+					<dim>128</dim>
 				</port>
 				<port id="2" precision="FP32">
-					<dim>512</dim>
+					<dim>128</dim>
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="532,input.151">
+				<port id="3" precision="FP32" names="549,input.179">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="253" name="__module.encoder.conv_act/aten::silu/Swish" type="Swish" version="opset4">
+		<layer id="291" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_23" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="533,input.153">
+				<port id="1" precision="FP32" names="550,input.181">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="254" name="self.encoder.conv_out.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="8, 512, 3, 3" offset="136507068" size="147456" />
+		<layer id="292" name="self.decoder.up_blocks.3.resnets.0.conv2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="128, 128, 3, 3" offset="194990364" size="589824" />
 			<output>
-				<port id="0" precision="FP32" names="self.encoder.conv_out.weight">
-					<dim>8</dim>
-					<dim>512</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.0.conv2.weight">
+					<dim>128</dim>
+					<dim>128</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="255" name="__module.encoder.conv_out/aten::_convolution/Convolution" type="Convolution" version="opset1">
+		<layer id="293" name="__module.decoder.up_blocks.3.resnets.0.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>512</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>8</dim>
-					<dim>512</dim>
+					<dim>128</dim>
+					<dim>128</dim>
 					<dim>3</dim>
 					<dim>3</dim>
 				</port>
@@ -4218,735 +4831,783 @@
 			<output>
 				<port id="2" precision="FP32">
 					<dim>-1</dim>
-					<dim>8</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="256" name="__module.encoder.conv_out/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 8, 1, 1" offset="136654524" size="32" />
+		<layer id="294" name="__module.decoder.up_blocks.3.resnets.0.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 128, 1, 1" offset="195580188" size="512" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
-					<dim>8</dim>
+					<dim>128</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="257" name="__module.encoder.conv_out/aten::_convolution/Add" type="Add" version="opset1">
+		<layer id="295" name="__module.decoder.up_blocks.3.resnets.0.conv2/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>8</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
 					<dim>1</dim>
-					<dim>8</dim>
+					<dim>128</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="540,input">
+				<port id="2" precision="FP32" names="558,hidden_states.43">
 					<dim>-1</dim>
-					<dim>8</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="258" name="self.quant_conv.weight" type="Const" version="opset1">
-			<data element_type="f32" shape="8, 8, 1, 1" offset="136654556" size="256" />
-			<output>
-				<port id="0" precision="FP32" names="self.quant_conv.weight">
-					<dim>8</dim>
-					<dim>8</dim>
-					<dim>1</dim>
-					<dim>1</dim>
-				</port>
-			</output>
-		</layer>
-		<layer id="259" name="__module.quant_conv/aten::_convolution/Convolution" type="Convolution" version="opset1">
-			<data strides="1, 1" dilations="1, 1" pads_begin="0, 0" pads_end="0, 0" auto_pad="explicit" />
+		<layer id="296" name="__module.decoder.up_blocks.3.resnets.0/aten::add/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>8</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>8</dim>
-					<dim>8</dim>
-					<dim>1</dim>
-					<dim>1</dim>
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32">
+				<port id="2" precision="FP32" names="566,567,input.185">
 					<dim>-1</dim>
-					<dim>8</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="260" name="__module.quant_conv/aten::_convolution/Reshape" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 8, 1, 1" offset="136654812" size="32" />
+		<layer id="297" name="self.decoder.up_blocks.3.resnets.1.norm1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="128" offset="195580700" size="512" />
 			<output>
-				<port id="0" precision="FP32">
-					<dim>1</dim>
-					<dim>8</dim>
-					<dim>1</dim>
-					<dim>1</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.1.norm1.weight">
+					<dim>128</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="261" name="__module.quant_conv/aten::_convolution/Add" type="Add" version="opset1">
-			<data auto_broadcast="numpy" />
+		<layer id="298" name="self.decoder.up_blocks.3.resnets.1.norm1.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="128" offset="195581212" size="512" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.1.norm1.bias">
+					<dim>128</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="299" name="__module.decoder.up_blocks.3.resnets.1.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>8</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>1</dim>
-					<dim>8</dim>
-					<dim>1</dim>
-					<dim>1</dim>
+					<dim>128</dim>
+				</port>
+				<port id="2" precision="FP32">
+					<dim>128</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="551,parameters">
+				<port id="3" precision="FP32" names="575,input.187">
 					<dim>-1</dim>
-					<dim>8</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="262" name="12" type="Const" version="opset1">
-			<data element_type="i64" shape="" offset="136654844" size="8" />
-			<output>
-				<port id="0" precision="I64" names="12" />
-			</output>
-		</layer>
-		<layer id="263" name="prim::ListUnpack/ShapeOf" type="ShapeOf" version="opset3">
-			<data output_type="i32" />
+		<layer id="300" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_24" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>8</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I32">
-					<dim>4</dim>
+				<port id="1" precision="FP32" names="576,input.189">
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="264" name="Constant_225588" type="Const" version="opset1">
-			<data element_type="i32" shape="1" offset="136654852" size="4" />
+		<layer id="301" name="self.decoder.up_blocks.3.resnets.1.conv1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="128, 128, 3, 3" offset="195581724" size="589824" />
 			<output>
-				<port id="0" precision="I32">
-					<dim>1</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.1.conv1.weight">
+					<dim>128</dim>
+					<dim>128</dim>
+					<dim>3</dim>
+					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="265" name="prim::ListUnpack/Gather" type="Gather" version="opset8">
-			<data batch_dims="0" />
+		<layer id="302" name="__module.decoder.up_blocks.3.resnets.1.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
-				<port id="0" precision="I32">
-					<dim>4</dim>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
-				<port id="1" precision="I64" />
-				<port id="2" precision="I32">
-					<dim>1</dim>
+				<port id="1" precision="FP32">
+					<dim>128</dim>
+					<dim>128</dim>
+					<dim>3</dim>
+					<dim>3</dim>
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="I32" />
-			</output>
-		</layer>
-		<layer id="266" name="prim::ListUnpack/Convert" type="Const" version="opset1">
-			<data element_type="i32" shape="" offset="136654856" size="4" />
-			<output>
-				<port id="0" precision="I32" />
-			</output>
-		</layer>
-		<layer id="267" name="prim::ListUnpack/Divide" type="Divide" version="opset1">
-			<data auto_broadcast="numpy" m_pythondiv="true" />
-			<input>
-				<port id="0" precision="I32" />
-				<port id="1" precision="I32" />
-			</input>
-			<output>
-				<port id="2" precision="I32" />
+				<port id="2" precision="FP32">
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
 			</output>
 		</layer>
-		<layer id="268" name="prim::ListUnpack/Mod" type="Mod" version="opset1">
-			<data auto_broadcast="numpy" />
-			<input>
-				<port id="0" precision="I32" />
-				<port id="1" precision="I32" />
-			</input>
+		<layer id="303" name="__module.decoder.up_blocks.3.resnets.1.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 128, 1, 1" offset="196171548" size="512" />
 			<output>
-				<port id="2" precision="I32" />
-			</output>
-		</layer>
-		<layer id="269" name="prim::ListUnpack/Greater" type="Greater" version="opset1">
-			<data auto_broadcast="numpy" />
-			<input>
-				<port id="0" precision="I32" />
-				<port id="1" precision="I32">
+				<port id="0" precision="FP32">
+					<dim>1</dim>
+					<dim>128</dim>
 					<dim>1</dim>
-				</port>
-			</input>
-			<output>
-				<port id="2" precision="BOOL">
 					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="270" name="prim::ListUnpack/Convert_0" type="Convert" version="opset1">
-			<data destination_type="i32" />
+		<layer id="304" name="__module.decoder.up_blocks.3.resnets.1.conv1/aten::_convolution/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
 			<input>
-				<port id="0" precision="BOOL">
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>1</dim>
+					<dim>128</dim>
+					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I32">
-					<dim>1</dim>
+				<port id="2" precision="FP32" names="583,input.191">
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="271" name="prim::ListUnpack/Add" type="Add" version="opset1">
-			<data auto_broadcast="numpy" />
-			<input>
-				<port id="0" precision="I32" />
-				<port id="1" precision="I32">
-					<dim>1</dim>
-				</port>
-			</input>
+		<layer id="305" name="self.decoder.up_blocks.3.resnets.1.norm2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="128" offset="196172060" size="512" />
 			<output>
-				<port id="2" precision="I32">
-					<dim>1</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.1.norm2.weight">
+					<dim>128</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="272" name="Constant_225589" type="Const" version="opset1">
-			<data element_type="i32" shape="1" offset="136654860" size="4" />
+		<layer id="306" name="self.decoder.up_blocks.3.resnets.1.norm2.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="128" offset="196172572" size="512" />
 			<output>
-				<port id="0" precision="I32">
-					<dim>1</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.1.norm2.bias">
+					<dim>128</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="273" name="prim::ListUnpack/Concat" type="Concat" version="opset1">
-			<data axis="0" />
+		<layer id="307" name="__module.decoder.up_blocks.3.resnets.1.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
-				<port id="0" precision="I32">
-					<dim>1</dim>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
-				<port id="1" precision="I32">
-					<dim>1</dim>
+				<port id="1" precision="FP32">
+					<dim>128</dim>
+				</port>
+				<port id="2" precision="FP32">
+					<dim>128</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="I32">
-					<dim>2</dim>
+				<port id="3" precision="FP32" names="586,input.193">
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="274" name="prim::ListUnpack/VariadicSplit" type="VariadicSplit" version="opset1">
+		<layer id="308" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_25" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>8</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
-				<port id="1" precision="I64" />
-				<port id="2" precision="I32">
-					<dim>2</dim>
-				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32" names="14">
+				<port id="1" precision="FP32" names="587,input.195">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
-				<port id="4" precision="FP32" names="15">
-					<dim>-1</dim>
-					<dim>4</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
+			</output>
+		</layer>
+		<layer id="309" name="self.decoder.up_blocks.3.resnets.1.conv2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="128, 128, 3, 3" offset="196173084" size="589824" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.1.conv2.weight">
+					<dim>128</dim>
+					<dim>128</dim>
+					<dim>3</dim>
+					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="275" name="aten::clamp/Minimum" type="Clamp" version="opset1">
-			<data min="-30" max="20" />
+		<layer id="310" name="__module.decoder.up_blocks.3.resnets.1.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
+				<port id="1" precision="FP32">
+					<dim>128</dim>
+					<dim>128</dim>
+					<dim>3</dim>
+					<dim>3</dim>
+				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="18">
+				<port id="2" precision="FP32">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="276" name="Constant_232771" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 1, 1, 1" offset="136654864" size="4" />
+		<layer id="311" name="__module.decoder.up_blocks.3.resnets.1.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 128, 1, 1" offset="196762908" size="512" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
-					<dim>1</dim>
+					<dim>128</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="277" name="aten::mul/Multiply" type="Multiply" version="opset1">
+		<layer id="312" name="__module.decoder.up_blocks.3.resnets.1.conv2/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
 					<dim>1</dim>
-					<dim>1</dim>
+					<dim>128</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="20">
+				<port id="2" precision="FP32" names="595,hidden_states.45">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="278" name="aten::exp/Exp" type="Exp" version="opset1">
+		<layer id="313" name="__module.decoder.up_blocks.3.resnets.1/aten::add/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
-			</input>
-			<output>
-				<port id="1" precision="FP32" names="21">
+				<port id="1" precision="FP32">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
-			</output>
-		</layer>
-		<layer id="279" name="aten::size/ShapeOf" type="ShapeOf" version="opset3">
-			<data output_type="i64" />
-			<input>
-				<port id="0" precision="FP32">
+			</input>
+			<output>
+				<port id="2" precision="FP32" names="596,597,input.199">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
-			</input>
-			<output>
-				<port id="1" precision="I64">
-					<dim>4</dim>
-				</port>
 			</output>
 		</layer>
-		<layer id="280" name="Constant_233102" type="Const" version="opset1">
-			<data element_type="i64" shape="1" offset="136654844" size="8" />
+		<layer id="314" name="self.decoder.up_blocks.3.resnets.2.norm1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="128" offset="196763420" size="512" />
 			<output>
-				<port id="0" precision="I64">
-					<dim>1</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.2.norm1.weight">
+					<dim>128</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="281" name="aten::size/Constant_1" type="Const" version="opset1">
-			<data element_type="i32" shape="" offset="136654852" size="4" />
+		<layer id="315" name="self.decoder.up_blocks.3.resnets.2.norm1.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="128" offset="196763932" size="512" />
 			<output>
-				<port id="0" precision="I32" />
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.2.norm1.bias">
+					<dim>128</dim>
+				</port>
 			</output>
 		</layer>
-		<layer id="282" name="aten::size/Gather_1" type="Gather" version="opset8">
-			<data batch_dims="0" />
+		<layer id="316" name="__module.decoder.up_blocks.3.resnets.2.norm1/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
-				<port id="0" precision="I64">
-					<dim>4</dim>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
-				<port id="1" precision="I64">
-					<dim>1</dim>
+				<port id="1" precision="FP32">
+					<dim>128</dim>
+				</port>
+				<port id="2" precision="FP32">
+					<dim>128</dim>
 				</port>
-				<port id="2" precision="I32" />
 			</input>
 			<output>
-				<port id="3" precision="I64" names="27,29">
-					<dim>1</dim>
+				<port id="3" precision="FP32" names="605,input.201">
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="283" name="ShapeOf_232978" type="ShapeOf" version="opset3">
-			<data output_type="i64" />
+		<layer id="317" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_26" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>8</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64">
-					<dim>4</dim>
+				<port id="1" precision="FP32" names="606,input.203">
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="284" name="Constant_233115" type="Const" version="opset1">
-			<data element_type="i64" shape="2" offset="117616300" size="16" />
+		<layer id="318" name="self.decoder.up_blocks.3.resnets.2.conv1.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="128, 128, 3, 3" offset="196764444" size="589824" />
 			<output>
-				<port id="0" precision="I64">
-					<dim>2</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.2.conv1.weight">
+					<dim>128</dim>
+					<dim>128</dim>
+					<dim>3</dim>
+					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="285" name="Constant_233116" type="Const" version="opset1">
-			<data element_type="i64" shape="" offset="117616284" size="8" />
-			<output>
-				<port id="0" precision="I64" />
-			</output>
-		</layer>
-		<layer id="286" name="Gather_233117" type="Gather" version="opset8">
-			<data batch_dims="0" />
+		<layer id="319" name="__module.decoder.up_blocks.3.resnets.2.conv1/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
-				<port id="0" precision="I64">
-					<dim>4</dim>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
-				<port id="1" precision="I64">
-					<dim>2</dim>
+				<port id="1" precision="FP32">
+					<dim>128</dim>
+					<dim>128</dim>
+					<dim>3</dim>
+					<dim>3</dim>
 				</port>
-				<port id="2" precision="I64" />
 			</input>
 			<output>
-				<port id="3" precision="I64">
-					<dim>2</dim>
+				<port id="2" precision="FP32">
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="287" name="Concat_224616" type="Concat" version="opset1">
-			<data axis="0" />
-			<input>
-				<port id="0" precision="I64">
+		<layer id="320" name="__module.decoder.up_blocks.3.resnets.2.conv1/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 128, 1, 1" offset="197354268" size="512" />
+			<output>
+				<port id="0" precision="FP32">
+					<dim>1</dim>
+					<dim>128</dim>
 					<dim>1</dim>
-				</port>
-				<port id="1" precision="I64">
 					<dim>1</dim>
-				</port>
-				<port id="2" precision="I64">
-					<dim>2</dim>
-				</port>
-			</input>
-			<output>
-				<port id="3" precision="I64">
-					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="288" name="aten::randn/Convert" type="Convert" version="opset1">
-			<data destination_type="i32" />
+		<layer id="321" name="__module.decoder.up_blocks.3.resnets.2.conv1/aten::_convolution/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
 			<input>
-				<port id="0" precision="I64">
-					<dim>4</dim>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>1</dim>
+					<dim>128</dim>
+					<dim>1</dim>
+					<dim>1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I32">
-					<dim>4</dim>
+				<port id="2" precision="FP32" names="613,input.205">
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="289" name="aten::randn/Constant_1" type="Const" version="opset1">
-			<data element_type="f32" shape="1" offset="136654868" size="4" />
+		<layer id="322" name="self.decoder.up_blocks.3.resnets.2.norm2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="128" offset="197354780" size="512" />
 			<output>
-				<port id="0" precision="FP32">
-					<dim>1</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.2.norm2.weight">
+					<dim>128</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="290" name="aten::randn/Constant_2" type="Const" version="opset1">
-			<data element_type="f32" shape="1" offset="136654872" size="4" />
+		<layer id="323" name="self.decoder.up_blocks.3.resnets.2.norm2.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="128" offset="197355292" size="512" />
 			<output>
-				<port id="0" precision="FP32">
-					<dim>1</dim>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.2.norm2.bias">
+					<dim>128</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="291" name="aten::randn/RandomUniform" type="RandomUniform" version="opset8">
-			<data output_type="f32" op_seed="1176214104" global_seed="0" alignment="tensorflow" />
-			<rt_info>
-				<attribute name="precise" version="0" />
-			</rt_info>
+		<layer id="324" name="__module.decoder.up_blocks.3.resnets.2.norm2/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
-				<port id="0" precision="I32">
-					<dim>4</dim>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>1</dim>
+					<dim>128</dim>
 				</port>
 				<port id="2" precision="FP32">
-					<dim>1</dim>
+					<dim>128</dim>
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32">
+				<port id="3" precision="FP32" names="616,input.207">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="292" name="aten::randn/Log" type="Log" version="opset1">
-			<rt_info>
-				<attribute name="precise" version="0" />
-			</rt_info>
+		<layer id="325" name="__module.decoder.mid_block.resnets.1.nonlinearity/aten::silu/Swish_27" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
+				<port id="1" precision="FP32" names="617,input.209">
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="326" name="self.decoder.up_blocks.3.resnets.2.conv2.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="128, 128, 3, 3" offset="197355804" size="589824" />
+			<output>
+				<port id="0" precision="FP32" names="self.decoder.up_blocks.3.resnets.2.conv2.weight">
+					<dim>128</dim>
+					<dim>128</dim>
+					<dim>3</dim>
+					<dim>3</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="327" name="__module.decoder.up_blocks.3.resnets.2.conv2/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
+			<input>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
 				<port id="1" precision="FP32">
+					<dim>128</dim>
+					<dim>128</dim>
+					<dim>3</dim>
+					<dim>3</dim>
+				</port>
+			</input>
+			<output>
+				<port id="2" precision="FP32">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="293" name="Constant_232772" type="Const" version="opset1">
-			<data element_type="f32" shape="1, 1, 1, 1" offset="136654876" size="4" />
+		<layer id="328" name="__module.decoder.up_blocks.3.resnets.2.conv2/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 128, 1, 1" offset="197945628" size="512" />
 			<output>
 				<port id="0" precision="FP32">
 					<dim>1</dim>
-					<dim>1</dim>
+					<dim>128</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="294" name="aten::randn/Multiply" type="Multiply" version="opset1">
+		<layer id="329" name="__module.decoder.up_blocks.3.resnets.2.conv2/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
 					<dim>1</dim>
-					<dim>1</dim>
+					<dim>128</dim>
 					<dim>1</dim>
 					<dim>1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32">
+				<port id="2" precision="FP32" names="625,hidden_states">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="295" name="aten::randn/Sqrt" type="Sqrt" version="opset1">
+		<layer id="330" name="__module.decoder.up_blocks.3.resnets.2/aten::add/Add" type="Add" version="opset1">
+			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
+				</port>
+				<port id="1" precision="FP32">
+					<dim>-1</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32">
+				<port id="2" precision="FP32" names="626,627,input.213">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="296" name="Constant_232704" type="Const" version="opset1">
-			<data element_type="f32" shape="1" offset="136654880" size="4" />
+		<layer id="331" name="self.decoder.conv_norm_out.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="128" offset="197946140" size="512" />
 			<output>
-				<port id="0" precision="FP32">
-					<dim>1</dim>
+				<port id="0" precision="FP32" names="self.decoder.conv_norm_out.weight">
+					<dim>128</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="297" name="Constant_232705" type="Const" version="opset1">
-			<data element_type="f32" shape="1" offset="136654884" size="4" />
+		<layer id="332" name="self.decoder.conv_norm_out.bias" type="Const" version="opset1">
+			<data element_type="f32" shape="128" offset="197946652" size="512" />
 			<output>
-				<port id="0" precision="FP32">
-					<dim>1</dim>
+				<port id="0" precision="FP32" names="self.decoder.conv_norm_out.bias">
+					<dim>128</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="298" name="aten::randn/Multiply_2" type="RandomUniform" version="opset8">
-			<data output_type="f32" op_seed="1176224104" global_seed="0" alignment="tensorflow" />
+		<layer id="333" name="__module.decoder.conv_norm_out/aten::group_norm/GroupNormalization" type="GroupNormalization" version="opset12">
+			<data num_groups="32" epsilon="9.9999999747524271e-07" />
 			<input>
-				<port id="0" precision="I32">
-					<dim>4</dim>
+				<port id="0" precision="FP32">
+					<dim>-1</dim>
+					<dim>128</dim>
+					<dim>-1</dim>
+					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>1</dim>
+					<dim>128</dim>
 				</port>
 				<port id="2" precision="FP32">
-					<dim>1</dim>
+					<dim>128</dim>
 				</port>
 			</input>
 			<output>
-				<port id="3" precision="FP32">
+				<port id="3" precision="FP32" names="630,input.215">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="299" name="aten::randn/Cos" type="Cos" version="opset1">
+		<layer id="334" name="__module.decoder.conv_act/aten::silu/Swish" type="Swish" version="opset4">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32">
+				<port id="1" precision="FP32" names="631,input">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="300" name="aten::randn/Multiply_3" type="Multiply" version="opset1">
-			<data auto_broadcast="numpy" />
-			<input>
-				<port id="0" precision="FP32">
-					<dim>-1</dim>
-					<dim>4</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
-				</port>
-				<port id="1" precision="FP32">
-					<dim>-1</dim>
-					<dim>4</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
-				</port>
-			</input>
+		<layer id="335" name="self.decoder.conv_out.weight" type="Const" version="opset1">
+			<data element_type="f32" shape="3, 128, 3, 3" offset="197947164" size="13824" />
 			<output>
-				<port id="2" precision="FP32" names="44,52,sample">
-					<dim>-1</dim>
-					<dim>4</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
+				<port id="0" precision="FP32" names="self.decoder.conv_out.weight">
+					<dim>3</dim>
+					<dim>128</dim>
+					<dim>3</dim>
+					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="301" name="aten::mul/Multiply_1" type="Multiply" version="opset1">
-			<data auto_broadcast="numpy" />
+		<layer id="336" name="__module.decoder.conv_out/aten::_convolution/Convolution" type="Convolution" version="opset1">
+			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>128</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>-1</dim>
-					<dim>4</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
+					<dim>3</dim>
+					<dim>128</dim>
+					<dim>3</dim>
+					<dim>3</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="53">
+				<port id="2" precision="FP32">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>3</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="302" name="aten::add/Add" type="Add" version="opset1">
+		<layer id="337" name="__module.decoder.conv_out/aten::_convolution/Reshape" type="Const" version="opset1">
+			<data element_type="f32" shape="1, 3, 1, 1" offset="197960988" size="12" />
+			<output>
+				<port id="0" precision="FP32">
+					<dim>1</dim>
+					<dim>3</dim>
+					<dim>1</dim>
+					<dim>1</dim>
+				</port>
+			</output>
+		</layer>
+		<layer id="338" name="__module.decoder.conv_out/aten::_convolution/Add" type="Add" version="opset1">
 			<data auto_broadcast="numpy" />
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>3</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 				<port id="1" precision="FP32">
-					<dim>-1</dim>
-					<dim>4</dim>
-					<dim>-1</dim>
-					<dim>-1</dim>
+					<dim>1</dim>
+					<dim>3</dim>
+					<dim>1</dim>
+					<dim>1</dim>
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="latent_sample">
+				<port id="2" precision="FP32" names="sample">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>3</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="303" name="Result_226474" type="Result" version="opset1">
+		<layer id="339" name="Result_237730" type="Result" version="opset1">
 			<input>
 				<port id="0" precision="FP32">
 					<dim>-1</dim>
-					<dim>4</dim>
+					<dim>3</dim>
 					<dim>-1</dim>
 					<dim>-1</dim>
 				</port>
@@ -4958,96 +5619,98 @@
 		<edge from-layer="1" from-port="0" to-layer="2" to-port="1" />
 		<edge from-layer="2" from-port="2" to-layer="4" to-port="0" />
 		<edge from-layer="3" from-port="0" to-layer="4" to-port="1" />
-		<edge from-layer="4" from-port="2" to-layer="7" to-port="0" />
-		<edge from-layer="4" from-port="2" to-layer="21" to-port="0" />
-		<edge from-layer="5" from-port="0" to-layer="7" to-port="1" />
-		<edge from-layer="6" from-port="0" to-layer="7" to-port="2" />
-		<edge from-layer="7" from-port="3" to-layer="8" to-port="0" />
-		<edge from-layer="8" from-port="1" to-layer="10" to-port="0" />
-		<edge from-layer="9" from-port="0" to-layer="10" to-port="1" />
-		<edge from-layer="10" from-port="2" to-layer="12" to-port="0" />
-		<edge from-layer="11" from-port="0" to-layer="12" to-port="1" />
-		<edge from-layer="12" from-port="2" to-layer="15" to-port="0" />
-		<edge from-layer="13" from-port="0" to-layer="15" to-port="1" />
-		<edge from-layer="14" from-port="0" to-layer="15" to-port="2" />
-		<edge from-layer="15" from-port="3" to-layer="16" to-port="0" />
-		<edge from-layer="16" from-port="1" to-layer="18" to-port="0" />
-		<edge from-layer="17" from-port="0" to-layer="18" to-port="1" />
-		<edge from-layer="18" from-port="2" to-layer="20" to-port="0" />
-		<edge from-layer="19" from-port="0" to-layer="20" to-port="1" />
-		<edge from-layer="20" from-port="2" to-layer="21" to-port="1" />
-		<edge from-layer="21" from-port="2" to-layer="24" to-port="0" />
-		<edge from-layer="21" from-port="2" to-layer="38" to-port="0" />
-		<edge from-layer="22" from-port="0" to-layer="24" to-port="1" />
-		<edge from-layer="23" from-port="0" to-layer="24" to-port="2" />
-		<edge from-layer="24" from-port="3" to-layer="25" to-port="0" />
-		<edge from-layer="25" from-port="1" to-layer="27" to-port="0" />
+		<edge from-layer="4" from-port="2" to-layer="6" to-port="0" />
+		<edge from-layer="5" from-port="0" to-layer="6" to-port="1" />
+		<edge from-layer="6" from-port="2" to-layer="8" to-port="0" />
+		<edge from-layer="7" from-port="0" to-layer="8" to-port="1" />
+		<edge from-layer="8" from-port="2" to-layer="11" to-port="0" />
+		<edge from-layer="8" from-port="2" to-layer="25" to-port="0" />
+		<edge from-layer="9" from-port="0" to-layer="11" to-port="1" />
+		<edge from-layer="10" from-port="0" to-layer="11" to-port="2" />
+		<edge from-layer="11" from-port="3" to-layer="12" to-port="0" />
+		<edge from-layer="12" from-port="1" to-layer="14" to-port="0" />
+		<edge from-layer="13" from-port="0" to-layer="14" to-port="1" />
+		<edge from-layer="14" from-port="2" to-layer="16" to-port="0" />
+		<edge from-layer="15" from-port="0" to-layer="16" to-port="1" />
+		<edge from-layer="16" from-port="2" to-layer="19" to-port="0" />
+		<edge from-layer="17" from-port="0" to-layer="19" to-port="1" />
+		<edge from-layer="18" from-port="0" to-layer="19" to-port="2" />
+		<edge from-layer="19" from-port="3" to-layer="20" to-port="0" />
+		<edge from-layer="20" from-port="1" to-layer="22" to-port="0" />
+		<edge from-layer="21" from-port="0" to-layer="22" to-port="1" />
+		<edge from-layer="22" from-port="2" to-layer="24" to-port="0" />
+		<edge from-layer="23" from-port="0" to-layer="24" to-port="1" />
+		<edge from-layer="24" from-port="2" to-layer="25" to-port="1" />
+		<edge from-layer="25" from-port="2" to-layer="27" to-port="0" />
+		<edge from-layer="25" from-port="2" to-layer="80" to-port="1" />
+		<edge from-layer="25" from-port="2" to-layer="70" to-port="0" />
 		<edge from-layer="26" from-port="0" to-layer="27" to-port="1" />
 		<edge from-layer="27" from-port="2" to-layer="29" to-port="0" />
 		<edge from-layer="28" from-port="0" to-layer="29" to-port="1" />
-		<edge from-layer="29" from-port="2" to-layer="32" to-port="0" />
-		<edge from-layer="30" from-port="0" to-layer="32" to-port="1" />
-		<edge from-layer="31" from-port="0" to-layer="32" to-port="2" />
-		<edge from-layer="32" from-port="3" to-layer="33" to-port="0" />
-		<edge from-layer="33" from-port="1" to-layer="35" to-port="0" />
-		<edge from-layer="34" from-port="0" to-layer="35" to-port="1" />
-		<edge from-layer="35" from-port="2" to-layer="37" to-port="0" />
-		<edge from-layer="36" from-port="0" to-layer="37" to-port="1" />
-		<edge from-layer="37" from-port="2" to-layer="38" to-port="1" />
+		<edge from-layer="29" from-port="2" to-layer="31" to-port="0" />
+		<edge from-layer="30" from-port="0" to-layer="31" to-port="1" />
+		<edge from-layer="31" from-port="2" to-layer="34" to-port="0" />
+		<edge from-layer="32" from-port="0" to-layer="34" to-port="1" />
+		<edge from-layer="33" from-port="0" to-layer="34" to-port="2" />
+		<edge from-layer="34" from-port="3" to-layer="36" to-port="0" />
+		<edge from-layer="34" from-port="3" to-layer="44" to-port="0" />
+		<edge from-layer="34" from-port="3" to-layer="52" to-port="0" />
+		<edge from-layer="35" from-port="0" to-layer="36" to-port="1" />
+		<edge from-layer="36" from-port="2" to-layer="38" to-port="0" />
+		<edge from-layer="37" from-port="0" to-layer="38" to-port="1" />
 		<edge from-layer="38" from-port="2" to-layer="40" to-port="0" />
 		<edge from-layer="39" from-port="0" to-layer="40" to-port="1" />
 		<edge from-layer="40" from-port="2" to-layer="42" to-port="0" />
 		<edge from-layer="41" from-port="0" to-layer="42" to-port="1" />
-		<edge from-layer="42" from-port="2" to-layer="44" to-port="0" />
-		<edge from-layer="42" from-port="2" to-layer="49" to-port="0" />
+		<edge from-layer="42" from-port="2" to-layer="59" to-port="0" />
 		<edge from-layer="43" from-port="0" to-layer="44" to-port="1" />
 		<edge from-layer="44" from-port="2" to-layer="46" to-port="0" />
 		<edge from-layer="45" from-port="0" to-layer="46" to-port="1" />
-		<edge from-layer="46" from-port="2" to-layer="63" to-port="0" />
-		<edge from-layer="47" from-port="0" to-layer="49" to-port="1" />
-		<edge from-layer="48" from-port="0" to-layer="49" to-port="2" />
-		<edge from-layer="49" from-port="3" to-layer="50" to-port="0" />
-		<edge from-layer="50" from-port="1" to-layer="52" to-port="0" />
+		<edge from-layer="46" from-port="2" to-layer="48" to-port="0" />
+		<edge from-layer="47" from-port="0" to-layer="48" to-port="1" />
+		<edge from-layer="48" from-port="2" to-layer="50" to-port="0" />
+		<edge from-layer="49" from-port="0" to-layer="50" to-port="1" />
+		<edge from-layer="50" from-port="2" to-layer="59" to-port="1" />
 		<edge from-layer="51" from-port="0" to-layer="52" to-port="1" />
 		<edge from-layer="52" from-port="2" to-layer="54" to-port="0" />
 		<edge from-layer="53" from-port="0" to-layer="54" to-port="1" />
-		<edge from-layer="54" from-port="2" to-layer="57" to-port="0" />
-		<edge from-layer="55" from-port="0" to-layer="57" to-port="1" />
-		<edge from-layer="56" from-port="0" to-layer="57" to-port="2" />
-		<edge from-layer="57" from-port="3" to-layer="58" to-port="0" />
-		<edge from-layer="58" from-port="1" to-layer="60" to-port="0" />
-		<edge from-layer="59" from-port="0" to-layer="60" to-port="1" />
-		<edge from-layer="60" from-port="2" to-layer="62" to-port="0" />
-		<edge from-layer="61" from-port="0" to-layer="62" to-port="1" />
-		<edge from-layer="62" from-port="2" to-layer="63" to-port="1" />
-		<edge from-layer="63" from-port="2" to-layer="66" to-port="0" />
-		<edge from-layer="63" from-port="2" to-layer="80" to-port="0" />
-		<edge from-layer="64" from-port="0" to-layer="66" to-port="1" />
-		<edge from-layer="65" from-port="0" to-layer="66" to-port="2" />
-		<edge from-layer="66" from-port="3" to-layer="67" to-port="0" />
-		<edge from-layer="67" from-port="1" to-layer="69" to-port="0" />
+		<edge from-layer="54" from-port="2" to-layer="56" to-port="0" />
+		<edge from-layer="55" from-port="0" to-layer="56" to-port="1" />
+		<edge from-layer="56" from-port="2" to-layer="58" to-port="0" />
+		<edge from-layer="57" from-port="0" to-layer="58" to-port="1" />
+		<edge from-layer="58" from-port="2" to-layer="59" to-port="2" />
+		<edge from-layer="59" from-port="3" to-layer="61" to-port="0" />
+		<edge from-layer="60" from-port="0" to-layer="61" to-port="1" />
+		<edge from-layer="61" from-port="2" to-layer="63" to-port="0" />
+		<edge from-layer="62" from-port="0" to-layer="63" to-port="1" />
+		<edge from-layer="63" from-port="2" to-layer="65" to-port="0" />
+		<edge from-layer="64" from-port="0" to-layer="65" to-port="1" />
+		<edge from-layer="65" from-port="2" to-layer="67" to-port="0" />
+		<edge from-layer="66" from-port="0" to-layer="67" to-port="1" />
+		<edge from-layer="67" from-port="2" to-layer="69" to-port="0" />
 		<edge from-layer="68" from-port="0" to-layer="69" to-port="1" />
-		<edge from-layer="69" from-port="2" to-layer="71" to-port="0" />
-		<edge from-layer="70" from-port="0" to-layer="71" to-port="1" />
-		<edge from-layer="71" from-port="2" to-layer="74" to-port="0" />
-		<edge from-layer="72" from-port="0" to-layer="74" to-port="1" />
-		<edge from-layer="73" from-port="0" to-layer="74" to-port="2" />
-		<edge from-layer="74" from-port="3" to-layer="75" to-port="0" />
-		<edge from-layer="75" from-port="1" to-layer="77" to-port="0" />
-		<edge from-layer="76" from-port="0" to-layer="77" to-port="1" />
-		<edge from-layer="77" from-port="2" to-layer="79" to-port="0" />
-		<edge from-layer="78" from-port="0" to-layer="79" to-port="1" />
-		<edge from-layer="79" from-port="2" to-layer="80" to-port="1" />
-		<edge from-layer="80" from-port="2" to-layer="82" to-port="0" />
-		<edge from-layer="81" from-port="0" to-layer="82" to-port="1" />
-		<edge from-layer="82" from-port="2" to-layer="84" to-port="0" />
-		<edge from-layer="83" from-port="0" to-layer="84" to-port="1" />
-		<edge from-layer="84" from-port="2" to-layer="86" to-port="0" />
-		<edge from-layer="84" from-port="2" to-layer="91" to-port="0" />
+		<edge from-layer="69" from-port="2" to-layer="79" to-port="0" />
+		<edge from-layer="70" from-port="1" to-layer="73" to-port="0" />
+		<edge from-layer="70" from-port="1" to-layer="77" to-port="0" />
+		<edge from-layer="71" from-port="0" to-layer="73" to-port="1" />
+		<edge from-layer="72" from-port="0" to-layer="73" to-port="2" />
+		<edge from-layer="73" from-port="3" to-layer="78" to-port="0" />
+		<edge from-layer="74" from-port="0" to-layer="78" to-port="1" />
+		<edge from-layer="75" from-port="0" to-layer="77" to-port="1" />
+		<edge from-layer="76" from-port="0" to-layer="77" to-port="2" />
+		<edge from-layer="77" from-port="3" to-layer="78" to-port="2" />
+		<edge from-layer="78" from-port="3" to-layer="79" to-port="1" />
+		<edge from-layer="79" from-port="2" to-layer="80" to-port="0" />
+		<edge from-layer="80" from-port="2" to-layer="83" to-port="0" />
+		<edge from-layer="80" from-port="2" to-layer="97" to-port="0" />
+		<edge from-layer="81" from-port="0" to-layer="83" to-port="1" />
+		<edge from-layer="82" from-port="0" to-layer="83" to-port="2" />
+		<edge from-layer="83" from-port="3" to-layer="84" to-port="0" />
+		<edge from-layer="84" from-port="1" to-layer="86" to-port="0" />
 		<edge from-layer="85" from-port="0" to-layer="86" to-port="1" />
 		<edge from-layer="86" from-port="2" to-layer="88" to-port="0" />
 		<edge from-layer="87" from-port="0" to-layer="88" to-port="1" />
-		<edge from-layer="88" from-port="2" to-layer="105" to-port="0" />
+		<edge from-layer="88" from-port="2" to-layer="91" to-port="0" />
 		<edge from-layer="89" from-port="0" to-layer="91" to-port="1" />
 		<edge from-layer="90" from-port="0" to-layer="91" to-port="2" />
 		<edge from-layer="91" from-port="3" to-layer="92" to-port="0" />
@@ -5055,17 +5718,17 @@
 		<edge from-layer="93" from-port="0" to-layer="94" to-port="1" />
 		<edge from-layer="94" from-port="2" to-layer="96" to-port="0" />
 		<edge from-layer="95" from-port="0" to-layer="96" to-port="1" />
-		<edge from-layer="96" from-port="2" to-layer="99" to-port="0" />
-		<edge from-layer="97" from-port="0" to-layer="99" to-port="1" />
-		<edge from-layer="98" from-port="0" to-layer="99" to-port="2" />
-		<edge from-layer="99" from-port="3" to-layer="100" to-port="0" />
-		<edge from-layer="100" from-port="1" to-layer="102" to-port="0" />
-		<edge from-layer="101" from-port="0" to-layer="102" to-port="1" />
-		<edge from-layer="102" from-port="2" to-layer="104" to-port="0" />
-		<edge from-layer="103" from-port="0" to-layer="104" to-port="1" />
-		<edge from-layer="104" from-port="2" to-layer="105" to-port="1" />
+		<edge from-layer="96" from-port="2" to-layer="97" to-port="1" />
+		<edge from-layer="97" from-port="2" to-layer="100" to-port="0" />
+		<edge from-layer="97" from-port="2" to-layer="114" to-port="0" />
+		<edge from-layer="98" from-port="0" to-layer="100" to-port="1" />
+		<edge from-layer="99" from-port="0" to-layer="100" to-port="2" />
+		<edge from-layer="100" from-port="3" to-layer="101" to-port="0" />
+		<edge from-layer="101" from-port="1" to-layer="103" to-port="0" />
+		<edge from-layer="102" from-port="0" to-layer="103" to-port="1" />
+		<edge from-layer="103" from-port="2" to-layer="105" to-port="0" />
+		<edge from-layer="104" from-port="0" to-layer="105" to-port="1" />
 		<edge from-layer="105" from-port="2" to-layer="108" to-port="0" />
-		<edge from-layer="105" from-port="2" to-layer="122" to-port="0" />
 		<edge from-layer="106" from-port="0" to-layer="108" to-port="1" />
 		<edge from-layer="107" from-port="0" to-layer="108" to-port="2" />
 		<edge from-layer="108" from-port="3" to-layer="109" to-port="0" />
@@ -5073,215 +5736,243 @@
 		<edge from-layer="110" from-port="0" to-layer="111" to-port="1" />
 		<edge from-layer="111" from-port="2" to-layer="113" to-port="0" />
 		<edge from-layer="112" from-port="0" to-layer="113" to-port="1" />
-		<edge from-layer="113" from-port="2" to-layer="116" to-port="0" />
-		<edge from-layer="114" from-port="0" to-layer="116" to-port="1" />
-		<edge from-layer="115" from-port="0" to-layer="116" to-port="2" />
-		<edge from-layer="116" from-port="3" to-layer="117" to-port="0" />
-		<edge from-layer="117" from-port="1" to-layer="119" to-port="0" />
-		<edge from-layer="118" from-port="0" to-layer="119" to-port="1" />
-		<edge from-layer="119" from-port="2" to-layer="121" to-port="0" />
-		<edge from-layer="120" from-port="0" to-layer="121" to-port="1" />
-		<edge from-layer="121" from-port="2" to-layer="122" to-port="1" />
-		<edge from-layer="122" from-port="2" to-layer="124" to-port="0" />
-		<edge from-layer="123" from-port="0" to-layer="124" to-port="1" />
-		<edge from-layer="124" from-port="2" to-layer="126" to-port="0" />
-		<edge from-layer="125" from-port="0" to-layer="126" to-port="1" />
-		<edge from-layer="126" from-port="2" to-layer="129" to-port="0" />
-		<edge from-layer="126" from-port="2" to-layer="143" to-port="0" />
-		<edge from-layer="127" from-port="0" to-layer="129" to-port="1" />
-		<edge from-layer="128" from-port="0" to-layer="129" to-port="2" />
-		<edge from-layer="129" from-port="3" to-layer="130" to-port="0" />
-		<edge from-layer="130" from-port="1" to-layer="132" to-port="0" />
-		<edge from-layer="131" from-port="0" to-layer="132" to-port="1" />
-		<edge from-layer="132" from-port="2" to-layer="134" to-port="0" />
-		<edge from-layer="133" from-port="0" to-layer="134" to-port="1" />
-		<edge from-layer="134" from-port="2" to-layer="137" to-port="0" />
-		<edge from-layer="135" from-port="0" to-layer="137" to-port="1" />
-		<edge from-layer="136" from-port="0" to-layer="137" to-port="2" />
-		<edge from-layer="137" from-port="3" to-layer="138" to-port="0" />
-		<edge from-layer="138" from-port="1" to-layer="140" to-port="0" />
-		<edge from-layer="139" from-port="0" to-layer="140" to-port="1" />
-		<edge from-layer="140" from-port="2" to-layer="142" to-port="0" />
-		<edge from-layer="141" from-port="0" to-layer="142" to-port="1" />
-		<edge from-layer="142" from-port="2" to-layer="143" to-port="1" />
-		<edge from-layer="143" from-port="2" to-layer="146" to-port="0" />
-		<edge from-layer="143" from-port="2" to-layer="160" to-port="0" />
-		<edge from-layer="144" from-port="0" to-layer="146" to-port="1" />
-		<edge from-layer="145" from-port="0" to-layer="146" to-port="2" />
-		<edge from-layer="146" from-port="3" to-layer="147" to-port="0" />
-		<edge from-layer="147" from-port="1" to-layer="149" to-port="0" />
-		<edge from-layer="148" from-port="0" to-layer="149" to-port="1" />
-		<edge from-layer="149" from-port="2" to-layer="151" to-port="0" />
-		<edge from-layer="150" from-port="0" to-layer="151" to-port="1" />
-		<edge from-layer="151" from-port="2" to-layer="154" to-port="0" />
-		<edge from-layer="152" from-port="0" to-layer="154" to-port="1" />
-		<edge from-layer="153" from-port="0" to-layer="154" to-port="2" />
-		<edge from-layer="154" from-port="3" to-layer="155" to-port="0" />
-		<edge from-layer="155" from-port="1" to-layer="157" to-port="0" />
-		<edge from-layer="156" from-port="0" to-layer="157" to-port="1" />
-		<edge from-layer="157" from-port="2" to-layer="159" to-port="0" />
-		<edge from-layer="158" from-port="0" to-layer="159" to-port="1" />
-		<edge from-layer="159" from-port="2" to-layer="160" to-port="1" />
-		<edge from-layer="160" from-port="2" to-layer="163" to-port="0" />
-		<edge from-layer="160" from-port="2" to-layer="177" to-port="0" />
-		<edge from-layer="161" from-port="0" to-layer="163" to-port="1" />
-		<edge from-layer="162" from-port="0" to-layer="163" to-port="2" />
-		<edge from-layer="163" from-port="3" to-layer="164" to-port="0" />
-		<edge from-layer="164" from-port="1" to-layer="166" to-port="0" />
-		<edge from-layer="165" from-port="0" to-layer="166" to-port="1" />
-		<edge from-layer="166" from-port="2" to-layer="168" to-port="0" />
-		<edge from-layer="167" from-port="0" to-layer="168" to-port="1" />
-		<edge from-layer="168" from-port="2" to-layer="171" to-port="0" />
-		<edge from-layer="169" from-port="0" to-layer="171" to-port="1" />
-		<edge from-layer="170" from-port="0" to-layer="171" to-port="2" />
-		<edge from-layer="171" from-port="3" to-layer="172" to-port="0" />
-		<edge from-layer="172" from-port="1" to-layer="174" to-port="0" />
-		<edge from-layer="173" from-port="0" to-layer="174" to-port="1" />
-		<edge from-layer="174" from-port="2" to-layer="176" to-port="0" />
-		<edge from-layer="175" from-port="0" to-layer="176" to-port="1" />
-		<edge from-layer="176" from-port="2" to-layer="177" to-port="1" />
-		<edge from-layer="177" from-port="2" to-layer="179" to-port="0" />
-		<edge from-layer="177" from-port="2" to-layer="232" to-port="1" />
-		<edge from-layer="177" from-port="2" to-layer="222" to-port="0" />
-		<edge from-layer="178" from-port="0" to-layer="179" to-port="1" />
-		<edge from-layer="179" from-port="2" to-layer="181" to-port="0" />
-		<edge from-layer="180" from-port="0" to-layer="181" to-port="1" />
-		<edge from-layer="181" from-port="2" to-layer="183" to-port="0" />
-		<edge from-layer="182" from-port="0" to-layer="183" to-port="1" />
-		<edge from-layer="183" from-port="2" to-layer="186" to-port="0" />
-		<edge from-layer="184" from-port="0" to-layer="186" to-port="1" />
-		<edge from-layer="185" from-port="0" to-layer="186" to-port="2" />
-		<edge from-layer="186" from-port="3" to-layer="188" to-port="0" />
-		<edge from-layer="186" from-port="3" to-layer="196" to-port="0" />
-		<edge from-layer="186" from-port="3" to-layer="204" to-port="0" />
+		<edge from-layer="113" from-port="2" to-layer="114" to-port="1" />
+		<edge from-layer="114" from-port="2" to-layer="117" to-port="0" />
+		<edge from-layer="114" from-port="2" to-layer="131" to-port="0" />
+		<edge from-layer="115" from-port="0" to-layer="117" to-port="1" />
+		<edge from-layer="116" from-port="0" to-layer="117" to-port="2" />
+		<edge from-layer="117" from-port="3" to-layer="118" to-port="0" />
+		<edge from-layer="118" from-port="1" to-layer="120" to-port="0" />
+		<edge from-layer="119" from-port="0" to-layer="120" to-port="1" />
+		<edge from-layer="120" from-port="2" to-layer="122" to-port="0" />
+		<edge from-layer="121" from-port="0" to-layer="122" to-port="1" />
+		<edge from-layer="122" from-port="2" to-layer="125" to-port="0" />
+		<edge from-layer="123" from-port="0" to-layer="125" to-port="1" />
+		<edge from-layer="124" from-port="0" to-layer="125" to-port="2" />
+		<edge from-layer="125" from-port="3" to-layer="126" to-port="0" />
+		<edge from-layer="126" from-port="1" to-layer="128" to-port="0" />
+		<edge from-layer="127" from-port="0" to-layer="128" to-port="1" />
+		<edge from-layer="128" from-port="2" to-layer="130" to-port="0" />
+		<edge from-layer="129" from-port="0" to-layer="130" to-port="1" />
+		<edge from-layer="130" from-port="2" to-layer="131" to-port="1" />
+		<edge from-layer="131" from-port="2" to-layer="134" to-port="0" />
+		<edge from-layer="131" from-port="2" to-layer="148" to-port="0" />
+		<edge from-layer="132" from-port="0" to-layer="134" to-port="1" />
+		<edge from-layer="133" from-port="0" to-layer="134" to-port="2" />
+		<edge from-layer="134" from-port="3" to-layer="135" to-port="0" />
+		<edge from-layer="135" from-port="1" to-layer="137" to-port="0" />
+		<edge from-layer="136" from-port="0" to-layer="137" to-port="1" />
+		<edge from-layer="137" from-port="2" to-layer="139" to-port="0" />
+		<edge from-layer="138" from-port="0" to-layer="139" to-port="1" />
+		<edge from-layer="139" from-port="2" to-layer="142" to-port="0" />
+		<edge from-layer="140" from-port="0" to-layer="142" to-port="1" />
+		<edge from-layer="141" from-port="0" to-layer="142" to-port="2" />
+		<edge from-layer="142" from-port="3" to-layer="143" to-port="0" />
+		<edge from-layer="143" from-port="1" to-layer="145" to-port="0" />
+		<edge from-layer="144" from-port="0" to-layer="145" to-port="1" />
+		<edge from-layer="145" from-port="2" to-layer="147" to-port="0" />
+		<edge from-layer="146" from-port="0" to-layer="147" to-port="1" />
+		<edge from-layer="147" from-port="2" to-layer="148" to-port="1" />
+		<edge from-layer="148" from-port="2" to-layer="151" to-port="0" />
+		<edge from-layer="149" from-port="0" to-layer="151" to-port="1" />
+		<edge from-layer="150" from-port="0" to-layer="151" to-port="2" />
+		<edge from-layer="151" from-port="3" to-layer="153" to-port="0" />
+		<edge from-layer="152" from-port="0" to-layer="153" to-port="1" />
+		<edge from-layer="153" from-port="2" to-layer="155" to-port="0" />
+		<edge from-layer="154" from-port="0" to-layer="155" to-port="1" />
+		<edge from-layer="155" from-port="2" to-layer="158" to-port="0" />
+		<edge from-layer="155" from-port="2" to-layer="172" to-port="0" />
+		<edge from-layer="156" from-port="0" to-layer="158" to-port="1" />
+		<edge from-layer="157" from-port="0" to-layer="158" to-port="2" />
+		<edge from-layer="158" from-port="3" to-layer="159" to-port="0" />
+		<edge from-layer="159" from-port="1" to-layer="161" to-port="0" />
+		<edge from-layer="160" from-port="0" to-layer="161" to-port="1" />
+		<edge from-layer="161" from-port="2" to-layer="163" to-port="0" />
+		<edge from-layer="162" from-port="0" to-layer="163" to-port="1" />
+		<edge from-layer="163" from-port="2" to-layer="166" to-port="0" />
+		<edge from-layer="164" from-port="0" to-layer="166" to-port="1" />
+		<edge from-layer="165" from-port="0" to-layer="166" to-port="2" />
+		<edge from-layer="166" from-port="3" to-layer="167" to-port="0" />
+		<edge from-layer="167" from-port="1" to-layer="169" to-port="0" />
+		<edge from-layer="168" from-port="0" to-layer="169" to-port="1" />
+		<edge from-layer="169" from-port="2" to-layer="171" to-port="0" />
+		<edge from-layer="170" from-port="0" to-layer="171" to-port="1" />
+		<edge from-layer="171" from-port="2" to-layer="172" to-port="1" />
+		<edge from-layer="172" from-port="2" to-layer="175" to-port="0" />
+		<edge from-layer="172" from-port="2" to-layer="189" to-port="0" />
+		<edge from-layer="173" from-port="0" to-layer="175" to-port="1" />
+		<edge from-layer="174" from-port="0" to-layer="175" to-port="2" />
+		<edge from-layer="175" from-port="3" to-layer="176" to-port="0" />
+		<edge from-layer="176" from-port="1" to-layer="178" to-port="0" />
+		<edge from-layer="177" from-port="0" to-layer="178" to-port="1" />
+		<edge from-layer="178" from-port="2" to-layer="180" to-port="0" />
+		<edge from-layer="179" from-port="0" to-layer="180" to-port="1" />
+		<edge from-layer="180" from-port="2" to-layer="183" to-port="0" />
+		<edge from-layer="181" from-port="0" to-layer="183" to-port="1" />
+		<edge from-layer="182" from-port="0" to-layer="183" to-port="2" />
+		<edge from-layer="183" from-port="3" to-layer="184" to-port="0" />
+		<edge from-layer="184" from-port="1" to-layer="186" to-port="0" />
+		<edge from-layer="185" from-port="0" to-layer="186" to-port="1" />
+		<edge from-layer="186" from-port="2" to-layer="188" to-port="0" />
 		<edge from-layer="187" from-port="0" to-layer="188" to-port="1" />
-		<edge from-layer="188" from-port="2" to-layer="190" to-port="0" />
-		<edge from-layer="189" from-port="0" to-layer="190" to-port="1" />
-		<edge from-layer="190" from-port="2" to-layer="192" to-port="0" />
-		<edge from-layer="191" from-port="0" to-layer="192" to-port="1" />
-		<edge from-layer="192" from-port="2" to-layer="194" to-port="0" />
-		<edge from-layer="193" from-port="0" to-layer="194" to-port="1" />
-		<edge from-layer="194" from-port="2" to-layer="211" to-port="0" />
-		<edge from-layer="195" from-port="0" to-layer="196" to-port="1" />
-		<edge from-layer="196" from-port="2" to-layer="198" to-port="0" />
-		<edge from-layer="197" from-port="0" to-layer="198" to-port="1" />
-		<edge from-layer="198" from-port="2" to-layer="200" to-port="0" />
-		<edge from-layer="199" from-port="0" to-layer="200" to-port="1" />
-		<edge from-layer="200" from-port="2" to-layer="202" to-port="0" />
-		<edge from-layer="201" from-port="0" to-layer="202" to-port="1" />
-		<edge from-layer="202" from-port="2" to-layer="211" to-port="1" />
-		<edge from-layer="203" from-port="0" to-layer="204" to-port="1" />
-		<edge from-layer="204" from-port="2" to-layer="206" to-port="0" />
-		<edge from-layer="205" from-port="0" to-layer="206" to-port="1" />
-		<edge from-layer="206" from-port="2" to-layer="208" to-port="0" />
-		<edge from-layer="207" from-port="0" to-layer="208" to-port="1" />
-		<edge from-layer="208" from-port="2" to-layer="210" to-port="0" />
-		<edge from-layer="209" from-port="0" to-layer="210" to-port="1" />
-		<edge from-layer="210" from-port="2" to-layer="211" to-port="2" />
-		<edge from-layer="211" from-port="3" to-layer="213" to-port="0" />
+		<edge from-layer="188" from-port="2" to-layer="189" to-port="1" />
+		<edge from-layer="189" from-port="2" to-layer="192" to-port="0" />
+		<edge from-layer="189" from-port="2" to-layer="206" to-port="0" />
+		<edge from-layer="190" from-port="0" to-layer="192" to-port="1" />
+		<edge from-layer="191" from-port="0" to-layer="192" to-port="2" />
+		<edge from-layer="192" from-port="3" to-layer="193" to-port="0" />
+		<edge from-layer="193" from-port="1" to-layer="195" to-port="0" />
+		<edge from-layer="194" from-port="0" to-layer="195" to-port="1" />
+		<edge from-layer="195" from-port="2" to-layer="197" to-port="0" />
+		<edge from-layer="196" from-port="0" to-layer="197" to-port="1" />
+		<edge from-layer="197" from-port="2" to-layer="200" to-port="0" />
+		<edge from-layer="198" from-port="0" to-layer="200" to-port="1" />
+		<edge from-layer="199" from-port="0" to-layer="200" to-port="2" />
+		<edge from-layer="200" from-port="3" to-layer="201" to-port="0" />
+		<edge from-layer="201" from-port="1" to-layer="203" to-port="0" />
+		<edge from-layer="202" from-port="0" to-layer="203" to-port="1" />
+		<edge from-layer="203" from-port="2" to-layer="205" to-port="0" />
+		<edge from-layer="204" from-port="0" to-layer="205" to-port="1" />
+		<edge from-layer="205" from-port="2" to-layer="206" to-port="1" />
+		<edge from-layer="206" from-port="2" to-layer="209" to-port="0" />
+		<edge from-layer="207" from-port="0" to-layer="209" to-port="1" />
+		<edge from-layer="208" from-port="0" to-layer="209" to-port="2" />
+		<edge from-layer="209" from-port="3" to-layer="211" to-port="0" />
+		<edge from-layer="210" from-port="0" to-layer="211" to-port="1" />
+		<edge from-layer="211" from-port="2" to-layer="213" to-port="0" />
 		<edge from-layer="212" from-port="0" to-layer="213" to-port="1" />
 		<edge from-layer="213" from-port="2" to-layer="215" to-port="0" />
+		<edge from-layer="213" from-port="2" to-layer="220" to-port="0" />
 		<edge from-layer="214" from-port="0" to-layer="215" to-port="1" />
 		<edge from-layer="215" from-port="2" to-layer="217" to-port="0" />
 		<edge from-layer="216" from-port="0" to-layer="217" to-port="1" />
-		<edge from-layer="217" from-port="2" to-layer="219" to-port="0" />
-		<edge from-layer="218" from-port="0" to-layer="219" to-port="1" />
-		<edge from-layer="219" from-port="2" to-layer="221" to-port="0" />
-		<edge from-layer="220" from-port="0" to-layer="221" to-port="1" />
-		<edge from-layer="221" from-port="2" to-layer="231" to-port="0" />
-		<edge from-layer="222" from-port="1" to-layer="225" to-port="0" />
-		<edge from-layer="222" from-port="1" to-layer="229" to-port="0" />
-		<edge from-layer="223" from-port="0" to-layer="225" to-port="1" />
-		<edge from-layer="224" from-port="0" to-layer="225" to-port="2" />
-		<edge from-layer="225" from-port="3" to-layer="287" to-port="0" />
-		<edge from-layer="225" from-port="3" to-layer="230" to-port="0" />
-		<edge from-layer="226" from-port="0" to-layer="230" to-port="1" />
-		<edge from-layer="227" from-port="0" to-layer="229" to-port="1" />
-		<edge from-layer="228" from-port="0" to-layer="229" to-port="2" />
-		<edge from-layer="229" from-port="3" to-layer="230" to-port="2" />
-		<edge from-layer="230" from-port="3" to-layer="231" to-port="1" />
-		<edge from-layer="231" from-port="2" to-layer="232" to-port="0" />
-		<edge from-layer="232" from-port="2" to-layer="235" to-port="0" />
-		<edge from-layer="232" from-port="2" to-layer="249" to-port="0" />
-		<edge from-layer="233" from-port="0" to-layer="235" to-port="1" />
-		<edge from-layer="234" from-port="0" to-layer="235" to-port="2" />
-		<edge from-layer="235" from-port="3" to-layer="236" to-port="0" />
-		<edge from-layer="236" from-port="1" to-layer="238" to-port="0" />
-		<edge from-layer="237" from-port="0" to-layer="238" to-port="1" />
-		<edge from-layer="238" from-port="2" to-layer="240" to-port="0" />
+		<edge from-layer="217" from-port="2" to-layer="234" to-port="0" />
+		<edge from-layer="218" from-port="0" to-layer="220" to-port="1" />
+		<edge from-layer="219" from-port="0" to-layer="220" to-port="2" />
+		<edge from-layer="220" from-port="3" to-layer="221" to-port="0" />
+		<edge from-layer="221" from-port="1" to-layer="223" to-port="0" />
+		<edge from-layer="222" from-port="0" to-layer="223" to-port="1" />
+		<edge from-layer="223" from-port="2" to-layer="225" to-port="0" />
+		<edge from-layer="224" from-port="0" to-layer="225" to-port="1" />
+		<edge from-layer="225" from-port="2" to-layer="228" to-port="0" />
+		<edge from-layer="226" from-port="0" to-layer="228" to-port="1" />
+		<edge from-layer="227" from-port="0" to-layer="228" to-port="2" />
+		<edge from-layer="228" from-port="3" to-layer="229" to-port="0" />
+		<edge from-layer="229" from-port="1" to-layer="231" to-port="0" />
+		<edge from-layer="230" from-port="0" to-layer="231" to-port="1" />
+		<edge from-layer="231" from-port="2" to-layer="233" to-port="0" />
+		<edge from-layer="232" from-port="0" to-layer="233" to-port="1" />
+		<edge from-layer="233" from-port="2" to-layer="234" to-port="1" />
+		<edge from-layer="234" from-port="2" to-layer="237" to-port="0" />
+		<edge from-layer="234" from-port="2" to-layer="251" to-port="0" />
+		<edge from-layer="235" from-port="0" to-layer="237" to-port="1" />
+		<edge from-layer="236" from-port="0" to-layer="237" to-port="2" />
+		<edge from-layer="237" from-port="3" to-layer="238" to-port="0" />
+		<edge from-layer="238" from-port="1" to-layer="240" to-port="0" />
 		<edge from-layer="239" from-port="0" to-layer="240" to-port="1" />
-		<edge from-layer="240" from-port="2" to-layer="243" to-port="0" />
-		<edge from-layer="241" from-port="0" to-layer="243" to-port="1" />
-		<edge from-layer="242" from-port="0" to-layer="243" to-port="2" />
-		<edge from-layer="243" from-port="3" to-layer="244" to-port="0" />
-		<edge from-layer="244" from-port="1" to-layer="246" to-port="0" />
-		<edge from-layer="245" from-port="0" to-layer="246" to-port="1" />
-		<edge from-layer="246" from-port="2" to-layer="248" to-port="0" />
+		<edge from-layer="240" from-port="2" to-layer="242" to-port="0" />
+		<edge from-layer="241" from-port="0" to-layer="242" to-port="1" />
+		<edge from-layer="242" from-port="2" to-layer="245" to-port="0" />
+		<edge from-layer="243" from-port="0" to-layer="245" to-port="1" />
+		<edge from-layer="244" from-port="0" to-layer="245" to-port="2" />
+		<edge from-layer="245" from-port="3" to-layer="246" to-port="0" />
+		<edge from-layer="246" from-port="1" to-layer="248" to-port="0" />
 		<edge from-layer="247" from-port="0" to-layer="248" to-port="1" />
-		<edge from-layer="248" from-port="2" to-layer="249" to-port="1" />
-		<edge from-layer="249" from-port="2" to-layer="252" to-port="0" />
-		<edge from-layer="250" from-port="0" to-layer="252" to-port="1" />
-		<edge from-layer="251" from-port="0" to-layer="252" to-port="2" />
-		<edge from-layer="252" from-port="3" to-layer="253" to-port="0" />
-		<edge from-layer="253" from-port="1" to-layer="255" to-port="0" />
-		<edge from-layer="254" from-port="0" to-layer="255" to-port="1" />
-		<edge from-layer="255" from-port="2" to-layer="257" to-port="0" />
+		<edge from-layer="248" from-port="2" to-layer="250" to-port="0" />
+		<edge from-layer="249" from-port="0" to-layer="250" to-port="1" />
+		<edge from-layer="250" from-port="2" to-layer="251" to-port="1" />
+		<edge from-layer="251" from-port="2" to-layer="254" to-port="0" />
+		<edge from-layer="251" from-port="2" to-layer="268" to-port="0" />
+		<edge from-layer="252" from-port="0" to-layer="254" to-port="1" />
+		<edge from-layer="253" from-port="0" to-layer="254" to-port="2" />
+		<edge from-layer="254" from-port="3" to-layer="255" to-port="0" />
+		<edge from-layer="255" from-port="1" to-layer="257" to-port="0" />
 		<edge from-layer="256" from-port="0" to-layer="257" to-port="1" />
 		<edge from-layer="257" from-port="2" to-layer="259" to-port="0" />
 		<edge from-layer="258" from-port="0" to-layer="259" to-port="1" />
-		<edge from-layer="259" from-port="2" to-layer="261" to-port="0" />
-		<edge from-layer="260" from-port="0" to-layer="261" to-port="1" />
-		<edge from-layer="261" from-port="2" to-layer="263" to-port="0" />
-		<edge from-layer="261" from-port="2" to-layer="274" to-port="0" />
-		<edge from-layer="261" from-port="2" to-layer="283" to-port="0" />
-		<edge from-layer="262" from-port="0" to-layer="265" to-port="1" />
-		<edge from-layer="262" from-port="0" to-layer="274" to-port="1" />
+		<edge from-layer="259" from-port="2" to-layer="262" to-port="0" />
+		<edge from-layer="260" from-port="0" to-layer="262" to-port="1" />
+		<edge from-layer="261" from-port="0" to-layer="262" to-port="2" />
+		<edge from-layer="262" from-port="3" to-layer="263" to-port="0" />
 		<edge from-layer="263" from-port="1" to-layer="265" to-port="0" />
-		<edge from-layer="264" from-port="0" to-layer="265" to-port="2" />
-		<edge from-layer="264" from-port="0" to-layer="269" to-port="1" />
-		<edge from-layer="265" from-port="3" to-layer="267" to-port="0" />
-		<edge from-layer="265" from-port="3" to-layer="268" to-port="0" />
+		<edge from-layer="264" from-port="0" to-layer="265" to-port="1" />
+		<edge from-layer="265" from-port="2" to-layer="267" to-port="0" />
 		<edge from-layer="266" from-port="0" to-layer="267" to-port="1" />
-		<edge from-layer="266" from-port="0" to-layer="268" to-port="1" />
-		<edge from-layer="267" from-port="2" to-layer="271" to-port="0" />
-		<edge from-layer="268" from-port="2" to-layer="269" to-port="0" />
-		<edge from-layer="269" from-port="2" to-layer="270" to-port="0" />
-		<edge from-layer="270" from-port="1" to-layer="271" to-port="1" />
-		<edge from-layer="271" from-port="2" to-layer="273" to-port="0" />
+		<edge from-layer="267" from-port="2" to-layer="268" to-port="1" />
+		<edge from-layer="268" from-port="2" to-layer="271" to-port="0" />
+		<edge from-layer="269" from-port="0" to-layer="271" to-port="1" />
+		<edge from-layer="270" from-port="0" to-layer="271" to-port="2" />
+		<edge from-layer="271" from-port="3" to-layer="273" to-port="0" />
 		<edge from-layer="272" from-port="0" to-layer="273" to-port="1" />
-		<edge from-layer="273" from-port="2" to-layer="274" to-port="2" />
-		<edge from-layer="274" from-port="4" to-layer="275" to-port="0" />
-		<edge from-layer="274" from-port="3" to-layer="279" to-port="0" />
-		<edge from-layer="274" from-port="3" to-layer="302" to-port="0" />
-		<edge from-layer="275" from-port="1" to-layer="277" to-port="0" />
+		<edge from-layer="273" from-port="2" to-layer="275" to-port="0" />
+		<edge from-layer="274" from-port="0" to-layer="275" to-port="1" />
+		<edge from-layer="275" from-port="2" to-layer="277" to-port="0" />
+		<edge from-layer="275" from-port="2" to-layer="282" to-port="0" />
 		<edge from-layer="276" from-port="0" to-layer="277" to-port="1" />
-		<edge from-layer="277" from-port="2" to-layer="278" to-port="0" />
-		<edge from-layer="278" from-port="1" to-layer="301" to-port="0" />
-		<edge from-layer="279" from-port="1" to-layer="282" to-port="0" />
+		<edge from-layer="277" from-port="2" to-layer="279" to-port="0" />
+		<edge from-layer="278" from-port="0" to-layer="279" to-port="1" />
+		<edge from-layer="279" from-port="2" to-layer="296" to-port="0" />
 		<edge from-layer="280" from-port="0" to-layer="282" to-port="1" />
 		<edge from-layer="281" from-port="0" to-layer="282" to-port="2" />
-		<edge from-layer="282" from-port="3" to-layer="287" to-port="1" />
-		<edge from-layer="283" from-port="1" to-layer="286" to-port="0" />
-		<edge from-layer="284" from-port="0" to-layer="286" to-port="1" />
-		<edge from-layer="285" from-port="0" to-layer="286" to-port="2" />
-		<edge from-layer="286" from-port="3" to-layer="287" to-port="2" />
-		<edge from-layer="287" from-port="3" to-layer="288" to-port="0" />
-		<edge from-layer="288" from-port="1" to-layer="291" to-port="0" />
-		<edge from-layer="288" from-port="1" to-layer="298" to-port="0" />
-		<edge from-layer="289" from-port="0" to-layer="291" to-port="1" />
-		<edge from-layer="290" from-port="0" to-layer="291" to-port="2" />
-		<edge from-layer="291" from-port="3" to-layer="292" to-port="0" />
-		<edge from-layer="292" from-port="1" to-layer="294" to-port="0" />
-		<edge from-layer="293" from-port="0" to-layer="294" to-port="1" />
-		<edge from-layer="294" from-port="2" to-layer="295" to-port="0" />
-		<edge from-layer="295" from-port="1" to-layer="300" to-port="0" />
-		<edge from-layer="296" from-port="0" to-layer="298" to-port="1" />
-		<edge from-layer="297" from-port="0" to-layer="298" to-port="2" />
-		<edge from-layer="298" from-port="3" to-layer="299" to-port="0" />
-		<edge from-layer="299" from-port="1" to-layer="300" to-port="1" />
-		<edge from-layer="300" from-port="2" to-layer="301" to-port="1" />
-		<edge from-layer="301" from-port="2" to-layer="302" to-port="1" />
-		<edge from-layer="302" from-port="2" to-layer="303" to-port="0" />
+		<edge from-layer="282" from-port="3" to-layer="283" to-port="0" />
+		<edge from-layer="283" from-port="1" to-layer="285" to-port="0" />
+		<edge from-layer="284" from-port="0" to-layer="285" to-port="1" />
+		<edge from-layer="285" from-port="2" to-layer="287" to-port="0" />
+		<edge from-layer="286" from-port="0" to-layer="287" to-port="1" />
+		<edge from-layer="287" from-port="2" to-layer="290" to-port="0" />
+		<edge from-layer="288" from-port="0" to-layer="290" to-port="1" />
+		<edge from-layer="289" from-port="0" to-layer="290" to-port="2" />
+		<edge from-layer="290" from-port="3" to-layer="291" to-port="0" />
+		<edge from-layer="291" from-port="1" to-layer="293" to-port="0" />
+		<edge from-layer="292" from-port="0" to-layer="293" to-port="1" />
+		<edge from-layer="293" from-port="2" to-layer="295" to-port="0" />
+		<edge from-layer="294" from-port="0" to-layer="295" to-port="1" />
+		<edge from-layer="295" from-port="2" to-layer="296" to-port="1" />
+		<edge from-layer="296" from-port="2" to-layer="299" to-port="0" />
+		<edge from-layer="296" from-port="2" to-layer="313" to-port="0" />
+		<edge from-layer="297" from-port="0" to-layer="299" to-port="1" />
+		<edge from-layer="298" from-port="0" to-layer="299" to-port="2" />
+		<edge from-layer="299" from-port="3" to-layer="300" to-port="0" />
+		<edge from-layer="300" from-port="1" to-layer="302" to-port="0" />
+		<edge from-layer="301" from-port="0" to-layer="302" to-port="1" />
+		<edge from-layer="302" from-port="2" to-layer="304" to-port="0" />
+		<edge from-layer="303" from-port="0" to-layer="304" to-port="1" />
+		<edge from-layer="304" from-port="2" to-layer="307" to-port="0" />
+		<edge from-layer="305" from-port="0" to-layer="307" to-port="1" />
+		<edge from-layer="306" from-port="0" to-layer="307" to-port="2" />
+		<edge from-layer="307" from-port="3" to-layer="308" to-port="0" />
+		<edge from-layer="308" from-port="1" to-layer="310" to-port="0" />
+		<edge from-layer="309" from-port="0" to-layer="310" to-port="1" />
+		<edge from-layer="310" from-port="2" to-layer="312" to-port="0" />
+		<edge from-layer="311" from-port="0" to-layer="312" to-port="1" />
+		<edge from-layer="312" from-port="2" to-layer="313" to-port="1" />
+		<edge from-layer="313" from-port="2" to-layer="316" to-port="0" />
+		<edge from-layer="313" from-port="2" to-layer="330" to-port="0" />
+		<edge from-layer="314" from-port="0" to-layer="316" to-port="1" />
+		<edge from-layer="315" from-port="0" to-layer="316" to-port="2" />
+		<edge from-layer="316" from-port="3" to-layer="317" to-port="0" />
+		<edge from-layer="317" from-port="1" to-layer="319" to-port="0" />
+		<edge from-layer="318" from-port="0" to-layer="319" to-port="1" />
+		<edge from-layer="319" from-port="2" to-layer="321" to-port="0" />
+		<edge from-layer="320" from-port="0" to-layer="321" to-port="1" />
+		<edge from-layer="321" from-port="2" to-layer="324" to-port="0" />
+		<edge from-layer="322" from-port="0" to-layer="324" to-port="1" />
+		<edge from-layer="323" from-port="0" to-layer="324" to-port="2" />
+		<edge from-layer="324" from-port="3" to-layer="325" to-port="0" />
+		<edge from-layer="325" from-port="1" to-layer="327" to-port="0" />
+		<edge from-layer="326" from-port="0" to-layer="327" to-port="1" />
+		<edge from-layer="327" from-port="2" to-layer="329" to-port="0" />
+		<edge from-layer="328" from-port="0" to-layer="329" to-port="1" />
+		<edge from-layer="329" from-port="2" to-layer="330" to-port="1" />
+		<edge from-layer="330" from-port="2" to-layer="333" to-port="0" />
+		<edge from-layer="331" from-port="0" to-layer="333" to-port="1" />
+		<edge from-layer="332" from-port="0" to-layer="333" to-port="2" />
+		<edge from-layer="333" from-port="3" to-layer="334" to-port="0" />
+		<edge from-layer="334" from-port="1" to-layer="336" to-port="0" />
+		<edge from-layer="335" from-port="0" to-layer="336" to-port="1" />
+		<edge from-layer="336" from-port="2" to-layer="338" to-port="0" />
+		<edge from-layer="337" from-port="0" to-layer="338" to-port="1" />
+		<edge from-layer="338" from-port="2" to-layer="339" to-port="0" />
 	</edges>
 	<rt_info>
 		<Runtime_version value="2024.4.0-16579-c3152d32c9c-releases/2024/4" />