IgorSwat commited on
Commit
31de14c
·
1 Parent(s): 9adc4c0

Add MLX model

Browse files
.gitattributes CHANGED
@@ -36,3 +36,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
36
  xnnpack/whisper_tiny_xnnpack.pte filter=lfs diff=lfs merge=lfs -text
37
  xnnpack/whisper_tiny_xnnpack_fp32.pte filter=lfs diff=lfs merge=lfs -text
38
  coreml/whisper_tiny_coreml_fp16.pte filter=lfs diff=lfs merge=lfs -text
 
 
36
  xnnpack/whisper_tiny_xnnpack.pte filter=lfs diff=lfs merge=lfs -text
37
  xnnpack/whisper_tiny_xnnpack_fp32.pte filter=lfs diff=lfs merge=lfs -text
38
  coreml/whisper_tiny_coreml_fp16.pte filter=lfs diff=lfs merge=lfs -text
39
+ mlx/whisper_tiny_mlx_bf16.pte filter=lfs diff=lfs merge=lfs -text
mlx/config.json ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
3
+ "model": "whisper_tiny",
4
+ "family": "whisper",
5
+ "capabilities": [
6
+ "speech-to-text"
7
+ ],
8
+ "backend": "mlx",
9
+ "license": "apache-2.0",
10
+ "tokenizer": "../tokenizer.json",
11
+ "variants": [
12
+ {
13
+ "file": "whisper_tiny_mlx_bf16.pte",
14
+ "precision": "bf16",
15
+ "quantized": false,
16
+ "default": true,
17
+ "methods": {
18
+ "decode": {
19
+ "inputs": [
20
+ {
21
+ "shape": [
22
+ 1,
23
+ 128
24
+ ],
25
+ "dtype": "int64"
26
+ },
27
+ {
28
+ "shape": [
29
+ 128
30
+ ],
31
+ "dtype": "int64"
32
+ },
33
+ {
34
+ "shape": [
35
+ 1,
36
+ 1500,
37
+ 384
38
+ ],
39
+ "dtype": "float32"
40
+ }
41
+ ],
42
+ "outputs": [
43
+ {
44
+ "shape": [
45
+ 1,
46
+ 128,
47
+ 51865
48
+ ],
49
+ "dtype": "float32"
50
+ }
51
+ ]
52
+ },
53
+ "encode": {
54
+ "inputs": [
55
+ {
56
+ "shape": [
57
+ 480000
58
+ ],
59
+ "dtype": "float32"
60
+ }
61
+ ],
62
+ "outputs": [
63
+ {
64
+ "shape": [
65
+ 1,
66
+ 1500,
67
+ 384
68
+ ],
69
+ "dtype": "float32"
70
+ }
71
+ ]
72
+ }
73
+ }
74
+ }
75
+ ]
76
+ }
mlx/whisper_tiny_mlx_bf16.pte ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:35c52e6e99b78054e62342a5500c9d548e7df60fb049db7d6b93906400122897
3
+ size 76425216