psvensk commited on
Commit
f34047f
·
verified ·
1 Parent(s): cb4c2f6

Upload folder using huggingface_hub

Browse files
medium_quantized/AudioEncoder.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:579c22ab4b44712e7e08bb242a3c6947fb3ff5b925f563e170fa34b333eca96f
3
+ size 243
medium_quantized/AudioEncoder.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:77bad4c3b410b106d71b6651fa35eacc0a193e34b67d5394db01ccf2bf19de92
3
+ size 409
medium_quantized/AudioEncoder.mlmodelc/metadata.json ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "metadataOutputVersion" : "3.0",
4
+ "storagePrecision" : "Mixed (Float16, Palettized (6 bits))",
5
+ "outputSchema" : [
6
+ {
7
+ "hasShapeFlexibility" : "0",
8
+ "isOptional" : "0",
9
+ "dataType" : "Float16",
10
+ "formattedType" : "MultiArray (Float16 1 × 1024 × 1 × 1500)",
11
+ "shortDescription" : "",
12
+ "shape" : "[1, 1024, 1, 1500]",
13
+ "name" : "encoder_output_embeds",
14
+ "type" : "MultiArray"
15
+ }
16
+ ],
17
+ "modelParameters" : [
18
+
19
+ ],
20
+ "specificationVersion" : 7,
21
+ "mlProgramOperationTypeHistogram" : {
22
+ "Concat" : 408,
23
+ "Ios16.mul" : 2304,
24
+ "Ios16.layerNorm" : 49,
25
+ "SliceByIndex" : 3456,
26
+ "Ios16.constexprLutToDense" : 146,
27
+ "Transpose" : 24,
28
+ "Ios16.einsum" : 4608,
29
+ "Ios16.conv" : 146,
30
+ "Ios16.add" : 49,
31
+ "Ios16.softmax" : 2304,
32
+ "Ios16.gelu" : 26,
33
+ "Ios16.batchNorm" : 49
34
+ },
35
+ "computePrecision" : "Mixed (Float16, Int32)",
36
+ "isUpdatable" : "0",
37
+ "stateSchema" : [
38
+
39
+ ],
40
+ "availability" : {
41
+ "macOS" : "13.0",
42
+ "tvOS" : "16.0",
43
+ "visionOS" : "1.0",
44
+ "watchOS" : "9.0",
45
+ "iOS" : "16.0",
46
+ "macCatalyst" : "16.0"
47
+ },
48
+ "modelType" : {
49
+ "name" : "MLModelType_mlProgram"
50
+ },
51
+ "userDefinedMetadata" : {
52
+ "com.github.apple.coremltools.conversion_date" : "2026-06-10",
53
+ "com.github.apple.coremltools.source" : "torch==2.12.0",
54
+ "com.github.apple.coremltools.version" : "9.0",
55
+ "com.github.apple.coremltools.source_dialect" : "TorchScript"
56
+ },
57
+ "inputSchema" : [
58
+ {
59
+ "hasShapeFlexibility" : "0",
60
+ "isOptional" : "0",
61
+ "dataType" : "Float16",
62
+ "formattedType" : "MultiArray (Float16 1 × 80 × 1 × 3000)",
63
+ "shortDescription" : "",
64
+ "shape" : "[1, 80, 1, 3000]",
65
+ "name" : "melspectrogram_features",
66
+ "type" : "MultiArray"
67
+ }
68
+ ],
69
+ "generatedClassName" : "AudioEncoder_mixedBitPalettized_6_0_bit",
70
+ "method" : "predict"
71
+ }
72
+ ]
medium_quantized/AudioEncoder.mlmodelc/model.mil ADDED
The diff for this file is too large to render. See raw diff
 
medium_quantized/AudioEncoder.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3a81bde5b1cc52822b2adbbcd2bd7f1dca7e1c63cd625c3c9a3950eebb5ee235
3
+ size 232761856
medium_quantized/MelSpectrogram.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a9b62a622d99c51a03f7ad3cc128a3e5a6f51af0d984f9771bf5f6807ea796f0
3
+ size 243
medium_quantized/MelSpectrogram.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bbb99c3af322f271b90d4486e07b1a749bb2e47c7eecb5b2878cf7adc45c830a
3
+ size 390
medium_quantized/MelSpectrogram.mlmodelc/metadata.json ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "metadataOutputVersion" : "3.0",
4
+ "storagePrecision" : "Float16",
5
+ "outputSchema" : [
6
+ {
7
+ "hasShapeFlexibility" : "0",
8
+ "isOptional" : "0",
9
+ "dataType" : "Float16",
10
+ "formattedType" : "MultiArray (Float16 1 × 80 × 1 × 3000)",
11
+ "shortDescription" : "",
12
+ "shape" : "[1, 80, 1, 3000]",
13
+ "name" : "melspectrogram_features",
14
+ "type" : "MultiArray"
15
+ }
16
+ ],
17
+ "modelParameters" : [
18
+
19
+ ],
20
+ "specificationVersion" : 7,
21
+ "mlProgramOperationTypeHistogram" : {
22
+ "Ios16.reshape" : 2,
23
+ "Ios16.mul" : 2,
24
+ "SliceByIndex" : 1,
25
+ "Ios16.sub" : 1,
26
+ "Ios16.log" : 1,
27
+ "Ios16.square" : 2,
28
+ "Ios16.add" : 3,
29
+ "Squeeze" : 2,
30
+ "Ios16.matmul" : 1,
31
+ "Ios16.conv" : 2,
32
+ "Ios16.maximum" : 1,
33
+ "ExpandDims" : 4,
34
+ "Ios16.reduceMax" : 1,
35
+ "Identity" : 1,
36
+ "Pad" : 1
37
+ },
38
+ "computePrecision" : "Mixed (Float16, Int32)",
39
+ "isUpdatable" : "0",
40
+ "stateSchema" : [
41
+
42
+ ],
43
+ "availability" : {
44
+ "macOS" : "13.0",
45
+ "tvOS" : "16.0",
46
+ "visionOS" : "1.0",
47
+ "watchOS" : "9.0",
48
+ "iOS" : "16.0",
49
+ "macCatalyst" : "16.0"
50
+ },
51
+ "modelType" : {
52
+ "name" : "MLModelType_mlProgram"
53
+ },
54
+ "userDefinedMetadata" : {
55
+ "com.github.apple.coremltools.conversion_date" : "2026-06-10",
56
+ "com.github.apple.coremltools.source" : "torch==2.12.0",
57
+ "com.github.apple.coremltools.version" : "9.0",
58
+ "com.github.apple.coremltools.source_dialect" : "TorchScript"
59
+ },
60
+ "inputSchema" : [
61
+ {
62
+ "hasShapeFlexibility" : "0",
63
+ "isOptional" : "0",
64
+ "dataType" : "Float16",
65
+ "formattedType" : "MultiArray (Float16 480000)",
66
+ "shortDescription" : "",
67
+ "shape" : "[480000]",
68
+ "name" : "audio",
69
+ "type" : "MultiArray"
70
+ }
71
+ ],
72
+ "generatedClassName" : "MelSpectrogram",
73
+ "method" : "predict"
74
+ }
75
+ ]
medium_quantized/MelSpectrogram.mlmodelc/model.mil ADDED
@@ -0,0 +1,66 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ program(1.0)
2
+ [buildInfo = dict<tensor<string, []>, tensor<string, []>>({{"coremlc-component-MIL", "3520.4.1"}, {"coremlc-version", "3520.5.1"}, {"coremltools-component-torch", "2.12.0"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "9.0"}})]
3
+ {
4
+ func main<ios16>(tensor<fp16, [480000]> audio) {
5
+ tensor<int32, [3]> var_10 = const()[name = tensor<string, []>("op_10"), val = tensor<int32, [3]>([1, 1, 480000])];
6
+ tensor<fp16, [1, 1, 480000]> input_1_cast_fp16 = reshape(shape = var_10, x = audio)[name = tensor<string, []>("input_1_cast_fp16")];
7
+ tensor<int32, [6]> input_3_pad_0 = const()[name = tensor<string, []>("input_3_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 200, 200])];
8
+ tensor<string, []> input_3_mode_0 = const()[name = tensor<string, []>("input_3_mode_0"), val = tensor<string, []>("reflect")];
9
+ tensor<fp16, []> const_1_to_fp16 = const()[name = tensor<string, []>("const_1_to_fp16"), val = tensor<fp16, []>(0x0p+0)];
10
+ tensor<fp16, [1, 1, 480400]> input_3_cast_fp16 = pad(constant_val = const_1_to_fp16, mode = input_3_mode_0, pad = input_3_pad_0, x = input_1_cast_fp16)[name = tensor<string, []>("input_3_cast_fp16")];
11
+ tensor<int32, [1]> var_22 = const()[name = tensor<string, []>("op_22"), val = tensor<int32, [1]>([480400])];
12
+ tensor<fp16, [480400]> input_cast_fp16 = reshape(shape = var_22, x = input_3_cast_fp16)[name = tensor<string, []>("input_cast_fp16")];
13
+ tensor<int32, [1]> expand_dims_0_axes_0 = const()[name = tensor<string, []>("expand_dims_0_axes_0"), val = tensor<int32, [1]>([0])];
14
+ tensor<fp16, [1, 480400]> expand_dims_0_cast_fp16 = expand_dims(axes = expand_dims_0_axes_0, x = input_cast_fp16)[name = tensor<string, []>("expand_dims_0_cast_fp16")];
15
+ tensor<int32, [1]> expand_dims_3 = const()[name = tensor<string, []>("expand_dims_3"), val = tensor<int32, [1]>([160])];
16
+ tensor<int32, [1]> expand_dims_4_axes_0 = const()[name = tensor<string, []>("expand_dims_4_axes_0"), val = tensor<int32, [1]>([1])];
17
+ tensor<fp16, [1, 1, 480400]> expand_dims_4_cast_fp16 = expand_dims(axes = expand_dims_4_axes_0, x = expand_dims_0_cast_fp16)[name = tensor<string, []>("expand_dims_4_cast_fp16")];
18
+ tensor<string, []> conv_0_pad_type_0 = const()[name = tensor<string, []>("conv_0_pad_type_0"), val = tensor<string, []>("valid")];
19
+ tensor<int32, [2]> conv_0_pad_0 = const()[name = tensor<string, []>("conv_0_pad_0"), val = tensor<int32, [2]>([0, 0])];
20
+ tensor<int32, [1]> conv_0_dilations_0 = const()[name = tensor<string, []>("conv_0_dilations_0"), val = tensor<int32, [1]>([1])];
21
+ tensor<int32, []> conv_0_groups_0 = const()[name = tensor<string, []>("conv_0_groups_0"), val = tensor<int32, []>(1)];
22
+ tensor<fp16, [201, 1, 400]> expand_dims_1_to_fp16 = const()[name = tensor<string, []>("expand_dims_1_to_fp16"), val = tensor<fp16, [201, 1, 400]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(64)))];
23
+ tensor<fp16, [1, 201, 3001]> conv_0_cast_fp16 = conv(dilations = conv_0_dilations_0, groups = conv_0_groups_0, pad = conv_0_pad_0, pad_type = conv_0_pad_type_0, strides = expand_dims_3, weight = expand_dims_1_to_fp16, x = expand_dims_4_cast_fp16)[name = tensor<string, []>("conv_0_cast_fp16")];
24
+ tensor<string, []> conv_1_pad_type_0 = const()[name = tensor<string, []>("conv_1_pad_type_0"), val = tensor<string, []>("valid")];
25
+ tensor<int32, [2]> conv_1_pad_0 = const()[name = tensor<string, []>("conv_1_pad_0"), val = tensor<int32, [2]>([0, 0])];
26
+ tensor<int32, [1]> conv_1_dilations_0 = const()[name = tensor<string, []>("conv_1_dilations_0"), val = tensor<int32, [1]>([1])];
27
+ tensor<int32, []> conv_1_groups_0 = const()[name = tensor<string, []>("conv_1_groups_0"), val = tensor<int32, []>(1)];
28
+ tensor<fp16, [201, 1, 400]> expand_dims_2_to_fp16 = const()[name = tensor<string, []>("expand_dims_2_to_fp16"), val = tensor<fp16, [201, 1, 400]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(160960)))];
29
+ tensor<fp16, [1, 201, 3001]> conv_1_cast_fp16 = conv(dilations = conv_1_dilations_0, groups = conv_1_groups_0, pad = conv_1_pad_0, pad_type = conv_1_pad_type_0, strides = expand_dims_3, weight = expand_dims_2_to_fp16, x = expand_dims_4_cast_fp16)[name = tensor<string, []>("conv_1_cast_fp16")];
30
+ tensor<int32, [1]> squeeze_0_axes_0 = const()[name = tensor<string, []>("squeeze_0_axes_0"), val = tensor<int32, [1]>([0])];
31
+ tensor<fp16, [201, 3001]> squeeze_0_cast_fp16 = squeeze(axes = squeeze_0_axes_0, x = conv_0_cast_fp16)[name = tensor<string, []>("squeeze_0_cast_fp16")];
32
+ tensor<int32, [1]> squeeze_1_axes_0 = const()[name = tensor<string, []>("squeeze_1_axes_0"), val = tensor<int32, [1]>([0])];
33
+ tensor<fp16, [201, 3001]> squeeze_1_cast_fp16 = squeeze(axes = squeeze_1_axes_0, x = conv_1_cast_fp16)[name = tensor<string, []>("squeeze_1_cast_fp16")];
34
+ tensor<fp16, [201, 3001]> square_0_cast_fp16 = square(x = squeeze_0_cast_fp16)[name = tensor<string, []>("square_0_cast_fp16")];
35
+ tensor<fp16, [201, 3001]> square_1_cast_fp16 = square(x = squeeze_1_cast_fp16)[name = tensor<string, []>("square_1_cast_fp16")];
36
+ tensor<fp16, [201, 3001]> add_1_cast_fp16 = add(x = square_0_cast_fp16, y = square_1_cast_fp16)[name = tensor<string, []>("add_1_cast_fp16")];
37
+ tensor<fp16, [201, 3001]> magnitudes_1_cast_fp16 = identity(x = add_1_cast_fp16)[name = tensor<string, []>("magnitudes_1_cast_fp16")];
38
+ tensor<int32, [2]> magnitudes_begin_0 = const()[name = tensor<string, []>("magnitudes_begin_0"), val = tensor<int32, [2]>([0, 0])];
39
+ tensor<int32, [2]> magnitudes_end_0 = const()[name = tensor<string, []>("magnitudes_end_0"), val = tensor<int32, [2]>([201, 3000])];
40
+ tensor<bool, [2]> magnitudes_end_mask_0 = const()[name = tensor<string, []>("magnitudes_end_mask_0"), val = tensor<bool, [2]>([true, false])];
41
+ tensor<fp16, [201, 3000]> magnitudes_cast_fp16 = slice_by_index(begin = magnitudes_begin_0, end = magnitudes_end_0, end_mask = magnitudes_end_mask_0, x = magnitudes_1_cast_fp16)[name = tensor<string, []>("magnitudes_cast_fp16")];
42
+ tensor<bool, []> mel_spec_1_transpose_x_0 = const()[name = tensor<string, []>("mel_spec_1_transpose_x_0"), val = tensor<bool, []>(false)];
43
+ tensor<bool, []> mel_spec_1_transpose_y_0 = const()[name = tensor<string, []>("mel_spec_1_transpose_y_0"), val = tensor<bool, []>(false)];
44
+ tensor<fp16, [80, 201]> mel_filters_to_fp16 = const()[name = tensor<string, []>("mel_filters_to_fp16"), val = tensor<fp16, [80, 201]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(321856)))];
45
+ tensor<fp16, [80, 3000]> mel_spec_1_cast_fp16 = matmul(transpose_x = mel_spec_1_transpose_x_0, transpose_y = mel_spec_1_transpose_y_0, x = mel_filters_to_fp16, y = magnitudes_cast_fp16)[name = tensor<string, []>("mel_spec_1_cast_fp16")];
46
+ tensor<fp16, []> var_42_to_fp16 = const()[name = tensor<string, []>("op_42_to_fp16"), val = tensor<fp16, []>(0x1p-24)];
47
+ tensor<fp16, [80, 3000]> mel_spec_cast_fp16 = add(x = mel_spec_1_cast_fp16, y = var_42_to_fp16)[name = tensor<string, []>("mel_spec_cast_fp16")];
48
+ tensor<fp16, []> log_0_epsilon_0_to_fp16 = const()[name = tensor<string, []>("log_0_epsilon_0_to_fp16"), val = tensor<fp16, []>(0x0p+0)];
49
+ tensor<fp16, [80, 3000]> log_0_cast_fp16 = log(epsilon = log_0_epsilon_0_to_fp16, x = mel_spec_cast_fp16)[name = tensor<string, []>("log_0_cast_fp16")];
50
+ tensor<fp16, []> mul_0_y_0_to_fp16 = const()[name = tensor<string, []>("mul_0_y_0_to_fp16"), val = tensor<fp16, []>(0x1.bccp-2)];
51
+ tensor<fp16, [80, 3000]> mul_0_cast_fp16 = mul(x = log_0_cast_fp16, y = mul_0_y_0_to_fp16)[name = tensor<string, []>("mul_0_cast_fp16")];
52
+ tensor<bool, []> var_45_keep_dims_0 = const()[name = tensor<string, []>("op_45_keep_dims_0"), val = tensor<bool, []>(false)];
53
+ tensor<fp16, []> var_45_cast_fp16 = reduce_max(keep_dims = var_45_keep_dims_0, x = mul_0_cast_fp16)[name = tensor<string, []>("op_45_cast_fp16")];
54
+ tensor<fp16, []> var_47_to_fp16 = const()[name = tensor<string, []>("op_47_to_fp16"), val = tensor<fp16, []>(0x1p+3)];
55
+ tensor<fp16, []> var_48_cast_fp16 = sub(x = var_45_cast_fp16, y = var_47_to_fp16)[name = tensor<string, []>("op_48_cast_fp16")];
56
+ tensor<fp16, [80, 3000]> log_spec_3_cast_fp16 = maximum(x = mul_0_cast_fp16, y = var_48_cast_fp16)[name = tensor<string, []>("log_spec_3_cast_fp16")];
57
+ tensor<fp16, []> var_51_to_fp16 = const()[name = tensor<string, []>("op_51_to_fp16"), val = tensor<fp16, []>(0x1p+2)];
58
+ tensor<fp16, [80, 3000]> var_52_cast_fp16 = add(x = log_spec_3_cast_fp16, y = var_51_to_fp16)[name = tensor<string, []>("op_52_cast_fp16")];
59
+ tensor<fp16, []> _inversed_log_spec_y_0_to_fp16 = const()[name = tensor<string, []>("_inversed_log_spec_y_0_to_fp16"), val = tensor<fp16, []>(0x1p-2)];
60
+ tensor<fp16, [80, 3000]> _inversed_log_spec_cast_fp16 = mul(x = var_52_cast_fp16, y = _inversed_log_spec_y_0_to_fp16)[name = tensor<string, []>("_inversed_log_spec_cast_fp16")];
61
+ tensor<int32, [1]> var_56_axes_0 = const()[name = tensor<string, []>("op_56_axes_0"), val = tensor<int32, [1]>([0])];
62
+ tensor<fp16, [1, 80, 3000]> var_56_cast_fp16 = expand_dims(axes = var_56_axes_0, x = _inversed_log_spec_cast_fp16)[name = tensor<string, []>("op_56_cast_fp16")];
63
+ tensor<int32, [1]> var_63_axes_0 = const()[name = tensor<string, []>("op_63_axes_0"), val = tensor<int32, [1]>([2])];
64
+ tensor<fp16, [1, 80, 1, 3000]> melspectrogram_features = expand_dims(axes = var_63_axes_0, x = var_56_cast_fp16)[name = tensor<string, []>("op_63_cast_fp16")];
65
+ } -> (melspectrogram_features);
66
+ }
medium_quantized/MelSpectrogram.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:801024dbc7a89c677be1f8b285de3409e35f7d1786c9c8d9d0d6842ac57a1c83
3
+ size 354080
medium_quantized/TextDecoder.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1e6e455710f5f42e302383bc355a9e2b580ea737a7db2dd7f915fd4d0510a0a
3
+ size 243
medium_quantized/TextDecoder.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3480ee2b09b62c984cc1342c01d4937d84660a66e56aa4abfc290a1aec7fa8a3
3
+ size 699
medium_quantized/TextDecoder.mlmodelc/metadata.json ADDED
@@ -0,0 +1,170 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "metadataOutputVersion" : "3.0",
4
+ "storagePrecision" : "Mixed (Float16, Palettized (6 bits))",
5
+ "outputSchema" : [
6
+ {
7
+ "hasShapeFlexibility" : "0",
8
+ "isOptional" : "0",
9
+ "dataType" : "Float16",
10
+ "formattedType" : "MultiArray (Float16 1 × 1 × 51865)",
11
+ "shortDescription" : "",
12
+ "shape" : "[1, 1, 51865]",
13
+ "name" : "logits",
14
+ "type" : "MultiArray"
15
+ },
16
+ {
17
+ "hasShapeFlexibility" : "0",
18
+ "isOptional" : "0",
19
+ "dataType" : "Float16",
20
+ "formattedType" : "MultiArray (Float16 1 × 24576 × 1 × 1)",
21
+ "shortDescription" : "",
22
+ "shape" : "[1, 24576, 1, 1]",
23
+ "name" : "key_cache_updates",
24
+ "type" : "MultiArray"
25
+ },
26
+ {
27
+ "hasShapeFlexibility" : "0",
28
+ "isOptional" : "0",
29
+ "dataType" : "Float16",
30
+ "formattedType" : "MultiArray (Float16 1 × 24576 × 1 × 1)",
31
+ "shortDescription" : "",
32
+ "shape" : "[1, 24576, 1, 1]",
33
+ "name" : "value_cache_updates",
34
+ "type" : "MultiArray"
35
+ },
36
+ {
37
+ "hasShapeFlexibility" : "0",
38
+ "isOptional" : "0",
39
+ "dataType" : "Float16",
40
+ "formattedType" : "MultiArray (Float16 1 × 1500)",
41
+ "shortDescription" : "",
42
+ "shape" : "[1, 1500]",
43
+ "name" : "alignment_heads_weights",
44
+ "type" : "MultiArray"
45
+ }
46
+ ],
47
+ "modelParameters" : [
48
+
49
+ ],
50
+ "specificationVersion" : 7,
51
+ "mlProgramOperationTypeHistogram" : {
52
+ "Transpose" : 1,
53
+ "Squeeze" : 1,
54
+ "Ios16.gather" : 2,
55
+ "Ios16.reduceMean" : 1,
56
+ "Ios16.softmax" : 48,
57
+ "Split" : 2,
58
+ "Ios16.linear" : 1,
59
+ "Ios16.add" : 145,
60
+ "Concat" : 3,
61
+ "ExpandDims" : 6,
62
+ "Ios16.sub" : 1,
63
+ "Ios16.conv" : 240,
64
+ "Ios16.gelu" : 24,
65
+ "Ios16.constexprLutToDense" : 239,
66
+ "Ios16.matmul" : 96,
67
+ "Ios16.layerNorm" : 73,
68
+ "SliceByIndex" : 12,
69
+ "Ios16.batchNorm" : 73,
70
+ "Ios16.reshape" : 192,
71
+ "Ios16.mul" : 144
72
+ },
73
+ "computePrecision" : "Mixed (Float16, Int32)",
74
+ "isUpdatable" : "0",
75
+ "stateSchema" : [
76
+
77
+ ],
78
+ "availability" : {
79
+ "macOS" : "13.0",
80
+ "tvOS" : "16.0",
81
+ "visionOS" : "1.0",
82
+ "watchOS" : "9.0",
83
+ "iOS" : "16.0",
84
+ "macCatalyst" : "16.0"
85
+ },
86
+ "modelType" : {
87
+ "name" : "MLModelType_mlProgram"
88
+ },
89
+ "userDefinedMetadata" : {
90
+ "com.github.apple.coremltools.conversion_date" : "2026-06-10",
91
+ "com.github.apple.coremltools.source" : "torch==2.12.0",
92
+ "com.github.apple.coremltools.version" : "9.0",
93
+ "com.github.apple.coremltools.source_dialect" : "TorchScript"
94
+ },
95
+ "inputSchema" : [
96
+ {
97
+ "hasShapeFlexibility" : "0",
98
+ "isOptional" : "0",
99
+ "dataType" : "Int32",
100
+ "formattedType" : "MultiArray (Int32 1)",
101
+ "shortDescription" : "",
102
+ "shape" : "[1]",
103
+ "name" : "input_ids",
104
+ "type" : "MultiArray"
105
+ },
106
+ {
107
+ "hasShapeFlexibility" : "0",
108
+ "isOptional" : "0",
109
+ "dataType" : "Int32",
110
+ "formattedType" : "MultiArray (Int32 1)",
111
+ "shortDescription" : "",
112
+ "shape" : "[1]",
113
+ "name" : "cache_length",
114
+ "type" : "MultiArray"
115
+ },
116
+ {
117
+ "hasShapeFlexibility" : "0",
118
+ "isOptional" : "0",
119
+ "dataType" : "Float16",
120
+ "formattedType" : "MultiArray (Float16 1 × 24576 × 1 × 448)",
121
+ "shortDescription" : "",
122
+ "shape" : "[1, 24576, 1, 448]",
123
+ "name" : "key_cache",
124
+ "type" : "MultiArray"
125
+ },
126
+ {
127
+ "hasShapeFlexibility" : "0",
128
+ "isOptional" : "0",
129
+ "dataType" : "Float16",
130
+ "formattedType" : "MultiArray (Float16 1 × 24576 × 1 × 448)",
131
+ "shortDescription" : "",
132
+ "shape" : "[1, 24576, 1, 448]",
133
+ "name" : "value_cache",
134
+ "type" : "MultiArray"
135
+ },
136
+ {
137
+ "hasShapeFlexibility" : "0",
138
+ "isOptional" : "0",
139
+ "dataType" : "Float16",
140
+ "formattedType" : "MultiArray (Float16 1 × 448)",
141
+ "shortDescription" : "",
142
+ "shape" : "[1, 448]",
143
+ "name" : "kv_cache_update_mask",
144
+ "type" : "MultiArray"
145
+ },
146
+ {
147
+ "hasShapeFlexibility" : "0",
148
+ "isOptional" : "0",
149
+ "dataType" : "Float16",
150
+ "formattedType" : "MultiArray (Float16 1 × 1024 × 1 × 1500)",
151
+ "shortDescription" : "",
152
+ "shape" : "[1, 1024, 1, 1500]",
153
+ "name" : "encoder_output_embeds",
154
+ "type" : "MultiArray"
155
+ },
156
+ {
157
+ "hasShapeFlexibility" : "0",
158
+ "isOptional" : "0",
159
+ "dataType" : "Float16",
160
+ "formattedType" : "MultiArray (Float16 1 × 448)",
161
+ "shortDescription" : "",
162
+ "shape" : "[1, 448]",
163
+ "name" : "decoder_key_padding_mask",
164
+ "type" : "MultiArray"
165
+ }
166
+ ],
167
+ "generatedClassName" : "TextDecoder_mixedBitPalettized_6_0_bit",
168
+ "method" : "predict"
169
+ }
170
+ ]
medium_quantized/TextDecoder.mlmodelc/model.mil ADDED
The diff for this file is too large to render. See raw diff
 
medium_quantized/TextDecoder.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc92ff4057f048cf8c68e1348d18d009b1fae3271aaa857c943151a4bf126de9
3
+ size 411468402
medium_quantized/config.json ADDED
@@ -0,0 +1,269 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/leonardo_work/EUHPC_A01_006/experiments_whisper/stage1_results/medium_250210_124500/",
3
+ "activation_dropout": 0.1,
4
+ "activation_function": "gelu",
5
+ "apply_spec_augment": false,
6
+ "architectures": [
7
+ "WhisperForConditionalGeneration"
8
+ ],
9
+ "attention_dropout": 0.0,
10
+ "begin_suppress_tokens": null,
11
+ "bos_token_id": 50257,
12
+ "classifier_proj_size": 256,
13
+ "d_model": 1024,
14
+ "decoder_attention_heads": 16,
15
+ "decoder_ffn_dim": 4096,
16
+ "decoder_layerdrop": 0.0,
17
+ "decoder_layers": 24,
18
+ "decoder_start_token_id": 50258,
19
+ "dropout": 0.0,
20
+ "encoder_attention_heads": 16,
21
+ "encoder_ffn_dim": 4096,
22
+ "encoder_layerdrop": 0.0,
23
+ "encoder_layers": 24,
24
+ "eos_token_id": 50257,
25
+ "forced_decoder_ids": null,
26
+ "init_std": 0.02,
27
+ "is_encoder_decoder": true,
28
+ "mask_feature_length": 10,
29
+ "mask_feature_min_masks": 0,
30
+ "mask_feature_prob": 0.0,
31
+ "mask_time_length": 10,
32
+ "mask_time_min_masks": 2,
33
+ "mask_time_prob": 0.05,
34
+ "max_length": null,
35
+ "max_source_positions": 1500,
36
+ "max_target_positions": 448,
37
+ "median_filter_width": 7,
38
+ "model_type": "whisper",
39
+ "num_hidden_layers": 24,
40
+ "num_mel_bins": 80,
41
+ "pad_token_id": 50257,
42
+ "scale_embedding": false,
43
+ "torch_dtype": "float16",
44
+ "transformers_version": "4.45.2",
45
+ "use_cache": true,
46
+ "use_weighted_layer_sum": false,
47
+ "vocab_size": 51865,
48
+ "alignment_heads": [
49
+ [
50
+ 13,
51
+ 15
52
+ ],
53
+ [
54
+ 15,
55
+ 4
56
+ ],
57
+ [
58
+ 15,
59
+ 15
60
+ ],
61
+ [
62
+ 16,
63
+ 1
64
+ ],
65
+ [
66
+ 20,
67
+ 0
68
+ ],
69
+ [
70
+ 23,
71
+ 4
72
+ ]
73
+ ],
74
+ "lang_ids": [
75
+ 50259,
76
+ 50260,
77
+ 50261,
78
+ 50262,
79
+ 50263,
80
+ 50264,
81
+ 50265,
82
+ 50266,
83
+ 50267,
84
+ 50268,
85
+ 50269,
86
+ 50270,
87
+ 50271,
88
+ 50272,
89
+ 50273,
90
+ 50274,
91
+ 50275,
92
+ 50276,
93
+ 50277,
94
+ 50278,
95
+ 50279,
96
+ 50280,
97
+ 50281,
98
+ 50282,
99
+ 50283,
100
+ 50284,
101
+ 50285,
102
+ 50286,
103
+ 50287,
104
+ 50288,
105
+ 50289,
106
+ 50290,
107
+ 50291,
108
+ 50292,
109
+ 50293,
110
+ 50294,
111
+ 50295,
112
+ 50296,
113
+ 50297,
114
+ 50298,
115
+ 50299,
116
+ 50300,
117
+ 50301,
118
+ 50302,
119
+ 50303,
120
+ 50304,
121
+ 50305,
122
+ 50306,
123
+ 50307,
124
+ 50308,
125
+ 50309,
126
+ 50310,
127
+ 50311,
128
+ 50312,
129
+ 50313,
130
+ 50314,
131
+ 50315,
132
+ 50316,
133
+ 50317,
134
+ 50318,
135
+ 50319,
136
+ 50320,
137
+ 50321,
138
+ 50322,
139
+ 50323,
140
+ 50324,
141
+ 50325,
142
+ 50326,
143
+ 50327,
144
+ 50328,
145
+ 50329,
146
+ 50330,
147
+ 50331,
148
+ 50332,
149
+ 50333,
150
+ 50334,
151
+ 50335,
152
+ 50336,
153
+ 50337,
154
+ 50338,
155
+ 50339,
156
+ 50340,
157
+ 50341,
158
+ 50342,
159
+ 50343,
160
+ 50344,
161
+ 50345,
162
+ 50346,
163
+ 50347,
164
+ 50348,
165
+ 50349,
166
+ 50350,
167
+ 50351,
168
+ 50352,
169
+ 50353,
170
+ 50354,
171
+ 50355,
172
+ 50356,
173
+ 50357
174
+ ],
175
+ "suppress_ids": [
176
+ 1,
177
+ 2,
178
+ 7,
179
+ 8,
180
+ 9,
181
+ 10,
182
+ 14,
183
+ 25,
184
+ 26,
185
+ 27,
186
+ 28,
187
+ 29,
188
+ 31,
189
+ 58,
190
+ 59,
191
+ 60,
192
+ 61,
193
+ 62,
194
+ 63,
195
+ 90,
196
+ 91,
197
+ 92,
198
+ 93,
199
+ 359,
200
+ 503,
201
+ 522,
202
+ 542,
203
+ 873,
204
+ 893,
205
+ 902,
206
+ 918,
207
+ 922,
208
+ 931,
209
+ 1350,
210
+ 1853,
211
+ 1982,
212
+ 2460,
213
+ 2627,
214
+ 3246,
215
+ 3253,
216
+ 3268,
217
+ 3536,
218
+ 3846,
219
+ 3961,
220
+ 4183,
221
+ 4667,
222
+ 6585,
223
+ 6647,
224
+ 7273,
225
+ 9061,
226
+ 9383,
227
+ 10428,
228
+ 10929,
229
+ 11938,
230
+ 12033,
231
+ 12331,
232
+ 12562,
233
+ 13793,
234
+ 14157,
235
+ 14635,
236
+ 15265,
237
+ 15618,
238
+ 16553,
239
+ 16604,
240
+ 18362,
241
+ 18956,
242
+ 20075,
243
+ 21675,
244
+ 22520,
245
+ 26130,
246
+ 26161,
247
+ 26435,
248
+ 28279,
249
+ 29464,
250
+ 31650,
251
+ 32302,
252
+ 32470,
253
+ 36865,
254
+ 42863,
255
+ 47425,
256
+ 49870,
257
+ 50254,
258
+ 50258,
259
+ 50358,
260
+ 50359,
261
+ 50360,
262
+ 50361,
263
+ 50362
264
+ ],
265
+ "suppress_ids_begin": [
266
+ 220,
267
+ 50257
268
+ ]
269
+ }
medium_quantized/generation_config.json ADDED
@@ -0,0 +1,240 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alignment_heads": [
3
+ [
4
+ 13,
5
+ 15
6
+ ],
7
+ [
8
+ 15,
9
+ 4
10
+ ],
11
+ [
12
+ 15,
13
+ 15
14
+ ],
15
+ [
16
+ 16,
17
+ 1
18
+ ],
19
+ [
20
+ 20,
21
+ 0
22
+ ],
23
+ [
24
+ 23,
25
+ 4
26
+ ]
27
+ ],
28
+ "begin_suppress_tokens": [
29
+ 220,
30
+ 50257
31
+ ],
32
+ "bos_token_id": 50257,
33
+ "decoder_start_token_id": 50258,
34
+ "eos_token_id": 50257,
35
+ "is_multilingual": true,
36
+ "lang_to_id": {
37
+ "<|af|>": 50327,
38
+ "<|am|>": 50334,
39
+ "<|ar|>": 50272,
40
+ "<|as|>": 50350,
41
+ "<|az|>": 50304,
42
+ "<|ba|>": 50355,
43
+ "<|be|>": 50330,
44
+ "<|bg|>": 50292,
45
+ "<|bn|>": 50302,
46
+ "<|bo|>": 50347,
47
+ "<|br|>": 50309,
48
+ "<|bs|>": 50315,
49
+ "<|ca|>": 50270,
50
+ "<|cs|>": 50283,
51
+ "<|cy|>": 50297,
52
+ "<|da|>": 50285,
53
+ "<|de|>": 50261,
54
+ "<|el|>": 50281,
55
+ "<|en|>": 50259,
56
+ "<|es|>": 50262,
57
+ "<|et|>": 50307,
58
+ "<|eu|>": 50310,
59
+ "<|fa|>": 50300,
60
+ "<|fi|>": 50277,
61
+ "<|fo|>": 50338,
62
+ "<|fr|>": 50265,
63
+ "<|gl|>": 50319,
64
+ "<|gu|>": 50333,
65
+ "<|haw|>": 50352,
66
+ "<|ha|>": 50354,
67
+ "<|he|>": 50279,
68
+ "<|hi|>": 50276,
69
+ "<|hr|>": 50291,
70
+ "<|ht|>": 50339,
71
+ "<|hu|>": 50286,
72
+ "<|hy|>": 50312,
73
+ "<|id|>": 50275,
74
+ "<|is|>": 50311,
75
+ "<|it|>": 50274,
76
+ "<|ja|>": 50266,
77
+ "<|jw|>": 50356,
78
+ "<|ka|>": 50329,
79
+ "<|kk|>": 50316,
80
+ "<|km|>": 50323,
81
+ "<|kn|>": 50306,
82
+ "<|ko|>": 50264,
83
+ "<|la|>": 50294,
84
+ "<|lb|>": 50345,
85
+ "<|ln|>": 50353,
86
+ "<|lo|>": 50336,
87
+ "<|lt|>": 50293,
88
+ "<|lv|>": 50301,
89
+ "<|mg|>": 50349,
90
+ "<|mi|>": 50295,
91
+ "<|mk|>": 50308,
92
+ "<|ml|>": 50296,
93
+ "<|mn|>": 50314,
94
+ "<|mr|>": 50320,
95
+ "<|ms|>": 50282,
96
+ "<|mt|>": 50343,
97
+ "<|my|>": 50346,
98
+ "<|ne|>": 50313,
99
+ "<|nl|>": 50271,
100
+ "<|nn|>": 50342,
101
+ "<|no|>": 50288,
102
+ "<|oc|>": 50328,
103
+ "<|pa|>": 50321,
104
+ "<|pl|>": 50269,
105
+ "<|ps|>": 50340,
106
+ "<|pt|>": 50267,
107
+ "<|ro|>": 50284,
108
+ "<|ru|>": 50263,
109
+ "<|sa|>": 50344,
110
+ "<|sd|>": 50332,
111
+ "<|si|>": 50322,
112
+ "<|sk|>": 50298,
113
+ "<|sl|>": 50305,
114
+ "<|sn|>": 50324,
115
+ "<|so|>": 50326,
116
+ "<|sq|>": 50317,
117
+ "<|sr|>": 50303,
118
+ "<|su|>": 50357,
119
+ "<|sv|>": 50273,
120
+ "<|sw|>": 50318,
121
+ "<|ta|>": 50287,
122
+ "<|te|>": 50299,
123
+ "<|tg|>": 50331,
124
+ "<|th|>": 50289,
125
+ "<|tk|>": 50341,
126
+ "<|tl|>": 50348,
127
+ "<|tr|>": 50268,
128
+ "<|tt|>": 50351,
129
+ "<|uk|>": 50280,
130
+ "<|ur|>": 50290,
131
+ "<|uz|>": 50337,
132
+ "<|vi|>": 50278,
133
+ "<|yi|>": 50335,
134
+ "<|yo|>": 50325,
135
+ "<|zh|>": 50260
136
+ },
137
+ "language": "<|sv|>",
138
+ "max_initial_timestamp_index": 50,
139
+ "max_length": 448,
140
+ "no_timestamps_token_id": 50363,
141
+ "pad_token_id": 50257,
142
+ "prev_sot_token_id": 50361,
143
+ "return_timestamps": false,
144
+ "suppress_tokens": [
145
+ 1,
146
+ 2,
147
+ 7,
148
+ 8,
149
+ 9,
150
+ 10,
151
+ 14,
152
+ 25,
153
+ 26,
154
+ 27,
155
+ 28,
156
+ 29,
157
+ 31,
158
+ 58,
159
+ 59,
160
+ 60,
161
+ 61,
162
+ 62,
163
+ 63,
164
+ 90,
165
+ 91,
166
+ 92,
167
+ 93,
168
+ 359,
169
+ 503,
170
+ 522,
171
+ 542,
172
+ 873,
173
+ 893,
174
+ 902,
175
+ 918,
176
+ 922,
177
+ 931,
178
+ 1350,
179
+ 1853,
180
+ 1982,
181
+ 2460,
182
+ 2627,
183
+ 3246,
184
+ 3253,
185
+ 3268,
186
+ 3536,
187
+ 3846,
188
+ 3961,
189
+ 4183,
190
+ 4667,
191
+ 6585,
192
+ 6647,
193
+ 7273,
194
+ 9061,
195
+ 9383,
196
+ 10428,
197
+ 10929,
198
+ 11938,
199
+ 12033,
200
+ 12331,
201
+ 12562,
202
+ 13793,
203
+ 14157,
204
+ 14635,
205
+ 15265,
206
+ 15618,
207
+ 16553,
208
+ 16604,
209
+ 18362,
210
+ 18956,
211
+ 20075,
212
+ 21675,
213
+ 22520,
214
+ 26130,
215
+ 26161,
216
+ 26435,
217
+ 28279,
218
+ 29464,
219
+ 31650,
220
+ 32302,
221
+ 32470,
222
+ 36865,
223
+ 42863,
224
+ 47425,
225
+ 49870,
226
+ 50254,
227
+ 50258,
228
+ 50358,
229
+ 50359,
230
+ 50360,
231
+ 50361,
232
+ 50362
233
+ ],
234
+ "task": "transcribe",
235
+ "task_to_id": {
236
+ "transcribe": 50359,
237
+ "translate": 50358
238
+ },
239
+ "transformers_version": "4.45.2"
240
+ }
medium_quantized/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
medium_quantized/vocab.json ADDED
The diff for this file is too large to render. See raw diff