Update VAE Encoder.

See https://github.com/apple/ml-stable-diffusion/issues/176

original/compiled/VAEEncoder.mlmodelc/analytics/coremldata.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:925a46e40af84f2a065c43d3d4bbe0d5b49766a08c75df4c46a870f2a191868c
+oid sha256:657e981b075d9079558e097e0c7589c72e13685615ab8e3ea8d9b1500990784a
 size 207

original/compiled/VAEEncoder.mlmodelc/coremldata.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:69c400c21fa4f0a15b9645d0f03ba36de9d758e0ca7ee076c079f449c3007ff1
-size 1274
+oid sha256:69796e1d32753e2b81d53715c58d8ae756827b24bdbe302811ff2acd98f9ccb0
+size 761

original/compiled/VAEEncoder.mlmodelc/metadata.json

@@ -10,7 +10,7 @@
         "formattedType" : "MultiArray (Float32)",
         "shortDescription" : "The latent embeddings from the unet model from the input image.",
         "shape" : "[]",
-        "name" : "latent_dist",
+        "name" : "latent",
         "type" : "MultiArray"
       }
     ],
@@ -23,26 +23,23 @@
     "storagePrecision" : "Float16",
     "license" : "OpenRAIL (https:\/\/huggingface.co\/spaces\/CompVis\/stable-diffusion-license)",
     "mlProgramOperationTypeHistogram" : {
-      "Transpose" : 7,
-      "Ios16.exp" : 1,
-      "Ios16.reduceMean" : 44,
-      "Ios16.softmax" : 1,
-      "Split" : 1,
-      "Ios16.linear" : 4,
-      "Ios16.add" : 35,
-      "Ios16.realDiv" : 22,
-      "Ios16.square" : 22,
       "Pad" : 3,
-      "Ios16.sub" : 22,
       "Ios16.cast" : 1,
-      "Ios16.clip" : 1,
+      "Ios16.mul" : 1,
+      "Ios16.sqrt" : 22,
+      "Ios16.sub" : 22,
+      "Transpose" : 7,
       "Ios16.conv" : 28,
+      "Ios16.add" : 33,
+      "Ios16.linear" : 4,
       "Ios16.matmul" : 2,
-      "Ios16.reshape" : 54,
+      "Ios16.realDiv" : 22,
+      "Ios16.reduceMean" : 44,
+      "Ios16.softmax" : 1,
       "Ios16.batchNorm" : 22,
-      "Ios16.silu" : 21,
-      "Ios16.sqrt" : 22,
-      "Ios16.mul" : 6
+      "Ios16.square" : 22,
+      "Ios16.reshape" : 54,
+      "Ios16.silu" : 21
     },
     "computePrecision" : "Mixed (Float32, Float16, Int32)",
     "isUpdatable" : "0",
@@ -62,55 +59,15 @@
         "isOptional" : "0",
         "dataType" : "Float16",
         "formattedType" : "MultiArray (Float16 1 × 3 × 512 × 512)",
-        "shortDescription" : "An image of the correct size to create the latent space with, image2image and in-painting.",
+        "shortDescription" : "The input image to base the initial latents on normalized to range [-1, 1]",
         "shape" : "[1, 3, 512, 512]",
-        "name" : "sample",
-        "type" : "MultiArray"
-      },
-      {
-        "hasShapeFlexibility" : "0",
-        "isOptional" : "0",
-        "dataType" : "Float16",
-        "formattedType" : "MultiArray (Float16 1 × 4 × 64 × 64)",
-        "shortDescription" : "Latent noise for `DiagonalGaussianDistribution` operation.",
-        "shape" : "[1, 4, 64, 64]",
-        "name" : "diagonal_noise",
-        "type" : "MultiArray"
-      },
-      {
-        "hasShapeFlexibility" : "0",
-        "isOptional" : "0",
-        "dataType" : "Float16",
-        "formattedType" : "MultiArray (Float16 1 × 4 × 64 × 64)",
-        "shortDescription" : "Latent noise for use with strength parameter of image2image",
-        "shape" : "[1, 4, 64, 64]",
-        "name" : "noise",
-        "type" : "MultiArray"
-      },
-      {
-        "hasShapeFlexibility" : "0",
-        "isOptional" : "0",
-        "dataType" : "Float16",
-        "formattedType" : "MultiArray (Float16 1 × 1)",
-        "shortDescription" : "Precalculated `sqrt_alphas_cumprod` value based on strength and the current schedular's alphasCumprod values",
-        "shape" : "[1, 1]",
-        "name" : "sqrt_alphas_cumprod",
-        "type" : "MultiArray"
-      },
-      {
-        "hasShapeFlexibility" : "0",
-        "isOptional" : "0",
-        "dataType" : "Float16",
-        "formattedType" : "MultiArray (Float16 1 × 1)",
-        "shortDescription" : "Precalculated `sqrt_one_minus_alphas_cumprod` value based on strength and the current schedular's alphasCumprod values",
-        "shape" : "[1, 1]",
-        "name" : "sqrt_one_minus_alphas_cumprod",
+        "name" : "z",
         "type" : "MultiArray"
       }
     ],
     "userDefinedMetadata" : {
-      "com.github.apple.coremltools.version" : "6.1",
-      "com.github.apple.coremltools.source" : "torch==1.13.1"
+      "com.github.apple.coremltools.version" : "6.3.0",
+      "com.github.apple.coremltools.source" : "torch==2.0.0"
     },
     "generatedClassName" : "Stable_Diffusion_version_runwayml_stable_diffusion_v1_5_vae_encoder",
     "method" : "predict"

original/compiled/VAEEncoder.mlmodelc/model.mil

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3cd529b933adc7e28b4057051eb816d753241246416bceacf25eb844e2b64d4a
-size 137752
+oid sha256:d33d5d475d2457cb48abae239a2ba0a6ff14a6fa344633e4866c74c6f7907724
+size 135576

original/compiled/VAEEncoder.mlmodelc/weights/weight.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:57d5438c221fc6c65bd4249f621cfd15deab77ee03495f5fb99662426ef74d51
+oid sha256:4c31357d1d5f8896ad66eba657d641e2238175458c70864b642323bf1bbdb68a
 size 68370240

original/packages/Stable_Diffusion_version_runwayml_stable-diffusion-v1-5_vae_encoder.mlpackage/Data/com.apple.CoreML/model.mlmodel

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:573fa4f8e8eb1601aa2ba87a789bfba6c9ee3a0d6c09db67a5c7c22c96a9194f
-size 117450
+oid sha256:c75354abd80e360ce305171d168706895c51dc069d6e1496c0433c90fc5eba56
+size 114904

original/packages/Stable_Diffusion_version_runwayml_stable-diffusion-v1-5_vae_encoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:57d5438c221fc6c65bd4249f621cfd15deab77ee03495f5fb99662426ef74d51
+oid sha256:4c31357d1d5f8896ad66eba657d641e2238175458c70864b642323bf1bbdb68a
 size 68370240

original/packages/Stable_Diffusion_version_runwayml_stable-diffusion-v1-5_vae_encoder.mlpackage/Manifest.json

@@ -1,18 +1,18 @@
 {
     "fileFormatVersion": "1.0.0",
     "itemInfoEntries": {
-        "3D939470-6F09-4758-9FFC-5CDD63158DC7": {
+        "A43D6EFD-002C-4F30-B88B-84BB2A88335C": {
             "author": "com.apple.CoreML",
             "description": "CoreML Model Specification",
             "name": "model.mlmodel",
             "path": "com.apple.CoreML/model.mlmodel"
         },
-        "E607BD7B-4F90-4362-B733-CE538A162148": {
+        "E020E8AA-E89B-41E5-AE98-824B42849511": {
             "author": "com.apple.CoreML",
             "description": "CoreML Model Weights",
             "name": "weights",
             "path": "com.apple.CoreML/weights"
         }
     },
-    "rootModelIdentifier": "3D939470-6F09-4758-9FFC-5CDD63158DC7"
+    "rootModelIdentifier": "A43D6EFD-002C-4F30-B88B-84BB2A88335C"
 }

split_einsum/compiled/VAEEncoder.mlmodelc/analytics/coremldata.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:925a46e40af84f2a065c43d3d4bbe0d5b49766a08c75df4c46a870f2a191868c
+oid sha256:657e981b075d9079558e097e0c7589c72e13685615ab8e3ea8d9b1500990784a
 size 207

split_einsum/compiled/VAEEncoder.mlmodelc/coremldata.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:69c400c21fa4f0a15b9645d0f03ba36de9d758e0ca7ee076c079f449c3007ff1
-size 1274
+oid sha256:69796e1d32753e2b81d53715c58d8ae756827b24bdbe302811ff2acd98f9ccb0
+size 761

split_einsum/compiled/VAEEncoder.mlmodelc/metadata.json

@@ -10,7 +10,7 @@
         "formattedType" : "MultiArray (Float32)",
         "shortDescription" : "The latent embeddings from the unet model from the input image.",
         "shape" : "[]",
-        "name" : "latent_dist",
+        "name" : "latent",
         "type" : "MultiArray"
       }
     ],
@@ -23,26 +23,23 @@
     "storagePrecision" : "Float16",
     "license" : "OpenRAIL (https:\/\/huggingface.co\/spaces\/CompVis\/stable-diffusion-license)",
     "mlProgramOperationTypeHistogram" : {
-      "Transpose" : 7,
-      "Ios16.exp" : 1,
-      "Ios16.reduceMean" : 44,
-      "Ios16.softmax" : 1,
-      "Split" : 1,
-      "Ios16.linear" : 4,
-      "Ios16.add" : 35,
-      "Ios16.realDiv" : 22,
-      "Ios16.square" : 22,
       "Pad" : 3,
-      "Ios16.sub" : 22,
       "Ios16.cast" : 1,
-      "Ios16.clip" : 1,
+      "Ios16.mul" : 1,
+      "Ios16.sqrt" : 22,
+      "Ios16.sub" : 22,
+      "Transpose" : 7,
       "Ios16.conv" : 28,
+      "Ios16.add" : 33,
+      "Ios16.linear" : 4,
       "Ios16.matmul" : 2,
-      "Ios16.reshape" : 54,
+      "Ios16.realDiv" : 22,
+      "Ios16.reduceMean" : 44,
+      "Ios16.softmax" : 1,
       "Ios16.batchNorm" : 22,
-      "Ios16.silu" : 21,
-      "Ios16.sqrt" : 22,
-      "Ios16.mul" : 6
+      "Ios16.square" : 22,
+      "Ios16.reshape" : 54,
+      "Ios16.silu" : 21
     },
     "computePrecision" : "Mixed (Float32, Float16, Int32)",
     "isUpdatable" : "0",
@@ -62,55 +59,15 @@
         "isOptional" : "0",
         "dataType" : "Float16",
         "formattedType" : "MultiArray (Float16 1 × 3 × 512 × 512)",
-        "shortDescription" : "An image of the correct size to create the latent space with, image2image and in-painting.",
+        "shortDescription" : "The input image to base the initial latents on normalized to range [-1, 1]",
         "shape" : "[1, 3, 512, 512]",
-        "name" : "sample",
-        "type" : "MultiArray"
-      },
-      {
-        "hasShapeFlexibility" : "0",
-        "isOptional" : "0",
-        "dataType" : "Float16",
-        "formattedType" : "MultiArray (Float16 1 × 4 × 64 × 64)",
-        "shortDescription" : "Latent noise for `DiagonalGaussianDistribution` operation.",
-        "shape" : "[1, 4, 64, 64]",
-        "name" : "diagonal_noise",
-        "type" : "MultiArray"
-      },
-      {
-        "hasShapeFlexibility" : "0",
-        "isOptional" : "0",
-        "dataType" : "Float16",
-        "formattedType" : "MultiArray (Float16 1 × 4 × 64 × 64)",
-        "shortDescription" : "Latent noise for use with strength parameter of image2image",
-        "shape" : "[1, 4, 64, 64]",
-        "name" : "noise",
-        "type" : "MultiArray"
-      },
-      {
-        "hasShapeFlexibility" : "0",
-        "isOptional" : "0",
-        "dataType" : "Float16",
-        "formattedType" : "MultiArray (Float16 1 × 1)",
-        "shortDescription" : "Precalculated `sqrt_alphas_cumprod` value based on strength and the current schedular's alphasCumprod values",
-        "shape" : "[1, 1]",
-        "name" : "sqrt_alphas_cumprod",
-        "type" : "MultiArray"
-      },
-      {
-        "hasShapeFlexibility" : "0",
-        "isOptional" : "0",
-        "dataType" : "Float16",
-        "formattedType" : "MultiArray (Float16 1 × 1)",
-        "shortDescription" : "Precalculated `sqrt_one_minus_alphas_cumprod` value based on strength and the current schedular's alphasCumprod values",
-        "shape" : "[1, 1]",
-        "name" : "sqrt_one_minus_alphas_cumprod",
+        "name" : "z",
         "type" : "MultiArray"
       }
     ],
     "userDefinedMetadata" : {
-      "com.github.apple.coremltools.version" : "6.1",
-      "com.github.apple.coremltools.source" : "torch==1.13.1"
+      "com.github.apple.coremltools.version" : "6.3.0",
+      "com.github.apple.coremltools.source" : "torch==2.0.0"
     },
     "generatedClassName" : "Stable_Diffusion_version_runwayml_stable_diffusion_v1_5_vae_encoder",
     "method" : "predict"

split_einsum/compiled/VAEEncoder.mlmodelc/model.mil

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3cd529b933adc7e28b4057051eb816d753241246416bceacf25eb844e2b64d4a
-size 137752
+oid sha256:d33d5d475d2457cb48abae239a2ba0a6ff14a6fa344633e4866c74c6f7907724
+size 135576

split_einsum/compiled/VAEEncoder.mlmodelc/weights/weight.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d34df7191f1b851ccbfc509c49afd294d2e7fc7bfc5103e79516388e16b3b726
+oid sha256:4c31357d1d5f8896ad66eba657d641e2238175458c70864b642323bf1bbdb68a
 size 68370240

split_einsum/packages/Stable_Diffusion_version_runwayml_stable-diffusion-v1-5_vae_encoder.mlpackage/Data/com.apple.CoreML/model.mlmodel

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:573fa4f8e8eb1601aa2ba87a789bfba6c9ee3a0d6c09db67a5c7c22c96a9194f
-size 117450
+oid sha256:c75354abd80e360ce305171d168706895c51dc069d6e1496c0433c90fc5eba56
+size 114904

split_einsum/packages/Stable_Diffusion_version_runwayml_stable-diffusion-v1-5_vae_encoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d34df7191f1b851ccbfc509c49afd294d2e7fc7bfc5103e79516388e16b3b726
+oid sha256:4c31357d1d5f8896ad66eba657d641e2238175458c70864b642323bf1bbdb68a
 size 68370240

split_einsum/packages/Stable_Diffusion_version_runwayml_stable-diffusion-v1-5_vae_encoder.mlpackage/Manifest.json

@@ -1,18 +1,18 @@
 {
     "fileFormatVersion": "1.0.0",
     "itemInfoEntries": {
-        "7EACD8CB-E3C8-49A1-B9FF-04B1EDACDC47": {
+        "A43D6EFD-002C-4F30-B88B-84BB2A88335C": {
             "author": "com.apple.CoreML",
             "description": "CoreML Model Specification",
             "name": "model.mlmodel",
             "path": "com.apple.CoreML/model.mlmodel"
         },
-        "86199C84-B673-4184-A06D-15D834AAA033": {
+        "E020E8AA-E89B-41E5-AE98-824B42849511": {
             "author": "com.apple.CoreML",
             "description": "CoreML Model Weights",
             "name": "weights",
             "path": "com.apple.CoreML/weights"
         }
     },
-    "rootModelIdentifier": "7EACD8CB-E3C8-49A1-B9FF-04B1EDACDC47"
+    "rootModelIdentifier": "A43D6EFD-002C-4F30-B88B-84BB2A88335C"
 }