mickekring commited on
Commit
89cddf5
·
verified ·
1 Parent(s): 3365a26

Add small/TextDecoder.mlmodelc

Browse files
small/TextDecoder.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e0e20e0319e66a38bc3c43d77d0471d00543a2203432d6261ac2a13f992c6cc4
3
+ size 243
small/TextDecoder.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:810065a9f13fddd8c69660ad7dc1cb27da9b10afc751cd8dd95283fe65962437
3
+ size 635
small/TextDecoder.mlmodelc/metadata.json ADDED
@@ -0,0 +1,168 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 × 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 × 9216 × 1 × 1)",
21
+ "shortDescription" : "",
22
+ "shape" : "[1, 9216, 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 × 9216 × 1 × 1)",
31
+ "shortDescription" : "",
32
+ "shape" : "[1, 9216, 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
+ "Ios16.linear" : 1,
53
+ "Concat" : 3,
54
+ "Ios16.reduceMean" : 1,
55
+ "Ios16.mul" : 72,
56
+ "Ios16.layerNorm" : 37,
57
+ "SliceByIndex" : 20,
58
+ "Ios16.sub" : 1,
59
+ "Transpose" : 1,
60
+ "Ios16.conv" : 120,
61
+ "Ios16.add" : 73,
62
+ "Squeeze" : 1,
63
+ "Ios16.matmul" : 48,
64
+ "Ios16.softmax" : 24,
65
+ "Ios16.gelu" : 12,
66
+ "ExpandDims" : 6,
67
+ "Ios16.batchNorm" : 37,
68
+ "Split" : 2,
69
+ "Ios16.gather" : 2,
70
+ "Ios16.reshape" : 96
71
+ },
72
+ "computePrecision" : "Mixed (Float16, Int32)",
73
+ "isUpdatable" : "0",
74
+ "stateSchema" : [
75
+
76
+ ],
77
+ "availability" : {
78
+ "macOS" : "13.0",
79
+ "tvOS" : "16.0",
80
+ "visionOS" : "1.0",
81
+ "watchOS" : "9.0",
82
+ "iOS" : "16.0",
83
+ "macCatalyst" : "16.0"
84
+ },
85
+ "modelType" : {
86
+ "name" : "MLModelType_mlProgram"
87
+ },
88
+ "userDefinedMetadata" : {
89
+ "com.github.apple.coremltools.source_dialect" : "TorchScript",
90
+ "com.github.apple.coremltools.source" : "torch==2.6.0",
91
+ "com.github.apple.coremltools.version" : "8.3.0"
92
+ },
93
+ "inputSchema" : [
94
+ {
95
+ "hasShapeFlexibility" : "0",
96
+ "isOptional" : "0",
97
+ "dataType" : "Int32",
98
+ "formattedType" : "MultiArray (Int32 1)",
99
+ "shortDescription" : "",
100
+ "shape" : "[1]",
101
+ "name" : "input_ids",
102
+ "type" : "MultiArray"
103
+ },
104
+ {
105
+ "hasShapeFlexibility" : "0",
106
+ "isOptional" : "0",
107
+ "dataType" : "Int32",
108
+ "formattedType" : "MultiArray (Int32 1)",
109
+ "shortDescription" : "",
110
+ "shape" : "[1]",
111
+ "name" : "cache_length",
112
+ "type" : "MultiArray"
113
+ },
114
+ {
115
+ "hasShapeFlexibility" : "0",
116
+ "isOptional" : "0",
117
+ "dataType" : "Float16",
118
+ "formattedType" : "MultiArray (Float16 1 × 9216 × 1 × 448)",
119
+ "shortDescription" : "",
120
+ "shape" : "[1, 9216, 1, 448]",
121
+ "name" : "key_cache",
122
+ "type" : "MultiArray"
123
+ },
124
+ {
125
+ "hasShapeFlexibility" : "0",
126
+ "isOptional" : "0",
127
+ "dataType" : "Float16",
128
+ "formattedType" : "MultiArray (Float16 1 × 9216 × 1 × 448)",
129
+ "shortDescription" : "",
130
+ "shape" : "[1, 9216, 1, 448]",
131
+ "name" : "value_cache",
132
+ "type" : "MultiArray"
133
+ },
134
+ {
135
+ "hasShapeFlexibility" : "0",
136
+ "isOptional" : "0",
137
+ "dataType" : "Float16",
138
+ "formattedType" : "MultiArray (Float16 1 × 448)",
139
+ "shortDescription" : "",
140
+ "shape" : "[1, 448]",
141
+ "name" : "kv_cache_update_mask",
142
+ "type" : "MultiArray"
143
+ },
144
+ {
145
+ "hasShapeFlexibility" : "0",
146
+ "isOptional" : "0",
147
+ "dataType" : "Float16",
148
+ "formattedType" : "MultiArray (Float16 1 × 768 × 1 × 1500)",
149
+ "shortDescription" : "",
150
+ "shape" : "[1, 768, 1, 1500]",
151
+ "name" : "encoder_output_embeds",
152
+ "type" : "MultiArray"
153
+ },
154
+ {
155
+ "hasShapeFlexibility" : "0",
156
+ "isOptional" : "0",
157
+ "dataType" : "Float16",
158
+ "formattedType" : "MultiArray (Float16 1 × 448)",
159
+ "shortDescription" : "",
160
+ "shape" : "[1, 448]",
161
+ "name" : "decoder_key_padding_mask",
162
+ "type" : "MultiArray"
163
+ }
164
+ ],
165
+ "generatedClassName" : "TextDecoder",
166
+ "method" : "predict"
167
+ }
168
+ ]
small/TextDecoder.mlmodelc/model.mil ADDED
The diff for this file is too large to render. See raw diff
 
small/TextDecoder.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0a817dd347e3f0301a3325803afa75f497ad51defa98d40edbf946fe88ceb2fc
3
+ size 307287346