alexwengg's picture
Upload 24 files
c1b4251 verified
raw
history blame
4.39 kB
[
{
"metadataOutputVersion" : "3.0",
"storagePrecision" : "Float32",
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 1 × 242 × 512)",
"shortDescription" : "",
"shape" : "[1, 242, 512]",
"name" : "pre_encoder_embs",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Int32",
"formattedType" : "MultiArray (Int32 1)",
"shortDescription" : "",
"shape" : "[1]",
"name" : "pre_encoder_lengths",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 1 × 14 × 512)",
"shortDescription" : "",
"shape" : "[1, 14, 512]",
"name" : "chunk_embs_in",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Int32",
"formattedType" : "MultiArray (Int32 1)",
"shortDescription" : "",
"shape" : "[1]",
"name" : "chunk_lens_in",
"type" : "MultiArray"
}
],
"modelParameters" : [
],
"specificationVersion" : 7,
"mlProgramOperationTypeHistogram" : {
"Concat" : 1,
"Ios16.cast" : 12,
"Ios16.mul" : 12,
"Ios16.floorDiv" : 3,
"Ios16.less" : 5,
"Ios16.sub" : 5,
"Transpose" : 1,
"Ios16.conv" : 5,
"Ios16.add" : 13,
"Ios16.relu" : 3,
"Tile" : 8,
"Ios16.linear" : 1,
"Ios16.greaterEqual" : 2,
"ExpandDims" : 17,
"Ios16.minimum" : 1,
"Ios16.maximum" : 1,
"Ios16.reshape" : 1,
"Ios16.gatherAlongAxis" : 1
},
"computePrecision" : "Mixed (Float32, Int32)",
"isUpdatable" : "0",
"stateSchema" : [
],
"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.conversion_date" : "2025-12-29",
"com.github.apple.coremltools.source" : "torch==2.9.1",
"com.github.apple.coremltools.version" : "9.0",
"com.github.apple.coremltools.source_dialect" : "TorchScript"
},
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 1 × 112 × 128)",
"shortDescription" : "",
"shape" : "[1, 112, 128]",
"name" : "chunk",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Int32",
"formattedType" : "MultiArray (Int32 1)",
"shortDescription" : "",
"shape" : "[1]",
"name" : "chunk_lengths",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 1 × 188 × 512)",
"shortDescription" : "",
"shape" : "[1, 188, 512]",
"name" : "spkcache",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Int32",
"formattedType" : "MultiArray (Int32 1)",
"shortDescription" : "",
"shape" : "[1]",
"name" : "spkcache_lengths",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 1 × 40 × 512)",
"shortDescription" : "",
"shape" : "[1, 40, 512]",
"name" : "fifo",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Int32",
"formattedType" : "MultiArray (Int32 1)",
"shortDescription" : "",
"shape" : "[1]",
"name" : "fifo_lengths",
"type" : "MultiArray"
}
],
"generatedClassName" : "Pipeline_PreEncoder",
"method" : "predict"
}
]