diff --git "a/vae_encoder.xml" "b/vae_encoder.xml"
--- "a/vae_encoder.xml"
+++ "b/vae_encoder.xml"
@@ -15,7 +15,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="1" name="Constant_13488" type="Const" version="opset1">
+		<layer id="1" name="Constant_13640" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 1, 512" offset="0" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -25,7 +25,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="2" name="Constant_13483" type="Const" version="opset1">
+		<layer id="2" name="Constant_13635" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 1, 512" offset="2048" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -35,13 +35,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="3" name="m.encoder.conv_in.weight" type="Const" version="opset1">
+		<layer id="3" name="vae.encoder.conv_in.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="128, 3, 3, 3" offset="4096" size="13824"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.conv_in.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.conv_in.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.conv_in.weight">
+				<port id="0" precision="FP32" names="vae.encoder.conv_in.weight">
 					<dim>128</dim>
 					<dim>3</dim>
 					<dim>3</dim>
@@ -49,10 +49,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="4" name="Convolution_114" type="Convolution" version="opset1">
+		<layer id="4" name="Conv_8/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_114"/>
+				<attribute name="fused_names" version="0" value="Conv_8/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -88,10 +88,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="6" name="onnx::Cast_249" type="Add" version="opset1">
+		<layer id="6" name="Conv_8" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_133, Reshape_134, input, onnx::Cast_249"/>
+				<attribute name="fused_names" version="0" value="Cast_9, Concat_133, Conv_8, Reshape_134"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -108,7 +108,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input,onnx::Cast_249">
+				<port id="2" precision="FP32" names="input,x">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -116,21 +116,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="7" name="onnx::Reshape_251" type="Const" version="opset1">
+		<layer id="7" name="Constant_10" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_251"/>
+				<attribute name="fused_names" version="0" value="Constant_10"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_251">
+				<port id="0" precision="I64" names="onnx::Reshape_115">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="8" name="onnx::InstanceNormalization_252" type="Reshape" version="opset1">
+		<layer id="8" name="Reshape_11" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_252"/>
+				<attribute name="fused_names" version="0" value="Reshape_11"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -144,7 +144,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_252">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_116">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>1048576</dim>
@@ -165,7 +165,7 @@
 		<layer id="10" name="MVN_173" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_192, Concat_237, MVN_173, Multiply_220, Reshape_193, Reshape_238, onnx::Reshape_255"/>
+				<attribute name="fused_names" version="0" value="Concat_192, Concat_237, InstanceNormalization_14, MVN_173, Multiply_220, Reshape_193, Reshape_238"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -178,17 +178,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_255">
+				<port id="2" precision="FP32" names="onnx::Reshape_119">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>1048576</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="11" name="onnx::Reshape_256" type="ShapeOf" version="opset3">
+		<layer id="11" name="Shape_15" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_256"/>
+				<attribute name="fused_names" version="0" value="Shape_15"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -199,15 +199,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_256">
+				<port id="1" precision="I64" names="onnx::Reshape_120">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="12" name="onnx::Mul_257" type="Reshape" version="opset1">
+		<layer id="12" name="Reshape_16" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_257"/>
+				<attribute name="fused_names" version="0" value="Reshape_16"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -220,7 +220,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_257">
+				<port id="2" precision="FP32" names="onnx::Mul_121">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -228,7 +228,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="13" name="Constant_13445" type="Const" version="opset1">
+		<layer id="13" name="Constant_13597" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 128, 1, 1" offset="18464" size="512"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -239,10 +239,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="14" name="onnx::Add_260" type="Multiply" version="opset1">
+		<layer id="14" name="Mul_17" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_260"/>
+				<attribute name="fused_names" version="0" value="Mul_17"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -259,7 +259,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_260">
+				<port id="2" precision="FP32" names="onnx::Add_124">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -267,7 +267,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="15" name="Constant_13446" type="Const" version="opset1">
+		<layer id="15" name="Constant_13598" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 128, 1, 1" offset="18976" size="512"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -278,10 +278,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="16" name="onnx::Cast_263" type="Add" version="opset1">
+		<layer id="16" name="Add_18" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.4, onnx::Cast_263"/>
+				<attribute name="fused_names" version="0" value="Add_18, Cast_19"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -298,7 +298,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.4,onnx::Cast_263">
+				<port id="2" precision="FP32" names="input.3,onnx::Cast_127">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -306,9 +306,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="17" name="input.8" type="Swish" version="opset4">
+		<layer id="17" name="Mul_21" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.8, onnx::Mul_265"/>
+				<attribute name="fused_names" version="0" value="Mul_21, Sigmoid_20"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -319,7 +319,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.8">
+				<port id="1" precision="FP32" names="input.7">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -327,13 +327,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="18" name="m.encoder.down_blocks.0.resnets.0.conv1.weight" type="Const" version="opset1">
+		<layer id="18" name="vae.encoder.down_blocks.0.resnets.0.conv1.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="128, 128, 3, 3" offset="19488" size="589824"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.0.resnets.0.conv1.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.0.resnets.0.conv1.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.0.resnets.0.conv1.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.0.resnets.0.conv1.weight">
 					<dim>128</dim>
 					<dim>128</dim>
 					<dim>3</dim>
@@ -341,10 +341,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="19" name="Convolution_278" type="Convolution" version="opset1">
+		<layer id="19" name="Conv_22/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_278"/>
+				<attribute name="fused_names" version="0" value="Conv_22/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -380,10 +380,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="21" name="onnx::Cast_267" type="Add" version="opset1">
+		<layer id="21" name="Conv_22" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_297, Reshape_298, input.12, onnx::Cast_267"/>
+				<attribute name="fused_names" version="0" value="Cast_23, Concat_297, Conv_22, Reshape_298"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -400,7 +400,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.12,onnx::Cast_267">
+				<port id="2" precision="FP32" names="h,input.11">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -408,21 +408,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="22" name="onnx::Reshape_269" type="Const" version="opset1">
+		<layer id="22" name="Constant_24" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_269"/>
+				<attribute name="fused_names" version="0" value="Constant_24"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_269">
+				<port id="0" precision="I64" names="onnx::Reshape_133">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="23" name="onnx::InstanceNormalization_270" type="Reshape" version="opset1">
+		<layer id="23" name="Reshape_25" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_270"/>
+				<attribute name="fused_names" version="0" value="Reshape_25"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -436,7 +436,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_270">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_134">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>1048576</dim>
@@ -457,7 +457,7 @@
 		<layer id="25" name="MVN_337" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_356, Concat_401, MVN_337, Multiply_384, Reshape_357, Reshape_402, onnx::Reshape_273"/>
+				<attribute name="fused_names" version="0" value="Concat_356, Concat_401, InstanceNormalization_28, MVN_337, Multiply_384, Reshape_357, Reshape_402"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -470,17 +470,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_273">
+				<port id="2" precision="FP32" names="onnx::Reshape_137">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>1048576</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="26" name="onnx::Reshape_274" type="ShapeOf" version="opset3">
+		<layer id="26" name="Shape_29" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_274"/>
+				<attribute name="fused_names" version="0" value="Shape_29"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -491,15 +491,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_274">
+				<port id="1" precision="I64" names="onnx::Reshape_138">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="27" name="onnx::Mul_275" type="Reshape" version="opset1">
+		<layer id="27" name="Reshape_30" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_275"/>
+				<attribute name="fused_names" version="0" value="Reshape_30"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -512,7 +512,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_275">
+				<port id="2" precision="FP32" names="onnx::Mul_139">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -520,7 +520,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="28" name="Constant_13447" type="Const" version="opset1">
+		<layer id="28" name="Constant_13599" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 128, 1, 1" offset="609824" size="512"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -531,10 +531,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="29" name="onnx::Add_278" type="Multiply" version="opset1">
+		<layer id="29" name="Mul_31" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_278"/>
+				<attribute name="fused_names" version="0" value="Mul_31"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -551,7 +551,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_278">
+				<port id="2" precision="FP32" names="onnx::Add_142">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -559,7 +559,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="30" name="Constant_13448" type="Const" version="opset1">
+		<layer id="30" name="Constant_13600" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 128, 1, 1" offset="610336" size="512"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -570,10 +570,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="31" name="onnx::Cast_281" type="Add" version="opset1">
+		<layer id="31" name="Add_32" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.16, onnx::Cast_281"/>
+				<attribute name="fused_names" version="0" value="Add_32, Cast_33"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -590,7 +590,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.16,onnx::Cast_281">
+				<port id="2" precision="FP32" names="input.15,onnx::Cast_145">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -598,9 +598,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="32" name="input.20" type="Swish" version="opset4">
+		<layer id="32" name="Mul_35" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.20, onnx::Mul_283"/>
+				<attribute name="fused_names" version="0" value="Mul_35, Sigmoid_34"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -611,7 +611,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.20">
+				<port id="1" precision="FP32" names="input.19">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -619,13 +619,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="33" name="m.encoder.down_blocks.0.resnets.0.conv2.weight" type="Const" version="opset1">
+		<layer id="33" name="vae.encoder.down_blocks.0.resnets.0.conv2.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="128, 128, 3, 3" offset="610848" size="589824"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.0.resnets.0.conv2.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.0.resnets.0.conv2.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.0.resnets.0.conv2.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.0.resnets.0.conv2.weight">
 					<dim>128</dim>
 					<dim>128</dim>
 					<dim>3</dim>
@@ -633,10 +633,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="34" name="Convolution_442" type="Convolution" version="opset1">
+		<layer id="34" name="Conv_36/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_442"/>
+				<attribute name="fused_names" version="0" value="Conv_36/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -672,10 +672,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="36" name="onnx::Add_285" type="Add" version="opset1">
+		<layer id="36" name="Conv_36" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_461, Reshape_462, onnx::Add_285"/>
+				<attribute name="fused_names" version="0" value="Concat_461, Conv_36, Reshape_462"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -692,7 +692,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_285">
+				<port id="2" precision="FP32" names="h.3">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -700,10 +700,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="37" name="onnx::Div_286" type="Add" version="opset1">
+		<layer id="37" name="Add_37" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.24, onnx::Cast_288, onnx::Div_286"/>
+				<attribute name="fused_names" version="0" value="Add_37, Cast_40, Div_39"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -720,7 +720,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.24,onnx::Cast_288,onnx::Div_286">
+				<port id="2" precision="FP32" names="input.23,onnx::Div_150,x.4">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -728,21 +728,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="38" name="onnx::Reshape_290" type="Const" version="opset1">
+		<layer id="38" name="Constant_41" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_290"/>
+				<attribute name="fused_names" version="0" value="Constant_41"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_290">
+				<port id="0" precision="I64" names="onnx::Reshape_154">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="39" name="onnx::InstanceNormalization_291" type="Reshape" version="opset1">
+		<layer id="39" name="Reshape_42" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_291"/>
+				<attribute name="fused_names" version="0" value="Reshape_42"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -756,7 +756,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_291">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_155">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>1048576</dim>
@@ -777,7 +777,7 @@
 		<layer id="41" name="MVN_504" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_523, Concat_568, MVN_504, Multiply_551, Reshape_524, Reshape_569, onnx::Reshape_294"/>
+				<attribute name="fused_names" version="0" value="Concat_523, Concat_568, InstanceNormalization_45, MVN_504, Multiply_551, Reshape_524, Reshape_569"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -790,17 +790,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_294">
+				<port id="2" precision="FP32" names="onnx::Reshape_158">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>1048576</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="42" name="onnx::Reshape_295" type="ShapeOf" version="opset3">
+		<layer id="42" name="Shape_46" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_295"/>
+				<attribute name="fused_names" version="0" value="Shape_46"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -811,15 +811,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_295">
+				<port id="1" precision="I64" names="onnx::Reshape_159">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="43" name="onnx::Mul_296" type="Reshape" version="opset1">
+		<layer id="43" name="Reshape_47" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_296"/>
+				<attribute name="fused_names" version="0" value="Reshape_47"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -832,7 +832,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_296">
+				<port id="2" precision="FP32" names="onnx::Mul_160">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -840,7 +840,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="44" name="Constant_13449" type="Const" version="opset1">
+		<layer id="44" name="Constant_13601" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 128, 1, 1" offset="1201184" size="512"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -851,10 +851,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="45" name="onnx::Add_299" type="Multiply" version="opset1">
+		<layer id="45" name="Mul_48" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_299"/>
+				<attribute name="fused_names" version="0" value="Mul_48"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -871,7 +871,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_299">
+				<port id="2" precision="FP32" names="onnx::Add_163">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -879,7 +879,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="46" name="Constant_13450" type="Const" version="opset1">
+		<layer id="46" name="Constant_13602" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 128, 1, 1" offset="1201696" size="512"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -890,10 +890,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="47" name="onnx::Cast_302" type="Add" version="opset1">
+		<layer id="47" name="Add_49" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.28, onnx::Cast_302"/>
+				<attribute name="fused_names" version="0" value="Add_49, Cast_50"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -910,7 +910,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.28,onnx::Cast_302">
+				<port id="2" precision="FP32" names="input.27,onnx::Cast_166">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -918,9 +918,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="48" name="input.32" type="Swish" version="opset4">
+		<layer id="48" name="Mul_52" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.32, onnx::Mul_304"/>
+				<attribute name="fused_names" version="0" value="Mul_52, Sigmoid_51"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -931,7 +931,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.32">
+				<port id="1" precision="FP32" names="input.31">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -939,13 +939,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="49" name="m.encoder.down_blocks.0.resnets.1.conv1.weight" type="Const" version="opset1">
+		<layer id="49" name="vae.encoder.down_blocks.0.resnets.1.conv1.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="128, 128, 3, 3" offset="1202208" size="589824"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.0.resnets.1.conv1.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.0.resnets.1.conv1.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.0.resnets.1.conv1.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.0.resnets.1.conv1.weight">
 					<dim>128</dim>
 					<dim>128</dim>
 					<dim>3</dim>
@@ -953,10 +953,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="50" name="Convolution_609" type="Convolution" version="opset1">
+		<layer id="50" name="Conv_53/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_609"/>
+				<attribute name="fused_names" version="0" value="Conv_53/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -992,10 +992,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="52" name="onnx::Cast_306" type="Add" version="opset1">
+		<layer id="52" name="Conv_53" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_628, Reshape_629, input.36, onnx::Cast_306"/>
+				<attribute name="fused_names" version="0" value="Cast_54, Concat_628, Conv_53, Reshape_629"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1012,7 +1012,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.36,onnx::Cast_306">
+				<port id="2" precision="FP32" names="h.7,input.35">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -1020,21 +1020,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="53" name="onnx::Reshape_308" type="Const" version="opset1">
+		<layer id="53" name="Constant_55" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_308"/>
+				<attribute name="fused_names" version="0" value="Constant_55"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_308">
+				<port id="0" precision="I64" names="onnx::Reshape_172">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="54" name="onnx::InstanceNormalization_309" type="Reshape" version="opset1">
+		<layer id="54" name="Reshape_56" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_309"/>
+				<attribute name="fused_names" version="0" value="Reshape_56"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1048,7 +1048,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_309">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_173">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>1048576</dim>
@@ -1069,7 +1069,7 @@
 		<layer id="56" name="MVN_668" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_687, Concat_732, MVN_668, Multiply_715, Reshape_688, Reshape_733, onnx::Reshape_312"/>
+				<attribute name="fused_names" version="0" value="Concat_687, Concat_732, InstanceNormalization_59, MVN_668, Multiply_715, Reshape_688, Reshape_733"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1082,17 +1082,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_312">
+				<port id="2" precision="FP32" names="onnx::Reshape_176">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>1048576</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="57" name="onnx::Reshape_313" type="ShapeOf" version="opset3">
+		<layer id="57" name="Shape_60" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_313"/>
+				<attribute name="fused_names" version="0" value="Shape_60"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1103,15 +1103,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_313">
+				<port id="1" precision="I64" names="onnx::Reshape_177">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="58" name="onnx::Mul_314" type="Reshape" version="opset1">
+		<layer id="58" name="Reshape_61" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_314"/>
+				<attribute name="fused_names" version="0" value="Reshape_61"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1124,7 +1124,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_314">
+				<port id="2" precision="FP32" names="onnx::Mul_178">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -1132,7 +1132,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="59" name="Constant_13451" type="Const" version="opset1">
+		<layer id="59" name="Constant_13603" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 128, 1, 1" offset="1792544" size="512"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -1143,10 +1143,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="60" name="onnx::Add_317" type="Multiply" version="opset1">
+		<layer id="60" name="Mul_62" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_317"/>
+				<attribute name="fused_names" version="0" value="Mul_62"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1163,7 +1163,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_317">
+				<port id="2" precision="FP32" names="onnx::Add_181">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -1171,7 +1171,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="61" name="Constant_13452" type="Const" version="opset1">
+		<layer id="61" name="Constant_13604" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 128, 1, 1" offset="1793056" size="512"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -1182,10 +1182,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="62" name="onnx::Cast_320" type="Add" version="opset1">
+		<layer id="62" name="Add_63" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.40, onnx::Cast_320"/>
+				<attribute name="fused_names" version="0" value="Add_63, Cast_64"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1202,7 +1202,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.40,onnx::Cast_320">
+				<port id="2" precision="FP32" names="input.39,onnx::Cast_184">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -1210,9 +1210,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="63" name="input.44" type="Swish" version="opset4">
+		<layer id="63" name="Mul_66" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.44, onnx::Mul_322"/>
+				<attribute name="fused_names" version="0" value="Mul_66, Sigmoid_65"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1223,7 +1223,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.44">
+				<port id="1" precision="FP32" names="input.43">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -1231,13 +1231,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="64" name="m.encoder.down_blocks.0.resnets.1.conv2.weight" type="Const" version="opset1">
+		<layer id="64" name="vae.encoder.down_blocks.0.resnets.1.conv2.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="128, 128, 3, 3" offset="1793568" size="589824"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.0.resnets.1.conv2.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.0.resnets.1.conv2.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.0.resnets.1.conv2.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.0.resnets.1.conv2.weight">
 					<dim>128</dim>
 					<dim>128</dim>
 					<dim>3</dim>
@@ -1245,10 +1245,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="65" name="Convolution_773" type="Convolution" version="opset1">
+		<layer id="65" name="Conv_67/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_773"/>
+				<attribute name="fused_names" version="0" value="Conv_67/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1284,10 +1284,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="67" name="onnx::Add_324" type="Add" version="opset1">
+		<layer id="67" name="Conv_67" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_792, Reshape_793, onnx::Add_324"/>
+				<attribute name="fused_names" version="0" value="Concat_792, Conv_67, Reshape_793"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1304,7 +1304,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_324">
+				<port id="2" precision="FP32" names="h.11">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -1312,10 +1312,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="68" name="onnx::Div_325" type="Add" version="opset1">
+		<layer id="68" name="Add_68" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Div_325, onnx::Pad_327"/>
+				<attribute name="fused_names" version="0" value="Add_68, Div_70"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1332,7 +1332,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Div_325,onnx::Pad_327">
+				<port id="2" precision="FP32" names="onnx::Div_189,x.8">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>512</dim>
@@ -1340,13 +1340,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="69" name="m.encoder.down_blocks.0.downsamplers.0.conv.weight" type="Const" version="opset1">
+		<layer id="69" name="vae.encoder.down_blocks.0.downsamplers.0.conv.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="128, 128, 3, 3" offset="2383904" size="589824"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.0.downsamplers.0.conv.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.0.downsamplers.0.conv.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.0.downsamplers.0.conv.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.0.downsamplers.0.conv.weight">
 					<dim>128</dim>
 					<dim>128</dim>
 					<dim>3</dim>
@@ -1354,10 +1354,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="70" name="Convolution_925" type="Convolution" version="opset1">
+		<layer id="70" name="Conv_86/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="2, 2" dilations="1, 1" pads_begin="0, 0" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_925, Split_857, input.48"/>
+				<attribute name="fused_names" version="0" value="Conv_86/WithoutBiases, Pad_85, Split_857"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1393,10 +1393,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="72" name="onnx::Cast_352" type="Add" version="opset1">
+		<layer id="72" name="Conv_86" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_944, Reshape_945, input.52, onnx::Cast_352"/>
+				<attribute name="fused_names" version="0" value="Cast_87, Concat_944, Conv_86, Reshape_945"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1413,7 +1413,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.52,onnx::Cast_352">
+				<port id="2" precision="FP32" names="input.47,x.16">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>256</dim>
@@ -1421,13 +1421,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="73" name="m.encoder.down_blocks.1.resnets.0.conv_shortcut.weight" type="Const" version="opset1">
+		<layer id="73" name="vae.encoder.down_blocks.1.resnets.0.conv_shortcut.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="256, 128, 1, 1" offset="2974240" size="131072"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.1.resnets.0.conv_shortcut.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.1.resnets.0.conv_shortcut.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.1.resnets.0.conv_shortcut.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.1.resnets.0.conv_shortcut.weight">
 					<dim>256</dim>
 					<dim>128</dim>
 					<dim>1</dim>
@@ -1435,10 +1435,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="74" name="Convolution_1301" type="Convolution" version="opset1">
+		<layer id="74" name="Conv_115/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="0, 0" pads_end="0, 0" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_1301"/>
+				<attribute name="fused_names" version="0" value="Conv_115/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1474,10 +1474,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="76" name="onnx::Add_389" type="Add" version="opset1">
+		<layer id="76" name="Conv_115" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_1320, Reshape_1321, onnx::Add_389"/>
+				<attribute name="fused_names" version="0" value="Concat_1320, Conv_115, Reshape_1321"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1494,7 +1494,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_389">
+				<port id="2" precision="FP32" names="x.20">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -1502,21 +1502,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="77" name="onnx::Reshape_354" type="Const" version="opset1">
+		<layer id="77" name="Constant_88" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_354"/>
+				<attribute name="fused_names" version="0" value="Constant_88"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_354">
+				<port id="0" precision="I64" names="onnx::Reshape_217">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="78" name="onnx::InstanceNormalization_355" type="Reshape" version="opset1">
+		<layer id="78" name="Reshape_89" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_355"/>
+				<attribute name="fused_names" version="0" value="Reshape_89"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1530,7 +1530,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_355">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_218">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>262144</dim>
@@ -1551,7 +1551,7 @@
 		<layer id="80" name="MVN_984" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_1003, Concat_1048, MVN_984, Multiply_1031, Reshape_1004, Reshape_1049, onnx::Reshape_358"/>
+				<attribute name="fused_names" version="0" value="Concat_1003, Concat_1048, InstanceNormalization_92, MVN_984, Multiply_1031, Reshape_1004, Reshape_1049"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1564,17 +1564,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_358">
+				<port id="2" precision="FP32" names="onnx::Reshape_221">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>262144</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="81" name="onnx::Reshape_359" type="ShapeOf" version="opset3">
+		<layer id="81" name="Shape_93" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_359"/>
+				<attribute name="fused_names" version="0" value="Shape_93"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1585,15 +1585,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_359">
+				<port id="1" precision="I64" names="onnx::Reshape_222">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="82" name="onnx::Mul_360" type="Reshape" version="opset1">
+		<layer id="82" name="Reshape_94" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_360"/>
+				<attribute name="fused_names" version="0" value="Reshape_94"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1606,7 +1606,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_360">
+				<port id="2" precision="FP32" names="onnx::Mul_223">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>256</dim>
@@ -1614,7 +1614,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="83" name="Constant_13453" type="Const" version="opset1">
+		<layer id="83" name="Constant_13605" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 128, 1, 1" offset="3106336" size="512"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -1625,10 +1625,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="84" name="onnx::Add_363" type="Multiply" version="opset1">
+		<layer id="84" name="Mul_95" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_363"/>
+				<attribute name="fused_names" version="0" value="Mul_95"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1645,7 +1645,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_363">
+				<port id="2" precision="FP32" names="onnx::Add_226">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>256</dim>
@@ -1653,7 +1653,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="85" name="Constant_13454" type="Const" version="opset1">
+		<layer id="85" name="Constant_13606" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 128, 1, 1" offset="3106848" size="512"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -1664,10 +1664,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="86" name="onnx::Cast_366" type="Add" version="opset1">
+		<layer id="86" name="Add_96" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.56, onnx::Cast_366"/>
+				<attribute name="fused_names" version="0" value="Add_96, Cast_97"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1684,7 +1684,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.56,onnx::Cast_366">
+				<port id="2" precision="FP32" names="input.51,onnx::Cast_229">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>256</dim>
@@ -1692,9 +1692,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="87" name="input.60" type="Swish" version="opset4">
+		<layer id="87" name="Mul_99" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.60, onnx::Mul_368"/>
+				<attribute name="fused_names" version="0" value="Mul_99, Sigmoid_98"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1705,7 +1705,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.60">
+				<port id="1" precision="FP32" names="input.55">
 					<dim>1</dim>
 					<dim>128</dim>
 					<dim>256</dim>
@@ -1713,13 +1713,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="88" name="m.encoder.down_blocks.1.resnets.0.conv1.weight" type="Const" version="opset1">
+		<layer id="88" name="vae.encoder.down_blocks.1.resnets.0.conv1.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="256, 128, 3, 3" offset="3107360" size="1179648"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.1.resnets.0.conv1.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.1.resnets.0.conv1.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.1.resnets.0.conv1.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.1.resnets.0.conv1.weight">
 					<dim>256</dim>
 					<dim>128</dim>
 					<dim>3</dim>
@@ -1727,10 +1727,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="89" name="Convolution_1089" type="Convolution" version="opset1">
+		<layer id="89" name="Conv_100/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_1089"/>
+				<attribute name="fused_names" version="0" value="Conv_100/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1766,10 +1766,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="91" name="onnx::Cast_370" type="Add" version="opset1">
+		<layer id="91" name="Conv_100" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_1108, Reshape_1109, input.64, onnx::Cast_370"/>
+				<attribute name="fused_names" version="0" value="Cast_101, Concat_1108, Conv_100, Reshape_1109"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1786,7 +1786,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.64,onnx::Cast_370">
+				<port id="2" precision="FP32" names="h.15,input.59">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -1794,21 +1794,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="92" name="onnx::Reshape_372" type="Const" version="opset1">
+		<layer id="92" name="Constant_102" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_372"/>
+				<attribute name="fused_names" version="0" value="Constant_102"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_372">
+				<port id="0" precision="I64" names="onnx::Reshape_235">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="93" name="onnx::InstanceNormalization_373" type="Reshape" version="opset1">
+		<layer id="93" name="Reshape_103" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_373"/>
+				<attribute name="fused_names" version="0" value="Reshape_103"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1822,7 +1822,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_373">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_236">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>524288</dim>
@@ -1843,7 +1843,7 @@
 		<layer id="95" name="MVN_1148" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_1167, Concat_1212, MVN_1148, Multiply_1195, Reshape_1168, Reshape_1213, onnx::Reshape_376"/>
+				<attribute name="fused_names" version="0" value="Concat_1167, Concat_1212, InstanceNormalization_106, MVN_1148, Multiply_1195, Reshape_1168, Reshape_1213"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1856,17 +1856,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_376">
+				<port id="2" precision="FP32" names="onnx::Reshape_239">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>524288</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="96" name="onnx::Reshape_377" type="ShapeOf" version="opset3">
+		<layer id="96" name="Shape_107" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_377"/>
+				<attribute name="fused_names" version="0" value="Shape_107"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1877,15 +1877,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_377">
+				<port id="1" precision="I64" names="onnx::Reshape_240">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="97" name="onnx::Mul_378" type="Reshape" version="opset1">
+		<layer id="97" name="Reshape_108" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_378"/>
+				<attribute name="fused_names" version="0" value="Reshape_108"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1898,7 +1898,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_378">
+				<port id="2" precision="FP32" names="onnx::Mul_241">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -1906,7 +1906,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="98" name="Constant_13455" type="Const" version="opset1">
+		<layer id="98" name="Constant_13607" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 256, 1, 1" offset="4288032" size="1024"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -1917,10 +1917,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="99" name="onnx::Add_381" type="Multiply" version="opset1">
+		<layer id="99" name="Mul_109" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_381"/>
+				<attribute name="fused_names" version="0" value="Mul_109"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1937,7 +1937,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_381">
+				<port id="2" precision="FP32" names="onnx::Add_244">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -1945,7 +1945,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="100" name="Constant_13456" type="Const" version="opset1">
+		<layer id="100" name="Constant_13608" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 256, 1, 1" offset="4289056" size="1024"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -1956,10 +1956,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="101" name="onnx::Cast_384" type="Add" version="opset1">
+		<layer id="101" name="Add_110" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.68, onnx::Cast_384"/>
+				<attribute name="fused_names" version="0" value="Add_110, Cast_111"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1976,7 +1976,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.68,onnx::Cast_384">
+				<port id="2" precision="FP32" names="input.63,onnx::Cast_247">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -1984,9 +1984,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="102" name="input.72" type="Swish" version="opset4">
+		<layer id="102" name="Mul_113" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.72, onnx::Mul_386"/>
+				<attribute name="fused_names" version="0" value="Mul_113, Sigmoid_112"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -1997,7 +1997,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.72">
+				<port id="1" precision="FP32" names="input.67">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -2005,13 +2005,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="103" name="m.encoder.down_blocks.1.resnets.0.conv2.weight" type="Const" version="opset1">
+		<layer id="103" name="vae.encoder.down_blocks.1.resnets.0.conv2.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="256, 256, 3, 3" offset="4290080" size="2359296"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.1.resnets.0.conv2.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.1.resnets.0.conv2.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.1.resnets.0.conv2.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.1.resnets.0.conv2.weight">
 					<dim>256</dim>
 					<dim>256</dim>
 					<dim>3</dim>
@@ -2019,10 +2019,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="104" name="Convolution_1253" type="Convolution" version="opset1">
+		<layer id="104" name="Conv_114/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_1253"/>
+				<attribute name="fused_names" version="0" value="Conv_114/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2058,10 +2058,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="106" name="onnx::Add_388" type="Add" version="opset1">
+		<layer id="106" name="Conv_114" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_1272, Reshape_1273, onnx::Add_388"/>
+				<attribute name="fused_names" version="0" value="Concat_1272, Conv_114, Reshape_1273"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2078,7 +2078,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_388">
+				<port id="2" precision="FP32" names="h.19">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -2086,10 +2086,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="107" name="onnx::Div_390" type="Add" version="opset1">
+		<layer id="107" name="Add_116" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.76, onnx::Cast_392, onnx::Div_390"/>
+				<attribute name="fused_names" version="0" value="Add_116, Cast_119, Div_118"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2106,7 +2106,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.76,onnx::Cast_392,onnx::Div_390">
+				<port id="2" precision="FP32" names="input.71,onnx::Div_253,x.24">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -2114,21 +2114,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="108" name="onnx::Reshape_394" type="Const" version="opset1">
+		<layer id="108" name="Constant_120" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_394"/>
+				<attribute name="fused_names" version="0" value="Constant_120"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_394">
+				<port id="0" precision="I64" names="onnx::Reshape_257">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="109" name="onnx::InstanceNormalization_395" type="Reshape" version="opset1">
+		<layer id="109" name="Reshape_121" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_395"/>
+				<attribute name="fused_names" version="0" value="Reshape_121"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2142,7 +2142,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_395">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_258">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>524288</dim>
@@ -2163,7 +2163,7 @@
 		<layer id="111" name="MVN_1363" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_1382, Concat_1427, MVN_1363, Multiply_1410, Reshape_1383, Reshape_1428, onnx::Reshape_398"/>
+				<attribute name="fused_names" version="0" value="Concat_1382, Concat_1427, InstanceNormalization_124, MVN_1363, Multiply_1410, Reshape_1383, Reshape_1428"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2176,17 +2176,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_398">
+				<port id="2" precision="FP32" names="onnx::Reshape_261">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>524288</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="112" name="onnx::Reshape_399" type="ShapeOf" version="opset3">
+		<layer id="112" name="Shape_125" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_399"/>
+				<attribute name="fused_names" version="0" value="Shape_125"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2197,15 +2197,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_399">
+				<port id="1" precision="I64" names="onnx::Reshape_262">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="113" name="onnx::Mul_400" type="Reshape" version="opset1">
+		<layer id="113" name="Reshape_126" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_400"/>
+				<attribute name="fused_names" version="0" value="Reshape_126"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2218,7 +2218,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_400">
+				<port id="2" precision="FP32" names="onnx::Mul_263">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -2226,7 +2226,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="114" name="Constant_13457" type="Const" version="opset1">
+		<layer id="114" name="Constant_13609" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 256, 1, 1" offset="6650400" size="1024"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -2237,10 +2237,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="115" name="onnx::Add_403" type="Multiply" version="opset1">
+		<layer id="115" name="Mul_127" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_403"/>
+				<attribute name="fused_names" version="0" value="Mul_127"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2257,7 +2257,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_403">
+				<port id="2" precision="FP32" names="onnx::Add_266">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -2265,7 +2265,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="116" name="Constant_13458" type="Const" version="opset1">
+		<layer id="116" name="Constant_13610" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 256, 1, 1" offset="6651424" size="1024"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -2276,10 +2276,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="117" name="onnx::Cast_406" type="Add" version="opset1">
+		<layer id="117" name="Add_128" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.80, onnx::Cast_406"/>
+				<attribute name="fused_names" version="0" value="Add_128, Cast_129"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2296,7 +2296,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.80,onnx::Cast_406">
+				<port id="2" precision="FP32" names="input.75,onnx::Cast_269">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -2304,9 +2304,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="118" name="input.84" type="Swish" version="opset4">
+		<layer id="118" name="Mul_131" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.84, onnx::Mul_408"/>
+				<attribute name="fused_names" version="0" value="Mul_131, Sigmoid_130"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2317,7 +2317,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.84">
+				<port id="1" precision="FP32" names="input.79">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -2325,13 +2325,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="119" name="m.encoder.down_blocks.1.resnets.1.conv1.weight" type="Const" version="opset1">
+		<layer id="119" name="vae.encoder.down_blocks.1.resnets.1.conv1.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="256, 256, 3, 3" offset="6652448" size="2359296"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.1.resnets.1.conv1.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.1.resnets.1.conv1.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.1.resnets.1.conv1.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.1.resnets.1.conv1.weight">
 					<dim>256</dim>
 					<dim>256</dim>
 					<dim>3</dim>
@@ -2339,10 +2339,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="120" name="Convolution_1468" type="Convolution" version="opset1">
+		<layer id="120" name="Conv_132/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_1468"/>
+				<attribute name="fused_names" version="0" value="Conv_132/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2378,10 +2378,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="122" name="onnx::Cast_410" type="Add" version="opset1">
+		<layer id="122" name="Conv_132" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_1487, Reshape_1488, input.88, onnx::Cast_410"/>
+				<attribute name="fused_names" version="0" value="Cast_133, Concat_1487, Conv_132, Reshape_1488"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2398,7 +2398,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.88,onnx::Cast_410">
+				<port id="2" precision="FP32" names="h.23,input.83">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -2406,21 +2406,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="123" name="onnx::Reshape_412" type="Const" version="opset1">
+		<layer id="123" name="Constant_134" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_412"/>
+				<attribute name="fused_names" version="0" value="Constant_134"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_412">
+				<port id="0" precision="I64" names="onnx::Reshape_275">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="124" name="onnx::InstanceNormalization_413" type="Reshape" version="opset1">
+		<layer id="124" name="Reshape_135" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_413"/>
+				<attribute name="fused_names" version="0" value="Reshape_135"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2434,7 +2434,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_413">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_276">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>524288</dim>
@@ -2455,7 +2455,7 @@
 		<layer id="126" name="MVN_1527" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_1546, Concat_1591, MVN_1527, Multiply_1574, Reshape_1547, Reshape_1592, onnx::Reshape_416"/>
+				<attribute name="fused_names" version="0" value="Concat_1546, Concat_1591, InstanceNormalization_138, MVN_1527, Multiply_1574, Reshape_1547, Reshape_1592"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2468,17 +2468,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_416">
+				<port id="2" precision="FP32" names="onnx::Reshape_279">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>524288</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="127" name="onnx::Reshape_417" type="ShapeOf" version="opset3">
+		<layer id="127" name="Shape_139" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_417"/>
+				<attribute name="fused_names" version="0" value="Shape_139"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2489,15 +2489,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_417">
+				<port id="1" precision="I64" names="onnx::Reshape_280">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="128" name="onnx::Mul_418" type="Reshape" version="opset1">
+		<layer id="128" name="Reshape_140" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_418"/>
+				<attribute name="fused_names" version="0" value="Reshape_140"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2510,7 +2510,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_418">
+				<port id="2" precision="FP32" names="onnx::Mul_281">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -2518,7 +2518,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="129" name="Constant_13459" type="Const" version="opset1">
+		<layer id="129" name="Constant_13611" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 256, 1, 1" offset="9012768" size="1024"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -2529,10 +2529,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="130" name="onnx::Add_421" type="Multiply" version="opset1">
+		<layer id="130" name="Mul_141" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_421"/>
+				<attribute name="fused_names" version="0" value="Mul_141"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2549,7 +2549,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_421">
+				<port id="2" precision="FP32" names="onnx::Add_284">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -2557,7 +2557,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="131" name="Constant_13460" type="Const" version="opset1">
+		<layer id="131" name="Constant_13612" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 256, 1, 1" offset="9013792" size="1024"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -2568,10 +2568,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="132" name="onnx::Cast_424" type="Add" version="opset1">
+		<layer id="132" name="Add_142" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.92, onnx::Cast_424"/>
+				<attribute name="fused_names" version="0" value="Add_142, Cast_143"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2588,7 +2588,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.92,onnx::Cast_424">
+				<port id="2" precision="FP32" names="input.87,onnx::Cast_287">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -2596,9 +2596,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="133" name="input.96" type="Swish" version="opset4">
+		<layer id="133" name="Mul_145" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.96, onnx::Mul_426"/>
+				<attribute name="fused_names" version="0" value="Mul_145, Sigmoid_144"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2609,7 +2609,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.96">
+				<port id="1" precision="FP32" names="input.91">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -2617,13 +2617,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="134" name="m.encoder.down_blocks.1.resnets.1.conv2.weight" type="Const" version="opset1">
+		<layer id="134" name="vae.encoder.down_blocks.1.resnets.1.conv2.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="256, 256, 3, 3" offset="9014816" size="2359296"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.1.resnets.1.conv2.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.1.resnets.1.conv2.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.1.resnets.1.conv2.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.1.resnets.1.conv2.weight">
 					<dim>256</dim>
 					<dim>256</dim>
 					<dim>3</dim>
@@ -2631,10 +2631,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="135" name="Convolution_1632" type="Convolution" version="opset1">
+		<layer id="135" name="Conv_146/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_1632"/>
+				<attribute name="fused_names" version="0" value="Conv_146/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2670,10 +2670,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="137" name="onnx::Add_428" type="Add" version="opset1">
+		<layer id="137" name="Conv_146" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_1651, Reshape_1652, onnx::Add_428"/>
+				<attribute name="fused_names" version="0" value="Concat_1651, Conv_146, Reshape_1652"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2690,7 +2690,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_428">
+				<port id="2" precision="FP32" names="h.27">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -2698,10 +2698,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="138" name="onnx::Div_429" type="Add" version="opset1">
+		<layer id="138" name="Add_147" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Div_429, onnx::Pad_431"/>
+				<attribute name="fused_names" version="0" value="Add_147, Div_149"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2718,7 +2718,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Div_429,onnx::Pad_431">
+				<port id="2" precision="FP32" names="onnx::Div_292,x.28">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>256</dim>
@@ -2726,13 +2726,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="139" name="m.encoder.down_blocks.1.downsamplers.0.conv.weight" type="Const" version="opset1">
+		<layer id="139" name="vae.encoder.down_blocks.1.downsamplers.0.conv.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="256, 256, 3, 3" offset="11375136" size="2359296"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.1.downsamplers.0.conv.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.1.downsamplers.0.conv.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.1.downsamplers.0.conv.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.1.downsamplers.0.conv.weight">
 					<dim>256</dim>
 					<dim>256</dim>
 					<dim>3</dim>
@@ -2740,10 +2740,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="140" name="Convolution_1781" type="Convolution" version="opset1">
+		<layer id="140" name="Conv_165/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="2, 2" dilations="1, 1" pads_begin="0, 0" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_1781, Split_1713, input.100"/>
+				<attribute name="fused_names" version="0" value="Conv_165/WithoutBiases, Pad_164, Split_1713"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2779,10 +2779,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="142" name="onnx::Cast_456" type="Add" version="opset1">
+		<layer id="142" name="Conv_165" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_1800, Reshape_1801, input.104, onnx::Cast_456"/>
+				<attribute name="fused_names" version="0" value="Cast_166, Concat_1800, Conv_165, Reshape_1801"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2799,7 +2799,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.104,onnx::Cast_456">
+				<port id="2" precision="FP32" names="input.95,x.36">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>128</dim>
@@ -2807,13 +2807,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="143" name="m.encoder.down_blocks.2.resnets.0.conv_shortcut.weight" type="Const" version="opset1">
+		<layer id="143" name="vae.encoder.down_blocks.2.resnets.0.conv_shortcut.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="512, 256, 1, 1" offset="13735456" size="524288"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.2.resnets.0.conv_shortcut.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.2.resnets.0.conv_shortcut.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.2.resnets.0.conv_shortcut.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.2.resnets.0.conv_shortcut.weight">
 					<dim>512</dim>
 					<dim>256</dim>
 					<dim>1</dim>
@@ -2821,10 +2821,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="144" name="Convolution_2157" type="Convolution" version="opset1">
+		<layer id="144" name="Conv_194/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="0, 0" pads_end="0, 0" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_2157"/>
+				<attribute name="fused_names" version="0" value="Conv_194/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2860,10 +2860,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="146" name="onnx::Add_493" type="Add" version="opset1">
+		<layer id="146" name="Conv_194" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_2176, Reshape_2177, onnx::Add_493"/>
+				<attribute name="fused_names" version="0" value="Concat_2176, Conv_194, Reshape_2177"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2880,7 +2880,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_493">
+				<port id="2" precision="FP32" names="x.40">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -2888,21 +2888,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="147" name="onnx::Reshape_458" type="Const" version="opset1">
+		<layer id="147" name="Constant_167" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_458"/>
+				<attribute name="fused_names" version="0" value="Constant_167"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_458">
+				<port id="0" precision="I64" names="onnx::Reshape_320">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="148" name="onnx::InstanceNormalization_459" type="Reshape" version="opset1">
+		<layer id="148" name="Reshape_168" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_459"/>
+				<attribute name="fused_names" version="0" value="Reshape_168"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2916,7 +2916,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_459">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_321">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>131072</dim>
@@ -2937,7 +2937,7 @@
 		<layer id="150" name="MVN_1840" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_1859, Concat_1904, MVN_1840, Multiply_1887, Reshape_1860, Reshape_1905, onnx::Reshape_462"/>
+				<attribute name="fused_names" version="0" value="Concat_1859, Concat_1904, InstanceNormalization_171, MVN_1840, Multiply_1887, Reshape_1860, Reshape_1905"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2950,17 +2950,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_462">
+				<port id="2" precision="FP32" names="onnx::Reshape_324">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>131072</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="151" name="onnx::Reshape_463" type="ShapeOf" version="opset3">
+		<layer id="151" name="Shape_172" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_463"/>
+				<attribute name="fused_names" version="0" value="Shape_172"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2971,15 +2971,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_463">
+				<port id="1" precision="I64" names="onnx::Reshape_325">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="152" name="onnx::Mul_464" type="Reshape" version="opset1">
+		<layer id="152" name="Reshape_173" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_464"/>
+				<attribute name="fused_names" version="0" value="Reshape_173"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -2992,7 +2992,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_464">
+				<port id="2" precision="FP32" names="onnx::Mul_326">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>128</dim>
@@ -3000,7 +3000,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="153" name="Constant_13461" type="Const" version="opset1">
+		<layer id="153" name="Constant_13613" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 256, 1, 1" offset="14261792" size="1024"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -3011,10 +3011,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="154" name="onnx::Add_467" type="Multiply" version="opset1">
+		<layer id="154" name="Mul_174" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_467"/>
+				<attribute name="fused_names" version="0" value="Mul_174"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3031,7 +3031,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_467">
+				<port id="2" precision="FP32" names="onnx::Add_329">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>128</dim>
@@ -3039,7 +3039,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="155" name="Constant_13462" type="Const" version="opset1">
+		<layer id="155" name="Constant_13614" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 256, 1, 1" offset="14262816" size="1024"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -3050,10 +3050,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="156" name="onnx::Cast_470" type="Add" version="opset1">
+		<layer id="156" name="Add_175" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.108, onnx::Cast_470"/>
+				<attribute name="fused_names" version="0" value="Add_175, Cast_176"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3070,7 +3070,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.108,onnx::Cast_470">
+				<port id="2" precision="FP32" names="input.99,onnx::Cast_332">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>128</dim>
@@ -3078,9 +3078,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="157" name="input.112" type="Swish" version="opset4">
+		<layer id="157" name="Mul_178" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.112, onnx::Mul_472"/>
+				<attribute name="fused_names" version="0" value="Mul_178, Sigmoid_177"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3091,7 +3091,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.112">
+				<port id="1" precision="FP32" names="input.103">
 					<dim>1</dim>
 					<dim>256</dim>
 					<dim>128</dim>
@@ -3099,13 +3099,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="158" name="m.encoder.down_blocks.2.resnets.0.conv1.weight" type="Const" version="opset1">
+		<layer id="158" name="vae.encoder.down_blocks.2.resnets.0.conv1.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="512, 256, 3, 3" offset="14263840" size="4718592"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.2.resnets.0.conv1.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.2.resnets.0.conv1.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.2.resnets.0.conv1.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.2.resnets.0.conv1.weight">
 					<dim>512</dim>
 					<dim>256</dim>
 					<dim>3</dim>
@@ -3113,10 +3113,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="159" name="Convolution_1945" type="Convolution" version="opset1">
+		<layer id="159" name="Conv_179/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_1945"/>
+				<attribute name="fused_names" version="0" value="Conv_179/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3152,10 +3152,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="161" name="onnx::Cast_474" type="Add" version="opset1">
+		<layer id="161" name="Conv_179" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_1964, Reshape_1965, input.116, onnx::Cast_474"/>
+				<attribute name="fused_names" version="0" value="Cast_180, Concat_1964, Conv_179, Reshape_1965"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3172,7 +3172,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.116,onnx::Cast_474">
+				<port id="2" precision="FP32" names="h.31,input.107">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -3180,21 +3180,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="162" name="onnx::Reshape_476" type="Const" version="opset1">
+		<layer id="162" name="Constant_181" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_476"/>
+				<attribute name="fused_names" version="0" value="Constant_181"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_476">
+				<port id="0" precision="I64" names="onnx::Reshape_338">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="163" name="onnx::InstanceNormalization_477" type="Reshape" version="opset1">
+		<layer id="163" name="Reshape_182" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_477"/>
+				<attribute name="fused_names" version="0" value="Reshape_182"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3208,7 +3208,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_477">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_339">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>262144</dim>
@@ -3229,7 +3229,7 @@
 		<layer id="165" name="MVN_2004" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_2023, Concat_2068, MVN_2004, Multiply_2051, Reshape_2024, Reshape_2069, onnx::Reshape_480"/>
+				<attribute name="fused_names" version="0" value="Concat_2023, Concat_2068, InstanceNormalization_185, MVN_2004, Multiply_2051, Reshape_2024, Reshape_2069"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3242,17 +3242,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_480">
+				<port id="2" precision="FP32" names="onnx::Reshape_342">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>262144</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="166" name="onnx::Reshape_481" type="ShapeOf" version="opset3">
+		<layer id="166" name="Shape_186" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_481"/>
+				<attribute name="fused_names" version="0" value="Shape_186"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3263,15 +3263,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_481">
+				<port id="1" precision="I64" names="onnx::Reshape_343">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="167" name="onnx::Mul_482" type="Reshape" version="opset1">
+		<layer id="167" name="Reshape_187" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_482"/>
+				<attribute name="fused_names" version="0" value="Reshape_187"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3284,7 +3284,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_482">
+				<port id="2" precision="FP32" names="onnx::Mul_344">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -3292,7 +3292,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="168" name="Constant_13463" type="Const" version="opset1">
+		<layer id="168" name="Constant_13615" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="18984480" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -3303,10 +3303,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="169" name="onnx::Add_485" type="Multiply" version="opset1">
+		<layer id="169" name="Mul_188" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_485"/>
+				<attribute name="fused_names" version="0" value="Mul_188"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3323,7 +3323,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_485">
+				<port id="2" precision="FP32" names="onnx::Add_347">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -3331,7 +3331,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="170" name="Constant_13464" type="Const" version="opset1">
+		<layer id="170" name="Constant_13616" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="18986528" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -3342,10 +3342,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="171" name="onnx::Cast_488" type="Add" version="opset1">
+		<layer id="171" name="Add_189" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.120, onnx::Cast_488"/>
+				<attribute name="fused_names" version="0" value="Add_189, Cast_190"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3362,7 +3362,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.120,onnx::Cast_488">
+				<port id="2" precision="FP32" names="input.111,onnx::Cast_350">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -3370,9 +3370,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="172" name="input.124" type="Swish" version="opset4">
+		<layer id="172" name="Mul_192" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.124, onnx::Mul_490"/>
+				<attribute name="fused_names" version="0" value="Mul_192, Sigmoid_191"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3383,7 +3383,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.124">
+				<port id="1" precision="FP32" names="input.115">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -3391,13 +3391,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="173" name="m.encoder.down_blocks.2.resnets.0.conv2.weight" type="Const" version="opset1">
+		<layer id="173" name="vae.encoder.down_blocks.2.resnets.0.conv2.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="512, 512, 3, 3" offset="18988576" size="9437184"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.2.resnets.0.conv2.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.2.resnets.0.conv2.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.2.resnets.0.conv2.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.2.resnets.0.conv2.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -3405,10 +3405,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="174" name="Convolution_2109" type="Convolution" version="opset1">
+		<layer id="174" name="Conv_193/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_2109"/>
+				<attribute name="fused_names" version="0" value="Conv_193/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3444,10 +3444,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="176" name="onnx::Add_492" type="Add" version="opset1">
+		<layer id="176" name="Conv_193" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_2128, Reshape_2129, onnx::Add_492"/>
+				<attribute name="fused_names" version="0" value="Concat_2128, Conv_193, Reshape_2129"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3464,7 +3464,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_492">
+				<port id="2" precision="FP32" names="h.35">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -3472,10 +3472,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="177" name="onnx::Div_494" type="Add" version="opset1">
+		<layer id="177" name="Add_195" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.128, onnx::Cast_496, onnx::Div_494"/>
+				<attribute name="fused_names" version="0" value="Add_195, Cast_198, Div_197"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3492,7 +3492,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.128,onnx::Cast_496,onnx::Div_494">
+				<port id="2" precision="FP32" names="input.119,onnx::Div_356,x.44">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -3500,21 +3500,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="178" name="onnx::Reshape_498" type="Const" version="opset1">
+		<layer id="178" name="Constant_199" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_498"/>
+				<attribute name="fused_names" version="0" value="Constant_199"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_498">
+				<port id="0" precision="I64" names="onnx::Reshape_360">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="179" name="onnx::InstanceNormalization_499" type="Reshape" version="opset1">
+		<layer id="179" name="Reshape_200" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_499"/>
+				<attribute name="fused_names" version="0" value="Reshape_200"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3528,7 +3528,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_499">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_361">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>262144</dim>
@@ -3549,7 +3549,7 @@
 		<layer id="181" name="MVN_2219" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_2238, Concat_2283, MVN_2219, Multiply_2266, Reshape_2239, Reshape_2284, onnx::Reshape_502"/>
+				<attribute name="fused_names" version="0" value="Concat_2238, Concat_2283, InstanceNormalization_203, MVN_2219, Multiply_2266, Reshape_2239, Reshape_2284"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3562,17 +3562,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_502">
+				<port id="2" precision="FP32" names="onnx::Reshape_364">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>262144</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="182" name="onnx::Reshape_503" type="ShapeOf" version="opset3">
+		<layer id="182" name="Shape_204" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_503"/>
+				<attribute name="fused_names" version="0" value="Shape_204"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3583,15 +3583,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_503">
+				<port id="1" precision="I64" names="onnx::Reshape_365">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="183" name="onnx::Mul_504" type="Reshape" version="opset1">
+		<layer id="183" name="Reshape_205" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_504"/>
+				<attribute name="fused_names" version="0" value="Reshape_205"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3604,7 +3604,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_504">
+				<port id="2" precision="FP32" names="onnx::Mul_366">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -3612,7 +3612,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="184" name="Constant_13465" type="Const" version="opset1">
+		<layer id="184" name="Constant_13617" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="28427808" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -3623,10 +3623,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="185" name="onnx::Add_507" type="Multiply" version="opset1">
+		<layer id="185" name="Mul_206" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_507"/>
+				<attribute name="fused_names" version="0" value="Mul_206"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3643,7 +3643,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_507">
+				<port id="2" precision="FP32" names="onnx::Add_369">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -3651,7 +3651,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="186" name="Constant_13466" type="Const" version="opset1">
+		<layer id="186" name="Constant_13618" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="28429856" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -3662,10 +3662,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="187" name="onnx::Cast_510" type="Add" version="opset1">
+		<layer id="187" name="Add_207" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.132, onnx::Cast_510"/>
+				<attribute name="fused_names" version="0" value="Add_207, Cast_208"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3682,7 +3682,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.132,onnx::Cast_510">
+				<port id="2" precision="FP32" names="input.123,onnx::Cast_372">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -3690,9 +3690,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="188" name="input.136" type="Swish" version="opset4">
+		<layer id="188" name="Mul_210" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.136, onnx::Mul_512"/>
+				<attribute name="fused_names" version="0" value="Mul_210, Sigmoid_209"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3703,7 +3703,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.136">
+				<port id="1" precision="FP32" names="input.127">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -3711,13 +3711,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="189" name="m.encoder.down_blocks.2.resnets.1.conv1.weight" type="Const" version="opset1">
+		<layer id="189" name="vae.encoder.down_blocks.2.resnets.1.conv1.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="512, 512, 3, 3" offset="28431904" size="9437184"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.2.resnets.1.conv1.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.2.resnets.1.conv1.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.2.resnets.1.conv1.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.2.resnets.1.conv1.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -3725,10 +3725,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="190" name="Convolution_2324" type="Convolution" version="opset1">
+		<layer id="190" name="Conv_211/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_2324"/>
+				<attribute name="fused_names" version="0" value="Conv_211/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3764,10 +3764,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="192" name="onnx::Cast_514" type="Add" version="opset1">
+		<layer id="192" name="Conv_211" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_2343, Reshape_2344, input.140, onnx::Cast_514"/>
+				<attribute name="fused_names" version="0" value="Cast_212, Concat_2343, Conv_211, Reshape_2344"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3784,7 +3784,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.140,onnx::Cast_514">
+				<port id="2" precision="FP32" names="h.39,input.131">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -3792,21 +3792,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="193" name="onnx::Reshape_516" type="Const" version="opset1">
+		<layer id="193" name="Constant_213" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_516"/>
+				<attribute name="fused_names" version="0" value="Constant_213"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_516">
+				<port id="0" precision="I64" names="onnx::Reshape_378">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="194" name="onnx::InstanceNormalization_517" type="Reshape" version="opset1">
+		<layer id="194" name="Reshape_214" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_517"/>
+				<attribute name="fused_names" version="0" value="Reshape_214"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3820,7 +3820,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_517">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_379">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>262144</dim>
@@ -3841,7 +3841,7 @@
 		<layer id="196" name="MVN_2383" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_2402, Concat_2447, MVN_2383, Multiply_2430, Reshape_2403, Reshape_2448, onnx::Reshape_520"/>
+				<attribute name="fused_names" version="0" value="Concat_2402, Concat_2447, InstanceNormalization_217, MVN_2383, Multiply_2430, Reshape_2403, Reshape_2448"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3854,17 +3854,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_520">
+				<port id="2" precision="FP32" names="onnx::Reshape_382">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>262144</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="197" name="onnx::Reshape_521" type="ShapeOf" version="opset3">
+		<layer id="197" name="Shape_218" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_521"/>
+				<attribute name="fused_names" version="0" value="Shape_218"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3875,15 +3875,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_521">
+				<port id="1" precision="I64" names="onnx::Reshape_383">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="198" name="onnx::Mul_522" type="Reshape" version="opset1">
+		<layer id="198" name="Reshape_219" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_522"/>
+				<attribute name="fused_names" version="0" value="Reshape_219"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3896,7 +3896,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_522">
+				<port id="2" precision="FP32" names="onnx::Mul_384">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -3904,7 +3904,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="199" name="Constant_13467" type="Const" version="opset1">
+		<layer id="199" name="Constant_13619" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="37871136" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -3915,10 +3915,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="200" name="onnx::Add_525" type="Multiply" version="opset1">
+		<layer id="200" name="Mul_220" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_525"/>
+				<attribute name="fused_names" version="0" value="Mul_220"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3935,7 +3935,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_525">
+				<port id="2" precision="FP32" names="onnx::Add_387">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -3943,7 +3943,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="201" name="Constant_13468" type="Const" version="opset1">
+		<layer id="201" name="Constant_13620" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="37873184" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -3954,10 +3954,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="202" name="onnx::Cast_528" type="Add" version="opset1">
+		<layer id="202" name="Add_221" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.144, onnx::Cast_528"/>
+				<attribute name="fused_names" version="0" value="Add_221, Cast_222"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3974,7 +3974,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.144,onnx::Cast_528">
+				<port id="2" precision="FP32" names="input.135,onnx::Cast_390">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -3982,9 +3982,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="203" name="input.148" type="Swish" version="opset4">
+		<layer id="203" name="Mul_224" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.148, onnx::Mul_530"/>
+				<attribute name="fused_names" version="0" value="Mul_224, Sigmoid_223"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -3995,7 +3995,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.148">
+				<port id="1" precision="FP32" names="input.139">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -4003,13 +4003,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="204" name="m.encoder.down_blocks.2.resnets.1.conv2.weight" type="Const" version="opset1">
+		<layer id="204" name="vae.encoder.down_blocks.2.resnets.1.conv2.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="512, 512, 3, 3" offset="37875232" size="9437184"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.2.resnets.1.conv2.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.2.resnets.1.conv2.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.2.resnets.1.conv2.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.2.resnets.1.conv2.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -4017,10 +4017,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="205" name="Convolution_2488" type="Convolution" version="opset1">
+		<layer id="205" name="Conv_225/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_2488"/>
+				<attribute name="fused_names" version="0" value="Conv_225/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4056,10 +4056,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="207" name="onnx::Add_532" type="Add" version="opset1">
+		<layer id="207" name="Conv_225" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_2507, Reshape_2508, onnx::Add_532"/>
+				<attribute name="fused_names" version="0" value="Concat_2507, Conv_225, Reshape_2508"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4076,7 +4076,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_532">
+				<port id="2" precision="FP32" names="h.43">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -4084,10 +4084,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="208" name="onnx::Div_533" type="Add" version="opset1">
+		<layer id="208" name="Add_226" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Div_533, onnx::Pad_535"/>
+				<attribute name="fused_names" version="0" value="Add_226, Div_228"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4104,7 +4104,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Div_533,onnx::Pad_535">
+				<port id="2" precision="FP32" names="onnx::Div_395,x.48">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>128</dim>
@@ -4112,13 +4112,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="209" name="m.encoder.down_blocks.2.downsamplers.0.conv.weight" type="Const" version="opset1">
+		<layer id="209" name="vae.encoder.down_blocks.2.downsamplers.0.conv.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="512, 512, 3, 3" offset="47314464" size="9437184"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.2.downsamplers.0.conv.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.2.downsamplers.0.conv.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.2.downsamplers.0.conv.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.2.downsamplers.0.conv.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -4126,10 +4126,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="210" name="Convolution_2637" type="Convolution" version="opset1">
+		<layer id="210" name="Conv_244/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="2, 2" dilations="1, 1" pads_begin="0, 0" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_2637, Split_2569, input.152"/>
+				<attribute name="fused_names" version="0" value="Conv_244/WithoutBiases, Pad_243, Split_2569"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4165,10 +4165,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="212" name="onnx::Cast_560" type="Add" version="opset1">
+		<layer id="212" name="Conv_244" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_2656, Reshape_2657, input.156, onnx::Cast_560"/>
+				<attribute name="fused_names" version="0" value="Cast_245, Concat_2656, Conv_244, Reshape_2657"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4185,7 +4185,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.156,onnx::Cast_560">
+				<port id="2" precision="FP32" names="input.143,x.56">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -4193,21 +4193,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="213" name="onnx::Reshape_562" type="Const" version="opset1">
+		<layer id="213" name="Constant_246" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_562"/>
+				<attribute name="fused_names" version="0" value="Constant_246"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_562">
+				<port id="0" precision="I64" names="onnx::Reshape_423">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="214" name="onnx::InstanceNormalization_563" type="Reshape" version="opset1">
+		<layer id="214" name="Reshape_247" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_563"/>
+				<attribute name="fused_names" version="0" value="Reshape_247"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4221,7 +4221,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_563">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_424">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
@@ -4242,7 +4242,7 @@
 		<layer id="216" name="MVN_2696" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_2715, Concat_2760, MVN_2696, Multiply_2743, Reshape_2716, Reshape_2761, onnx::Reshape_566"/>
+				<attribute name="fused_names" version="0" value="Concat_2715, Concat_2760, InstanceNormalization_250, MVN_2696, Multiply_2743, Reshape_2716, Reshape_2761"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4255,17 +4255,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_566">
+				<port id="2" precision="FP32" names="onnx::Reshape_427">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="217" name="onnx::Reshape_567" type="ShapeOf" version="opset3">
+		<layer id="217" name="Shape_251" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_567"/>
+				<attribute name="fused_names" version="0" value="Shape_251"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4276,15 +4276,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_567">
+				<port id="1" precision="I64" names="onnx::Reshape_428">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="218" name="onnx::Mul_568" type="Reshape" version="opset1">
+		<layer id="218" name="Reshape_252" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_568"/>
+				<attribute name="fused_names" version="0" value="Reshape_252"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4297,7 +4297,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_568">
+				<port id="2" precision="FP32" names="onnx::Mul_429">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -4305,7 +4305,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="219" name="Constant_13469" type="Const" version="opset1">
+		<layer id="219" name="Constant_13621" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="56753696" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -4316,10 +4316,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="220" name="onnx::Add_571" type="Multiply" version="opset1">
+		<layer id="220" name="Mul_253" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_571"/>
+				<attribute name="fused_names" version="0" value="Mul_253"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4336,7 +4336,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_571">
+				<port id="2" precision="FP32" names="onnx::Add_432">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -4344,7 +4344,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="221" name="Constant_13470" type="Const" version="opset1">
+		<layer id="221" name="Constant_13622" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="56755744" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -4355,10 +4355,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="222" name="onnx::Cast_574" type="Add" version="opset1">
+		<layer id="222" name="Add_254" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.160, onnx::Cast_574"/>
+				<attribute name="fused_names" version="0" value="Add_254, Cast_255"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4375,7 +4375,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.160,onnx::Cast_574">
+				<port id="2" precision="FP32" names="input.147,onnx::Cast_435">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -4383,9 +4383,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="223" name="input.164" type="Swish" version="opset4">
+		<layer id="223" name="Mul_257" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.164, onnx::Mul_576"/>
+				<attribute name="fused_names" version="0" value="Mul_257, Sigmoid_256"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4396,7 +4396,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.164">
+				<port id="1" precision="FP32" names="input.151">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -4404,13 +4404,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="224" name="m.encoder.down_blocks.3.resnets.0.conv1.weight" type="Const" version="opset1">
+		<layer id="224" name="vae.encoder.down_blocks.3.resnets.0.conv1.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="512, 512, 3, 3" offset="56757792" size="9437184"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.3.resnets.0.conv1.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.3.resnets.0.conv1.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.3.resnets.0.conv1.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.3.resnets.0.conv1.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -4418,10 +4418,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="225" name="Convolution_2801" type="Convolution" version="opset1">
+		<layer id="225" name="Conv_258/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_2801"/>
+				<attribute name="fused_names" version="0" value="Conv_258/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4457,10 +4457,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="227" name="onnx::Cast_578" type="Add" version="opset1">
+		<layer id="227" name="Conv_258" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_2820, Reshape_2821, input.168, onnx::Cast_578"/>
+				<attribute name="fused_names" version="0" value="Cast_259, Concat_2820, Conv_258, Reshape_2821"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4477,7 +4477,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.168,onnx::Cast_578">
+				<port id="2" precision="FP32" names="h.47,input.155">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -4485,21 +4485,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="228" name="onnx::Reshape_580" type="Const" version="opset1">
+		<layer id="228" name="Constant_260" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_580"/>
+				<attribute name="fused_names" version="0" value="Constant_260"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_580">
+				<port id="0" precision="I64" names="onnx::Reshape_441">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="229" name="onnx::InstanceNormalization_581" type="Reshape" version="opset1">
+		<layer id="229" name="Reshape_261" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_581"/>
+				<attribute name="fused_names" version="0" value="Reshape_261"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4513,7 +4513,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_581">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_442">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
@@ -4534,7 +4534,7 @@
 		<layer id="231" name="MVN_2860" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_2879, Concat_2924, MVN_2860, Multiply_2907, Reshape_2880, Reshape_2925, onnx::Reshape_584"/>
+				<attribute name="fused_names" version="0" value="Concat_2879, Concat_2924, InstanceNormalization_264, MVN_2860, Multiply_2907, Reshape_2880, Reshape_2925"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4547,17 +4547,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_584">
+				<port id="2" precision="FP32" names="onnx::Reshape_445">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="232" name="onnx::Reshape_585" type="ShapeOf" version="opset3">
+		<layer id="232" name="Shape_265" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_585"/>
+				<attribute name="fused_names" version="0" value="Shape_265"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4568,15 +4568,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_585">
+				<port id="1" precision="I64" names="onnx::Reshape_446">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="233" name="onnx::Mul_586" type="Reshape" version="opset1">
+		<layer id="233" name="Reshape_266" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_586"/>
+				<attribute name="fused_names" version="0" value="Reshape_266"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4589,7 +4589,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_586">
+				<port id="2" precision="FP32" names="onnx::Mul_447">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -4597,7 +4597,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="234" name="Constant_13471" type="Const" version="opset1">
+		<layer id="234" name="Constant_13623" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="66197024" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -4608,10 +4608,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="235" name="onnx::Add_589" type="Multiply" version="opset1">
+		<layer id="235" name="Mul_267" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_589"/>
+				<attribute name="fused_names" version="0" value="Mul_267"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4628,7 +4628,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_589">
+				<port id="2" precision="FP32" names="onnx::Add_450">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -4636,7 +4636,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="236" name="Constant_13472" type="Const" version="opset1">
+		<layer id="236" name="Constant_13624" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="66199072" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -4647,10 +4647,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="237" name="onnx::Cast_592" type="Add" version="opset1">
+		<layer id="237" name="Add_268" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.172, onnx::Cast_592"/>
+				<attribute name="fused_names" version="0" value="Add_268, Cast_269"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4667,7 +4667,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.172,onnx::Cast_592">
+				<port id="2" precision="FP32" names="input.159,onnx::Cast_453">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -4675,9 +4675,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="238" name="input.176" type="Swish" version="opset4">
+		<layer id="238" name="Mul_271" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.176, onnx::Mul_594"/>
+				<attribute name="fused_names" version="0" value="Mul_271, Sigmoid_270"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4688,7 +4688,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.176">
+				<port id="1" precision="FP32" names="input.163">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -4696,13 +4696,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="239" name="m.encoder.down_blocks.3.resnets.0.conv2.weight" type="Const" version="opset1">
+		<layer id="239" name="vae.encoder.down_blocks.3.resnets.0.conv2.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="512, 512, 3, 3" offset="66201120" size="9437184"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.3.resnets.0.conv2.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.3.resnets.0.conv2.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.3.resnets.0.conv2.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.3.resnets.0.conv2.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -4710,10 +4710,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="240" name="Convolution_2965" type="Convolution" version="opset1">
+		<layer id="240" name="Conv_272/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_2965"/>
+				<attribute name="fused_names" version="0" value="Conv_272/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4749,10 +4749,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="242" name="onnx::Add_596" type="Add" version="opset1">
+		<layer id="242" name="Conv_272" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_2984, Reshape_2985, onnx::Add_596"/>
+				<attribute name="fused_names" version="0" value="Concat_2984, Conv_272, Reshape_2985"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4769,7 +4769,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_596">
+				<port id="2" precision="FP32" names="h.51">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -4777,10 +4777,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="243" name="onnx::Div_597" type="Add" version="opset1">
+		<layer id="243" name="Add_273" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.180, onnx::Cast_599, onnx::Div_597"/>
+				<attribute name="fused_names" version="0" value="Add_273, Cast_276, Div_275"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4797,7 +4797,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.180,onnx::Cast_599,onnx::Div_597">
+				<port id="2" precision="FP32" names="input.167,onnx::Div_458,x.60">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -4805,21 +4805,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="244" name="onnx::Reshape_601" type="Const" version="opset1">
+		<layer id="244" name="Constant_277" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_601"/>
+				<attribute name="fused_names" version="0" value="Constant_277"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_601">
+				<port id="0" precision="I64" names="onnx::Reshape_462">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="245" name="onnx::InstanceNormalization_602" type="Reshape" version="opset1">
+		<layer id="245" name="Reshape_278" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_602"/>
+				<attribute name="fused_names" version="0" value="Reshape_278"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4833,7 +4833,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_602">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_463">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
@@ -4854,7 +4854,7 @@
 		<layer id="247" name="MVN_3027" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_3046, Concat_3091, MVN_3027, Multiply_3074, Reshape_3047, Reshape_3092, onnx::Reshape_605"/>
+				<attribute name="fused_names" version="0" value="Concat_3046, Concat_3091, InstanceNormalization_281, MVN_3027, Multiply_3074, Reshape_3047, Reshape_3092"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4867,17 +4867,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_605">
+				<port id="2" precision="FP32" names="onnx::Reshape_466">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="248" name="onnx::Reshape_606" type="ShapeOf" version="opset3">
+		<layer id="248" name="Shape_282" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_606"/>
+				<attribute name="fused_names" version="0" value="Shape_282"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4888,15 +4888,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_606">
+				<port id="1" precision="I64" names="onnx::Reshape_467">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="249" name="onnx::Mul_607" type="Reshape" version="opset1">
+		<layer id="249" name="Reshape_283" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_607"/>
+				<attribute name="fused_names" version="0" value="Reshape_283"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4909,7 +4909,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_607">
+				<port id="2" precision="FP32" names="onnx::Mul_468">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -4917,7 +4917,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="250" name="Constant_13473" type="Const" version="opset1">
+		<layer id="250" name="Constant_13625" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="75640352" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -4928,10 +4928,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="251" name="onnx::Add_610" type="Multiply" version="opset1">
+		<layer id="251" name="Mul_284" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_610"/>
+				<attribute name="fused_names" version="0" value="Mul_284"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4948,7 +4948,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_610">
+				<port id="2" precision="FP32" names="onnx::Add_471">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -4956,7 +4956,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="252" name="Constant_13474" type="Const" version="opset1">
+		<layer id="252" name="Constant_13626" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="75642400" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -4967,10 +4967,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="253" name="onnx::Cast_613" type="Add" version="opset1">
+		<layer id="253" name="Add_285" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.184, onnx::Cast_613"/>
+				<attribute name="fused_names" version="0" value="Add_285, Cast_286"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -4987,7 +4987,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.184,onnx::Cast_613">
+				<port id="2" precision="FP32" names="input.171,onnx::Cast_474">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -4995,9 +4995,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="254" name="input.188" type="Swish" version="opset4">
+		<layer id="254" name="Mul_288" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.188, onnx::Mul_615"/>
+				<attribute name="fused_names" version="0" value="Mul_288, Sigmoid_287"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5008,7 +5008,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.188">
+				<port id="1" precision="FP32" names="input.175">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -5016,13 +5016,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="255" name="m.encoder.down_blocks.3.resnets.1.conv1.weight" type="Const" version="opset1">
+		<layer id="255" name="vae.encoder.down_blocks.3.resnets.1.conv1.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="512, 512, 3, 3" offset="75644448" size="9437184"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.3.resnets.1.conv1.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.3.resnets.1.conv1.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.3.resnets.1.conv1.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.3.resnets.1.conv1.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -5030,10 +5030,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="256" name="Convolution_3132" type="Convolution" version="opset1">
+		<layer id="256" name="Conv_289/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_3132"/>
+				<attribute name="fused_names" version="0" value="Conv_289/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5069,10 +5069,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="258" name="onnx::Cast_617" type="Add" version="opset1">
+		<layer id="258" name="Conv_289" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_3151, Reshape_3152, input.192, onnx::Cast_617"/>
+				<attribute name="fused_names" version="0" value="Cast_290, Concat_3151, Conv_289, Reshape_3152"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5089,7 +5089,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.192,onnx::Cast_617">
+				<port id="2" precision="FP32" names="h.55,input.179">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -5097,21 +5097,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="259" name="onnx::Reshape_619" type="Const" version="opset1">
+		<layer id="259" name="Constant_291" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_619"/>
+				<attribute name="fused_names" version="0" value="Constant_291"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_619">
+				<port id="0" precision="I64" names="onnx::Reshape_480">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="260" name="onnx::InstanceNormalization_620" type="Reshape" version="opset1">
+		<layer id="260" name="Reshape_292" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_620"/>
+				<attribute name="fused_names" version="0" value="Reshape_292"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5125,7 +5125,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_620">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_481">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
@@ -5146,7 +5146,7 @@
 		<layer id="262" name="MVN_3191" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_3210, Concat_3255, MVN_3191, Multiply_3238, Reshape_3211, Reshape_3256, onnx::Reshape_623"/>
+				<attribute name="fused_names" version="0" value="Concat_3210, Concat_3255, InstanceNormalization_295, MVN_3191, Multiply_3238, Reshape_3211, Reshape_3256"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5159,17 +5159,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_623">
+				<port id="2" precision="FP32" names="onnx::Reshape_484">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="263" name="onnx::Reshape_624" type="ShapeOf" version="opset3">
+		<layer id="263" name="Shape_296" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_624"/>
+				<attribute name="fused_names" version="0" value="Shape_296"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5180,15 +5180,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_624">
+				<port id="1" precision="I64" names="onnx::Reshape_485">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="264" name="onnx::Mul_625" type="Reshape" version="opset1">
+		<layer id="264" name="Reshape_297" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_625"/>
+				<attribute name="fused_names" version="0" value="Reshape_297"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5201,7 +5201,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_625">
+				<port id="2" precision="FP32" names="onnx::Mul_486">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -5209,7 +5209,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="265" name="Constant_13475" type="Const" version="opset1">
+		<layer id="265" name="Constant_13627" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="85083680" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -5220,10 +5220,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="266" name="onnx::Add_628" type="Multiply" version="opset1">
+		<layer id="266" name="Mul_298" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_628"/>
+				<attribute name="fused_names" version="0" value="Mul_298"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5240,7 +5240,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_628">
+				<port id="2" precision="FP32" names="onnx::Add_489">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -5248,7 +5248,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="267" name="Constant_13476" type="Const" version="opset1">
+		<layer id="267" name="Constant_13628" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="85085728" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -5259,10 +5259,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="268" name="onnx::Cast_631" type="Add" version="opset1">
+		<layer id="268" name="Add_299" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.196, onnx::Cast_631"/>
+				<attribute name="fused_names" version="0" value="Add_299, Cast_300"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5279,7 +5279,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.196,onnx::Cast_631">
+				<port id="2" precision="FP32" names="input.183,onnx::Cast_492">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -5287,9 +5287,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="269" name="input.200" type="Swish" version="opset4">
+		<layer id="269" name="Mul_302" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.200, onnx::Mul_633"/>
+				<attribute name="fused_names" version="0" value="Mul_302, Sigmoid_301"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5300,7 +5300,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.200">
+				<port id="1" precision="FP32" names="input.187">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -5308,13 +5308,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="270" name="m.encoder.down_blocks.3.resnets.1.conv2.weight" type="Const" version="opset1">
+		<layer id="270" name="vae.encoder.down_blocks.3.resnets.1.conv2.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="512, 512, 3, 3" offset="85087776" size="9437184"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.down_blocks.3.resnets.1.conv2.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.down_blocks.3.resnets.1.conv2.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.down_blocks.3.resnets.1.conv2.weight">
+				<port id="0" precision="FP32" names="vae.encoder.down_blocks.3.resnets.1.conv2.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -5322,10 +5322,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="271" name="Convolution_3296" type="Convolution" version="opset1">
+		<layer id="271" name="Conv_303/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_3296"/>
+				<attribute name="fused_names" version="0" value="Conv_303/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5361,10 +5361,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="273" name="onnx::Add_635" type="Add" version="opset1">
+		<layer id="273" name="Conv_303" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_3315, Reshape_3316, onnx::Add_635"/>
+				<attribute name="fused_names" version="0" value="Concat_3315, Conv_303, Reshape_3316"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5381,7 +5381,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_635">
+				<port id="2" precision="FP32" names="h.59">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -5389,10 +5389,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="274" name="onnx::Div_636" type="Add" version="opset1">
+		<layer id="274" name="Add_304" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.204, onnx::Cast_638, onnx::Div_636"/>
+				<attribute name="fused_names" version="0" value="Add_304, Cast_307, Div_306"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5409,7 +5409,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.204,onnx::Cast_638,onnx::Div_636">
+				<port id="2" precision="FP32" names="input.191,onnx::Div_497,x.64">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -5417,21 +5417,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="275" name="onnx::Reshape_640" type="Const" version="opset1">
+		<layer id="275" name="Constant_308" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_640"/>
+				<attribute name="fused_names" version="0" value="Constant_308"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_640">
+				<port id="0" precision="I64" names="onnx::Reshape_501">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="276" name="onnx::InstanceNormalization_641" type="Reshape" version="opset1">
+		<layer id="276" name="Reshape_309" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_641"/>
+				<attribute name="fused_names" version="0" value="Reshape_309"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5445,7 +5445,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_641">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_502">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
@@ -5466,7 +5466,7 @@
 		<layer id="278" name="MVN_3358" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_3377, Concat_3422, MVN_3358, Multiply_3405, Reshape_3378, Reshape_3423, onnx::Reshape_644"/>
+				<attribute name="fused_names" version="0" value="Concat_3377, Concat_3422, InstanceNormalization_312, MVN_3358, Multiply_3405, Reshape_3378, Reshape_3423"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5479,17 +5479,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_644">
+				<port id="2" precision="FP32" names="onnx::Reshape_505">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="279" name="onnx::Reshape_645" type="ShapeOf" version="opset3">
+		<layer id="279" name="Shape_313" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_645"/>
+				<attribute name="fused_names" version="0" value="Shape_313"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5500,15 +5500,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_645">
+				<port id="1" precision="I64" names="onnx::Reshape_506">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="280" name="onnx::Mul_646" type="Reshape" version="opset1">
+		<layer id="280" name="Reshape_314" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_646"/>
+				<attribute name="fused_names" version="0" value="Reshape_314"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5521,7 +5521,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_646">
+				<port id="2" precision="FP32" names="onnx::Mul_507">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -5529,7 +5529,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="281" name="Constant_13477" type="Const" version="opset1">
+		<layer id="281" name="Constant_13629" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="94527008" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -5540,10 +5540,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="282" name="onnx::Add_649" type="Multiply" version="opset1">
+		<layer id="282" name="Mul_315" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_649"/>
+				<attribute name="fused_names" version="0" value="Mul_315"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5560,7 +5560,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_649">
+				<port id="2" precision="FP32" names="onnx::Add_510">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -5568,7 +5568,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="283" name="Constant_13478" type="Const" version="opset1">
+		<layer id="283" name="Constant_13630" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="94529056" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -5579,10 +5579,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="284" name="onnx::Cast_652" type="Add" version="opset1">
+		<layer id="284" name="Add_316" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.208, onnx::Cast_652"/>
+				<attribute name="fused_names" version="0" value="Add_316, Cast_317"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5599,7 +5599,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.208,onnx::Cast_652">
+				<port id="2" precision="FP32" names="input.195,onnx::Cast_513">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -5607,9 +5607,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="285" name="input.212" type="Swish" version="opset4">
+		<layer id="285" name="Mul_319" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.212, onnx::Mul_654"/>
+				<attribute name="fused_names" version="0" value="Mul_319, Sigmoid_318"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5620,7 +5620,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.212">
+				<port id="1" precision="FP32" names="input.199">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -5628,13 +5628,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="286" name="m.encoder.mid_block.resnets.0.conv1.weight" type="Const" version="opset1">
+		<layer id="286" name="vae.encoder.mid_block.resnets.0.conv1.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="512, 512, 3, 3" offset="94531104" size="9437184"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.mid_block.resnets.0.conv1.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.mid_block.resnets.0.conv1.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.mid_block.resnets.0.conv1.weight">
+				<port id="0" precision="FP32" names="vae.encoder.mid_block.resnets.0.conv1.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -5642,10 +5642,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="287" name="Convolution_3463" type="Convolution" version="opset1">
+		<layer id="287" name="Conv_320/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_3463"/>
+				<attribute name="fused_names" version="0" value="Conv_320/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5681,10 +5681,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="289" name="onnx::Cast_656" type="Add" version="opset1">
+		<layer id="289" name="Conv_320" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_3482, Reshape_3483, input.216, onnx::Cast_656"/>
+				<attribute name="fused_names" version="0" value="Cast_321, Concat_3482, Conv_320, Reshape_3483"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5701,7 +5701,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.216,onnx::Cast_656">
+				<port id="2" precision="FP32" names="h.63,input.203">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -5709,21 +5709,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="290" name="onnx::Reshape_658" type="Const" version="opset1">
+		<layer id="290" name="Constant_322" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_658"/>
+				<attribute name="fused_names" version="0" value="Constant_322"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_658">
+				<port id="0" precision="I64" names="onnx::Reshape_519">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="291" name="onnx::InstanceNormalization_659" type="Reshape" version="opset1">
+		<layer id="291" name="Reshape_323" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_659"/>
+				<attribute name="fused_names" version="0" value="Reshape_323"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5737,7 +5737,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_659">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_520">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
@@ -5758,7 +5758,7 @@
 		<layer id="293" name="MVN_3522" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_3541, Concat_3586, MVN_3522, Multiply_3569, Reshape_3542, Reshape_3587, onnx::Reshape_662"/>
+				<attribute name="fused_names" version="0" value="Concat_3541, Concat_3586, InstanceNormalization_326, MVN_3522, Multiply_3569, Reshape_3542, Reshape_3587"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5771,17 +5771,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_662">
+				<port id="2" precision="FP32" names="onnx::Reshape_523">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="294" name="onnx::Reshape_663" type="ShapeOf" version="opset3">
+		<layer id="294" name="Shape_327" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_663"/>
+				<attribute name="fused_names" version="0" value="Shape_327"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5792,15 +5792,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_663">
+				<port id="1" precision="I64" names="onnx::Reshape_524">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="295" name="onnx::Mul_664" type="Reshape" version="opset1">
+		<layer id="295" name="Reshape_328" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_664"/>
+				<attribute name="fused_names" version="0" value="Reshape_328"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5813,7 +5813,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_664">
+				<port id="2" precision="FP32" names="onnx::Mul_525">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -5821,7 +5821,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="296" name="Constant_13479" type="Const" version="opset1">
+		<layer id="296" name="Constant_13631" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="103970336" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -5832,10 +5832,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="297" name="onnx::Add_667" type="Multiply" version="opset1">
+		<layer id="297" name="Mul_329" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_667"/>
+				<attribute name="fused_names" version="0" value="Mul_329"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5852,7 +5852,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_667">
+				<port id="2" precision="FP32" names="onnx::Add_528">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -5860,7 +5860,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="298" name="Constant_13480" type="Const" version="opset1">
+		<layer id="298" name="Constant_13632" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="103972384" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -5871,10 +5871,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="299" name="onnx::Cast_670" type="Add" version="opset1">
+		<layer id="299" name="Add_330" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.220, onnx::Cast_670"/>
+				<attribute name="fused_names" version="0" value="Add_330, Cast_331"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5891,7 +5891,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.220,onnx::Cast_670">
+				<port id="2" precision="FP32" names="input.207,onnx::Cast_531">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -5899,9 +5899,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="300" name="input.224" type="Swish" version="opset4">
+		<layer id="300" name="Mul_333" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.224, onnx::Mul_672"/>
+				<attribute name="fused_names" version="0" value="Mul_333, Sigmoid_332"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5912,7 +5912,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.224">
+				<port id="1" precision="FP32" names="input.211">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -5920,13 +5920,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="301" name="m.encoder.mid_block.resnets.0.conv2.weight" type="Const" version="opset1">
+		<layer id="301" name="vae.encoder.mid_block.resnets.0.conv2.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="512, 512, 3, 3" offset="103974432" size="9437184"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.mid_block.resnets.0.conv2.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.mid_block.resnets.0.conv2.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.mid_block.resnets.0.conv2.weight">
+				<port id="0" precision="FP32" names="vae.encoder.mid_block.resnets.0.conv2.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -5934,10 +5934,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="302" name="Convolution_3627" type="Convolution" version="opset1">
+		<layer id="302" name="Conv_334/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_3627"/>
+				<attribute name="fused_names" version="0" value="Conv_334/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5973,10 +5973,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="304" name="onnx::Add_674" type="Add" version="opset1">
+		<layer id="304" name="Conv_334" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_3646, Reshape_3647, onnx::Add_674"/>
+				<attribute name="fused_names" version="0" value="Concat_3646, Conv_334, Reshape_3647"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -5993,7 +5993,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_674">
+				<port id="2" precision="FP32" names="h.67">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -6001,10 +6001,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="305" name="onnx::Div_675" type="Add" version="opset1">
+		<layer id="305" name="Add_335" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.228, onnx::Div_675"/>
+				<attribute name="fused_names" version="0" value="Add_335, Div_337"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6021,7 +6021,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.228,onnx::Div_675">
+				<port id="2" precision="FP32" names="hidden_states,onnx::Div_536">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -6029,21 +6029,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="306" name="onnx::Reshape_690" type="Const" version="opset1">
+		<layer id="306" name="Constant_350" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_690"/>
+				<attribute name="fused_names" version="0" value="Constant_350"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_690">
+				<port id="0" precision="I64" names="onnx::Reshape_551">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="307" name="onnx::InstanceNormalization_691" type="Reshape" version="opset1">
+		<layer id="307" name="Reshape_351" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_691"/>
+				<attribute name="fused_names" version="0" value="Reshape_351"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6057,7 +6057,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_691">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_552">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
@@ -6078,7 +6078,7 @@
 		<layer id="309" name="MVN_3704" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_3723, Concat_3768, MVN_3704, Multiply_3751, Reshape_3724, Reshape_3769, onnx::Reshape_694"/>
+				<attribute name="fused_names" version="0" value="Concat_3723, Concat_3768, InstanceNormalization_354, MVN_3704, Multiply_3751, Reshape_3724, Reshape_3769"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6091,17 +6091,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_694">
+				<port id="2" precision="FP32" names="onnx::Reshape_555">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="310" name="onnx::Reshape_695" type="ShapeOf" version="opset3">
+		<layer id="310" name="Shape_355" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Gather_678, onnx::Gather_681, onnx::Gather_684, onnx::Gather_687, onnx::Reshape_695, onnx::Reshape_816"/>
+				<attribute name="fused_names" version="0" value="Concat_447, Shape_338, Shape_341, Shape_344, Shape_347, Shape_355"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6112,15 +6112,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Gather_678,onnx::Gather_681,onnx::Gather_684,onnx::Gather_687,onnx::Reshape_695,onnx::Reshape_816">
+				<port id="1" precision="I64" names="onnx::Gather_539,onnx::Gather_542,onnx::Gather_545,onnx::Gather_548,onnx::Reshape_556,onnx::Reshape_670">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="311" name="onnx::Mul_696" type="Reshape" version="opset1">
+		<layer id="311" name="Reshape_356" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_696"/>
+				<attribute name="fused_names" version="0" value="Reshape_356"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6133,7 +6133,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_696">
+				<port id="2" precision="FP32" names="onnx::Mul_557">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -6141,7 +6141,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="312" name="Constant_13481" type="Const" version="opset1">
+		<layer id="312" name="Constant_13633" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="113413664" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -6152,10 +6152,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="313" name="onnx::Add_699" type="Multiply" version="opset1">
+		<layer id="313" name="Mul_357" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_699"/>
+				<attribute name="fused_names" version="0" value="Mul_357"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6172,7 +6172,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_699">
+				<port id="2" precision="FP32" names="onnx::Add_560">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -6180,7 +6180,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="314" name="Constant_13482" type="Const" version="opset1">
+		<layer id="314" name="Constant_13634" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="113415712" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -6191,10 +6191,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="315" name="onnx::Reshape_702" type="Add" version="opset1">
+		<layer id="315" name="Add_358" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_702"/>
+				<attribute name="fused_names" version="0" value="Add_358"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6211,7 +6211,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_702">
+				<port id="2" precision="FP32" names="hidden_states.3">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -6219,10 +6219,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="316" name="Constant_11761" type="Const" version="opset1">
+		<layer id="316" name="Constant_11913" type="Const" version="opset1">
 			<data element_type="i64" shape="2" offset="113417760" size="16"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_710"/>
+				<attribute name="fused_names" version="0" value="Concat_366"/>
 			</rt_info>
 			<output>
 				<port id="0" precision="I64">
@@ -6230,16 +6230,16 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="317" name="Constant_11762" type="Const" version="opset1">
+		<layer id="317" name="Constant_11914" type="Const" version="opset1">
 			<data element_type="i64" shape="" offset="113417776" size="8"/>
 			<output>
 				<port id="0" precision="I64"/>
 			</output>
 		</layer>
-		<layer id="318" name="Gather_11763" type="Gather" version="opset8">
+		<layer id="318" name="Gather_11915" type="Gather" version="opset8">
 			<data batch_dims="0"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_710"/>
+				<attribute name="fused_names" version="0" value="Concat_366"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="I64">
@@ -6256,13 +6256,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="319" name="onnx::Gather_685" type="Const" version="opset1">
+		<layer id="319" name="Constant_345" type="Const" version="opset1">
 			<data element_type="i64" shape="" offset="18456" size="8"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Gather_685"/>
+				<attribute name="fused_names" version="0" value="Constant_345"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Gather_685"/>
+				<port id="0" precision="I64" names="onnx::Gather_546"/>
 			</output>
 		</layer>
 		<layer id="320" name="Constant_3688" type="Const" version="opset1">
@@ -6274,10 +6274,10 @@
 				<port id="0" precision="I64"/>
 			</output>
 		</layer>
-		<layer id="321" name="onnx::Mul_686" type="Gather" version="opset8">
+		<layer id="321" name="Gather_346" type="Gather" version="opset8">
 			<data batch_dims="0"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Constant_3688, onnx::Gather_685, onnx::Mul_686"/>
+				<attribute name="fused_names" version="0" value="Constant_345, Constant_3688, Gather_346"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="I64">
@@ -6287,16 +6287,16 @@
 				<port id="2" precision="I64"/>
 			</input>
 			<output>
-				<port id="3" precision="I64" names="onnx::Mul_686"/>
+				<port id="3" precision="I64" names="onnx::Mul_547"/>
 			</output>
 		</layer>
-		<layer id="322" name="onnx::Gather_688" type="Const" version="opset1">
+		<layer id="322" name="Constant_348" type="Const" version="opset1">
 			<data element_type="i64" shape="" offset="113417784" size="8"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Gather_688"/>
+				<attribute name="fused_names" version="0" value="Constant_348"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Gather_688"/>
+				<port id="0" precision="I64" names="onnx::Gather_549"/>
 			</output>
 		</layer>
 		<layer id="323" name="Constant_3692" type="Const" version="opset1">
@@ -6308,10 +6308,10 @@
 				<port id="0" precision="I64"/>
 			</output>
 		</layer>
-		<layer id="324" name="onnx::Mul_689" type="Gather" version="opset8">
+		<layer id="324" name="Gather_349" type="Gather" version="opset8">
 			<data batch_dims="0"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Constant_3692, onnx::Gather_688, onnx::Mul_689"/>
+				<attribute name="fused_names" version="0" value="Constant_348, Constant_3692, Gather_349"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="I64">
@@ -6321,36 +6321,36 @@
 				<port id="2" precision="I64"/>
 			</input>
 			<output>
-				<port id="3" precision="I64" names="onnx::Mul_689"/>
+				<port id="3" precision="I64" names="onnx::Mul_550"/>
 			</output>
 		</layer>
-		<layer id="325" name="onnx::Unsqueeze_703" type="Multiply" version="opset1">
+		<layer id="325" name="Mul_359" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Unsqueeze_703"/>
+				<attribute name="fused_names" version="0" value="Mul_359"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="I64"/>
 				<port id="1" precision="I64"/>
 			</input>
 			<output>
-				<port id="2" precision="I64" names="onnx::Unsqueeze_703"/>
+				<port id="2" precision="I64" names="onnx::Unsqueeze_564"/>
 			</output>
 		</layer>
-		<layer id="326" name="onnx::Unsqueeze_708" type="Const" version="opset1">
+		<layer id="326" name="Constant_364" type="Const" version="opset1">
 			<data element_type="i64" shape="1" offset="113417776" size="8"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Unsqueeze_708"/>
+				<attribute name="fused_names" version="0" value="Constant_364"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Unsqueeze_708">
+				<port id="0" precision="I64" names="onnx::Unsqueeze_569">
 					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="327" name="onnx::Concat_709" type="Unsqueeze" version="opset1">
+		<layer id="327" name="Unsqueeze_365" type="Unsqueeze" version="opset1">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Concat_709, onnx::Unsqueeze_708"/>
+				<attribute name="fused_names" version="0" value="Constant_364, Unsqueeze_365"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="I64"/>
@@ -6359,15 +6359,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="I64" names="onnx::Concat_709">
+				<port id="2" precision="I64" names="onnx::Concat_570">
 					<dim>1</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="328" name="onnx::Reshape_710" type="Concat" version="opset1">
+		<layer id="328" name="Concat_366" type="Concat" version="opset1">
 			<data axis="0"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_710"/>
+				<attribute name="fused_names" version="0" value="Concat_366"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="I64">
@@ -6378,15 +6378,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="I64" names="onnx::Reshape_710">
+				<port id="2" precision="I64" names="onnx::Reshape_571">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="329" name="onnx::Transpose_711" type="Reshape" version="opset1">
+		<layer id="329" name="Reshape_367" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Transpose_711"/>
+				<attribute name="fused_names" version="0" value="Reshape_367"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6400,7 +6400,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Transpose_711">
+				<port id="2" precision="FP32" names="onnx::Transpose_572">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>4096</dim>
@@ -6418,9 +6418,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="331" name="onnx::MatMul_712" type="Transpose" version="opset1">
+		<layer id="331" name="Transpose_368" type="Transpose" version="opset1">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::MatMul_712"/>
+				<attribute name="fused_names" version="0" value="Transpose_368"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6433,17 +6433,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::MatMul_712">
+				<port id="2" precision="FP32" names="input.215">
 					<dim>1</dim>
 					<dim>4096</dim>
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="332" name="Constant_13329" type="Const" version="opset1">
+		<layer id="332" name="Constant_13481" type="Const" version="opset1">
 			<data element_type="f32" shape="512, 512" offset="113417816" size="1048576"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_714, onnx::MatMul_930"/>
+				<attribute name="fused_names" version="0" value="MatMul_369, onnx::MatMul_784"/>
 			</rt_info>
 			<output>
 				<port id="0" precision="FP32">
@@ -6452,10 +6452,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="333" name="onnx::Add_714" type="MatMul" version="opset1">
+		<layer id="333" name="MatMul_369" type="MatMul" version="opset1">
 			<data transpose_a="false" transpose_b="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_714, onnx::MatMul_930"/>
+				<attribute name="fused_names" version="0" value="MatMul_369, onnx::MatMul_784"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6469,17 +6469,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_714">
+				<port id="2" precision="FP32" names="onnx::Add_575">
 					<dim>1</dim>
 					<dim>4096</dim>
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="334" name="query_proj" type="Add" version="opset1">
+		<layer id="334" name="Add_370" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="query_proj"/>
+				<attribute name="fused_names" version="0" value="Add_370"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6494,17 +6494,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="query_proj">
+				<port id="2" precision="FP32" names="projection">
 					<dim>1</dim>
 					<dim>4096</dim>
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="335" name="onnx::Reshape_738" type="Const" version="opset1">
+		<layer id="335" name="Concat_385" type="Const" version="opset1">
 			<data element_type="i64" shape="4" offset="114466392" size="32"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Concat_935, onnx::Concat_936, onnx::Reshape_738"/>
+				<attribute name="fused_names" version="0" value="Concat_385, onnx::Concat_789, onnx::Concat_790"/>
 			</rt_info>
 			<output>
 				<port id="0" precision="I64">
@@ -6512,10 +6512,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="336" name="onnx::Transpose_739" type="Reshape" version="opset1">
+		<layer id="336" name="Reshape_386" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Transpose_739"/>
+				<attribute name="fused_names" version="0" value="Reshape_386"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6528,7 +6528,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Transpose_739">
+				<port id="2" precision="FP32" names="onnx::Transpose_598">
 					<dim>1</dim>
 					<dim>4096</dim>
 					<dim>1</dim>
@@ -6536,7 +6536,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="337" name="Constant_13331" type="Const" version="opset1">
+		<layer id="337" name="Constant_13483" type="Const" version="opset1">
 			<data element_type="i64" shape="4" offset="114466424" size="32"/>
 			<output>
 				<port id="0" precision="I64">
@@ -6544,10 +6544,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="338" name="onnx::Mul_740" type="Reshape" version="opset1">
+		<layer id="338" name="Transpose_387" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_740"/>
+				<attribute name="fused_names" version="0" value="Transpose_387"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6561,7 +6561,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_740">
+				<port id="2" precision="FP32" names="query_states">
 					<dim>1</dim>
 					<dim>1</dim>
 					<dim>4096</dim>
@@ -6569,7 +6569,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="339" name="Constant_13484" type="Const" version="opset1">
+		<layer id="339" name="Constant_13636" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 1, 1, 1" offset="114466456" size="4"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -6580,10 +6580,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="340" name="onnx::MatMul_779" type="Multiply" version="opset1">
+		<layer id="340" name="Mul_414" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::MatMul_779"/>
+				<attribute name="fused_names" version="0" value="Mul_414"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6600,7 +6600,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::MatMul_779">
+				<port id="2" precision="FP32" names="onnx::MatMul_634">
 					<dim>1</dim>
 					<dim>1</dim>
 					<dim>4096</dim>
@@ -6608,7 +6608,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="341" name="Constant_13485" type="Const" version="opset1">
+		<layer id="341" name="Constant_13637" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 1, 512" offset="114466460" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -6618,10 +6618,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="342" name="Constant_13343" type="Const" version="opset1">
+		<layer id="342" name="Constant_13495" type="Const" version="opset1">
 			<data element_type="f32" shape="512, 512" offset="114468508" size="1048576"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_717, onnx::MatMul_931"/>
+				<attribute name="fused_names" version="0" value="MatMul_371, onnx::MatMul_785"/>
 			</rt_info>
 			<output>
 				<port id="0" precision="FP32">
@@ -6630,10 +6630,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="343" name="onnx::Add_717" type="MatMul" version="opset1">
+		<layer id="343" name="MatMul_371" type="MatMul" version="opset1">
 			<data transpose_a="false" transpose_b="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_717, onnx::MatMul_931"/>
+				<attribute name="fused_names" version="0" value="MatMul_371, onnx::MatMul_785"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6647,17 +6647,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_717">
+				<port id="2" precision="FP32" names="onnx::Add_578">
 					<dim>1</dim>
 					<dim>4096</dim>
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="344" name="key_proj" type="Add" version="opset1">
+		<layer id="344" name="Add_372" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="key_proj"/>
+				<attribute name="fused_names" version="0" value="Add_372"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6672,17 +6672,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="key_proj">
+				<port id="2" precision="FP32" names="projection.3">
 					<dim>1</dim>
 					<dim>4096</dim>
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="345" name="onnx::Reshape_757" type="Const" version="opset1">
+		<layer id="345" name="Concat_398" type="Const" version="opset1">
 			<data element_type="i64" shape="4" offset="114466392" size="32"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Concat_935, onnx::Concat_936, onnx::Reshape_757"/>
+				<attribute name="fused_names" version="0" value="Concat_398, onnx::Concat_789, onnx::Concat_790"/>
 			</rt_info>
 			<output>
 				<port id="0" precision="I64">
@@ -6690,10 +6690,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="346" name="onnx::Transpose_758" type="Reshape" version="opset1">
+		<layer id="346" name="Reshape_399" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Transpose_758"/>
+				<attribute name="fused_names" version="0" value="Reshape_399"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6706,7 +6706,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Transpose_758">
+				<port id="2" precision="FP32" names="onnx::Transpose_615">
 					<dim>1</dim>
 					<dim>4096</dim>
 					<dim>1</dim>
@@ -6725,9 +6725,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="348" name="onnx::Mul_780" type="Transpose" version="opset1">
+		<layer id="348" name="Transpose_415" type="Transpose" version="opset1">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_780"/>
+				<attribute name="fused_names" version="0" value="Transpose_415"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6741,7 +6741,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_780">
+				<port id="2" precision="FP32" names="onnx::Mul_635">
 					<dim>1</dim>
 					<dim>1</dim>
 					<dim>512</dim>
@@ -6749,7 +6749,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="349" name="Constant_13486" type="Const" version="opset1">
+		<layer id="349" name="Constant_13638" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 1, 1, 1" offset="114466456" size="4"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -6760,10 +6760,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="350" name="onnx::MatMul_782" type="Multiply" version="opset1">
+		<layer id="350" name="Mul_417" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::MatMul_782"/>
+				<attribute name="fused_names" version="0" value="Mul_417"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6780,7 +6780,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::MatMul_782">
+				<port id="2" precision="FP32" names="onnx::MatMul_637">
 					<dim>1</dim>
 					<dim>1</dim>
 					<dim>512</dim>
@@ -6788,10 +6788,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="351" name="onnx::Cast_783" type="MatMul" version="opset1">
+		<layer id="351" name="MatMul_418" type="MatMul" version="opset1">
 			<data transpose_a="false" transpose_b="false"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Cast_783, onnx::Softmax_784"/>
+				<attribute name="fused_names" version="0" value="Cast_419, MatMul_418"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6808,7 +6808,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Cast_783,onnx::Softmax_784">
+				<port id="2" precision="FP32" names="attention_scores,attention_scores.3">
 					<dim>1</dim>
 					<dim>1</dim>
 					<dim>4096</dim>
@@ -6816,10 +6816,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="352" name="onnx::Cast_785" type="SoftMax" version="opset8">
+		<layer id="352" name="Softmax_420" type="SoftMax" version="opset8">
 			<data axis="-1"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Cast_785, onnx::MatMul_786"/>
+				<attribute name="fused_names" version="0" value="Cast_421, Softmax_420"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6830,7 +6830,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="onnx::Cast_785,onnx::MatMul_786">
+				<port id="1" precision="FP32" names="attention_probs,onnx::Cast_640">
 					<dim>1</dim>
 					<dim>1</dim>
 					<dim>4096</dim>
@@ -6838,7 +6838,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="353" name="Constant_13487" type="Const" version="opset1">
+		<layer id="353" name="Constant_13639" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 1, 512" offset="115517116" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -6848,10 +6848,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="354" name="Constant_13350" type="Const" version="opset1">
+		<layer id="354" name="Constant_13502" type="Const" version="opset1">
 			<data element_type="f32" shape="512, 512" offset="115519164" size="1048576"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_720, onnx::MatMul_932"/>
+				<attribute name="fused_names" version="0" value="MatMul_373, onnx::MatMul_786"/>
 			</rt_info>
 			<output>
 				<port id="0" precision="FP32">
@@ -6860,10 +6860,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="355" name="onnx::Add_720" type="MatMul" version="opset1">
+		<layer id="355" name="MatMul_373" type="MatMul" version="opset1">
 			<data transpose_a="false" transpose_b="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_720, onnx::MatMul_932"/>
+				<attribute name="fused_names" version="0" value="MatMul_373, onnx::MatMul_786"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6877,17 +6877,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_720">
+				<port id="2" precision="FP32" names="onnx::Add_581">
 					<dim>1</dim>
 					<dim>4096</dim>
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="356" name="value_proj" type="Add" version="opset1">
+		<layer id="356" name="Add_374" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="value_proj"/>
+				<attribute name="fused_names" version="0" value="Add_374"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6902,17 +6902,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="value_proj">
+				<port id="2" precision="FP32" names="projection.7">
 					<dim>1</dim>
 					<dim>4096</dim>
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="357" name="onnx::Reshape_775" type="Const" version="opset1">
+		<layer id="357" name="Concat_410" type="Const" version="opset1">
 			<data element_type="i64" shape="4" offset="114466392" size="32"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Concat_935, onnx::Concat_936, onnx::Reshape_775"/>
+				<attribute name="fused_names" version="0" value="Concat_410, onnx::Concat_789, onnx::Concat_790"/>
 			</rt_info>
 			<output>
 				<port id="0" precision="I64">
@@ -6920,10 +6920,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="358" name="onnx::Transpose_776" type="Reshape" version="opset1">
+		<layer id="358" name="Reshape_411" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Transpose_776"/>
+				<attribute name="fused_names" version="0" value="Reshape_411"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6936,7 +6936,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Transpose_776">
+				<port id="2" precision="FP32" names="onnx::Transpose_631">
 					<dim>1</dim>
 					<dim>4096</dim>
 					<dim>1</dim>
@@ -6944,7 +6944,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="359" name="Constant_13352" type="Const" version="opset1">
+		<layer id="359" name="Constant_13504" type="Const" version="opset1">
 			<data element_type="i64" shape="4" offset="114466424" size="32"/>
 			<output>
 				<port id="0" precision="I64">
@@ -6952,10 +6952,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="360" name="onnx::MatMul_777" type="Reshape" version="opset1">
+		<layer id="360" name="Transpose_412" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::MatMul_777"/>
+				<attribute name="fused_names" version="0" value="Transpose_412"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6969,7 +6969,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::MatMul_777">
+				<port id="2" precision="FP32" names="value_states">
 					<dim>1</dim>
 					<dim>1</dim>
 					<dim>4096</dim>
@@ -6977,10 +6977,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="361" name="onnx::Transpose_787" type="MatMul" version="opset1">
+		<layer id="361" name="MatMul_422" type="MatMul" version="opset1">
 			<data transpose_a="false" transpose_b="false"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Transpose_787"/>
+				<attribute name="fused_names" version="0" value="MatMul_422"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -6997,7 +6997,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Transpose_787">
+				<port id="2" precision="FP32" names="context_states">
 					<dim>1</dim>
 					<dim>1</dim>
 					<dim>4096</dim>
@@ -7005,7 +7005,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="362" name="Constant_13359" type="Const" version="opset1">
+		<layer id="362" name="Constant_13511" type="Const" version="opset1">
 			<data element_type="i64" shape="4" offset="116567740" size="32"/>
 			<output>
 				<port id="0" precision="I64">
@@ -7013,10 +7013,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="363" name="onnx::Shape_788" type="Reshape" version="opset1">
+		<layer id="363" name="Transpose_423" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Shape_788"/>
+				<attribute name="fused_names" version="0" value="Transpose_423"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7030,7 +7030,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Shape_788">
+				<port id="2" precision="FP32" names="onnx::Shape_643">
 					<dim>1</dim>
 					<dim>4096</dim>
 					<dim>1</dim>
@@ -7038,10 +7038,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="364" name="onnx::Reshape_802" type="Const" version="opset1">
+		<layer id="364" name="Concat_434" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="116567772" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Concat_939, onnx::Reshape_802"/>
+				<attribute name="fused_names" version="0" value="Concat_434, onnx::Concat_793"/>
 			</rt_info>
 			<output>
 				<port id="0" precision="I64">
@@ -7049,10 +7049,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="365" name="onnx::MatMul_803" type="Reshape" version="opset1">
+		<layer id="365" name="Reshape_435" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::MatMul_803"/>
+				<attribute name="fused_names" version="0" value="Reshape_435"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7066,17 +7066,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::MatMul_803">
+				<port id="2" precision="FP32" names="input.219">
 					<dim>1</dim>
 					<dim>4096</dim>
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="366" name="Constant_13371" type="Const" version="opset1">
+		<layer id="366" name="Constant_13523" type="Const" version="opset1">
 			<data element_type="f32" shape="512, 512" offset="116567796" size="1048576"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_805, onnx::MatMul_940"/>
+				<attribute name="fused_names" version="0" value="MatMul_436, onnx::MatMul_794"/>
 			</rt_info>
 			<output>
 				<port id="0" precision="FP32">
@@ -7085,10 +7085,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="367" name="onnx::Add_805" type="MatMul" version="opset1">
+		<layer id="367" name="MatMul_436" type="MatMul" version="opset1">
 			<data transpose_a="false" transpose_b="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_805, onnx::MatMul_940"/>
+				<attribute name="fused_names" version="0" value="MatMul_436, onnx::MatMul_794"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7102,17 +7102,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_805">
+				<port id="2" precision="FP32" names="onnx::Add_659">
 					<dim>1</dim>
 					<dim>4096</dim>
 					<dim>512</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="368" name="onnx::Transpose_806" type="Add" version="opset1">
+		<layer id="368" name="Add_437" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Transpose_806"/>
+				<attribute name="fused_names" version="0" value="Add_437"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7127,7 +7127,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Transpose_806">
+				<port id="2" precision="FP32" names="hidden_states.7">
 					<dim>1</dim>
 					<dim>4096</dim>
 					<dim>512</dim>
@@ -7145,9 +7145,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="370" name="onnx::Reshape_807" type="Transpose" version="opset1">
+		<layer id="370" name="Transpose_438" type="Transpose" version="opset1">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_807"/>
+				<attribute name="fused_names" version="0" value="Transpose_438"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7160,17 +7160,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_807">
+				<port id="2" precision="FP32" names="onnx::Reshape_661">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>4096</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="371" name="onnx::Add_817" type="Reshape" version="opset1">
+		<layer id="371" name="Reshape_448" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_817"/>
+				<attribute name="fused_names" version="0" value="Reshape_448"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7183,7 +7183,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_817">
+				<port id="2" precision="FP32" names="hidden_states.11">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -7191,10 +7191,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="372" name="onnx::Div_818" type="Add" version="opset1">
+		<layer id="372" name="Add_449" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.232, onnx::Cast_820, onnx::Div_818"/>
+				<attribute name="fused_names" version="0" value="Add_449, Cast_452, Div_451"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7211,7 +7211,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.232,onnx::Cast_820,onnx::Div_818">
+				<port id="2" precision="FP32" names="input.223,onnx::Div_672,x.68">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -7219,21 +7219,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="373" name="onnx::Reshape_822" type="Const" version="opset1">
+		<layer id="373" name="Constant_453" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_822"/>
+				<attribute name="fused_names" version="0" value="Constant_453"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_822">
+				<port id="0" precision="I64" names="onnx::Reshape_676">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="374" name="onnx::InstanceNormalization_823" type="Reshape" version="opset1">
+		<layer id="374" name="Reshape_454" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_823"/>
+				<attribute name="fused_names" version="0" value="Reshape_454"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7247,7 +7247,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_823">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_677">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
@@ -7268,7 +7268,7 @@
 		<layer id="376" name="MVN_4260" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_4279, Concat_4324, MVN_4260, Multiply_4307, Reshape_4280, Reshape_4325, onnx::Reshape_826"/>
+				<attribute name="fused_names" version="0" value="Concat_4279, Concat_4324, InstanceNormalization_457, MVN_4260, Multiply_4307, Reshape_4280, Reshape_4325"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7281,17 +7281,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_826">
+				<port id="2" precision="FP32" names="onnx::Reshape_680">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="377" name="onnx::Reshape_827" type="ShapeOf" version="opset3">
+		<layer id="377" name="Shape_458" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_827"/>
+				<attribute name="fused_names" version="0" value="Shape_458"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7302,15 +7302,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_827">
+				<port id="1" precision="I64" names="onnx::Reshape_681">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="378" name="onnx::Mul_828" type="Reshape" version="opset1">
+		<layer id="378" name="Reshape_459" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_828"/>
+				<attribute name="fused_names" version="0" value="Reshape_459"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7323,7 +7323,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_828">
+				<port id="2" precision="FP32" names="onnx::Mul_682">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -7331,7 +7331,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="379" name="Constant_13489" type="Const" version="opset1">
+		<layer id="379" name="Constant_13641" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="117616372" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -7342,10 +7342,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="380" name="onnx::Add_831" type="Multiply" version="opset1">
+		<layer id="380" name="Mul_460" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_831"/>
+				<attribute name="fused_names" version="0" value="Mul_460"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7362,7 +7362,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_831">
+				<port id="2" precision="FP32" names="onnx::Add_685">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -7370,7 +7370,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="381" name="Constant_13490" type="Const" version="opset1">
+		<layer id="381" name="Constant_13642" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="117618420" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -7381,10 +7381,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="382" name="onnx::Cast_834" type="Add" version="opset1">
+		<layer id="382" name="Add_461" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.236, onnx::Cast_834"/>
+				<attribute name="fused_names" version="0" value="Add_461, Cast_462"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7401,7 +7401,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.236,onnx::Cast_834">
+				<port id="2" precision="FP32" names="input.227,onnx::Cast_688">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -7409,9 +7409,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="383" name="input.240" type="Swish" version="opset4">
+		<layer id="383" name="Mul_464" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.240, onnx::Mul_836"/>
+				<attribute name="fused_names" version="0" value="Mul_464, Sigmoid_463"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7422,7 +7422,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.240">
+				<port id="1" precision="FP32" names="input.231">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -7430,13 +7430,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="384" name="m.encoder.mid_block.resnets.1.conv1.weight" type="Const" version="opset1">
+		<layer id="384" name="vae.encoder.mid_block.resnets.1.conv1.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="512, 512, 3, 3" offset="117620468" size="9437184"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.mid_block.resnets.1.conv1.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.mid_block.resnets.1.conv1.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.mid_block.resnets.1.conv1.weight">
+				<port id="0" precision="FP32" names="vae.encoder.mid_block.resnets.1.conv1.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -7444,10 +7444,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="385" name="Convolution_4365" type="Convolution" version="opset1">
+		<layer id="385" name="Conv_465/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_4365"/>
+				<attribute name="fused_names" version="0" value="Conv_465/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7483,10 +7483,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="387" name="onnx::Cast_838" type="Add" version="opset1">
+		<layer id="387" name="Conv_465" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_4384, Reshape_4385, input.244, onnx::Cast_838"/>
+				<attribute name="fused_names" version="0" value="Cast_466, Concat_4384, Conv_465, Reshape_4385"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7503,7 +7503,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.244,onnx::Cast_838">
+				<port id="2" precision="FP32" names="h.71,input.235">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -7511,21 +7511,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="388" name="onnx::Reshape_840" type="Const" version="opset1">
+		<layer id="388" name="Constant_467" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_840"/>
+				<attribute name="fused_names" version="0" value="Constant_467"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_840">
+				<port id="0" precision="I64" names="onnx::Reshape_694">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="389" name="onnx::InstanceNormalization_841" type="Reshape" version="opset1">
+		<layer id="389" name="Reshape_468" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_841"/>
+				<attribute name="fused_names" version="0" value="Reshape_468"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7539,7 +7539,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_841">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_695">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
@@ -7560,7 +7560,7 @@
 		<layer id="391" name="MVN_4424" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_4443, Concat_4488, MVN_4424, Multiply_4471, Reshape_4444, Reshape_4489, onnx::Reshape_844"/>
+				<attribute name="fused_names" version="0" value="Concat_4443, Concat_4488, InstanceNormalization_471, MVN_4424, Multiply_4471, Reshape_4444, Reshape_4489"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7573,17 +7573,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_844">
+				<port id="2" precision="FP32" names="onnx::Reshape_698">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="392" name="onnx::Reshape_845" type="ShapeOf" version="opset3">
+		<layer id="392" name="Shape_472" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_845"/>
+				<attribute name="fused_names" version="0" value="Shape_472"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7594,15 +7594,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_845">
+				<port id="1" precision="I64" names="onnx::Reshape_699">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="393" name="onnx::Mul_846" type="Reshape" version="opset1">
+		<layer id="393" name="Reshape_473" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_846"/>
+				<attribute name="fused_names" version="0" value="Reshape_473"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7615,7 +7615,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_846">
+				<port id="2" precision="FP32" names="onnx::Mul_700">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -7623,7 +7623,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="394" name="Constant_13491" type="Const" version="opset1">
+		<layer id="394" name="Constant_13643" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="127059700" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -7634,10 +7634,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="395" name="onnx::Add_849" type="Multiply" version="opset1">
+		<layer id="395" name="Mul_474" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_849"/>
+				<attribute name="fused_names" version="0" value="Mul_474"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7654,7 +7654,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_849">
+				<port id="2" precision="FP32" names="onnx::Add_703">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -7662,7 +7662,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="396" name="Constant_13492" type="Const" version="opset1">
+		<layer id="396" name="Constant_13644" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="127061748" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -7673,10 +7673,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="397" name="onnx::Cast_852" type="Add" version="opset1">
+		<layer id="397" name="Add_475" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.248, onnx::Cast_852"/>
+				<attribute name="fused_names" version="0" value="Add_475, Cast_476"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7693,7 +7693,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.248,onnx::Cast_852">
+				<port id="2" precision="FP32" names="input.239,onnx::Cast_706">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -7701,9 +7701,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="398" name="input.252" type="Swish" version="opset4">
+		<layer id="398" name="Mul_478" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.252, onnx::Mul_854"/>
+				<attribute name="fused_names" version="0" value="Mul_478, Sigmoid_477"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7714,7 +7714,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.252">
+				<port id="1" precision="FP32" names="input.243">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -7722,13 +7722,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="399" name="m.encoder.mid_block.resnets.1.conv2.weight" type="Const" version="opset1">
+		<layer id="399" name="vae.encoder.mid_block.resnets.1.conv2.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="512, 512, 3, 3" offset="127063796" size="9437184"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.mid_block.resnets.1.conv2.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.mid_block.resnets.1.conv2.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.mid_block.resnets.1.conv2.weight">
+				<port id="0" precision="FP32" names="vae.encoder.mid_block.resnets.1.conv2.weight">
 					<dim>512</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -7736,10 +7736,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="400" name="Convolution_4529" type="Convolution" version="opset1">
+		<layer id="400" name="Conv_479/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_4529"/>
+				<attribute name="fused_names" version="0" value="Conv_479/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7775,10 +7775,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="402" name="onnx::Add_856" type="Add" version="opset1">
+		<layer id="402" name="Conv_479" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_4548, Reshape_4549, onnx::Add_856"/>
+				<attribute name="fused_names" version="0" value="Concat_4548, Conv_479, Reshape_4549"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7795,7 +7795,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_856">
+				<port id="2" precision="FP32" names="h.75">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -7803,10 +7803,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="403" name="onnx::Div_857" type="Add" version="opset1">
+		<layer id="403" name="Add_480" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.256, onnx::Div_857"/>
+				<attribute name="fused_names" version="0" value="Add_480, Div_482"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7823,7 +7823,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.256,onnx::Div_857">
+				<port id="2" precision="FP32" names="input.247,onnx::Div_711">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -7831,21 +7831,21 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="404" name="onnx::Reshape_860" type="Const" version="opset1">
+		<layer id="404" name="Constant_483" type="Const" version="opset1">
 			<data element_type="i64" shape="3" offset="18432" size="24"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_860"/>
+				<attribute name="fused_names" version="0" value="Constant_483"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="I64" names="onnx::Reshape_860">
+				<port id="0" precision="I64" names="onnx::Reshape_714">
 					<dim>3</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="405" name="onnx::InstanceNormalization_861" type="Reshape" version="opset1">
+		<layer id="405" name="Reshape_484" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::InstanceNormalization_861"/>
+				<attribute name="fused_names" version="0" value="Reshape_484"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7859,7 +7859,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::InstanceNormalization_861">
+				<port id="2" precision="FP32" names="onnx::InstanceNormalization_715">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
@@ -7880,7 +7880,7 @@
 		<layer id="407" name="MVN_4590" type="MVN" version="opset6">
 			<data eps="9.9999999747524271e-07" normalize_variance="true" eps_mode="INSIDE_SQRT"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_4609, Concat_4654, MVN_4590, Multiply_4637, Reshape_4610, Reshape_4655, onnx::Reshape_864"/>
+				<attribute name="fused_names" version="0" value="Concat_4609, Concat_4654, InstanceNormalization_487, MVN_4590, Multiply_4637, Reshape_4610, Reshape_4655"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7893,17 +7893,17 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Reshape_864">
+				<port id="2" precision="FP32" names="onnx::Reshape_718">
 					<dim>1</dim>
 					<dim>32</dim>
 					<dim>65536</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="408" name="onnx::Reshape_865" type="ShapeOf" version="opset3">
+		<layer id="408" name="Shape_488" type="ShapeOf" version="opset3">
 			<data output_type="i64"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Reshape_865"/>
+				<attribute name="fused_names" version="0" value="Shape_488"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7914,15 +7914,15 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="I64" names="onnx::Reshape_865">
+				<port id="1" precision="I64" names="onnx::Reshape_719">
 					<dim>4</dim>
 				</port>
 			</output>
 		</layer>
-		<layer id="409" name="onnx::Mul_866" type="Reshape" version="opset1">
+		<layer id="409" name="Reshape_489" type="Reshape" version="opset1">
 			<data special_zero="true"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Mul_866"/>
+				<attribute name="fused_names" version="0" value="Reshape_489"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7935,7 +7935,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Mul_866">
+				<port id="2" precision="FP32" names="onnx::Mul_720">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -7943,7 +7943,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="410" name="Constant_13493" type="Const" version="opset1">
+		<layer id="410" name="Constant_13645" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="136503028" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -7954,10 +7954,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="411" name="onnx::Add_869" type="Multiply" version="opset1">
+		<layer id="411" name="Mul_490" type="Multiply" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="onnx::Add_869"/>
+				<attribute name="fused_names" version="0" value="Mul_490"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -7974,7 +7974,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="onnx::Add_869">
+				<port id="2" precision="FP32" names="onnx::Add_723">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -7982,7 +7982,7 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="412" name="Constant_13494" type="Const" version="opset1">
+		<layer id="412" name="Constant_13646" type="Const" version="opset1">
 			<data element_type="f32" shape="1, 512, 1, 1" offset="136505076" size="2048"/>
 			<output>
 				<port id="0" precision="FP32">
@@ -7993,10 +7993,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="413" name="input.260" type="Add" version="opset1">
+		<layer id="413" name="Add_491" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.260"/>
+				<attribute name="fused_names" version="0" value="Add_491"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -8013,7 +8013,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.260">
+				<port id="2" precision="FP32" names="input.251">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -8021,9 +8021,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="414" name="input.264" type="Swish" version="opset4">
+		<layer id="414" name="Mul_493" type="Swish" version="opset4">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="input.264, onnx::Mul_873"/>
+				<attribute name="fused_names" version="0" value="Mul_493, Sigmoid_492"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -8034,7 +8034,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="1" precision="FP32" names="input.264">
+				<port id="1" precision="FP32" names="input.255">
 					<dim>1</dim>
 					<dim>512</dim>
 					<dim>64</dim>
@@ -8042,13 +8042,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="415" name="m.encoder.conv_out.weight" type="Const" version="opset1">
+		<layer id="415" name="vae.encoder.conv_out.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="8, 512, 3, 3" offset="136507124" size="147456"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.encoder.conv_out.weight"/>
+				<attribute name="fused_names" version="0" value="vae.encoder.conv_out.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.encoder.conv_out.weight">
+				<port id="0" precision="FP32" names="vae.encoder.conv_out.weight">
 					<dim>8</dim>
 					<dim>512</dim>
 					<dim>3</dim>
@@ -8056,10 +8056,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="416" name="Convolution_4694" type="Convolution" version="opset1">
+		<layer id="416" name="Conv_494/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="1, 1" pads_end="1, 1" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_4694"/>
+				<attribute name="fused_names" version="0" value="Conv_494/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -8095,10 +8095,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="418" name="input.268" type="Add" version="opset1">
+		<layer id="418" name="Conv_494" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_4713, Reshape_4714, input.268"/>
+				<attribute name="fused_names" version="0" value="Concat_4713, Conv_494, Reshape_4714"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -8115,7 +8115,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="input.268">
+				<port id="2" precision="FP32" names="input.259">
 					<dim>1</dim>
 					<dim>8</dim>
 					<dim>64</dim>
@@ -8123,13 +8123,13 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="419" name="m.quant_conv.weight" type="Const" version="opset1">
+		<layer id="419" name="vae.quant_conv.weight" type="Const" version="opset1">
 			<data element_type="f32" shape="8, 8, 1, 1" offset="136654612" size="256"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="m.quant_conv.weight"/>
+				<attribute name="fused_names" version="0" value="vae.quant_conv.weight"/>
 			</rt_info>
 			<output>
-				<port id="0" precision="FP32" names="m.quant_conv.weight">
+				<port id="0" precision="FP32" names="vae.quant_conv.weight">
 					<dim>8</dim>
 					<dim>8</dim>
 					<dim>1</dim>
@@ -8137,10 +8137,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="420" name="Convolution_4742" type="Convolution" version="opset1">
+		<layer id="420" name="Conv_495/WithoutBiases" type="Convolution" version="opset1">
 			<data strides="1, 1" dilations="1, 1" pads_begin="0, 0" pads_end="0, 0" auto_pad="explicit"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Convolution_4742"/>
+				<attribute name="fused_names" version="0" value="Conv_495/WithoutBiases"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -8176,10 +8176,10 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="422" name="moments" type="Add" version="opset1">
+		<layer id="422" name="parameters" type="Add" version="opset1">
 			<data auto_broadcast="numpy"/>
 			<rt_info>
-				<attribute name="fused_names" version="0" value="Concat_4761, Reshape_4762, moments"/>
+				<attribute name="fused_names" version="0" value="Concat_4761, Reshape_4762, parameters"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -8196,7 +8196,7 @@
 				</port>
 			</input>
 			<output>
-				<port id="2" precision="FP32" names="moments">
+				<port id="2" precision="FP32" names="parameters">
 					<dim>1</dim>
 					<dim>8</dim>
 					<dim>64</dim>
@@ -8204,9 +8204,9 @@
 				</port>
 			</output>
 		</layer>
-		<layer id="423" name="moments/sink_port_0" type="Result" version="opset1">
+		<layer id="423" name="parameters/sink_port_0" type="Result" version="opset1">
 			<rt_info>
-				<attribute name="fused_names" version="0" value="moments/sink_port_0"/>
+				<attribute name="fused_names" version="0" value="parameters/sink_port_0"/>
 			</rt_info>
 			<input>
 				<port id="0" precision="FP32">
@@ -8225,9 +8225,9 @@
 		<edge from-layer="3" from-port="0" to-layer="4" to-port="1"/>
 		<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="37" to-port="0"/>
-		<edge from-layer="6" from-port="2" to-layer="11" to-port="0"/>
 		<edge from-layer="6" from-port="2" to-layer="8" to-port="0"/>
+		<edge from-layer="6" from-port="2" to-layer="11" to-port="0"/>
+		<edge from-layer="6" from-port="2" to-layer="37" 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="10" to-port="0"/>
 		<edge from-layer="9" from-port="0" to-layer="10" to-port="1"/>
@@ -8298,8 +8298,8 @@
 		<edge from-layer="70" from-port="2" to-layer="72" to-port="0"/>
 		<edge from-layer="71" from-port="0" to-layer="72" to-port="1"/>
 		<edge from-layer="72" from-port="2" to-layer="74" to-port="0"/>
-		<edge from-layer="72" from-port="2" to-layer="78" to-port="0"/>
 		<edge from-layer="72" from-port="2" to-layer="81" to-port="0"/>
+		<edge from-layer="72" from-port="2" to-layer="78" to-port="0"/>
 		<edge from-layer="73" from-port="0" to-layer="74" to-port="1"/>
 		<edge from-layer="74" from-port="2" to-layer="76" to-port="0"/>
 		<edge from-layer="75" from-port="0" to-layer="76" to-port="1"/>
@@ -8336,8 +8336,8 @@
 		<edge from-layer="105" from-port="0" to-layer="106" to-port="1"/>
 		<edge from-layer="106" from-port="2" to-layer="107" to-port="1"/>
 		<edge from-layer="107" from-port="2" to-layer="109" to-port="0"/>
-		<edge from-layer="107" from-port="2" to-layer="112" to-port="0"/>
 		<edge from-layer="107" from-port="2" to-layer="138" to-port="0"/>
+		<edge from-layer="107" from-port="2" to-layer="112" to-port="0"/>
 		<edge from-layer="108" from-port="0" to-layer="109" to-port="1"/>
 		<edge from-layer="109" from-port="2" to-layer="111" to-port="0"/>
 		<edge from-layer="110" from-port="0" to-layer="111" to-port="1"/>
@@ -8373,9 +8373,9 @@
 		<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="144" to-port="0"/>
 		<edge from-layer="142" from-port="2" to-layer="148" to-port="0"/>
 		<edge from-layer="142" from-port="2" to-layer="151" to-port="0"/>
-		<edge from-layer="142" from-port="2" to-layer="144" to-port="0"/>
 		<edge from-layer="143" from-port="0" to-layer="144" to-port="1"/>
 		<edge from-layer="144" from-port="2" to-layer="146" to-port="0"/>
 		<edge from-layer="145" from-port="0" to-layer="146" to-port="1"/>
@@ -8394,8 +8394,8 @@
 		<edge from-layer="158" from-port="0" to-layer="159" to-port="1"/>
 		<edge from-layer="159" from-port="2" 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="161" from-port="2" to-layer="166" to-port="0"/>
+		<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="165" to-port="0"/>
 		<edge from-layer="164" from-port="0" to-layer="165" to-port="1"/>
@@ -8428,8 +8428,8 @@
 		<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="192" from-port="2" to-layer="197" to-port="0"/>
+		<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="196" to-port="0"/>
 		<edge from-layer="195" from-port="0" to-layer="196" to-port="1"/>
@@ -8449,9 +8449,9 @@
 		<edge from-layer="209" from-port="0" to-layer="210" to-port="1"/>
 		<edge from-layer="210" from-port="2" to-layer="212" to-port="0"/>
 		<edge from-layer="211" from-port="0" to-layer="212" to-port="1"/>
-		<edge from-layer="212" from-port="2" to-layer="214" to-port="0"/>
 		<edge from-layer="212" from-port="2" to-layer="217" to-port="0"/>
 		<edge from-layer="212" from-port="2" to-layer="243" to-port="0"/>
+		<edge from-layer="212" from-port="2" to-layer="214" to-port="0"/>
 		<edge from-layer="213" from-port="0" to-layer="214" to-port="1"/>
 		<edge from-layer="214" from-port="2" to-layer="216" to-port="0"/>
 		<edge from-layer="215" from-port="0" to-layer="216" to-port="1"/>
@@ -8517,9 +8517,9 @@
 		<edge from-layer="271" from-port="2" 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="1"/>
+		<edge from-layer="274" from-port="2" to-layer="305" to-port="0"/>
 		<edge from-layer="274" from-port="2" to-layer="276" to-port="0"/>
 		<edge from-layer="274" from-port="2" to-layer="279" to-port="0"/>
-		<edge from-layer="274" from-port="2" to-layer="305" to-port="0"/>
 		<edge from-layer="275" from-port="0" to-layer="276" to-port="1"/>
 		<edge from-layer="276" from-port="2" to-layer="278" to-port="0"/>
 		<edge from-layer="277" from-port="0" to-layer="278" to-port="1"/>
@@ -8551,9 +8551,9 @@
 		<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="305" to-port="1"/>
+		<edge from-layer="305" from-port="2" to-layer="307" to-port="0"/>
 		<edge from-layer="305" from-port="2" to-layer="310" to-port="0"/>
 		<edge from-layer="305" from-port="2" to-layer="372" to-port="1"/>
-		<edge from-layer="305" from-port="2" to-layer="307" to-port="0"/>
 		<edge from-layer="306" from-port="0" to-layer="307" to-port="1"/>
 		<edge from-layer="307" from-port="2" to-layer="309" to-port="0"/>
 		<edge from-layer="308" from-port="0" to-layer="309" to-port="1"/>
@@ -8627,8 +8627,8 @@
 		<edge from-layer="370" from-port="2" to-layer="371" to-port="0"/>
 		<edge from-layer="371" from-port="2" to-layer="372" to-port="0"/>
 		<edge from-layer="372" from-port="2" to-layer="374" to-port="0"/>
-		<edge from-layer="372" from-port="2" to-layer="403" to-port="0"/>
 		<edge from-layer="372" from-port="2" to-layer="377" to-port="0"/>
+		<edge from-layer="372" from-port="2" to-layer="403" to-port="0"/>
 		<edge from-layer="373" from-port="0" to-layer="374" to-port="1"/>
 		<edge from-layer="374" from-port="2" to-layer="376" to-port="0"/>
 		<edge from-layer="375" from-port="0" to-layer="376" to-port="1"/>
@@ -8683,8 +8683,8 @@
 		<edge from-layer="422" from-port="2" to-layer="423" to-port="0"/>
 	</edges>
 	<meta_data>
-		<MO_version value="2022.1.0-7019-cdb9bec7210-releases/2022/1"/>
-		<Runtime_version value="2022.1.0-7019-cdb9bec7210-releases/2022/1"/>
+		<MO_version value="2022.2.0-7690-940e927a22b-refs/pull/1296/head"/>
+		<Runtime_version value="2022.2.0-7690-940e927a22b-refs/pull/1296/head"/>
 		<legacy_path value="False"/>
 		<cli_parameters>
 			<caffe_parser_path value="DIR"/>
@@ -8700,9 +8700,9 @@
 			<extensions value="DIR"/>
 			<framework value="onnx"/>
 			<freeze_placeholder_with_value value="{}"/>
-			<input_model value="DIR/vae_encoder.onnx"/>
+			<input_model value="DIR\vae_encoder.onnx"/>
 			<input_model_is_text value="False"/>
-			<k value="DIR/CustomLayersMapping.xml"/>
+			<k value="DIR\CustomLayersMapping.xml"/>
 			<layout value="()"/>
 			<layout_values value="{}"/>
 			<legacy_mxnet_model value="False"/>