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:eee76fbae1fa08a0d7bd18b2eb1f0324f9fdba01656f441a3e8357f08bb6345a
+oid sha256:132c6d52eacfe75bf06faebc63ba323965841e05c77b1c5e6158491100d96bd9
 size 207

original/compiled/VAEEncoder.mlmodelc/coremldata.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:854e552c6e9273ef52b93a0fd2013399421450f497fcd327eb8aa371a2e5f46d
-size 1284
+oid sha256:2ac1e4a8ea5cfe3a02d26fff80894c8eee09248a0776be6dd55aab7d44d97d1b
+size 771

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_stabilityai_stable_diffusion_2_base_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:3b7ad3fbf4898564c46e08732ce223712961270064549b06bdc95ec22a6e88b1
+oid sha256:069f194d78e0893a066afb3805726f4e28d13538f2168204008530e9d9774ee2
 size 68370240

original/packages/Stable_Diffusion_version_stabilityai_stable-diffusion-2-base_vae_encoder.mlpackage/Data/com.apple.CoreML/model.mlmodel

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:69544ea13c1a7b3115c01eb2e775a10136832d6ff91a074faf11b75f241d726c
-size 117460
+oid sha256:d148321b3515119341c54be79906feeaeb15e04e90368fab9ab434ad8ac8017e
+size 114914

original/packages/Stable_Diffusion_version_stabilityai_stable-diffusion-2-base_vae_encoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3b7ad3fbf4898564c46e08732ce223712961270064549b06bdc95ec22a6e88b1
+oid sha256:069f194d78e0893a066afb3805726f4e28d13538f2168204008530e9d9774ee2
 size 68370240

original/packages/Stable_Diffusion_version_stabilityai_stable-diffusion-2-base_vae_encoder.mlpackage/Manifest.json

@@ -1,18 +1,18 @@
 {
     "fileFormatVersion": "1.0.0",
     "itemInfoEntries": {
-        "38BC24FA-2978-4B03-AF04-231A43CBF74C": {
+        "08A44175-364B-4515-AA88-70B20EC944C2": {
             "author": "com.apple.CoreML",
             "description": "CoreML Model Weights",
             "name": "weights",
             "path": "com.apple.CoreML/weights"
         },
-        "4627B776-3BCE-4766-960C-5877AC8F5A80": {
+        "9E280C88-8CAA-4EEA-986E-E04DA7E31BF6": {
             "author": "com.apple.CoreML",
             "description": "CoreML Model Specification",
             "name": "model.mlmodel",
             "path": "com.apple.CoreML/model.mlmodel"
         }
     },
-    "rootModelIdentifier": "4627B776-3BCE-4766-960C-5877AC8F5A80"
+    "rootModelIdentifier": "9E280C88-8CAA-4EEA-986E-E04DA7E31BF6"
 }

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

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:eee76fbae1fa08a0d7bd18b2eb1f0324f9fdba01656f441a3e8357f08bb6345a
+oid sha256:132c6d52eacfe75bf06faebc63ba323965841e05c77b1c5e6158491100d96bd9
 size 207

split_einsum/compiled/VAEEncoder.mlmodelc/coremldata.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:854e552c6e9273ef52b93a0fd2013399421450f497fcd327eb8aa371a2e5f46d
-size 1284
+oid sha256:2ac1e4a8ea5cfe3a02d26fff80894c8eee09248a0776be6dd55aab7d44d97d1b
+size 771

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_stabilityai_stable_diffusion_2_base_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:5061f127d29e767db8e4703bdfaf6cf2e944d5d8a27e807c78828496de3ce267
+oid sha256:069f194d78e0893a066afb3805726f4e28d13538f2168204008530e9d9774ee2
 size 68370240

split_einsum/packages/Stable_Diffusion_version_stabilityai_stable-diffusion-2-base_vae_encoder.mlpackage/Data/com.apple.CoreML/model.mlmodel

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:69544ea13c1a7b3115c01eb2e775a10136832d6ff91a074faf11b75f241d726c
-size 117460
+oid sha256:d148321b3515119341c54be79906feeaeb15e04e90368fab9ab434ad8ac8017e
+size 114914

split_einsum/packages/Stable_Diffusion_version_stabilityai_stable-diffusion-2-base_vae_encoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5061f127d29e767db8e4703bdfaf6cf2e944d5d8a27e807c78828496de3ce267
+oid sha256:069f194d78e0893a066afb3805726f4e28d13538f2168204008530e9d9774ee2
 size 68370240

split_einsum/packages/Stable_Diffusion_version_stabilityai_stable-diffusion-2-base_vae_encoder.mlpackage/Manifest.json

@@ -1,18 +1,18 @@
 {
     "fileFormatVersion": "1.0.0",
     "itemInfoEntries": {
-        "260E53C9-7E53-4C87-9FA8-161CC0D21FDF": {
-            "author": "com.apple.CoreML",
-            "description": "CoreML Model Specification",
-            "name": "model.mlmodel",
-            "path": "com.apple.CoreML/model.mlmodel"
-        },
-        "70A48119-9715-487C-9953-FCA2BCE32D24": {
+        "08A44175-364B-4515-AA88-70B20EC944C2": {
             "author": "com.apple.CoreML",
             "description": "CoreML Model Weights",
             "name": "weights",
             "path": "com.apple.CoreML/weights"
+        },
+        "9E280C88-8CAA-4EEA-986E-E04DA7E31BF6": {
+            "author": "com.apple.CoreML",
+            "description": "CoreML Model Specification",
+            "name": "model.mlmodel",
+            "path": "com.apple.CoreML/model.mlmodel"
         }
     },
-    "rootModelIdentifier": "260E53C9-7E53-4C87-9FA8-161CC0D21FDF"
+    "rootModelIdentifier": "9E280C88-8CAA-4EEA-986E-E04DA7E31BF6"
 }