FluidInference/speaker-diarization-coreml

1[2  {3    "shortDescription" : "pyannote community-1 speaker embedding backend (WeSpeaker ResNet34 consuming 80×998 FBANK features + 589-frame weights, interpolates weights to 125-frame pooling layer internally)",4    "metadataOutputVersion" : "3.0",5    "outputSchema" : [6      {7        "hasShapeFlexibility" : "0",8        "isOptional" : "0",9        "dataType" : "Float32",10        "formattedType" : "MultiArray (Float32)",11        "shortDescription" : "",12        "shape" : "[]",13        "name" : "embedding",14        "type" : "MultiArray"15      }16    ],17    "version" : "pyannote-speaker-diarization-community-1",18    "modelParameters" : [19 20    ],21    "author" : "Fluid Inference",22    "specificationVersion" : 8,23    "storagePrecision" : "Float16",24    "license" : "CC-BY-4.0",25    "mlProgramOperationTypeHistogram" : {26      "Ios17.mul" : 4,27      "Ios17.sqrt" : 1,28      "Ios17.sub" : 2,29      "Ios17.conv" : 38,30      "Ios17.concat" : 1,31      "Ios17.clip" : 2,32      "Ios17.add" : 18,33      "Ios17.realDiv" : 4,34      "Ios16.relu" : 33,35      "Ios17.expandDims" : 7,36      "Ios16.reduceSum" : 4,37      "Ios16.reduceL2Norm" : 1,38      "Ios17.squeeze" : 2,39      "Ios17.reshape" : 2,40      "Ios17.cast" : 541    },42    "computePrecision" : "Mixed (Float16, Float32, Int32)",43    "stateSchema" : [44 45    ],46    "isUpdatable" : "0",47    "availability" : {48      "macOS" : "14.0",49      "tvOS" : "17.0",50      "visionOS" : "1.0",51      "watchOS" : "10.0",52      "iOS" : "17.0",53      "macCatalyst" : "17.0"54    },55    "modelType" : {56      "name" : "MLModelType_mlProgram"57    },58    "inputSchema" : [59      {60        "dataType" : "Float32",61        "hasShapeFlexibility" : "1",62        "isOptional" : "0",63        "shapeFlexibility" : "1...32 × 1 × 80 × 998",64        "shapeRange" : "[[1, 32], [1, 1], [80, 80], [998, 998]]",65        "formattedType" : "MultiArray (Float32 1 × 1 × 80 × 998)",66        "type" : "MultiArray",67        "shape" : "[1, 1, 80, 998]",68        "name" : "fbank_features",69        "shortDescription" : ""70      },71      {72        "dataType" : "Float32",73        "hasShapeFlexibility" : "1",74        "isOptional" : "0",75        "shapeFlexibility" : "1...32 × 589",76        "shapeRange" : "[[1, 32], [589, 589]]",77        "formattedType" : "MultiArray (Float32 1 × 589)",78        "type" : "MultiArray",79        "shape" : "[1, 589]",80        "name" : "weights",81        "shortDescription" : ""82      }83    ],84    "userDefinedMetadata" : {85      "com.github.apple.coremltools.conversion_date" : "2025-10-16",86      "com.github.apple.coremltools.source" : "torch==2.8.0",87      "com.github.apple.coremltools.version" : "9.0b1",88      "com.github.apple.coremltools.source_dialect" : "TorchScript"89    },90    "generatedClassName" : "embedding_community_1",91    "method" : "predict"92  }93]