Add SPLIT_EINSUM variant.

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6ee59ccda5036f123131d35c0c7a96678b33c793cdf81379afede2950fdfbf17
+size 207

split_einsum/compiled/TextEncoder.mlmodelc/coremldata.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2ec66e4fac048e8836ac2cc5103bb66149958908bb06e293f0e55b7f792843dd
+size 838

split_einsum/compiled/TextEncoder.mlmodelc/metadata.json

@@ -0,0 +1,85 @@
+[
+  {
+    "shortDescription" : "Stable Diffusion generates images conditioned on text and\/or other images as input through the diffusion process. Please refer to https:\/\/arxiv.org\/abs\/2112.10752 for details.",
+    "metadataOutputVersion" : "3.0",
+    "outputSchema" : [
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "The token embeddings as encoded by the Transformer model",
+        "shape" : "[]",
+        "name" : "last_hidden_state",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "The version of the `last_hidden_state` output after pooling",
+        "shape" : "[]",
+        "name" : "pooled_outputs",
+        "type" : "MultiArray"
+      }
+    ],
+    "version" : "stabilityai\/stable-diffusion-2-1-base",
+    "modelParameters" : [
+
+    ],
+    "author" : "Please refer to the Model Card available at huggingface.co\/stabilityai\/stable-diffusion-2-1-base",
+    "specificationVersion" : 7,
+    "storagePrecision" : "Float16",
+    "license" : "OpenRAIL (https:\/\/huggingface.co\/spaces\/CompVis\/stable-diffusion-license)",
+    "mlProgramOperationTypeHistogram" : {
+      "Ios16.cast" : 3,
+      "Ios16.mul" : 23,
+      "Ios16.layerNorm" : 47,
+      "Ios16.sub" : 1,
+      "BandPart" : 1,
+      "Stack" : 1,
+      "Transpose" : 115,
+      "Ios16.linear" : 138,
+      "Ios16.add" : 70,
+      "Ios16.matmul" : 46,
+      "Ios16.gelu" : 23,
+      "Ios16.softmax" : 23,
+      "ExpandDims" : 1,
+      "Ios16.gather" : 1,
+      "Ios16.gatherNd" : 1,
+      "Ios16.reshape" : 230,
+      "Ios16.reduceArgmax" : 1
+    },
+    "computePrecision" : "Mixed (Float16, Float32, Int32)",
+    "isUpdatable" : "0",
+    "availability" : {
+      "macOS" : "13.0",
+      "tvOS" : "16.0",
+      "watchOS" : "9.0",
+      "iOS" : "16.0",
+      "macCatalyst" : "16.0"
+    },
+    "modelType" : {
+      "name" : "MLModelType_mlProgram"
+    },
+    "inputSchema" : [
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32 1 × 77)",
+        "shortDescription" : "The token ids that represent the input text",
+        "shape" : "[1, 77]",
+        "name" : "input_ids",
+        "type" : "MultiArray"
+      }
+    ],
+    "userDefinedMetadata" : {
+      "com.github.apple.coremltools.version" : "6.1",
+      "com.github.apple.coremltools.source" : "torch==1.13.1"
+    },
+    "generatedClassName" : "Stable_Diffusion_version_stabilityai_stable_diffusion_2_1_base_text_encoder",
+    "method" : "predict"
+  }
+]

split_einsum/compiled/TextEncoder.mlmodelc/model.mil

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a79b7a70cc6a8780549c65b6a80a36e30e126b2ac63a656fce104fbc7799d32c
+size 320318

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8321973f654bd7c62070c79a32eb2dc8179aca16e5b7ebe16775fbc41260c082
+size 680811520

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:50ad9846ff88727fc5cff3e3944dd3f32fcf16e1879ae6f5ec6ab4576a08491c
+size 207

split_einsum/compiled/Unet.mlmodelc/coremldata.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:22c8cf48d32b10b2976f5b2aa15b2335fe7ff1541b7bb20dc0aaee5316db6a91
+size 1220

split_einsum/compiled/Unet.mlmodelc/metadata.json

@@ -0,0 +1,102 @@
+[
+  {
+    "shortDescription" : "Stable Diffusion generates images conditioned on text or other images as input through the diffusion process. Please refer to https:\/\/arxiv.org\/abs\/2112.10752 for details.",
+    "metadataOutputVersion" : "3.0",
+    "outputSchema" : [
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "Same shape and dtype as the `sample` input. The predicted noise to facilitate the reverse diffusion (denoising) process",
+        "shape" : "[]",
+        "name" : "noise_pred",
+        "type" : "MultiArray"
+      }
+    ],
+    "version" : "stabilityai\/stable-diffusion-2-1-base",
+    "modelParameters" : [
+
+    ],
+    "author" : "Please refer to the Model Card available at huggingface.co\/stabilityai\/stable-diffusion-2-1-base",
+    "specificationVersion" : 7,
+    "storagePrecision" : "Float16",
+    "license" : "OpenRAIL (https:\/\/huggingface.co\/spaces\/CompVis\/stable-diffusion-license)",
+    "mlProgramOperationTypeHistogram" : {
+      "Transpose" : 32,
+      "UpsampleNearestNeighbor" : 3,
+      "Ios16.reduceMean" : 218,
+      "Ios16.sin" : 1,
+      "Ios16.softmax" : 390,
+      "Split" : 16,
+      "Ios16.add" : 265,
+      "Concat" : 46,
+      "Ios16.realDiv" : 61,
+      "Ios16.square" : 61,
+      "ExpandDims" : 3,
+      "Ios16.sub" : 109,
+      "Ios16.cast" : 1,
+      "Ios16.conv" : 282,
+      "Ios16.einsum" : 780,
+      "Ios16.gelu" : 16,
+      "Ios16.reshape" : 154,
+      "Ios16.batchNorm" : 61,
+      "Ios16.rsqrt" : 48,
+      "Ios16.silu" : 47,
+      "Ios16.sqrt" : 61,
+      "SliceByIndex" : 1172,
+      "Ios16.mul" : 551,
+      "Ios16.cos" : 1
+    },
+    "computePrecision" : "Mixed (Float32, Float16, Int32)",
+    "isUpdatable" : "0",
+    "availability" : {
+      "macOS" : "13.0",
+      "tvOS" : "16.0",
+      "watchOS" : "9.0",
+      "iOS" : "16.0",
+      "macCatalyst" : "16.0"
+    },
+    "modelType" : {
+      "name" : "MLModelType_mlProgram"
+    },
+    "inputSchema" : [
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float16",
+        "formattedType" : "MultiArray (Float16 2 × 4 × 64 × 64)",
+        "shortDescription" : "The low resolution latent feature maps being denoised through reverse diffusion",
+        "shape" : "[2, 4, 64, 64]",
+        "name" : "sample",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float16",
+        "formattedType" : "MultiArray (Float16 2)",
+        "shortDescription" : "A value emitted by the associated scheduler object to condition the model on a given noise schedule",
+        "shape" : "[2]",
+        "name" : "timestep",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float16",
+        "formattedType" : "MultiArray (Float16 2 × 1024 × 1 × 77)",
+        "shortDescription" : "Output embeddings from the associated text_encoder model to condition to generated image on text. A maximum of 77 tokens (~40 words) are allowed. Longer text is truncated. Shorter text does not reduce computation.",
+        "shape" : "[2, 1024, 1, 77]",
+        "name" : "encoder_hidden_states",
+        "type" : "MultiArray"
+      }
+    ],
+    "userDefinedMetadata" : {
+      "com.github.apple.coremltools.version" : "6.1",
+      "com.github.apple.coremltools.source" : "torch==1.13.1"
+    },
+    "generatedClassName" : "Stable_Diffusion_version_stabilityai_stable_diffusion_2_1_base_unet",
+    "method" : "predict"
+  }
+]

split_einsum/compiled/Unet.mlmodelc/model.mil

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:51498bdb1d2167f35a358a3bc5f2417b4ffc1fe4eb936f64ab1a6ba8c9ba108f
+size 1889302

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f82da2385e2d92ca3a96ea75d6f3ec2db6b04dfc4d1f4254ad9059643d9699a4
+size 1732124416

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0270221011227ad9debf43fb4675178091ef88baa9b0100b2c6b7768622b8b54
+size 207

split_einsum/compiled/UnetChunk1.mlmodelc/coremldata.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cb7324b6d3791244534f175bc907aecbd5c4eb1691ac75310de194fce41bfd4c
+size 528

split_einsum/compiled/UnetChunk1.mlmodelc/metadata.json

@@ -0,0 +1,216 @@
+[
+  {
+    "metadataOutputVersion" : "3.0",
+    "storagePrecision" : "Float16",
+    "outputSchema" : [
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "",
+        "shape" : "[]",
+        "name" : "input_143_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "",
+        "shape" : "[]",
+        "name" : "input_169_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "",
+        "shape" : "[]",
+        "name" : "input_35_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "",
+        "shape" : "[]",
+        "name" : "hidden_states_149_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "",
+        "shape" : "[]",
+        "name" : "input_171_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "",
+        "shape" : "[]",
+        "name" : "input_253_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "",
+        "shape" : "[]",
+        "name" : "input_61_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "",
+        "shape" : "[]",
+        "name" : "input_115_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "",
+        "shape" : "[]",
+        "name" : "input_7_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "",
+        "shape" : "[]",
+        "name" : "input_63_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "",
+        "shape" : "[]",
+        "name" : "input_117_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "",
+        "shape" : "[]",
+        "name" : "input_89_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "",
+        "shape" : "[]",
+        "name" : "input_15_cast",
+        "type" : "MultiArray"
+      }
+    ],
+    "modelParameters" : [
+
+    ],
+    "specificationVersion" : 7,
+    "mlProgramOperationTypeHistogram" : {
+      "Transpose" : 14,
+      "Ios16.reduceMean" : 104,
+      "Ios16.sin" : 1,
+      "Ios16.softmax" : 180,
+      "Split" : 7,
+      "Ios16.add" : 124,
+      "Concat" : 18,
+      "Ios16.realDiv" : 31,
+      "Ios16.square" : 31,
+      "ExpandDims" : 3,
+      "Ios16.sub" : 52,
+      "Ios16.cast" : 13,
+      "Ios16.conv" : 129,
+      "Ios16.einsum" : 360,
+      "Ios16.gelu" : 7,
+      "Ios16.reshape" : 76,
+      "Ios16.batchNorm" : 31,
+      "Ios16.rsqrt" : 21,
+      "Ios16.silu" : 26,
+      "Ios16.sqrt" : 31,
+      "SliceByIndex" : 542,
+      "Ios16.mul" : 251,
+      "Ios16.cos" : 1
+    },
+    "computePrecision" : "Mixed (Float32, Float16, Int32)",
+    "isUpdatable" : "0",
+    "availability" : {
+      "macOS" : "13.0",
+      "tvOS" : "16.0",
+      "watchOS" : "9.0",
+      "iOS" : "16.0",
+      "macCatalyst" : "16.0"
+    },
+    "modelType" : {
+      "name" : "MLModelType_mlProgram"
+    },
+    "userDefinedMetadata" : {
+
+    },
+    "inputSchema" : [
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float16",
+        "formattedType" : "MultiArray (Float16 2 × 4 × 64 × 64)",
+        "shortDescription" : "",
+        "shape" : "[2, 4, 64, 64]",
+        "name" : "sample",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float16",
+        "formattedType" : "MultiArray (Float16 2)",
+        "shortDescription" : "",
+        "shape" : "[2]",
+        "name" : "timestep",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float16",
+        "formattedType" : "MultiArray (Float16 2 × 1024 × 1 × 77)",
+        "shortDescription" : "",
+        "shape" : "[2, 1024, 1, 77]",
+        "name" : "encoder_hidden_states",
+        "type" : "MultiArray"
+      }
+    ],
+    "generatedClassName" : "Stable_Diffusion_version_stabilityai_stable_diffusion_2_1_base_unet_chunk1",
+    "method" : "predict"
+  }
+]

split_einsum/compiled/UnetChunk1.mlmodelc/model.mil

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:345e510ac13f8ecd050cec2605cc22b11c8c57b6b2ac779c4f274bb15bb697c4
+size 878526

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a4e1bd4282f3d3aaa886c3f1c9b1ff828300f7ce25d6435485cc268311a399b2
+size 893578816

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

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

split_einsum/compiled/UnetChunk2.mlmodelc/coremldata.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:546fb96be244e776a2daf18253690b64fd52b7077903b87d3f0280cf48b02438
+size 594

split_einsum/compiled/UnetChunk2.mlmodelc/metadata.json

@@ -0,0 +1,204 @@
+[
+  {
+    "metadataOutputVersion" : "3.0",
+    "storagePrecision" : "Float16",
+    "outputSchema" : [
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "",
+        "shape" : "[]",
+        "name" : "noise_pred",
+        "type" : "MultiArray"
+      }
+    ],
+    "modelParameters" : [
+
+    ],
+    "specificationVersion" : 7,
+    "mlProgramOperationTypeHistogram" : {
+      "Transpose" : 18,
+      "UpsampleNearestNeighbor" : 3,
+      "Ios16.reduceMean" : 114,
+      "Ios16.softmax" : 210,
+      "Split" : 9,
+      "Ios16.add" : 141,
+      "Concat" : 28,
+      "Ios16.realDiv" : 30,
+      "Ios16.square" : 30,
+      "Ios16.sub" : 57,
+      "Ios16.cast" : 14,
+      "Ios16.conv" : 153,
+      "Ios16.einsum" : 420,
+      "Ios16.gelu" : 9,
+      "Ios16.reshape" : 78,
+      "Ios16.batchNorm" : 30,
+      "Ios16.rsqrt" : 27,
+      "Ios16.silu" : 21,
+      "Ios16.sqrt" : 30,
+      "SliceByIndex" : 630,
+      "Ios16.mul" : 300
+    },
+    "computePrecision" : "Mixed (Int32, Float16, Float32)",
+    "isUpdatable" : "0",
+    "availability" : {
+      "macOS" : "13.0",
+      "tvOS" : "16.0",
+      "watchOS" : "9.0",
+      "iOS" : "16.0",
+      "macCatalyst" : "16.0"
+    },
+    "modelType" : {
+      "name" : "MLModelType_mlProgram"
+    },
+    "userDefinedMetadata" : {
+
+    },
+    "inputSchema" : [
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float16",
+        "formattedType" : "MultiArray (Float16 2 × 1024 × 1 × 77)",
+        "shortDescription" : "",
+        "shape" : "[2, 1024, 1, 77]",
+        "name" : "encoder_hidden_states",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32 2 × 640 × 16 × 16)",
+        "shortDescription" : "",
+        "shape" : "[2, 640, 16, 16]",
+        "name" : "input_117_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32 2 × 1280 × 1 × 1)",
+        "shortDescription" : "",
+        "shape" : "[2, 1280, 1, 1]",
+        "name" : "input_15_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32 2 × 320 × 64 × 64)",
+        "shortDescription" : "",
+        "shape" : "[2, 320, 64, 64]",
+        "name" : "input_7_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32 2 × 640 × 32 × 32)",
+        "shortDescription" : "",
+        "shape" : "[2, 640, 32, 32]",
+        "name" : "input_89_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32 2 × 320 × 64 × 64)",
+        "shortDescription" : "",
+        "shape" : "[2, 320, 64, 64]",
+        "name" : "input_35_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32 2 × 320 × 64 × 64)",
+        "shortDescription" : "",
+        "shape" : "[2, 320, 64, 64]",
+        "name" : "input_61_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32 2 × 320 × 32 × 32)",
+        "shortDescription" : "",
+        "shape" : "[2, 320, 32, 32]",
+        "name" : "input_63_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32 2 × 1280 × 16 × 16)",
+        "shortDescription" : "",
+        "shape" : "[2, 1280, 16, 16]",
+        "name" : "input_143_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32 2 × 1280 × 16 × 16)",
+        "shortDescription" : "",
+        "shape" : "[2, 1280, 16, 16]",
+        "name" : "input_169_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32 2 × 1280 × 8 × 8)",
+        "shortDescription" : "",
+        "shape" : "[2, 1280, 8, 8]",
+        "name" : "hidden_states_149_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32 2 × 1280 × 8 × 8)",
+        "shortDescription" : "",
+        "shape" : "[2, 1280, 8, 8]",
+        "name" : "input_171_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32 2 × 2560 × 8 × 8)",
+        "shortDescription" : "",
+        "shape" : "[2, 2560, 8, 8]",
+        "name" : "input_253_cast",
+        "type" : "MultiArray"
+      },
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32 2 × 640 × 32 × 32)",
+        "shortDescription" : "",
+        "shape" : "[2, 640, 32, 32]",
+        "name" : "input_115_cast",
+        "type" : "MultiArray"
+      }
+    ],
+    "generatedClassName" : "Stable_Diffusion_version_stabilityai_stable_diffusion_2_1_base_unet_chunk2",
+    "method" : "predict"
+  }
+]

split_einsum/compiled/UnetChunk2.mlmodelc/model.mil

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1d9846f57d5585564dc70e68f8cf2243c17c9e38d5c1ac4494bce492d1647aad
+size 1018022

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:da0c12de8817340c887604e3252d80ad8df1d38440e5bfa47e9b26609e52afd7
+size 838545664

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

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

split_einsum/compiled/VAEDecoder.mlmodelc/coremldata.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f02192454e1eef80f93c2139aed5440dfd2704933b4354bfd22859abdadd1c16
+size 768

split_einsum/compiled/VAEDecoder.mlmodelc/metadata.json

@@ -0,0 +1,75 @@
+[
+  {
+    "shortDescription" : "Stable Diffusion generates images conditioned on text and\/or other images as input through the diffusion process. Please refer to https:\/\/arxiv.org\/abs\/2112.10752 for details.",
+    "metadataOutputVersion" : "3.0",
+    "outputSchema" : [
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "Generated image normalized to range [-1, 1]",
+        "shape" : "[]",
+        "name" : "image",
+        "type" : "MultiArray"
+      }
+    ],
+    "version" : "stabilityai\/stable-diffusion-2-1-base",
+    "modelParameters" : [
+
+    ],
+    "author" : "Please refer to the Model Card available at huggingface.co\/stabilityai\/stable-diffusion-2-1-base",
+    "specificationVersion" : 7,
+    "storagePrecision" : "Float16",
+    "license" : "OpenRAIL (https:\/\/huggingface.co\/spaces\/CompVis\/stable-diffusion-license)",
+    "mlProgramOperationTypeHistogram" : {
+      "Ios16.cast" : 1,
+      "Ios16.mul" : 1,
+      "Ios16.sqrt" : 30,
+      "Ios16.sub" : 30,
+      "Transpose" : 7,
+      "UpsampleNearestNeighbor" : 3,
+      "Ios16.conv" : 36,
+      "Ios16.add" : 45,
+      "Ios16.linear" : 4,
+      "Ios16.matmul" : 2,
+      "Ios16.realDiv" : 30,
+      "Ios16.reduceMean" : 60,
+      "Ios16.softmax" : 1,
+      "Ios16.batchNorm" : 30,
+      "Ios16.square" : 30,
+      "Ios16.reshape" : 70,
+      "Ios16.silu" : 29
+    },
+    "computePrecision" : "Mixed (Float32, Float16, Int32)",
+    "isUpdatable" : "0",
+    "availability" : {
+      "macOS" : "13.0",
+      "tvOS" : "16.0",
+      "watchOS" : "9.0",
+      "iOS" : "16.0",
+      "macCatalyst" : "16.0"
+    },
+    "modelType" : {
+      "name" : "MLModelType_mlProgram"
+    },
+    "inputSchema" : [
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float16",
+        "formattedType" : "MultiArray (Float16 1 × 4 × 64 × 64)",
+        "shortDescription" : "The denoised latent embeddings from the unet model after the last step of reverse diffusion",
+        "shape" : "[1, 4, 64, 64]",
+        "name" : "z",
+        "type" : "MultiArray"
+      }
+    ],
+    "userDefinedMetadata" : {
+      "com.github.apple.coremltools.version" : "6.1",
+      "com.github.apple.coremltools.source" : "torch==1.13.1"
+    },
+    "generatedClassName" : "Stable_Diffusion_version_stabilityai_stable_diffusion_2_1_base_vae_decoder",
+    "method" : "predict"
+  }
+]

split_einsum/compiled/VAEDecoder.mlmodelc/model.mil

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bf6aa65536d1a72a18d2066a82e1ba5fb002583ae0482625f911d27a49a4f1f6
+size 176862

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c25e5fc1dc7c96e447792dc68770929eb609167269e8e83d771744f83a16453a
+size 99039232

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9c368a34c610037cedd5e7ff5434cf0a5d59d57b62cb8fe2d753f9798f0c30e1
+size 207

split_einsum/compiled/VAEEncoder.mlmodelc/coremldata.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e85251c72a334672f9e763a7a24fc551805007aa14ac144147adbc07ce23a90d
+size 1288

split_einsum/compiled/VAEEncoder.mlmodelc/metadata.json

@@ -0,0 +1,118 @@
+[
+  {
+    "shortDescription" : "Stable Diffusion generates images conditioned on text and\/or other images as input through the diffusion process. Please refer to https:\/\/arxiv.org\/abs\/2112.10752 for details.",
+    "metadataOutputVersion" : "3.0",
+    "outputSchema" : [
+      {
+        "hasShapeFlexibility" : "0",
+        "isOptional" : "0",
+        "dataType" : "Float32",
+        "formattedType" : "MultiArray (Float32)",
+        "shortDescription" : "The latent embeddings from the unet model from the input image.",
+        "shape" : "[]",
+        "name" : "latent_dist",
+        "type" : "MultiArray"
+      }
+    ],
+    "version" : "stabilityai\/stable-diffusion-2-1-base",
+    "modelParameters" : [
+
+    ],
+    "author" : "Please refer to the Model Card available at huggingface.co\/stabilityai\/stable-diffusion-2-1-base",
+    "specificationVersion" : 7,
+    "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.conv" : 28,
+      "Ios16.matmul" : 2,
+      "Ios16.reshape" : 54,
+      "Ios16.batchNorm" : 22,
+      "Ios16.silu" : 21,
+      "Ios16.sqrt" : 22,
+      "Ios16.mul" : 6
+    },
+    "computePrecision" : "Mixed (Float32, Float16, Int32)",
+    "isUpdatable" : "0",
+    "availability" : {
+      "macOS" : "13.0",
+      "tvOS" : "16.0",
+      "watchOS" : "9.0",
+      "iOS" : "16.0",
+      "macCatalyst" : "16.0"
+    },
+    "modelType" : {
+      "name" : "MLModelType_mlProgram"
+    },
+    "inputSchema" : [
+      {
+        "hasShapeFlexibility" : "0",
+        "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.",
+        "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",
+        "type" : "MultiArray"
+      }
+    ],
+    "userDefinedMetadata" : {
+      "com.github.apple.coremltools.version" : "6.1",
+      "com.github.apple.coremltools.source" : "torch==1.13.1"
+    },
+    "generatedClassName" : "Stable_Diffusion_version_stabilityai_stable_diffusion_2_1_base_vae_encoder",
+    "method" : "predict"
+  }
+]

split_einsum/compiled/VAEEncoder.mlmodelc/model.mil

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

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5a426db5c4683418072eb114c571855373d8161a4a66a17f13666ac31e5fbe49
+size 68370240

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ad495e8ec7e03084589f3f037ef507f881a1554ae193d388525df3920c3639ae
+size 272312

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8321973f654bd7c62070c79a32eb2dc8179aca16e5b7ebe16775fbc41260c082
+size 680811520

split_einsum/packages/Stable_Diffusion_version_stabilityai_stable-diffusion-2-1-base_text_encoder.mlpackage/Manifest.json

@@ -0,0 +1,18 @@
+{
+    "fileFormatVersion": "1.0.0",
+    "itemInfoEntries": {
+        "68704C24-A886-429D-A4FF-5EBFAB51C7A1": {
+            "author": "com.apple.CoreML",
+            "description": "CoreML Model Weights",
+            "name": "weights",
+            "path": "com.apple.CoreML/weights"
+        },
+        "77100E23-571E-4D1A-B7DD-B06BB4841DD8": {
+            "author": "com.apple.CoreML",
+            "description": "CoreML Model Specification",
+            "name": "model.mlmodel",
+            "path": "com.apple.CoreML/model.mlmodel"
+        }
+    },
+    "rootModelIdentifier": "77100E23-571E-4D1A-B7DD-B06BB4841DD8"
+}

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:17874541e2d46c52a809a3d2498e9920f8556801ae645e9c8526d4f7a6f503e7
+size 1640010

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f82da2385e2d92ca3a96ea75d6f3ec2db6b04dfc4d1f4254ad9059643d9699a4
+size 1732124416

split_einsum/packages/Stable_Diffusion_version_stabilityai_stable-diffusion-2-1-base_unet.mlpackage/Manifest.json

@@ -0,0 +1,18 @@
+{
+    "fileFormatVersion": "1.0.0",
+    "itemInfoEntries": {
+        "31609C42-0554-496A-82E9-9C43FC6E2F6F": {
+            "author": "com.apple.CoreML",
+            "description": "CoreML Model Specification",
+            "name": "model.mlmodel",
+            "path": "com.apple.CoreML/model.mlmodel"
+        },
+        "5288BB4A-9079-491F-B580-94647E933B60": {
+            "author": "com.apple.CoreML",
+            "description": "CoreML Model Weights",
+            "name": "weights",
+            "path": "com.apple.CoreML/weights"
+        }
+    },
+    "rootModelIdentifier": "31609C42-0554-496A-82E9-9C43FC6E2F6F"
+}

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:274a73cf5c7c08e716179e84ac7fe507f6fab5c8579771c38e3bae7a827a23fc
+size 761806

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a4e1bd4282f3d3aaa886c3f1c9b1ff828300f7ce25d6435485cc268311a399b2
+size 893578816

split_einsum/packages/Stable_Diffusion_version_stabilityai_stable-diffusion-2-1-base_unet_chunk1.mlpackage/Manifest.json

@@ -0,0 +1,18 @@
+{
+    "fileFormatVersion": "1.0.0",
+    "itemInfoEntries": {
+        "4A43DB2E-1412-4D43-9668-B08163512BD0": {
+            "author": "com.apple.CoreML",
+            "description": "CoreML Model Specification",
+            "name": "model.mlmodel",
+            "path": "com.apple.CoreML/model.mlmodel"
+        },
+        "7A358A50-B497-41F8-AB9D-91FBE3FCEA9C": {
+            "author": "com.apple.CoreML",
+            "description": "CoreML Model Weights",
+            "name": "weights",
+            "path": "com.apple.CoreML/weights"
+        }
+    },
+    "rootModelIdentifier": "4A43DB2E-1412-4D43-9668-B08163512BD0"
+}

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f4e6f75be4541e9165bd271a3ebc14f866f3edb3957efb913a5f21158c54d4ef
+size 884523

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:da0c12de8817340c887604e3252d80ad8df1d38440e5bfa47e9b26609e52afd7
+size 838545664

split_einsum/packages/Stable_Diffusion_version_stabilityai_stable-diffusion-2-1-base_unet_chunk2.mlpackage/Manifest.json

@@ -0,0 +1,18 @@
+{
+    "fileFormatVersion": "1.0.0",
+    "itemInfoEntries": {
+        "091F66CF-129C-42D1-A5DD-EFBD005CF5BB": {
+            "author": "com.apple.CoreML",
+            "description": "CoreML Model Weights",
+            "name": "weights",
+            "path": "com.apple.CoreML/weights"
+        },
+        "E8870E32-4708-4BF5-A91F-CEFD7B3174A4": {
+            "author": "com.apple.CoreML",
+            "description": "CoreML Model Specification",
+            "name": "model.mlmodel",
+            "path": "com.apple.CoreML/model.mlmodel"
+        }
+    },
+    "rootModelIdentifier": "E8870E32-4708-4BF5-A91F-CEFD7B3174A4"
+}

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:31b00e2dc7274d47b8cf3d77501acbbc18970c61ce22dc89b2a197a1abf4a1f6
+size 149800

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c25e5fc1dc7c96e447792dc68770929eb609167269e8e83d771744f83a16453a
+size 99039232

split_einsum/packages/Stable_Diffusion_version_stabilityai_stable-diffusion-2-1-base_vae_decoder.mlpackage/Manifest.json

@@ -0,0 +1,18 @@
+{
+    "fileFormatVersion": "1.0.0",
+    "itemInfoEntries": {
+        "052A4366-11AD-4F01-B0BD-7F22E071CE76": {
+            "author": "com.apple.CoreML",
+            "description": "CoreML Model Specification",
+            "name": "model.mlmodel",
+            "path": "com.apple.CoreML/model.mlmodel"
+        },
+        "DF4CCEC4-9113-4D6E-8305-D0A78FDBB054": {
+            "author": "com.apple.CoreML",
+            "description": "CoreML Model Weights",
+            "name": "weights",
+            "path": "com.apple.CoreML/weights"
+        }
+    },
+    "rootModelIdentifier": "052A4366-11AD-4F01-B0BD-7F22E071CE76"
+}

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:44d30d1e5b083b72de0fe9b5bd11e5472efe8b70e583312657027d3737f0848b
+size 117464

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

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5a426db5c4683418072eb114c571855373d8161a4a66a17f13666ac31e5fbe49
+size 68370240

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

@@ -0,0 +1,18 @@
+{
+    "fileFormatVersion": "1.0.0",
+    "itemInfoEntries": {
+        "370605FE-15B3-4B65-B377-0CE007FCDD34": {
+            "author": "com.apple.CoreML",
+            "description": "CoreML Model Specification",
+            "name": "model.mlmodel",
+            "path": "com.apple.CoreML/model.mlmodel"
+        },
+        "C6E3D00F-CABF-4AB6-8E2A-98A453A1FB89": {
+            "author": "com.apple.CoreML",
+            "description": "CoreML Model Weights",
+            "name": "weights",
+            "path": "com.apple.CoreML/weights"
+        }
+    },
+    "rootModelIdentifier": "370605FE-15B3-4B65-B377-0CE007FCDD34"
+}