| [ |
| { |
| "metadataOutputVersion" : "3.0", |
| "storagePrecision" : "Float16", |
| "outputSchema" : [ |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float16", |
| "formattedType" : "MultiArray (Float16 1 × 1 × 51865)", |
| "shortDescription" : "", |
| "shape" : "[1, 1, 51865]", |
| "name" : "logits", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float16", |
| "formattedType" : "MultiArray (Float16 1 × 40960 × 1 × 1)", |
| "shortDescription" : "", |
| "shape" : "[1, 40960, 1, 1]", |
| "name" : "key_cache_updates", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float16", |
| "formattedType" : "MultiArray (Float16 1 × 40960 × 1 × 1)", |
| "shortDescription" : "", |
| "shape" : "[1, 40960, 1, 1]", |
| "name" : "value_cache_updates", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float16", |
| "formattedType" : "MultiArray (Float16 1 × 1500)", |
| "shortDescription" : "", |
| "shape" : "[1, 1500]", |
| "name" : "alignment_heads_weights", |
| "type" : "MultiArray" |
| } |
| ], |
| "modelParameters" : [ |
|
|
| ], |
| "specificationVersion" : 7, |
| "mlProgramOperationTypeHistogram" : { |
| "Ios16.linear" : 1, |
| "Concat" : 3, |
| "Ios16.reduceMean" : 1, |
| "Ios16.mul" : 192, |
| "Ios16.layerNorm" : 97, |
| "SliceByIndex" : 46, |
| "Ios16.sub" : 1, |
| "Transpose" : 1, |
| "Ios16.conv" : 320, |
| "Ios16.add" : 193, |
| "Squeeze" : 1, |
| "Ios16.matmul" : 128, |
| "Ios16.softmax" : 64, |
| "Ios16.gelu" : 32, |
| "ExpandDims" : 6, |
| "Ios16.batchNorm" : 97, |
| "Split" : 2, |
| "Ios16.gather" : 2, |
| "Ios16.reshape" : 256 |
| }, |
| "computePrecision" : "Mixed (Float16, 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.version" : "8.3.0", |
| "com.github.apple.coremltools.source" : "torch==2.5.0", |
| "com.github.apple.coremltools.source_dialect" : "TorchScript" |
| }, |
| "inputSchema" : [ |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Int32", |
| "formattedType" : "MultiArray (Int32 1)", |
| "shortDescription" : "", |
| "shape" : "[1]", |
| "name" : "input_ids", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Int32", |
| "formattedType" : "MultiArray (Int32 1)", |
| "shortDescription" : "", |
| "shape" : "[1]", |
| "name" : "cache_length", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float16", |
| "formattedType" : "MultiArray (Float16 1 × 40960 × 1 × 448)", |
| "shortDescription" : "", |
| "shape" : "[1, 40960, 1, 448]", |
| "name" : "key_cache", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float16", |
| "formattedType" : "MultiArray (Float16 1 × 40960 × 1 × 448)", |
| "shortDescription" : "", |
| "shape" : "[1, 40960, 1, 448]", |
| "name" : "value_cache", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float16", |
| "formattedType" : "MultiArray (Float16 1 × 448)", |
| "shortDescription" : "", |
| "shape" : "[1, 448]", |
| "name" : "kv_cache_update_mask", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float16", |
| "formattedType" : "MultiArray (Float16 1 × 1280 × 1 × 1500)", |
| "shortDescription" : "", |
| "shape" : "[1, 1280, 1, 1500]", |
| "name" : "encoder_output_embeds", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float16", |
| "formattedType" : "MultiArray (Float16 1 × 448)", |
| "shortDescription" : "", |
| "shape" : "[1, 448]", |
| "name" : "decoder_key_padding_mask", |
| "type" : "MultiArray" |
| } |
| ], |
| "generatedClassName" : "TextDecoder", |
| "method" : "predict" |
| } |
| ] |
