bweng commited on
Commit
8c74d96
·
verified ·
1 Parent(s): 02f55ec

Upload 5 files

Browse files
PldaRho.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0be6ce8767cc572779215d3a442a93b029c39ff34ecee9beb176999555547da2
3
+ size 243
PldaRho.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a2a06352a5114b37e1e9f5b472a6551b169a263c16d4b7aca43281b3eadbfe1a
3
+ size 525
PldaRho.mlmodelc/metadata.json ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "shortDescription" : "pyannote community-1 PLDA rho (features scaled by sqrt(phi) for VBx clustering)",
4
+ "metadataOutputVersion" : "3.0",
5
+ "outputSchema" : [
6
+ {
7
+ "hasShapeFlexibility" : "0",
8
+ "isOptional" : "0",
9
+ "dataType" : "Float32",
10
+ "formattedType" : "MultiArray (Float32 1 × 128)",
11
+ "shortDescription" : "",
12
+ "shape" : "[1, 128]",
13
+ "name" : "rho",
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" : "Float32",
24
+ "license" : "CC-BY-4.0",
25
+ "mlProgramOperationTypeHistogram" : {
26
+ "Ios16.reduceSum" : 2,
27
+ "Ios17.clip" : 2,
28
+ "Ios17.sqrt" : 2,
29
+ "Ios17.linear" : 2,
30
+ "Ios17.realDiv" : 2,
31
+ "Ios17.mul" : 5,
32
+ "Ios17.sub" : 3
33
+ },
34
+ "computePrecision" : "Mixed (Float32, Int32)",
35
+ "stateSchema" : [
36
+
37
+ ],
38
+ "isUpdatable" : "0",
39
+ "availability" : {
40
+ "macOS" : "14.0",
41
+ "tvOS" : "17.0",
42
+ "visionOS" : "1.0",
43
+ "watchOS" : "10.0",
44
+ "iOS" : "17.0",
45
+ "macCatalyst" : "17.0"
46
+ },
47
+ "modelType" : {
48
+ "name" : "MLModelType_mlProgram"
49
+ },
50
+ "inputSchema" : [
51
+ {
52
+ "hasShapeFlexibility" : "0",
53
+ "isOptional" : "0",
54
+ "dataType" : "Float32",
55
+ "formattedType" : "MultiArray (Float32 1 × 256)",
56
+ "shortDescription" : "",
57
+ "shape" : "[1, 256]",
58
+ "name" : "embeddings",
59
+ "type" : "MultiArray"
60
+ }
61
+ ],
62
+ "userDefinedMetadata" : {
63
+ "com.github.apple.coremltools.conversion_date" : "2025-10-02",
64
+ "com.github.apple.coremltools.source" : "torch==2.8.0",
65
+ "com.github.apple.coremltools.version" : "9.0b1",
66
+ "com.github.apple.coremltools.source_dialect" : "TorchScript"
67
+ },
68
+ "generatedClassName" : "plda_rho_community_1",
69
+ "method" : "predict"
70
+ }
71
+ ]
PldaRho.mlmodelc/model.mil ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ program(1.0)
2
+ [buildInfo = dict<tensor<string, []>, tensor<string, []>>({{"coremlc-component-MIL", "3500.14.1"}, {"coremlc-version", "3500.32.1"}, {"coremltools-component-torch", "2.8.0"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "9.0b1"}})]
3
+ {
4
+ func main<ios17>(tensor<fp32, [1, 256]> embeddings) {
5
+ tensor<fp32, [128]> sqrt_phi = const()[name = tensor<string, []>("sqrt_phi"), val = tensor<fp32, [128]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(64)))];
6
+ tensor<fp32, [128, 128]> transform_plda_tr = const()[name = tensor<string, []>("transform_plda_tr"), val = tensor<fp32, [128, 128]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(640)))];
7
+ tensor<fp32, [128]> transform_mu = const()[name = tensor<string, []>("transform_mu"), val = tensor<fp32, [128]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(66240)))];
8
+ tensor<fp32, []> transform_lda_dim_scale = const()[name = tensor<string, []>("transform_lda_dim_scale"), val = tensor<fp32, []>(0x1.6a09e6p+3)];
9
+ tensor<fp32, [128]> transform_mean2 = const()[name = tensor<string, []>("transform_mean2"), val = tensor<fp32, [128]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(66816)))];
10
+ tensor<fp32, []> transform_lda_scale = const()[name = tensor<string, []>("transform_lda_scale"), val = tensor<fp32, []>(0x1p+4)];
11
+ tensor<fp32, [256]> transform_mean1 = const()[name = tensor<string, []>("transform_mean1"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(67392)))];
12
+ tensor<fp32, []> var_4 = const()[name = tensor<string, []>("op_4"), val = tensor<fp32, []>(0x1.197998p-40)];
13
+ tensor<fp32, [1, 256]> x_1 = sub(x = embeddings, y = transform_mean1)[name = tensor<string, []>("x_1")];
14
+ tensor<fp32, [1, 256]> var_17 = mul(x = x_1, y = x_1)[name = tensor<string, []>("op_17")];
15
+ tensor<int32, [1]> var_19_axes_0 = const()[name = tensor<string, []>("op_19_axes_0"), val = tensor<int32, [1]>([-1])];
16
+ tensor<bool, []> var_19_keep_dims_0 = const()[name = tensor<string, []>("op_19_keep_dims_0"), val = tensor<bool, []>(true)];
17
+ tensor<fp32, [1, 1]> var_19 = reduce_sum(axes = var_19_axes_0, keep_dims = var_19_keep_dims_0, x = var_17)[name = tensor<string, []>("op_19")];
18
+ tensor<fp32, []> const_0 = const()[name = tensor<string, []>("const_0"), val = tensor<fp32, []>(0x1.fffffep+127)];
19
+ tensor<fp32, [1, 1]> clip_0 = clip(alpha = var_4, beta = const_0, x = var_19)[name = tensor<string, []>("clip_0")];
20
+ tensor<fp32, [1, 1]> norm_1 = sqrt(x = clip_0)[name = tensor<string, []>("norm_1")];
21
+ tensor<fp32, [1, 256]> normalized1 = real_div(x = x_1, y = norm_1)[name = tensor<string, []>("normalized1")];
22
+ tensor<fp32, [128, 256]> transpose_0 = const()[name = tensor<string, []>("transpose_0"), val = tensor<fp32, [128, 256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(68480)))];
23
+ tensor<fp32, [128]> var_23_bias_0 = const()[name = tensor<string, []>("op_23_bias_0"), val = tensor<fp32, [128]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(199616)))];
24
+ tensor<fp32, [1, 128]> var_23 = linear(bias = var_23_bias_0, weight = transpose_0, x = normalized1)[name = tensor<string, []>("op_23")];
25
+ tensor<fp32, [1, 128]> projected = mul(x = var_23, y = transform_lda_scale)[name = tensor<string, []>("projected")];
26
+ tensor<fp32, [1, 128]> x = sub(x = projected, y = transform_mean2)[name = tensor<string, []>("x")];
27
+ tensor<fp32, [1, 128]> var_26 = mul(x = x, y = x)[name = tensor<string, []>("op_26")];
28
+ tensor<int32, [1]> var_28_axes_0 = const()[name = tensor<string, []>("op_28_axes_0"), val = tensor<int32, [1]>([-1])];
29
+ tensor<bool, []> var_28_keep_dims_0 = const()[name = tensor<string, []>("op_28_keep_dims_0"), val = tensor<bool, []>(true)];
30
+ tensor<fp32, [1, 1]> var_28 = reduce_sum(axes = var_28_axes_0, keep_dims = var_28_keep_dims_0, x = var_26)[name = tensor<string, []>("op_28")];
31
+ tensor<fp32, []> const_1 = const()[name = tensor<string, []>("const_1"), val = tensor<fp32, []>(0x1.fffffep+127)];
32
+ tensor<fp32, [1, 1]> clip_1 = clip(alpha = var_4, beta = const_1, x = var_28)[name = tensor<string, []>("clip_1")];
33
+ tensor<fp32, [1, 1]> norm = sqrt(x = clip_1)[name = tensor<string, []>("norm")];
34
+ tensor<fp32, [1, 128]> var_31 = real_div(x = x, y = norm)[name = tensor<string, []>("op_31")];
35
+ tensor<fp32, [1, 128]> normalized2 = mul(x = var_31, y = transform_lda_dim_scale)[name = tensor<string, []>("normalized2")];
36
+ tensor<fp32, [1, 128]> plda_centered = sub(x = normalized2, y = transform_mu)[name = tensor<string, []>("plda_centered")];
37
+ tensor<fp32, [1, 128]> features = linear(bias = var_23_bias_0, weight = transform_plda_tr, x = plda_centered)[name = tensor<string, []>("features")];
38
+ tensor<fp32, [1, 128]> rho = mul(x = features, y = sqrt_phi)[name = tensor<string, []>("op_36")];
39
+ } -> (rho);
40
+ }
PldaRho.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:80f7d229202636d372428c90596f11a91545f07da77259f07153aaf225914a36
3
+ size 200192