NorbertKlockiewicz commited on
Commit
193632d
·
1 Parent(s): c16e1a0

Backfill constant metadata method values in config.json from .pte

Browse files
e2b/mlx/config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
3
+ "model": "gemma_4",
4
+ "family": "gemma",
5
+ "capabilities": [
6
+ "text-generation"
7
+ ],
8
+ "backend": "mlx",
9
+ "license": "gemma",
10
+ "variants": [
11
+ {
12
+ "file": "gemma4_e2b_mlx_int4.pte",
13
+ "precision": "int4",
14
+ "quantized": true,
15
+ "default": true,
16
+ "methods": {
17
+ "enable_dynamic_shape": true,
18
+ "get_bos_id": 2,
19
+ "get_eos_ids": [
20
+ 1,
21
+ 106,
22
+ 50
23
+ ],
24
+ "get_max_context_len": 2048,
25
+ "get_max_seq_len": 2048,
26
+ "use_kv_cache": true,
27
+ "forward": {
28
+ "inputs": [],
29
+ "outputs": []
30
+ }
31
+ }
32
+ }
33
+ ],
34
+ "size": "e2b",
35
+ "_methods_needing_shapes": [
36
+ "forward"
37
+ ]
38
+ }
e2b/vulkan/config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
3
+ "model": "gemma_4",
4
+ "family": "gemma",
5
+ "capabilities": [
6
+ "text-generation"
7
+ ],
8
+ "backend": "vulkan",
9
+ "license": "gemma",
10
+ "variants": [
11
+ {
12
+ "file": "gemma_4_e2b_vulkan_8da4w.pte",
13
+ "precision": "8da4w",
14
+ "quantized": true,
15
+ "default": true,
16
+ "methods": {
17
+ "enable_dynamic_shape": true,
18
+ "get_bos_id": 2,
19
+ "get_eos_ids": [
20
+ 1,
21
+ 106,
22
+ 50
23
+ ],
24
+ "get_max_context_len": 2048,
25
+ "get_max_seq_len": 128,
26
+ "get_vocab_size": 262144,
27
+ "use_kv_cache": true,
28
+ "forward": {
29
+ "inputs": [],
30
+ "outputs": []
31
+ }
32
+ }
33
+ }
34
+ ],
35
+ "size": "e2b",
36
+ "_methods_needing_shapes": [
37
+ "forward"
38
+ ]
39
+ }
e2b/xnnpack/config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
3
+ "model": "gemma_4",
4
+ "family": "gemma",
5
+ "capabilities": [
6
+ "text-generation"
7
+ ],
8
+ "backend": "xnnpack",
9
+ "license": "gemma",
10
+ "variants": [
11
+ {
12
+ "file": "gemma_4_e2b_xnnpack_8da4w.pte",
13
+ "precision": "8da4w",
14
+ "quantized": true,
15
+ "default": true,
16
+ "methods": {
17
+ "enable_dynamic_shape": true,
18
+ "get_bos_id": 2,
19
+ "get_eos_ids": [
20
+ 1,
21
+ 106,
22
+ 50
23
+ ],
24
+ "get_max_context_len": 4048,
25
+ "get_max_seq_len": 4048,
26
+ "get_vocab_size": 262144,
27
+ "use_kv_cache": true,
28
+ "forward": {
29
+ "inputs": [],
30
+ "outputs": []
31
+ }
32
+ }
33
+ }
34
+ ],
35
+ "size": "e2b",
36
+ "_methods_needing_shapes": [
37
+ "forward"
38
+ ]
39
+ }