| [ |
| { |
| "metadataOutputVersion" : "3.0", |
| "storagePrecision" : "Mixed (Float16, Palettized (8 bits), UInt8)", |
| "outputSchema" : [ |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float16", |
| "formattedType" : "MultiArray (Float16 1 × 1 × 151936)", |
| "shortDescription" : "", |
| "shape" : "[1, 1, 151936]", |
| "name" : "logits", |
| "type" : "MultiArray" |
| } |
| ], |
| "modelParameters" : [ |
|
|
| ], |
| "specificationVersion" : 9, |
| "mlProgramOperationTypeHistogram" : { |
| "Ios18.expandDims" : 29, |
| "Ios18.mul" : 283, |
| "Ios18.cos" : 1, |
| "Ios18.softmax" : 14, |
| "Ios18.matmul" : 56, |
| "Ios18.rsqrt" : 57, |
| "Ios16.reduceMean" : 57, |
| "Ios18.sin" : 1, |
| "Ios18.readState" : 84, |
| "Tile" : 28, |
| "Ios18.add" : 155, |
| "Ios16.reduceSum" : 1, |
| "Ios18.writeState" : 56, |
| "Ios18.reshape" : 143, |
| "Ios18.constexprLutToDense" : 99, |
| "Ios18.linear" : 99, |
| "Ios18.concat" : 28, |
| "Ios18.transpose" : 98, |
| "OneHot" : 1, |
| "Ios18.sub" : 29, |
| "Ios18.pow" : 57, |
| "Ios18.cast" : 4, |
| "Ios18.silu" : 14, |
| "Ios18.sliceByIndex" : 57 |
| }, |
| "computePrecision" : "Mixed (Float16, Float32, Int32)", |
| "isUpdatable" : "0", |
| "stateSchema" : [ |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "k_cache_0", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "v_cache_0", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "k_cache_1", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "v_cache_1", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "k_cache_2", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "v_cache_2", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "k_cache_3", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "v_cache_3", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "k_cache_4", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "v_cache_4", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "k_cache_5", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "v_cache_5", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "k_cache_6", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "v_cache_6", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "k_cache_7", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "v_cache_7", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "k_cache_8", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "v_cache_8", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "k_cache_9", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "v_cache_9", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "k_cache_10", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "v_cache_10", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "k_cache_11", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "v_cache_11", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "k_cache_12", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "v_cache_12", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "k_cache_13", |
| "type" : "State" |
| }, |
| { |
| "dataType" : "Float16", |
| "isOptional" : "0", |
| "formattedType" : "State (Float16 1 × 8 × 1024 × 128)", |
| "shortDescription" : "", |
| "shape" : "[1, 8, 1024, 128]", |
| "name" : "v_cache_13", |
| "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.version" : "8.3.0", |
| "com.github.apple.coremltools.source" : "torch==2.9.1", |
| "com.github.apple.coremltools.source_dialect" : "TorchScript" |
| }, |
| "inputSchema" : [ |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float32", |
| "formattedType" : "MultiArray (Float32 1 × 128 × 1024)", |
| "shortDescription" : "", |
| "shape" : "[1, 128, 1024]", |
| "name" : "input_embeds", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Int32", |
| "formattedType" : "MultiArray (Int32 128)", |
| "shortDescription" : "", |
| "shape" : "[128]", |
| "name" : "positions", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float32", |
| "formattedType" : "MultiArray (Float32 1 × 1 × 128 × 1024)", |
| "shortDescription" : "", |
| "shape" : "[1, 1, 128, 1024]", |
| "name" : "attention_mask", |
| "type" : "MultiArray" |
| } |
| ], |
| "generatedClassName" : "decoder_part2", |
| "method" : "predict" |
| } |
| ] |
