|
[ |
|
{ |
|
"metadataOutputVersion" : "3.0", |
|
"storagePrecision" : "Float16", |
|
"outputSchema" : [ |
|
{ |
|
"hasShapeFlexibility" : "0", |
|
"isOptional" : "0", |
|
"dataType" : "Float16", |
|
"formattedType" : "MultiArray (Float16)", |
|
"shortDescription" : "", |
|
"shape" : "[]", |
|
"name" : "dummy", |
|
"type" : "MultiArray" |
|
} |
|
], |
|
"modelParameters" : [ |
|
|
|
], |
|
"specificationVersion" : 9, |
|
"mlProgramOperationTypeHistogram" : { |
|
"Ios18.writeState" : 14, |
|
"Shape" : 12, |
|
"Ios18.linear" : 12, |
|
"Identity" : 1, |
|
"Ios18.gather" : 12, |
|
"Ios18.concat" : 12, |
|
"Ios18.sliceUpdate" : 14, |
|
"Ios18.cast" : 24, |
|
"Ios18.expandDims" : 12, |
|
"Ios18.readState" : 14 |
|
}, |
|
"computePrecision" : "Mixed (Float16, Int16, Int32, UInt16)", |
|
"isUpdatable" : "0", |
|
"stateSchema" : [ |
|
{ |
|
"dataType" : "Float16", |
|
"isOptional" : "0", |
|
"formattedType" : "State (Float16 6 × 1 × 448 × 512)", |
|
"shortDescription" : "", |
|
"shape" : "[6, 1, 448, 512]", |
|
"name" : "k_cache1", |
|
"type" : "State" |
|
}, |
|
{ |
|
"dataType" : "Float16", |
|
"isOptional" : "0", |
|
"formattedType" : "State (Float16 6 × 1 × 448 × 512)", |
|
"shortDescription" : "", |
|
"shape" : "[6, 1, 448, 512]", |
|
"name" : "v_cache1", |
|
"type" : "State" |
|
}, |
|
{ |
|
"dataType" : "Float16", |
|
"isOptional" : "0", |
|
"formattedType" : "State (Float16 6 × 1 × 1500 × 512)", |
|
"shortDescription" : "", |
|
"shape" : "[6, 1, 1500, 512]", |
|
"name" : "k_cache2", |
|
"type" : "State" |
|
}, |
|
{ |
|
"dataType" : "Float16", |
|
"isOptional" : "0", |
|
"formattedType" : "State (Float16 6 × 1 × 1500 × 512)", |
|
"shortDescription" : "", |
|
"shape" : "[6, 1, 1500, 512]", |
|
"name" : "v_cache2", |
|
"type" : "State" |
|
} |
|
], |
|
"availability" : { |
|
"macOS" : "15.0", |
|
"tvOS" : "18.0", |
|
"visionOS" : "2.0", |
|
"watchOS" : "11.0", |
|
"iOS" : "18.0", |
|
"macCatalyst" : "18.0" |
|
}, |
|
"modelType" : { |
|
"name" : "MLModelType_mlProgram" |
|
}, |
|
"userDefinedMetadata" : { |
|
"com.github.apple.coremltools.source_dialect" : "TorchScript", |
|
"com.github.apple.coremltools.source" : "torch==2.4.1", |
|
"com.github.apple.coremltools.version" : "8.0" |
|
}, |
|
"inputSchema" : [ |
|
{ |
|
"dataType" : "Float16", |
|
"hasShapeFlexibility" : "1", |
|
"isOptional" : "0", |
|
"shapeFlexibility" : "1 × 1...1500 × 512", |
|
"shapeRange" : "[[1, 1], [1, 1500], [512, 512]]", |
|
"formattedType" : "MultiArray (Float16 1 × 1 × 512)", |
|
"type" : "MultiArray", |
|
"shape" : "[1, 1, 512]", |
|
"name" : "audio_data", |
|
"shortDescription" : "" |
|
} |
|
], |
|
"generatedClassName" : "decoder_first", |
|
"method" : "predict" |
|
} |
|
] |