| [ |
| { |
| "metadataOutputVersion" : "3.0", |
| "shortDescription" : "StyleTTS2 KDiffusion single denoise step (StyleTransformer1d) @ T_tok=512.", |
| "outputSchema" : [ |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float16", |
| "formattedType" : "MultiArray (Float16 1 × 1 × 256)", |
| "shortDescription" : "", |
| "shape" : "[1, 1, 256]", |
| "name" : "denoised", |
| "type" : "MultiArray" |
| } |
| ], |
| "storagePrecision" : "Float16", |
| "modelParameters" : [ |
|
|
| ], |
| "specificationVersion" : 8, |
| "mlProgramOperationTypeHistogram" : { |
| "Ios17.cos" : 1, |
| "Ios17.reshape" : 20, |
| "Ios17.inverse" : 2, |
| "Ios16.softmax" : 3, |
| "Ios17.matmul" : 6, |
| "Ios17.sin" : 1, |
| "Ios17.transpose" : 27, |
| "Split" : 9, |
| "Ios16.reduceMean" : 1, |
| "Ios17.expandDims" : 2, |
| "Ios17.add" : 23, |
| "Tile" : 2, |
| "Ios16.reduceSum" : 1, |
| "Ios17.layerNorm" : 3, |
| "Ios17.log" : 1, |
| "Ios17.sqrt" : 1, |
| "Ios16.gelu" : 7, |
| "Ios17.cast" : 4, |
| "Ios17.conv" : 1, |
| "Ios17.realDiv" : 1, |
| "Ios17.linear" : 25, |
| "Stack" : 1, |
| "Ios17.concat" : 2, |
| "Ios17.mul" : 19 |
| }, |
| "computePrecision" : "Mixed (Float16, 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" |
| }, |
| "inputSchema" : [ |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float32", |
| "formattedType" : "MultiArray (Float32 1 × 1 × 256)", |
| "shortDescription" : "", |
| "shape" : "[1, 1, 256]", |
| "name" : "x_noisy", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float32", |
| "formattedType" : "MultiArray (Float32 1)", |
| "shortDescription" : "", |
| "shape" : "[1]", |
| "name" : "sigma", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float32", |
| "formattedType" : "MultiArray (Float32 1 × 512 × 768)", |
| "shortDescription" : "", |
| "shape" : "[1, 512, 768]", |
| "name" : "embedding", |
| "type" : "MultiArray" |
| }, |
| { |
| "hasShapeFlexibility" : "0", |
| "isOptional" : "0", |
| "dataType" : "Float32", |
| "formattedType" : "MultiArray (Float32 1 × 256)", |
| "shortDescription" : "", |
| "shape" : "[1, 256]", |
| "name" : "features", |
| "type" : "MultiArray" |
| } |
| ], |
| "userDefinedMetadata" : { |
| "com.github.apple.coremltools.version" : "8.3.0", |
| "com.github.apple.coremltools.source" : "torch==2.11.0", |
| "com.github.apple.coremltools.source_dialect" : "TorchScript" |
| }, |
| "generatedClassName" : "styletts2_diffusion_step_512", |
| "method" : "predict" |
| } |
| ] |