|
[ |
|
{ |
|
"metadataOutputVersion" : "3.0", |
|
"storagePrecision" : "Mixed (Float16, Int4)", |
|
"outputSchema" : [ |
|
{ |
|
"hasShapeFlexibility" : "0", |
|
"isOptional" : "0", |
|
"dataType" : "Float16", |
|
"formattedType" : "MultiArray (Float16)", |
|
"shortDescription" : "", |
|
"shape" : "[]", |
|
"name" : "logits", |
|
"type" : "MultiArray" |
|
} |
|
], |
|
"modelParameters" : [ |
|
|
|
], |
|
"specificationVersion" : 9, |
|
"mlProgramOperationTypeHistogram" : { |
|
"Ios18.expandDims" : 6, |
|
"Range1d" : 1, |
|
"Ios18.cos" : 1, |
|
"Ios18.mul" : 266, |
|
"Ios18.matmul" : 1, |
|
"Ios18.rsqrt" : 49, |
|
"Ios16.reduceMean" : 49, |
|
"Ios18.scaledDotProductAttention" : 24, |
|
"Ios18.sin" : 1, |
|
"Ios18.readState" : 50, |
|
"Ios18.gather" : 53, |
|
"Ios18.add" : 147, |
|
"Ios18.writeState" : 48, |
|
"Ios18.sliceUpdate" : 48, |
|
"Ios18.reshape" : 96, |
|
"Shape" : 52, |
|
"Ios18.linear" : 169, |
|
"Ios18.concat" : 123, |
|
"Ios18.transpose" : 97, |
|
"Ios18.sub" : 1, |
|
"Ios18.pow" : 49, |
|
"Ios18.cast" : 105, |
|
"Ios18.silu" : 24, |
|
"Ios18.sliceByIndex" : 216, |
|
"Ios18.constexprBlockwiseShiftScale" : 169 |
|
}, |
|
"computePrecision" : "Mixed (Float16, Float32, Int16, Int32, UInt16)", |
|
"isUpdatable" : "0", |
|
"stateSchema" : [ |
|
{ |
|
"dataType" : "Float16", |
|
"isOptional" : "0", |
|
"formattedType" : "State (Float16 24 × 1 × 32 × 2048 × 64)", |
|
"shortDescription" : "", |
|
"shape" : "[24, 1, 32, 2048, 64]", |
|
"name" : "key_cache", |
|
"type" : "State" |
|
}, |
|
{ |
|
"dataType" : "Float16", |
|
"isOptional" : "0", |
|
"formattedType" : "State (Float16 24 × 1 × 32 × 2048 × 64)", |
|
"shortDescription" : "", |
|
"shape" : "[24, 1, 32, 2048, 64]", |
|
"name" : "value_cache", |
|
"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.5.1", |
|
"com.github.apple.coremltools.version" : "8.1" |
|
}, |
|
"inputSchema" : [ |
|
{ |
|
"dataType" : "Int32", |
|
"hasShapeFlexibility" : "1", |
|
"isOptional" : "0", |
|
"shapeFlexibility" : "1 × 1...2048", |
|
"shapeRange" : "[[1, 1], [1, 2048]]", |
|
"formattedType" : "MultiArray (Int32 1 × 1)", |
|
"type" : "MultiArray", |
|
"shape" : "[1, 1]", |
|
"name" : "input_ids", |
|
"shortDescription" : "" |
|
}, |
|
{ |
|
"dataType" : "Float16", |
|
"hasShapeFlexibility" : "1", |
|
"isOptional" : "0", |
|
"shapeFlexibility" : "1 × 1 × 1...2048 × 1...2048", |
|
"shapeRange" : "[[1, 1], [1, 1], [1, 2048], [1, 2048]]", |
|
"formattedType" : "MultiArray (Float16 1 × 1 × 1 × 1)", |
|
"type" : "MultiArray", |
|
"shape" : "[1, 1, 1, 1]", |
|
"name" : "causal_mask", |
|
"shortDescription" : "" |
|
} |
|
], |
|
"generatedClassName" : "SmolLM2_1_7B_Instruct_4bit", |
|
"method" : "predict" |
|
} |
|
] |