coreml-stable-diffusion-xl-base-ios_split_einsum_compiled/UnetChunk2.mlmodelc/metadata.json

[
  {
    "metadataOutputVersion" : "3.0",
    "storagePrecision" : "Mixed (Float16, Palettized (1 bits), Palettized (2 bits), Palettized (4 bits), Palettized (6 bits), Palettized (8 bits))",
    "outputSchema" : [
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float16",
        "formattedType" : "MultiArray (Float16 2 × 4 × 96 × 96)",
        "shortDescription" : "",
        "shape" : "[2, 4, 96, 96]",
        "name" : "noise_pred",
        "type" : "MultiArray"
      }
    ],
    "modelParameters" : [

    ],
    "specificationVersion" : 7,
    "mlProgramOperationTypeHistogram" : {
      "Transpose" : 56,
      "UpsampleNearestNeighbor" : 2,
      "Ios16.reduceMean" : 212,
      "Ios16.softmax" : 1000,
      "Split" : 29,
      "Ios16.add" : 297,
      "Concat" : 64,
      "Ios16.realDiv" : 22,
      "Ios16.square" : 22,
      "Ios16.sub" : 106,
      "Ios16.cast" : 12,
      "Ios16.conv" : 327,
      "Ios16.constexprLutToDense" : 319,
      "Ios16.einsum" : 2000,
      "Ios16.gelu" : 29,
      "Ios16.batchNorm" : 22,
      "Ios16.reshape" : 55,
      "Ios16.rsqrt" : 84,
      "Ios16.silu" : 17,
      "Ios16.sqrt" : 22,
      "Ios16.mul" : 1281,
      "SliceByIndex" : 3000
    },
    "computePrecision" : "Mixed (Float16, Float32, Int32)",
    "isUpdatable" : "0",
    "availability" : {
      "macOS" : "13.0",
      "tvOS" : "16.0",
      "visionOS" : "1.0",
      "watchOS" : "9.0",
      "iOS" : "16.0",
      "macCatalyst" : "16.0"
    },
    "modelType" : {
      "name" : "MLModelType_mlProgram"
    },
    "userDefinedMetadata" : {
      "com.github.apple.coremltools.version" : "7.0b2",
      "com.github.apple.coremltools.source" : "milinternal"
    },
    "inputSchema" : [
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float16",
        "formattedType" : "MultiArray (Float16 2 × 2048 × 1 × 77)",
        "shortDescription" : "",
        "shape" : "[2, 2048, 1, 77]",
        "name" : "encoder_hidden_states",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2 × 640 × 48 × 48)",
        "shortDescription" : "",
        "shape" : "[2, 640, 48, 48]",
        "name" : "input_79_cast",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2 × 640 × 48 × 48)",
        "shortDescription" : "",
        "shape" : "[2, 640, 48, 48]",
        "name" : "input_113_cast",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2 × 1280 × 24 × 24)",
        "shortDescription" : "",
        "shape" : "[2, 1280, 24, 24]",
        "name" : "hidden_states_283_cast",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2 × 320 × 96 × 96)",
        "shortDescription" : "",
        "shape" : "[2, 320, 96, 96]",
        "name" : "input_43_cast",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2 × 1280 × 24 × 24)",
        "shortDescription" : "",
        "shape" : "[2, 1280, 24, 24]",
        "name" : "input_213_cast",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2 × 320 × 96 × 96)",
        "shortDescription" : "",
        "shape" : "[2, 320, 96, 96]",
        "name" : "input_13_cast",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2 × 640 × 24 × 24)",
        "shortDescription" : "",
        "shape" : "[2, 640, 24, 24]",
        "name" : "input_115_cast",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2 × 10240 × 1 × 576)",
        "shortDescription" : "",
        "shape" : "[2, 10240, 1, 576]",
        "name" : "var_37735_cast",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2 × 320 × 96 × 96)",
        "shortDescription" : "",
        "shape" : "[2, 320, 96, 96]",
        "name" : "input_29_cast",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2 × 320 × 48 × 48)",
        "shortDescription" : "",
        "shape" : "[2, 320, 48, 48]",
        "name" : "input_45_cast",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2 × 1280 × 1 × 1)",
        "shortDescription" : "",
        "shape" : "[2, 1280, 1, 1]",
        "name" : "input_21_cast",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2 × 1280 × 1 × 576)",
        "shortDescription" : "",
        "shape" : "[2, 1280, 1, 576]",
        "name" : "inputs_251_cast",
        "type" : "MultiArray"
      }
    ],
    "generatedClassName" : "Stable_Diffusion_version_stabilityai_stable_diffusion_xl_base_1_0_unet_chunk2",
    "method" : "predict"
  }
]