neonforestmist's picture
Publish shared-weight multifunction Core ML model
d5a927b verified
Raw
History Blame Contribute Delete
24.9 kB
[
{
"metadataOutputVersion" : "3.0",
"storagePrecision" : "Float16",
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 4 × 64 × 64)",
"shortDescription" : "",
"shape" : "[2, 4, 64, 64]",
"name" : "noise_pred",
"type" : "MultiArray"
}
],
"modelParameters" : [
],
"specificationVersion" : 9,
"functions" : [
{
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 2 × 768 × 1 × 77)",
"shortDescription" : "",
"shape" : "[2, 768, 1, 77]",
"name" : "encoder_hidden_states",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 1280 × 1 × 256)",
"shortDescription" : "",
"shape" : "[2, 1280, 1, 256]",
"name" : "inputs_23_cast_fp16",
"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_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 10240 × 1 × 256)",
"shortDescription" : "",
"shape" : "[2, 10240, 1, 256]",
"name" : "var_2337_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 320 × 32 × 32)",
"shortDescription" : "",
"shape" : "[2, 320, 32, 32]",
"name" : "input_37_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 640 × 32 × 32)",
"shortDescription" : "",
"shape" : "[2, 640, 32, 32]",
"name" : "input_63_cast_fp16",
"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_fp16",
"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_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 640 × 16 × 16)",
"shortDescription" : "",
"shape" : "[2, 640, 16, 16]",
"name" : "input_65_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 1280 × 16 × 16)",
"shortDescription" : "",
"shape" : "[2, 1280, 16, 16]",
"name" : "hidden_states_61_cast_fp16",
"type" : "MultiArray"
}
],
"computePrecision" : "Mixed (Float16, Float32, Int32)",
"storagePrecision" : "Float16",
"stateSchema" : [
],
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 4 × 64 × 64)",
"shortDescription" : "",
"shape" : "[2, 4, 64, 64]",
"name" : "noise_pred",
"type" : "MultiArray"
}
],
"name" : "base",
"mlProgramOperationTypeHistogram" : {
"Ios18.softmax" : 80,
"Ios18.mul" : 86,
"UpsampleNearestNeighbor" : 2,
"Ios18.batchNorm" : 16,
"Ios16.reduceMean" : 32,
"Ios18.realDiv" : 16,
"Split" : 6,
"Ios18.add" : 48,
"Ios18.layerNorm" : 15,
"Ios18.reshape" : 43,
"Ios18.conv" : 85,
"Ios18.gelu" : 6,
"Ios18.concat" : 15,
"Ios18.sub" : 16,
"Ios16.einsum" : 160,
"Ios18.silu" : 11,
"Ios18.cast" : 10,
"Ios18.transpose" : 10,
"Ios18.square" : 16,
"Ios18.sqrt" : 16,
"Ios18.sliceByIndex" : 240
}
},
{
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 2 × 768 × 1 × 77)",
"shortDescription" : "",
"shape" : "[2, 768, 1, 77]",
"name" : "encoder_hidden_states",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 320 × 32 × 32)",
"shortDescription" : "",
"shape" : "[2, 320, 32, 32]",
"name" : "input_53_cast_fp16",
"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_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 10240 × 1 × 256)",
"shortDescription" : "",
"shape" : "[2, 10240, 1, 256]",
"name" : "var_2913_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 320 × 64 × 64)",
"shortDescription" : "",
"shape" : "[2, 320, 64, 64]",
"name" : "input_51_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 640 × 32 × 32)",
"shortDescription" : "",
"shape" : "[2, 640, 32, 32]",
"name" : "input_95_cast_fp16",
"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_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 1280 × 1 × 256)",
"shortDescription" : "",
"shape" : "[2, 1280, 1, 256]",
"name" : "inputs_23_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 640 × 16 × 16)",
"shortDescription" : "",
"shape" : "[2, 640, 16, 16]",
"name" : "input_97_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 1280 × 16 × 16)",
"shortDescription" : "",
"shape" : "[2, 1280, 16, 16]",
"name" : "hidden_states_61_cast_fp16",
"type" : "MultiArray"
}
],
"computePrecision" : "Mixed (Float16, Float32, Int32)",
"storagePrecision" : "Float16",
"stateSchema" : [
],
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 4 × 64 × 64)",
"shortDescription" : "",
"shape" : "[2, 4, 64, 64]",
"name" : "noise_pred",
"type" : "MultiArray"
}
],
"name" : "monet",
"mlProgramOperationTypeHistogram" : {
"Ios18.softmax" : 80,
"Ios18.mul" : 86,
"UpsampleNearestNeighbor" : 2,
"Ios18.batchNorm" : 16,
"Ios16.reduceMean" : 32,
"Ios18.realDiv" : 16,
"Split" : 6,
"Ios18.add" : 88,
"Ios18.layerNorm" : 15,
"Ios18.reshape" : 43,
"Ios18.conv" : 165,
"Ios18.gelu" : 6,
"Ios18.concat" : 15,
"Ios18.sub" : 16,
"Ios16.einsum" : 160,
"Ios18.silu" : 11,
"Ios18.cast" : 10,
"Ios18.transpose" : 10,
"Ios18.square" : 16,
"Ios18.sqrt" : 16,
"Ios18.sliceByIndex" : 240
}
},
{
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 2 × 768 × 1 × 77)",
"shortDescription" : "",
"shape" : "[2, 768, 1, 77]",
"name" : "encoder_hidden_states",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 1280 × 1 × 256)",
"shortDescription" : "",
"shape" : "[2, 1280, 1, 256]",
"name" : "inputs_23_cast_fp16",
"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_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 10240 × 1 × 256)",
"shortDescription" : "",
"shape" : "[2, 10240, 1, 256]",
"name" : "var_2913_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 640 × 32 × 32)",
"shortDescription" : "",
"shape" : "[2, 640, 32, 32]",
"name" : "input_95_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 640 × 16 × 16)",
"shortDescription" : "",
"shape" : "[2, 640, 16, 16]",
"name" : "input_97_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 320 × 64 × 64)",
"shortDescription" : "",
"shape" : "[2, 320, 64, 64]",
"name" : "input_51_cast_fp16",
"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_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 320 × 32 × 32)",
"shortDescription" : "",
"shape" : "[2, 320, 32, 32]",
"name" : "input_53_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 1280 × 16 × 16)",
"shortDescription" : "",
"shape" : "[2, 1280, 16, 16]",
"name" : "hidden_states_61_cast_fp16",
"type" : "MultiArray"
}
],
"computePrecision" : "Mixed (Float16, Float32, Int32)",
"storagePrecision" : "Float16",
"stateSchema" : [
],
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 4 × 64 × 64)",
"shortDescription" : "",
"shape" : "[2, 4, 64, 64]",
"name" : "noise_pred",
"type" : "MultiArray"
}
],
"name" : "pointillism",
"mlProgramOperationTypeHistogram" : {
"Ios18.softmax" : 80,
"Ios18.mul" : 86,
"UpsampleNearestNeighbor" : 2,
"Ios18.batchNorm" : 16,
"Ios16.reduceMean" : 32,
"Ios18.realDiv" : 16,
"Split" : 6,
"Ios18.add" : 88,
"Ios18.layerNorm" : 15,
"Ios18.reshape" : 43,
"Ios18.conv" : 165,
"Ios18.gelu" : 6,
"Ios18.concat" : 15,
"Ios18.sub" : 16,
"Ios16.einsum" : 160,
"Ios18.silu" : 11,
"Ios18.cast" : 10,
"Ios18.transpose" : 10,
"Ios18.square" : 16,
"Ios18.sqrt" : 16,
"Ios18.sliceByIndex" : 240
}
},
{
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 2 × 768 × 1 × 77)",
"shortDescription" : "",
"shape" : "[2, 768, 1, 77]",
"name" : "encoder_hidden_states",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 10240 × 1 × 256)",
"shortDescription" : "",
"shape" : "[2, 10240, 1, 256]",
"name" : "var_2913_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 640 × 32 × 32)",
"shortDescription" : "",
"shape" : "[2, 640, 32, 32]",
"name" : "input_95_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 320 × 64 × 64)",
"shortDescription" : "",
"shape" : "[2, 320, 64, 64]",
"name" : "input_51_cast_fp16",
"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_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 320 × 32 × 32)",
"shortDescription" : "",
"shape" : "[2, 320, 32, 32]",
"name" : "input_53_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 640 × 16 × 16)",
"shortDescription" : "",
"shape" : "[2, 640, 16, 16]",
"name" : "input_97_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 1280 × 16 × 16)",
"shortDescription" : "",
"shape" : "[2, 1280, 16, 16]",
"name" : "hidden_states_61_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 1280 × 1 × 256)",
"shortDescription" : "",
"shape" : "[2, 1280, 1, 256]",
"name" : "inputs_23_cast_fp16",
"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_fp16",
"type" : "MultiArray"
}
],
"computePrecision" : "Mixed (Float16, Float32, Int32)",
"storagePrecision" : "Float16",
"stateSchema" : [
],
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 4 × 64 × 64)",
"shortDescription" : "",
"shape" : "[2, 4, 64, 64]",
"name" : "noise_pred",
"type" : "MultiArray"
}
],
"name" : "watercolor_anime",
"mlProgramOperationTypeHistogram" : {
"Ios18.softmax" : 80,
"Ios18.mul" : 86,
"UpsampleNearestNeighbor" : 2,
"Ios18.batchNorm" : 16,
"Ios16.reduceMean" : 32,
"Ios18.realDiv" : 16,
"Split" : 6,
"Ios18.add" : 88,
"Ios18.layerNorm" : 15,
"Ios18.reshape" : 43,
"Ios18.conv" : 165,
"Ios18.gelu" : 6,
"Ios18.concat" : 15,
"Ios18.sub" : 16,
"Ios16.einsum" : 160,
"Ios18.silu" : 11,
"Ios18.cast" : 10,
"Ios18.transpose" : 10,
"Ios18.square" : 16,
"Ios18.sqrt" : 16,
"Ios18.sliceByIndex" : 240
}
}
],
"mlProgramOperationTypeHistogram" : {
"Ios18.softmax" : 80,
"Ios18.mul" : 86,
"UpsampleNearestNeighbor" : 2,
"Ios18.batchNorm" : 16,
"Ios16.reduceMean" : 32,
"Ios18.realDiv" : 16,
"Split" : 6,
"Ios18.add" : 48,
"Ios18.layerNorm" : 15,
"Ios18.reshape" : 43,
"Ios18.conv" : 85,
"Ios18.gelu" : 6,
"Ios18.concat" : 15,
"Ios18.sub" : 16,
"Ios16.einsum" : 160,
"Ios18.silu" : 11,
"Ios18.cast" : 10,
"Ios18.transpose" : 10,
"Ios18.square" : 16,
"Ios18.sqrt" : 16,
"Ios18.sliceByIndex" : 240
},
"isUpdatable" : "0",
"stateSchema" : [
],
"availability" : {
"macOS" : "15.0",
"tvOS" : "18.0",
"visionOS" : "2.0",
"watchOS" : "11.0",
"iOS" : "18.0",
"macCatalyst" : "18.0"
},
"computePrecision" : "Mixed (Float16, Float32, Int32)",
"modelType" : {
"name" : "MLModelType_mlProgram"
},
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 2 × 768 × 1 × 77)",
"shortDescription" : "",
"shape" : "[2, 768, 1, 77]",
"name" : "encoder_hidden_states",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 1280 × 1 × 256)",
"shortDescription" : "",
"shape" : "[2, 1280, 1, 256]",
"name" : "inputs_23_cast_fp16",
"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_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 10240 × 1 × 256)",
"shortDescription" : "",
"shape" : "[2, 10240, 1, 256]",
"name" : "var_2337_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 320 × 32 × 32)",
"shortDescription" : "",
"shape" : "[2, 320, 32, 32]",
"name" : "input_37_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 640 × 32 × 32)",
"shortDescription" : "",
"shape" : "[2, 640, 32, 32]",
"name" : "input_63_cast_fp16",
"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_fp16",
"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_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 640 × 16 × 16)",
"shortDescription" : "",
"shape" : "[2, 640, 16, 16]",
"name" : "input_65_cast_fp16",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 2 × 1280 × 16 × 16)",
"shortDescription" : "",
"shape" : "[2, 1280, 16, 16]",
"name" : "hidden_states_61_cast_fp16",
"type" : "MultiArray"
}
],
"defaultFunctionName" : "base",
"generatedClassName" : "UnetChunk2",
"userDefinedMetadata" : {
},
"method" : "predict"
}
]