| [ | |
| { | |
| "metadataOutputVersion" : "3.0", | |
| "storagePrecision" : "Float32", | |
| "outputSchema" : [ | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)", | |
| "shortDescription" : "", | |
| "shape" : "[2, 1, 512, 16, 64]", | |
| "name" : "new_cache_1_internal_tensor_assign_2", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 1)", | |
| "shortDescription" : "", | |
| "shape" : "[1]", | |
| "name" : "var_445", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)", | |
| "shortDescription" : "", | |
| "shape" : "[2, 1, 512, 16, 64]", | |
| "name" : "new_cache_3_internal_tensor_assign_2", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 1)", | |
| "shortDescription" : "", | |
| "shape" : "[1]", | |
| "name" : "var_864", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)", | |
| "shortDescription" : "", | |
| "shape" : "[2, 1, 512, 16, 64]", | |
| "name" : "new_cache_5_internal_tensor_assign_2", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 1)", | |
| "shortDescription" : "", | |
| "shape" : "[1]", | |
| "name" : "var_1283", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)", | |
| "shortDescription" : "", | |
| "shape" : "[2, 1, 512, 16, 64]", | |
| "name" : "new_cache_7_internal_tensor_assign_2", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 1)", | |
| "shortDescription" : "", | |
| "shape" : "[1]", | |
| "name" : "var_1702", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)", | |
| "shortDescription" : "", | |
| "shape" : "[2, 1, 512, 16, 64]", | |
| "name" : "new_cache_9_internal_tensor_assign_2", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 1)", | |
| "shortDescription" : "", | |
| "shape" : "[1]", | |
| "name" : "var_2121", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)", | |
| "shortDescription" : "", | |
| "shape" : "[2, 1, 512, 16, 64]", | |
| "name" : "new_cache_internal_tensor_assign_2", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 1)", | |
| "shortDescription" : "", | |
| "shape" : "[1]", | |
| "name" : "var_2365", | |
| "type" : "MultiArray" | |
| } | |
| ], | |
| "modelParameters" : [ | |
| ], | |
| "specificationVersion" : 8, | |
| "mlProgramOperationTypeHistogram" : { | |
| "Ios17.cos" : 6, | |
| "Range1d" : 12, | |
| "Ios17.notEqual" : 10, | |
| "Ios17.floor" : 6, | |
| "Ios17.reshape" : 116, | |
| "Ios17.logicalAnd" : 5, | |
| "Ios16.softmax" : 5, | |
| "Ios17.scatter" : 12, | |
| "Ios17.matmul" : 10, | |
| "Ios17.sin" : 6, | |
| "Ios17.transpose" : 20, | |
| "Select" : 15, | |
| "Ios17.expandDims" : 5, | |
| "Ios17.add" : 49, | |
| "Tile" : 6, | |
| "Ios17.lessEqual" : 5, | |
| "Ios17.sliceByIndex" : 73, | |
| "Ios17.scatterAlongAxis" : 12, | |
| "Ios17.logicalNot" : 5, | |
| "Ios17.layerNorm" : 11, | |
| "Ios17.less" : 5, | |
| "Ios17.sub" : 17, | |
| "Ios17.cast" : 6, | |
| "Ios16.gelu" : 5, | |
| "Ios17.linear" : 21, | |
| "Stack" : 11, | |
| "Ios17.mul" : 67 | |
| }, | |
| "computePrecision" : "Mixed (Float32, Int32)", | |
| "isUpdatable" : "0", | |
| "stateSchema" : [ | |
| ], | |
| "availability" : { | |
| "macOS" : "14.0", | |
| "tvOS" : "17.0", | |
| "visionOS" : "1.0", | |
| "watchOS" : "10.0", | |
| "iOS" : "17.0", | |
| "macCatalyst" : "17.0" | |
| }, | |
| "modelType" : { | |
| "name" : "MLModelType_mlProgram" | |
| }, | |
| "userDefinedMetadata" : { | |
| "com.github.apple.coremltools.conversion_date" : "2026-02-06", | |
| "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 × 175 × 1024)", | |
| "shortDescription" : "", | |
| "shape" : "[1, 175, 1024]", | |
| "name" : "conditioning", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)", | |
| "shortDescription" : "", | |
| "shape" : "[2, 1, 512, 16, 64]", | |
| "name" : "cache0", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 1)", | |
| "shortDescription" : "", | |
| "shape" : "[1]", | |
| "name" : "position0", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)", | |
| "shortDescription" : "", | |
| "shape" : "[2, 1, 512, 16, 64]", | |
| "name" : "cache1", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 1)", | |
| "shortDescription" : "", | |
| "shape" : "[1]", | |
| "name" : "position1", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)", | |
| "shortDescription" : "", | |
| "shape" : "[2, 1, 512, 16, 64]", | |
| "name" : "cache2", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 1)", | |
| "shortDescription" : "", | |
| "shape" : "[1]", | |
| "name" : "position2", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)", | |
| "shortDescription" : "", | |
| "shape" : "[2, 1, 512, 16, 64]", | |
| "name" : "cache3", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 1)", | |
| "shortDescription" : "", | |
| "shape" : "[1]", | |
| "name" : "position3", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)", | |
| "shortDescription" : "", | |
| "shape" : "[2, 1, 512, 16, 64]", | |
| "name" : "cache4", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 1)", | |
| "shortDescription" : "", | |
| "shape" : "[1]", | |
| "name" : "position4", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 2 × 1 × 512 × 16 × 64)", | |
| "shortDescription" : "", | |
| "shape" : "[2, 1, 512, 16, 64]", | |
| "name" : "cache5", | |
| "type" : "MultiArray" | |
| }, | |
| { | |
| "hasShapeFlexibility" : "0", | |
| "isOptional" : "0", | |
| "dataType" : "Float32", | |
| "formattedType" : "MultiArray (Float32 1)", | |
| "shortDescription" : "", | |
| "shape" : "[1]", | |
| "name" : "position5", | |
| "type" : "MultiArray" | |
| } | |
| ], | |
| "generatedClassName" : "cond_step_175", | |
| "method" : "predict" | |
| } | |
| ] |