altalt-org commited on
Commit
88e66e7
·
verified ·
1 Parent(s): d146370

Upload folder using huggingface_hub

Browse files
model-coreml-head.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9397022412985f760f1c0afbcdeb49259736d030f3b7d59520a6a0f918f8c197
3
+ size 243
model-coreml-head.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b49e9209eecb0e3beb5342c1d5321a43b4ed7fe1cfa9c0795cc70c8e669d8b3
3
+ size 721
model-coreml-head.mlmodelc/metadata.json ADDED
@@ -0,0 +1,101 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "shortDescription" : "SortFormer head (conformer + transformer + prediction) for speaker diarization",
4
+ "metadataOutputVersion" : "3.0",
5
+ "outputSchema" : [
6
+ {
7
+ "hasShapeFlexibility" : "0",
8
+ "isOptional" : "0",
9
+ "dataType" : "Float32",
10
+ "formattedType" : "MultiArray (Float32 1 × 400 × 4)",
11
+ "shortDescription" : "Speaker probabilities [1, T, 4]",
12
+ "shape" : "[1, 400, 4]",
13
+ "name" : "speaker_preds",
14
+ "type" : "MultiArray"
15
+ }
16
+ ],
17
+ "version" : "1.0",
18
+ "modelParameters" : [
19
+
20
+ ],
21
+ "author" : "sortformer-ggml",
22
+ "specificationVersion" : 7,
23
+ "storagePrecision" : "Float16",
24
+ "license" : "Apache-2.0",
25
+ "mlProgramOperationTypeHistogram" : {
26
+ "Transpose" : 192,
27
+ "Ios16.softmax" : 35,
28
+ "Split" : 17,
29
+ "Ios16.linear" : 247,
30
+ "Select" : 51,
31
+ "Ios16.add" : 173,
32
+ "Tile" : 1,
33
+ "Ios16.sigmoid" : 18,
34
+ "Ios16.logicalAnd" : 2,
35
+ "Pad" : 34,
36
+ "ExpandDims" : 8,
37
+ "Ios16.sub" : 1,
38
+ "Ios16.cast" : 4,
39
+ "Ios16.less" : 2,
40
+ "Ios16.conv" : 51,
41
+ "Ios16.matmul" : 87,
42
+ "Ios16.layerNorm" : 121,
43
+ "SliceByIndex" : 34,
44
+ "Ios16.reshape" : 174,
45
+ "Ios16.silu" : 51,
46
+ "Ios16.relu" : 20,
47
+ "Ios16.mul" : 107,
48
+ "Ios16.logicalNot" : 2
49
+ },
50
+ "computePrecision" : "Mixed (Float16, Float32, Int32)",
51
+ "stateSchema" : [
52
+
53
+ ],
54
+ "isUpdatable" : "0",
55
+ "availability" : {
56
+ "macOS" : "13.0",
57
+ "tvOS" : "16.0",
58
+ "visionOS" : "1.0",
59
+ "watchOS" : "9.0",
60
+ "iOS" : "16.0",
61
+ "macCatalyst" : "16.0"
62
+ },
63
+ "modelType" : {
64
+ "name" : "MLModelType_mlProgram"
65
+ },
66
+ "inputSchema" : [
67
+ {
68
+ "hasShapeFlexibility" : "0",
69
+ "isOptional" : "0",
70
+ "dataType" : "Float32",
71
+ "formattedType" : "MultiArray (Float32 1 × 400 × 512)",
72
+ "shortDescription" : "Pre-encoder embeddings [1, T, 512], T <= 400",
73
+ "shape" : "[1, 400, 512]",
74
+ "name" : "pre_encoder_embs",
75
+ "type" : "MultiArray"
76
+ },
77
+ {
78
+ "hasShapeFlexibility" : "0",
79
+ "isOptional" : "0",
80
+ "dataType" : "Int32",
81
+ "formattedType" : "MultiArray (Int32 1)",
82
+ "shortDescription" : "Valid sequence length",
83
+ "shape" : "[1]",
84
+ "name" : "pre_encoder_lengths",
85
+ "type" : "MultiArray"
86
+ }
87
+ ],
88
+ "userDefinedMetadata" : {
89
+ "com.github.apple.coremltools.source" : "torch==2.10.0",
90
+ "com.github.apple.coremltools.version" : "9.0",
91
+ "com.github.apple.coremltools.conversion_date" : "2026-02-03",
92
+ "com.github.apple.coremltools.source_dialect" : "TorchScript",
93
+ "n_speakers" : "4",
94
+ "precision" : "fp16",
95
+ "max_seq_len" : "400",
96
+ "fc_d_model" : "512"
97
+ },
98
+ "generatedClassName" : "model_coreml_head",
99
+ "method" : "predict"
100
+ }
101
+ ]
model-coreml-head.mlmodelc/model.mil ADDED
The diff for this file is too large to render. See raw diff
 
model-coreml-head.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0e3bd110a1e6bfa86386b684ef7efcda7f9f2e20052f1a022f586661e38569b1
3
+ size 235929152