Upload Sequoia model
Browse filesAdds a multi-function 512 context model.
This view is limited to 50 files because it contains too many changes.
See raw diff
- sequoia/Llama-2-7b-hf_chunk1.mlmodelc/analytics/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk1.mlmodelc/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk1.mlmodelc/metadata.json +274 -0
- sequoia/Llama-2-7b-hf_chunk1.mlmodelc/model.mil +98 -0
- sequoia/Llama-2-7b-hf_chunk1.mlmodelc/weights/weight.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk10.mlmodelc/analytics/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk10.mlmodelc/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk10.mlmodelc/metadata.json +500 -0
- sequoia/Llama-2-7b-hf_chunk10.mlmodelc/model.mil +0 -0
- sequoia/Llama-2-7b-hf_chunk10.mlmodelc/weights/weight.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk11.mlmodelc/analytics/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk11.mlmodelc/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk11.mlmodelc/metadata.json +500 -0
- sequoia/Llama-2-7b-hf_chunk11.mlmodelc/model.mil +0 -0
- sequoia/Llama-2-7b-hf_chunk11.mlmodelc/weights/weight.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk12.mlmodelc/analytics/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk12.mlmodelc/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk12.mlmodelc/metadata.json +423 -0
- sequoia/Llama-2-7b-hf_chunk12.mlmodelc/model.mil +0 -0
- sequoia/Llama-2-7b-hf_chunk12.mlmodelc/weights/weight.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk2.mlmodelc/analytics/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk2.mlmodelc/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk2.mlmodelc/metadata.json +500 -0
- sequoia/Llama-2-7b-hf_chunk2.mlmodelc/model.mil +0 -0
- sequoia/Llama-2-7b-hf_chunk2.mlmodelc/weights/weight.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk3.mlmodelc/analytics/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk3.mlmodelc/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk3.mlmodelc/metadata.json +500 -0
- sequoia/Llama-2-7b-hf_chunk3.mlmodelc/model.mil +0 -0
- sequoia/Llama-2-7b-hf_chunk3.mlmodelc/weights/weight.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk4.mlmodelc/analytics/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk4.mlmodelc/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk4.mlmodelc/metadata.json +500 -0
- sequoia/Llama-2-7b-hf_chunk4.mlmodelc/model.mil +0 -0
- sequoia/Llama-2-7b-hf_chunk4.mlmodelc/weights/weight.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk5.mlmodelc/analytics/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk5.mlmodelc/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk5.mlmodelc/metadata.json +500 -0
- sequoia/Llama-2-7b-hf_chunk5.mlmodelc/model.mil +0 -0
- sequoia/Llama-2-7b-hf_chunk5.mlmodelc/weights/weight.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk6.mlmodelc/analytics/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk6.mlmodelc/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk6.mlmodelc/metadata.json +500 -0
- sequoia/Llama-2-7b-hf_chunk6.mlmodelc/model.mil +0 -0
- sequoia/Llama-2-7b-hf_chunk6.mlmodelc/weights/weight.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk7.mlmodelc/analytics/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk7.mlmodelc/coremldata.bin +3 -0
- sequoia/Llama-2-7b-hf_chunk7.mlmodelc/metadata.json +500 -0
- sequoia/Llama-2-7b-hf_chunk7.mlmodelc/model.mil +0 -0
- sequoia/Llama-2-7b-hf_chunk7.mlmodelc/weights/weight.bin +3 -0
sequoia/Llama-2-7b-hf_chunk1.mlmodelc/analytics/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e729e06a5dac91d54425432e10c01d40645eefd035e7d3569e6aaf5acc4a1493
|
3 |
+
size 243
|
sequoia/Llama-2-7b-hf_chunk1.mlmodelc/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1a55bcffcb4e191cd6358ad92d705948cd757010e873528f66b6e21943904acd
|
3 |
+
size 485
|
sequoia/Llama-2-7b-hf_chunk1.mlmodelc/metadata.json
ADDED
@@ -0,0 +1,274 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 × 4096 × 1 × 512)",
|
11 |
+
"shortDescription" : "",
|
12 |
+
"shape" : "[1, 4096, 1, 512]",
|
13 |
+
"name" : "x",
|
14 |
+
"type" : "MultiArray"
|
15 |
+
},
|
16 |
+
{
|
17 |
+
"hasShapeFlexibility" : "0",
|
18 |
+
"isOptional" : "0",
|
19 |
+
"dataType" : "Float16",
|
20 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
21 |
+
"shortDescription" : "",
|
22 |
+
"shape" : "[128, 512]",
|
23 |
+
"name" : "cos",
|
24 |
+
"type" : "MultiArray"
|
25 |
+
},
|
26 |
+
{
|
27 |
+
"hasShapeFlexibility" : "0",
|
28 |
+
"isOptional" : "0",
|
29 |
+
"dataType" : "Float16",
|
30 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
31 |
+
"shortDescription" : "",
|
32 |
+
"shape" : "[128, 512]",
|
33 |
+
"name" : "sin",
|
34 |
+
"type" : "MultiArray"
|
35 |
+
},
|
36 |
+
{
|
37 |
+
"hasShapeFlexibility" : "0",
|
38 |
+
"isOptional" : "0",
|
39 |
+
"dataType" : "Float16",
|
40 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
41 |
+
"shortDescription" : "",
|
42 |
+
"shape" : "[1, 1, 512, 512]",
|
43 |
+
"name" : "mask",
|
44 |
+
"type" : "MultiArray"
|
45 |
+
}
|
46 |
+
],
|
47 |
+
"modelParameters" : [
|
48 |
+
|
49 |
+
],
|
50 |
+
"specificationVersion" : 9,
|
51 |
+
"functions" : [
|
52 |
+
{
|
53 |
+
"inputSchema" : [
|
54 |
+
{
|
55 |
+
"hasShapeFlexibility" : "0",
|
56 |
+
"isOptional" : "0",
|
57 |
+
"dataType" : "Int32",
|
58 |
+
"formattedType" : "MultiArray (Int32 1 × 512)",
|
59 |
+
"shortDescription" : "",
|
60 |
+
"shape" : "[1, 512]",
|
61 |
+
"name" : "input_ids",
|
62 |
+
"type" : "MultiArray"
|
63 |
+
},
|
64 |
+
{
|
65 |
+
"hasShapeFlexibility" : "0",
|
66 |
+
"isOptional" : "0",
|
67 |
+
"dataType" : "Int32",
|
68 |
+
"formattedType" : "MultiArray (Int32 1)",
|
69 |
+
"shortDescription" : "",
|
70 |
+
"shape" : "[1]",
|
71 |
+
"name" : "full_sequence_length",
|
72 |
+
"type" : "MultiArray"
|
73 |
+
}
|
74 |
+
],
|
75 |
+
"computePrecision" : "Mixed (Float16, Int16, Int32, UInt16)",
|
76 |
+
"storagePrecision" : "Float16",
|
77 |
+
"stateSchema" : [
|
78 |
+
|
79 |
+
],
|
80 |
+
"outputSchema" : [
|
81 |
+
{
|
82 |
+
"hasShapeFlexibility" : "0",
|
83 |
+
"isOptional" : "0",
|
84 |
+
"dataType" : "Float16",
|
85 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
86 |
+
"shortDescription" : "",
|
87 |
+
"shape" : "[1, 4096, 1, 512]",
|
88 |
+
"name" : "x",
|
89 |
+
"type" : "MultiArray"
|
90 |
+
},
|
91 |
+
{
|
92 |
+
"hasShapeFlexibility" : "0",
|
93 |
+
"isOptional" : "0",
|
94 |
+
"dataType" : "Float16",
|
95 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
96 |
+
"shortDescription" : "",
|
97 |
+
"shape" : "[128, 512]",
|
98 |
+
"name" : "cos",
|
99 |
+
"type" : "MultiArray"
|
100 |
+
},
|
101 |
+
{
|
102 |
+
"hasShapeFlexibility" : "0",
|
103 |
+
"isOptional" : "0",
|
104 |
+
"dataType" : "Float16",
|
105 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
106 |
+
"shortDescription" : "",
|
107 |
+
"shape" : "[128, 512]",
|
108 |
+
"name" : "sin",
|
109 |
+
"type" : "MultiArray"
|
110 |
+
},
|
111 |
+
{
|
112 |
+
"hasShapeFlexibility" : "0",
|
113 |
+
"isOptional" : "0",
|
114 |
+
"dataType" : "Float16",
|
115 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
116 |
+
"shortDescription" : "",
|
117 |
+
"shape" : "[1, 1, 512, 512]",
|
118 |
+
"name" : "mask",
|
119 |
+
"type" : "MultiArray"
|
120 |
+
}
|
121 |
+
],
|
122 |
+
"name" : "input_512_context_512",
|
123 |
+
"mlProgramOperationTypeHistogram" : {
|
124 |
+
"Select" : 2,
|
125 |
+
"Ios18.maximum" : 1,
|
126 |
+
"Ios18.gather" : 3,
|
127 |
+
"Ios18.sub" : 3,
|
128 |
+
"Ios18.transpose" : 1,
|
129 |
+
"Ios18.less" : 2,
|
130 |
+
"Ios18.cast" : 2,
|
131 |
+
"Ios18.expandDims" : 4,
|
132 |
+
"Tile" : 2
|
133 |
+
}
|
134 |
+
},
|
135 |
+
{
|
136 |
+
"inputSchema" : [
|
137 |
+
{
|
138 |
+
"hasShapeFlexibility" : "0",
|
139 |
+
"isOptional" : "0",
|
140 |
+
"dataType" : "Int32",
|
141 |
+
"formattedType" : "MultiArray (Int32 1 × 1)",
|
142 |
+
"shortDescription" : "",
|
143 |
+
"shape" : "[1, 1]",
|
144 |
+
"name" : "input_ids",
|
145 |
+
"type" : "MultiArray"
|
146 |
+
},
|
147 |
+
{
|
148 |
+
"hasShapeFlexibility" : "0",
|
149 |
+
"isOptional" : "0",
|
150 |
+
"dataType" : "Int32",
|
151 |
+
"formattedType" : "MultiArray (Int32 1)",
|
152 |
+
"shortDescription" : "",
|
153 |
+
"shape" : "[1]",
|
154 |
+
"name" : "full_sequence_length",
|
155 |
+
"type" : "MultiArray"
|
156 |
+
}
|
157 |
+
],
|
158 |
+
"computePrecision" : "Mixed (Float16, Int16, Int32, UInt16)",
|
159 |
+
"storagePrecision" : "Float16",
|
160 |
+
"stateSchema" : [
|
161 |
+
|
162 |
+
],
|
163 |
+
"outputSchema" : [
|
164 |
+
{
|
165 |
+
"hasShapeFlexibility" : "0",
|
166 |
+
"isOptional" : "0",
|
167 |
+
"dataType" : "Float16",
|
168 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 1)",
|
169 |
+
"shortDescription" : "",
|
170 |
+
"shape" : "[1, 4096, 1, 1]",
|
171 |
+
"name" : "x",
|
172 |
+
"type" : "MultiArray"
|
173 |
+
},
|
174 |
+
{
|
175 |
+
"hasShapeFlexibility" : "0",
|
176 |
+
"isOptional" : "0",
|
177 |
+
"dataType" : "Float16",
|
178 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
179 |
+
"shortDescription" : "",
|
180 |
+
"shape" : "[128, 1]",
|
181 |
+
"name" : "cos",
|
182 |
+
"type" : "MultiArray"
|
183 |
+
},
|
184 |
+
{
|
185 |
+
"hasShapeFlexibility" : "0",
|
186 |
+
"isOptional" : "0",
|
187 |
+
"dataType" : "Float16",
|
188 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
189 |
+
"shortDescription" : "",
|
190 |
+
"shape" : "[128, 1]",
|
191 |
+
"name" : "sin",
|
192 |
+
"type" : "MultiArray"
|
193 |
+
},
|
194 |
+
{
|
195 |
+
"hasShapeFlexibility" : "0",
|
196 |
+
"isOptional" : "0",
|
197 |
+
"dataType" : "Float16",
|
198 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 1 × 512)",
|
199 |
+
"shortDescription" : "",
|
200 |
+
"shape" : "[1, 1, 1, 512]",
|
201 |
+
"name" : "mask",
|
202 |
+
"type" : "MultiArray"
|
203 |
+
}
|
204 |
+
],
|
205 |
+
"name" : "input_1_context_512",
|
206 |
+
"mlProgramOperationTypeHistogram" : {
|
207 |
+
"Select" : 1,
|
208 |
+
"Ios18.maximum" : 1,
|
209 |
+
"Ios18.gather" : 3,
|
210 |
+
"Ios18.sub" : 3,
|
211 |
+
"Ios18.transpose" : 1,
|
212 |
+
"Ios18.less" : 1,
|
213 |
+
"Ios18.cast" : 2,
|
214 |
+
"Ios18.expandDims" : 4
|
215 |
+
}
|
216 |
+
}
|
217 |
+
],
|
218 |
+
"mlProgramOperationTypeHistogram" : {
|
219 |
+
"Select" : 2,
|
220 |
+
"Ios18.maximum" : 1,
|
221 |
+
"Ios18.gather" : 3,
|
222 |
+
"Ios18.sub" : 3,
|
223 |
+
"Ios18.transpose" : 1,
|
224 |
+
"Ios18.less" : 2,
|
225 |
+
"Ios18.cast" : 2,
|
226 |
+
"Ios18.expandDims" : 4,
|
227 |
+
"Tile" : 2
|
228 |
+
},
|
229 |
+
"isUpdatable" : "0",
|
230 |
+
"stateSchema" : [
|
231 |
+
|
232 |
+
],
|
233 |
+
"availability" : {
|
234 |
+
"macOS" : "15.0",
|
235 |
+
"tvOS" : "18.0",
|
236 |
+
"visionOS" : "2.0",
|
237 |
+
"watchOS" : "11.0",
|
238 |
+
"iOS" : "18.0",
|
239 |
+
"macCatalyst" : "18.0"
|
240 |
+
},
|
241 |
+
"computePrecision" : "Mixed (Float16, Int16, Int32, UInt16)",
|
242 |
+
"modelType" : {
|
243 |
+
"name" : "MLModelType_mlProgram"
|
244 |
+
},
|
245 |
+
"inputSchema" : [
|
246 |
+
{
|
247 |
+
"hasShapeFlexibility" : "0",
|
248 |
+
"isOptional" : "0",
|
249 |
+
"dataType" : "Int32",
|
250 |
+
"formattedType" : "MultiArray (Int32 1 × 512)",
|
251 |
+
"shortDescription" : "",
|
252 |
+
"shape" : "[1, 512]",
|
253 |
+
"name" : "input_ids",
|
254 |
+
"type" : "MultiArray"
|
255 |
+
},
|
256 |
+
{
|
257 |
+
"hasShapeFlexibility" : "0",
|
258 |
+
"isOptional" : "0",
|
259 |
+
"dataType" : "Int32",
|
260 |
+
"formattedType" : "MultiArray (Int32 1)",
|
261 |
+
"shortDescription" : "",
|
262 |
+
"shape" : "[1]",
|
263 |
+
"name" : "full_sequence_length",
|
264 |
+
"type" : "MultiArray"
|
265 |
+
}
|
266 |
+
],
|
267 |
+
"defaultFunctionName" : "input_512_context_512",
|
268 |
+
"generatedClassName" : "Llama_2_7b_hf_2024_07_02_20_36_17_merged_chunk1",
|
269 |
+
"userDefinedMetadata" : {
|
270 |
+
|
271 |
+
},
|
272 |
+
"method" : "predict"
|
273 |
+
}
|
274 |
+
]
|
sequoia/Llama-2-7b-hf_chunk1.mlmodelc/model.mil
ADDED
@@ -0,0 +1,98 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
program(1.3)
|
2 |
+
[buildInfo = dict<string, string>({{"coremlc-component-MIL", "3400.34.1"}, {"coremlc-version", "3400.42.1"}})]
|
3 |
+
{
|
4 |
+
func input_1_context_512<ios18>(tensor<int32, [1]> full_sequence_length, tensor<int32, [1, 1]> input_ids) {
|
5 |
+
tensor<int32, [1]> T = const()[name = string("T"), val = tensor<int32, [1]>([1])];
|
6 |
+
int32 x_axis_0 = const()[name = string("x_axis_0"), val = int32(0)];
|
7 |
+
int32 x_batch_dims_0 = const()[name = string("x_batch_dims_0"), val = int32(0)];
|
8 |
+
bool x_validate_indices_0 = const()[name = string("x_validate_indices_0"), val = bool(false)];
|
9 |
+
tensor<fp16, [32000, 4096]> wte_weight_to_fp16 = const()[name = string("wte_weight_to_fp16"), val = tensor<fp16, [32000, 4096]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))];
|
10 |
+
string input_ids_to_int16_dtype_0 = const()[name = string("input_ids_to_int16_dtype_0"), val = string("int16")];
|
11 |
+
tensor<int16, [1, 1]> input_ids_to_int16 = cast(dtype = input_ids_to_int16_dtype_0, x = input_ids)[name = string("cast_6")];
|
12 |
+
tensor<fp16, [1, 1, 4096]> x_cast_fp16_cast_uint16 = gather(axis = x_axis_0, batch_dims = x_batch_dims_0, indices = input_ids_to_int16, validate_indices = x_validate_indices_0, x = wte_weight_to_fp16)[name = string("x_cast_fp16_cast_uint16")];
|
13 |
+
tensor<int32, [3]> var_16_perm_0 = const()[name = string("op_16_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
|
14 |
+
tensor<int32, [1]> var_18_axes_0 = const()[name = string("op_18_axes_0"), val = tensor<int32, [1]>([2])];
|
15 |
+
tensor<fp16, [1, 4096, 1]> var_16_cast_fp16 = transpose(perm = var_16_perm_0, x = x_cast_fp16_cast_uint16)[name = string("transpose_0")];
|
16 |
+
tensor<fp16, [1, 4096, 1, 1]> x = expand_dims(axes = var_18_axes_0, x = var_16_cast_fp16)[name = string("op_18_cast_fp16")];
|
17 |
+
tensor<int32, [1]> pos_offset = sub(x = T, y = full_sequence_length)[name = string("pos_offset")];
|
18 |
+
tensor<int32, [1]> var_26 = const()[name = string("op_26"), val = tensor<int32, [1]>([0])];
|
19 |
+
tensor<int32, [1]> input_pos_1 = sub(x = var_26, y = pos_offset)[name = string("input_pos_1")];
|
20 |
+
tensor<int32, [1]> var_34 = const()[name = string("op_34"), val = tensor<int32, [1]>([0])];
|
21 |
+
tensor<int32, [1]> input_pos = maximum(x = input_pos_1, y = var_34)[name = string("input_pos")];
|
22 |
+
int32 var_45 = const()[name = string("op_45"), val = int32(1)];
|
23 |
+
int32 var_46_batch_dims_0 = const()[name = string("op_46_batch_dims_0"), val = int32(0)];
|
24 |
+
bool var_46_validate_indices_0 = const()[name = string("op_46_validate_indices_0"), val = bool(false)];
|
25 |
+
tensor<fp16, [128, 512]> var_44_to_fp16 = const()[name = string("op_44_to_fp16"), val = tensor<fp16, [128, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(262144128)))];
|
26 |
+
string input_pos_to_uint16_dtype_0 = const()[name = string("input_pos_to_uint16_dtype_0"), val = string("uint16")];
|
27 |
+
tensor<uint16, [1]> input_pos_to_uint16 = cast(dtype = input_pos_to_uint16_dtype_0, x = input_pos)[name = string("cast_5")];
|
28 |
+
tensor<fp16, [128, 1]> cos = gather(axis = var_45, batch_dims = var_46_batch_dims_0, indices = input_pos_to_uint16, validate_indices = var_46_validate_indices_0, x = var_44_to_fp16)[name = string("op_46_cast_fp16_cast_uint16")];
|
29 |
+
int32 var_56 = const()[name = string("op_56"), val = int32(1)];
|
30 |
+
int32 var_57_batch_dims_0 = const()[name = string("op_57_batch_dims_0"), val = int32(0)];
|
31 |
+
bool var_57_validate_indices_0 = const()[name = string("op_57_validate_indices_0"), val = bool(false)];
|
32 |
+
tensor<fp16, [128, 512]> var_55_to_fp16 = const()[name = string("op_55_to_fp16"), val = tensor<fp16, [128, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(262275264)))];
|
33 |
+
tensor<fp16, [128, 1]> sin = gather(axis = var_56, batch_dims = var_57_batch_dims_0, indices = input_pos_to_uint16, validate_indices = var_57_validate_indices_0, x = var_55_to_fp16)[name = string("op_57_cast_fp16_cast_uint16")];
|
34 |
+
tensor<int32, [512]> var_104 = const()[name = string("op_104"), val = tensor<int32, [512]>([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 470, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 484, 485, 486, 487, 488, 489, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511])];
|
35 |
+
int32 var_105 = const()[name = string("op_105"), val = int32(512)];
|
36 |
+
tensor<int32, [1]> var_107 = sub(x = var_105, y = full_sequence_length)[name = string("op_107")];
|
37 |
+
tensor<bool, [512]> var_108 = less(x = var_104, y = var_107)[name = string("op_108")];
|
38 |
+
tensor<int32, [1]> expand_dims_0_axes_0 = const()[name = string("expand_dims_0_axes_0"), val = tensor<int32, [1]>([0])];
|
39 |
+
tensor<bool, [1, 512]> expand_dims_0 = expand_dims(axes = expand_dims_0_axes_0, x = var_108)[name = string("expand_dims_0")];
|
40 |
+
tensor<fp16, [1, 512]> all_mask_to_fp16 = const()[name = string("all_mask_to_fp16"), val = tensor<fp16, [1, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(263455104)))];
|
41 |
+
tensor<fp16, [1, 512]> m_1_to_fp16 = const()[name = string("m_1_to_fp16"), val = tensor<fp16, [1, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(263456192)))];
|
42 |
+
tensor<fp16, [1, 512]> m_cast_fp16 = select(a = all_mask_to_fp16, b = m_1_to_fp16, cond = expand_dims_0)[name = string("m_cast_fp16")];
|
43 |
+
tensor<int32, [1]> var_111_axes_0 = const()[name = string("op_111_axes_0"), val = tensor<int32, [1]>([0])];
|
44 |
+
tensor<fp16, [1, 1, 512]> var_111_cast_fp16 = expand_dims(axes = var_111_axes_0, x = m_cast_fp16)[name = string("op_111_cast_fp16")];
|
45 |
+
tensor<int32, [1]> var_113_axes_0 = const()[name = string("op_113_axes_0"), val = tensor<int32, [1]>([0])];
|
46 |
+
tensor<fp16, [1, 1, 1, 512]> mask = expand_dims(axes = var_113_axes_0, x = var_111_cast_fp16)[name = string("op_113_cast_fp16")];
|
47 |
+
} -> (x, cos, sin, mask);
|
48 |
+
func input_512_context_512<ios18>(tensor<int32, [1]> full_sequence_length, tensor<int32, [1, 512]> input_ids) {
|
49 |
+
tensor<int32, [1]> T = const()[name = string("T"), val = tensor<int32, [1]>([512])];
|
50 |
+
int32 x_axis_0 = const()[name = string("x_axis_0"), val = int32(0)];
|
51 |
+
int32 x_batch_dims_0 = const()[name = string("x_batch_dims_0"), val = int32(0)];
|
52 |
+
bool x_validate_indices_0 = const()[name = string("x_validate_indices_0"), val = bool(false)];
|
53 |
+
tensor<fp16, [32000, 4096]> wte_weight_to_fp16 = const()[name = string("wte_weight_to_fp16"), val = tensor<fp16, [32000, 4096]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))];
|
54 |
+
string input_ids_to_int16_dtype_0 = const()[name = string("input_ids_to_int16_dtype_0"), val = string("int16")];
|
55 |
+
tensor<int16, [1, 512]> input_ids_to_int16 = cast(dtype = input_ids_to_int16_dtype_0, x = input_ids)[name = string("cast_6")];
|
56 |
+
tensor<fp16, [1, 512, 4096]> x_cast_fp16_cast_uint16 = gather(axis = x_axis_0, batch_dims = x_batch_dims_0, indices = input_ids_to_int16, validate_indices = x_validate_indices_0, x = wte_weight_to_fp16)[name = string("x_cast_fp16_cast_uint16")];
|
57 |
+
tensor<int32, [3]> var_16_perm_0 = const()[name = string("op_16_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
|
58 |
+
tensor<int32, [1]> var_18_axes_0 = const()[name = string("op_18_axes_0"), val = tensor<int32, [1]>([2])];
|
59 |
+
tensor<fp16, [1, 4096, 512]> var_16_cast_fp16 = transpose(perm = var_16_perm_0, x = x_cast_fp16_cast_uint16)[name = string("transpose_0")];
|
60 |
+
tensor<fp16, [1, 4096, 1, 512]> x = expand_dims(axes = var_18_axes_0, x = var_16_cast_fp16)[name = string("op_18_cast_fp16")];
|
61 |
+
tensor<int32, [1]> pos_offset = sub(x = T, y = full_sequence_length)[name = string("pos_offset")];
|
62 |
+
tensor<int32, [512]> var_26 = const()[name = string("op_26"), val = tensor<int32, [512]>([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 470, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 484, 485, 486, 487, 488, 489, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511])];
|
63 |
+
tensor<int32, [512]> input_pos_1 = sub(x = var_26, y = pos_offset)[name = string("input_pos_1")];
|
64 |
+
tensor<int32, [512]> var_34 = const()[name = string("op_34"), val = tensor<int32, [512]>([0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0])];
|
65 |
+
tensor<int32, [512]> input_pos = maximum(x = input_pos_1, y = var_34)[name = string("input_pos")];
|
66 |
+
int32 var_45 = const()[name = string("op_45"), val = int32(1)];
|
67 |
+
int32 var_46_batch_dims_0 = const()[name = string("op_46_batch_dims_0"), val = int32(0)];
|
68 |
+
bool var_46_validate_indices_0 = const()[name = string("op_46_validate_indices_0"), val = bool(false)];
|
69 |
+
tensor<fp16, [128, 512]> var_44_to_fp16 = const()[name = string("op_44_to_fp16"), val = tensor<fp16, [128, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(262144128)))];
|
70 |
+
string input_pos_to_uint16_dtype_0 = const()[name = string("input_pos_to_uint16_dtype_0"), val = string("uint16")];
|
71 |
+
tensor<uint16, [512]> input_pos_to_uint16 = cast(dtype = input_pos_to_uint16_dtype_0, x = input_pos)[name = string("cast_5")];
|
72 |
+
tensor<fp16, [128, 512]> cos = gather(axis = var_45, batch_dims = var_46_batch_dims_0, indices = input_pos_to_uint16, validate_indices = var_46_validate_indices_0, x = var_44_to_fp16)[name = string("op_46_cast_fp16_cast_uint16")];
|
73 |
+
int32 var_56 = const()[name = string("op_56"), val = int32(1)];
|
74 |
+
int32 var_57_batch_dims_0 = const()[name = string("op_57_batch_dims_0"), val = int32(0)];
|
75 |
+
bool var_57_validate_indices_0 = const()[name = string("op_57_validate_indices_0"), val = bool(false)];
|
76 |
+
tensor<fp16, [128, 512]> var_55_to_fp16 = const()[name = string("op_55_to_fp16"), val = tensor<fp16, [128, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(262275264)))];
|
77 |
+
tensor<fp16, [128, 512]> sin = gather(axis = var_56, batch_dims = var_57_batch_dims_0, indices = input_pos_to_uint16, validate_indices = var_57_validate_indices_0, x = var_55_to_fp16)[name = string("op_57_cast_fp16_cast_uint16")];
|
78 |
+
tensor<int32, [512, 1]> var_92 = const()[name = string("op_92"), val = tensor<int32, [512, 1]>([[0], [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14], [15], [16], [17], [18], [19], [20], [21], [22], [23], [24], [25], [26], [27], [28], [29], [30], [31], [32], [33], [34], [35], [36], [37], [38], [39], [40], [41], [42], [43], [44], [45], [46], [47], [48], [49], [50], [51], [52], [53], [54], [55], [56], [57], [58], [59], [60], [61], [62], [63], [64], [65], [66], [67], [68], [69], [70], [71], [72], [73], [74], [75], [76], [77], [78], [79], [80], [81], [82], [83], [84], [85], [86], [87], [88], [89], [90], [91], [92], [93], [94], [95], [96], [97], [98], [99], [100], [101], [102], [103], [104], [105], [106], [107], [108], [109], [110], [111], [112], [113], [114], [115], [116], [117], [118], [119], [120], [121], [122], [123], [124], [125], [126], [127], [128], [129], [130], [131], [132], [133], [134], [135], [136], [137], [138], [139], [140], [141], [142], [143], [144], [145], [146], [147], [148], [149], [150], [151], [152], [153], [154], [155], [156], [157], [158], [159], [160], [161], [162], [163], [164], [165], [166], [167], [168], [169], [170], [171], [172], [173], [174], [175], [176], [177], [178], [179], [180], [181], [182], [183], [184], [185], [186], [187], [188], [189], [190], [191], [192], [193], [194], [195], [196], [197], [198], [199], [200], [201], [202], [203], [204], [205], [206], [207], [208], [209], [210], [211], [212], [213], [214], [215], [216], [217], [218], [219], [220], [221], [222], [223], [224], [225], [226], [227], [228], [229], [230], [231], [232], [233], [234], [235], [236], [237], [238], [239], [240], [241], [242], [243], [244], [245], [246], [247], [248], [249], [250], [251], [252], [253], [254], [255], [256], [257], [258], [259], [260], [261], [262], [263], [264], [265], [266], [267], [268], [269], [270], [271], [272], [273], [274], [275], [276], [277], [278], [279], [280], [281], [282], [283], [284], [285], [286], [287], [288], [289], [290], [291], [292], [293], [294], [295], [296], [297], [298], [299], [300], [301], [302], [303], [304], [305], [306], [307], [308], [309], [310], [311], [312], [313], [314], [315], [316], [317], [318], [319], [320], [321], [322], [323], [324], [325], [326], [327], [328], [329], [330], [331], [332], [333], [334], [335], [336], [337], [338], [339], [340], [341], [342], [343], [344], [345], [346], [347], [348], [349], [350], [351], [352], [353], [354], [355], [356], [357], [358], [359], [360], [361], [362], [363], [364], [365], [366], [367], [368], [369], [370], [371], [372], [373], [374], [375], [376], [377], [378], [379], [380], [381], [382], [383], [384], [385], [386], [387], [388], [389], [390], [391], [392], [393], [394], [395], [396], [397], [398], [399], [400], [401], [402], [403], [404], [405], [406], [407], [408], [409], [410], [411], [412], [413], [414], [415], [416], [417], [418], [419], [420], [421], [422], [423], [424], [425], [426], [427], [428], [429], [430], [431], [432], [433], [434], [435], [436], [437], [438], [439], [440], [441], [442], [443], [444], [445], [446], [447], [448], [449], [450], [451], [452], [453], [454], [455], [456], [457], [458], [459], [460], [461], [462], [463], [464], [465], [466], [467], [468], [469], [470], [471], [472], [473], [474], [475], [476], [477], [478], [479], [480], [481], [482], [483], [484], [485], [486], [487], [488], [489], [490], [491], [492], [493], [494], [495], [496], [497], [498], [499], [500], [501], [502], [503], [504], [505], [506], [507], [508], [509], [510], [511]])];
|
79 |
+
tensor<bool, [512, 1]> var_95 = less(x = var_92, y = pos_offset)[name = string("op_95")];
|
80 |
+
tensor<int32, [2]> var_95_after_broadcast_reps_0 = const()[name = string("op_95_after_broadcast_reps_0"), val = tensor<int32, [2]>([1, 512])];
|
81 |
+
tensor<bool, [512, 512]> var_95_after_broadcast = tile(reps = var_95_after_broadcast_reps_0, x = var_95)[name = string("op_95_after_broadcast")];
|
82 |
+
tensor<fp16, [512, 512]> all_mask_to_fp16 = const()[name = string("all_mask_to_fp16"), val = tensor<fp16, [512, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(262406400)))];
|
83 |
+
tensor<fp16, [512, 512]> m_1_to_fp16 = const()[name = string("m_1_to_fp16"), val = tensor<fp16, [512, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(262930752)))];
|
84 |
+
tensor<fp16, [512, 512]> m_3_cast_fp16 = select(a = all_mask_to_fp16, b = m_1_to_fp16, cond = var_95_after_broadcast)[name = string("m_3_cast_fp16")];
|
85 |
+
int32 var_106 = const()[name = string("op_106"), val = int32(512)];
|
86 |
+
tensor<int32, [1]> var_108 = sub(x = var_106, y = full_sequence_length)[name = string("op_108")];
|
87 |
+
tensor<bool, [512]> var_109 = less(x = var_26, y = var_108)[name = string("op_109")];
|
88 |
+
tensor<int32, [1]> expand_dims_0_axes_0 = const()[name = string("expand_dims_0_axes_0"), val = tensor<int32, [1]>([0])];
|
89 |
+
tensor<bool, [1, 512]> expand_dims_0 = expand_dims(axes = expand_dims_0_axes_0, x = var_109)[name = string("expand_dims_0")];
|
90 |
+
tensor<int32, [2]> var_109_after_broadcast_reps_0 = const()[name = string("op_109_after_broadcast_reps_0"), val = tensor<int32, [2]>([512, 1])];
|
91 |
+
tensor<bool, [512, 512]> var_109_after_broadcast = tile(reps = var_109_after_broadcast_reps_0, x = expand_dims_0)[name = string("op_109_after_broadcast")];
|
92 |
+
tensor<fp16, [512, 512]> m_cast_fp16 = select(a = all_mask_to_fp16, b = m_3_cast_fp16, cond = var_109_after_broadcast)[name = string("m_cast_fp16")];
|
93 |
+
tensor<int32, [1]> var_112_axes_0 = const()[name = string("op_112_axes_0"), val = tensor<int32, [1]>([0])];
|
94 |
+
tensor<fp16, [1, 512, 512]> var_112_cast_fp16 = expand_dims(axes = var_112_axes_0, x = m_cast_fp16)[name = string("op_112_cast_fp16")];
|
95 |
+
tensor<int32, [1]> var_114_axes_0 = const()[name = string("op_114_axes_0"), val = tensor<int32, [1]>([0])];
|
96 |
+
tensor<fp16, [1, 1, 512, 512]> mask = expand_dims(axes = var_114_axes_0, x = var_112_cast_fp16)[name = string("op_114_cast_fp16")];
|
97 |
+
} -> (x, cos, sin, mask);
|
98 |
+
}
|
sequoia/Llama-2-7b-hf_chunk1.mlmodelc/weights/weight.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:63ea75c6154c60560d9edb4d2e2f028afa38a3927bb7277b7d01558bc198e965
|
3 |
+
size 263457280
|
sequoia/Llama-2-7b-hf_chunk10.mlmodelc/analytics/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:84e317a82cdf4e96f808f63e77f10098844d47ad522545181edfac4d287c9c92
|
3 |
+
size 243
|
sequoia/Llama-2-7b-hf_chunk10.mlmodelc/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e430d0795ff5c384187174f5718a2c13d0070f5d6a811831e18862497865a86d
|
3 |
+
size 1037
|
sequoia/Llama-2-7b-hf_chunk10.mlmodelc/metadata.json
ADDED
@@ -0,0 +1,500 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 × 4096 × 1 × 512)",
|
11 |
+
"shortDescription" : "",
|
12 |
+
"shape" : "[1, 4096, 1, 512]",
|
13 |
+
"name" : "new_x",
|
14 |
+
"type" : "MultiArray"
|
15 |
+
},
|
16 |
+
{
|
17 |
+
"hasShapeFlexibility" : "0",
|
18 |
+
"isOptional" : "0",
|
19 |
+
"dataType" : "Float16",
|
20 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
21 |
+
"shortDescription" : "",
|
22 |
+
"shape" : "[1, 32, 128, 511]",
|
23 |
+
"name" : "new_k_cache_0",
|
24 |
+
"type" : "MultiArray"
|
25 |
+
},
|
26 |
+
{
|
27 |
+
"hasShapeFlexibility" : "0",
|
28 |
+
"isOptional" : "0",
|
29 |
+
"dataType" : "Float16",
|
30 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
31 |
+
"shortDescription" : "",
|
32 |
+
"shape" : "[1, 32, 128, 511]",
|
33 |
+
"name" : "new_k_cache_1",
|
34 |
+
"type" : "MultiArray"
|
35 |
+
},
|
36 |
+
{
|
37 |
+
"hasShapeFlexibility" : "0",
|
38 |
+
"isOptional" : "0",
|
39 |
+
"dataType" : "Float16",
|
40 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
41 |
+
"shortDescription" : "",
|
42 |
+
"shape" : "[1, 32, 128, 511]",
|
43 |
+
"name" : "new_k_cache_2",
|
44 |
+
"type" : "MultiArray"
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"hasShapeFlexibility" : "0",
|
48 |
+
"isOptional" : "0",
|
49 |
+
"dataType" : "Float16",
|
50 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
51 |
+
"shortDescription" : "",
|
52 |
+
"shape" : "[1, 32, 128, 511]",
|
53 |
+
"name" : "new_v_cache_0",
|
54 |
+
"type" : "MultiArray"
|
55 |
+
},
|
56 |
+
{
|
57 |
+
"hasShapeFlexibility" : "0",
|
58 |
+
"isOptional" : "0",
|
59 |
+
"dataType" : "Float16",
|
60 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
61 |
+
"shortDescription" : "",
|
62 |
+
"shape" : "[1, 32, 128, 511]",
|
63 |
+
"name" : "new_v_cache_1",
|
64 |
+
"type" : "MultiArray"
|
65 |
+
},
|
66 |
+
{
|
67 |
+
"hasShapeFlexibility" : "0",
|
68 |
+
"isOptional" : "0",
|
69 |
+
"dataType" : "Float16",
|
70 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
71 |
+
"shortDescription" : "",
|
72 |
+
"shape" : "[1, 32, 128, 511]",
|
73 |
+
"name" : "new_v_cache_2",
|
74 |
+
"type" : "MultiArray"
|
75 |
+
}
|
76 |
+
],
|
77 |
+
"modelParameters" : [
|
78 |
+
|
79 |
+
],
|
80 |
+
"specificationVersion" : 9,
|
81 |
+
"functions" : [
|
82 |
+
{
|
83 |
+
"inputSchema" : [
|
84 |
+
{
|
85 |
+
"hasShapeFlexibility" : "0",
|
86 |
+
"isOptional" : "0",
|
87 |
+
"dataType" : "Float16",
|
88 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
89 |
+
"shortDescription" : "",
|
90 |
+
"shape" : "[1, 4096, 1, 512]",
|
91 |
+
"name" : "x",
|
92 |
+
"type" : "MultiArray"
|
93 |
+
},
|
94 |
+
{
|
95 |
+
"hasShapeFlexibility" : "0",
|
96 |
+
"isOptional" : "0",
|
97 |
+
"dataType" : "Float16",
|
98 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
99 |
+
"shortDescription" : "",
|
100 |
+
"shape" : "[128, 512]",
|
101 |
+
"name" : "cos",
|
102 |
+
"type" : "MultiArray"
|
103 |
+
},
|
104 |
+
{
|
105 |
+
"hasShapeFlexibility" : "0",
|
106 |
+
"isOptional" : "0",
|
107 |
+
"dataType" : "Float16",
|
108 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
109 |
+
"shortDescription" : "",
|
110 |
+
"shape" : "[128, 512]",
|
111 |
+
"name" : "sin",
|
112 |
+
"type" : "MultiArray"
|
113 |
+
},
|
114 |
+
{
|
115 |
+
"hasShapeFlexibility" : "0",
|
116 |
+
"isOptional" : "0",
|
117 |
+
"dataType" : "Float16",
|
118 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
119 |
+
"shortDescription" : "",
|
120 |
+
"shape" : "[1, 1, 512, 512]",
|
121 |
+
"name" : "mask",
|
122 |
+
"type" : "MultiArray"
|
123 |
+
}
|
124 |
+
],
|
125 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
126 |
+
"storagePrecision" : "Float16",
|
127 |
+
"stateSchema" : [
|
128 |
+
|
129 |
+
],
|
130 |
+
"outputSchema" : [
|
131 |
+
{
|
132 |
+
"hasShapeFlexibility" : "0",
|
133 |
+
"isOptional" : "0",
|
134 |
+
"dataType" : "Float16",
|
135 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
136 |
+
"shortDescription" : "",
|
137 |
+
"shape" : "[1, 4096, 1, 512]",
|
138 |
+
"name" : "new_x",
|
139 |
+
"type" : "MultiArray"
|
140 |
+
},
|
141 |
+
{
|
142 |
+
"hasShapeFlexibility" : "0",
|
143 |
+
"isOptional" : "0",
|
144 |
+
"dataType" : "Float16",
|
145 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
146 |
+
"shortDescription" : "",
|
147 |
+
"shape" : "[1, 32, 128, 511]",
|
148 |
+
"name" : "new_k_cache_0",
|
149 |
+
"type" : "MultiArray"
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"hasShapeFlexibility" : "0",
|
153 |
+
"isOptional" : "0",
|
154 |
+
"dataType" : "Float16",
|
155 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
156 |
+
"shortDescription" : "",
|
157 |
+
"shape" : "[1, 32, 128, 511]",
|
158 |
+
"name" : "new_k_cache_1",
|
159 |
+
"type" : "MultiArray"
|
160 |
+
},
|
161 |
+
{
|
162 |
+
"hasShapeFlexibility" : "0",
|
163 |
+
"isOptional" : "0",
|
164 |
+
"dataType" : "Float16",
|
165 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
166 |
+
"shortDescription" : "",
|
167 |
+
"shape" : "[1, 32, 128, 511]",
|
168 |
+
"name" : "new_k_cache_2",
|
169 |
+
"type" : "MultiArray"
|
170 |
+
},
|
171 |
+
{
|
172 |
+
"hasShapeFlexibility" : "0",
|
173 |
+
"isOptional" : "0",
|
174 |
+
"dataType" : "Float16",
|
175 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
176 |
+
"shortDescription" : "",
|
177 |
+
"shape" : "[1, 32, 128, 511]",
|
178 |
+
"name" : "new_v_cache_0",
|
179 |
+
"type" : "MultiArray"
|
180 |
+
},
|
181 |
+
{
|
182 |
+
"hasShapeFlexibility" : "0",
|
183 |
+
"isOptional" : "0",
|
184 |
+
"dataType" : "Float16",
|
185 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
186 |
+
"shortDescription" : "",
|
187 |
+
"shape" : "[1, 32, 128, 511]",
|
188 |
+
"name" : "new_v_cache_1",
|
189 |
+
"type" : "MultiArray"
|
190 |
+
},
|
191 |
+
{
|
192 |
+
"hasShapeFlexibility" : "0",
|
193 |
+
"isOptional" : "0",
|
194 |
+
"dataType" : "Float16",
|
195 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
196 |
+
"shortDescription" : "",
|
197 |
+
"shape" : "[1, 32, 128, 511]",
|
198 |
+
"name" : "new_v_cache_2",
|
199 |
+
"type" : "MultiArray"
|
200 |
+
}
|
201 |
+
],
|
202 |
+
"name" : "input_512_context_512",
|
203 |
+
"mlProgramOperationTypeHistogram" : {
|
204 |
+
"Ios18.constexprLutToDense" : 21,
|
205 |
+
"Ios18.conv" : 21,
|
206 |
+
"Ios18.matmul" : 6,
|
207 |
+
"Ios18.expandDims" : 6,
|
208 |
+
"Ios18.concat" : 18,
|
209 |
+
"Ios18.add" : 15,
|
210 |
+
"Ios18.realDiv" : 6,
|
211 |
+
"Ios18.silu" : 3,
|
212 |
+
"Ios18.softmax" : 3,
|
213 |
+
"Ios18.sliceByIndex" : 18,
|
214 |
+
"Ios16.reduceL2Norm" : 6,
|
215 |
+
"Ios18.squeeze" : 6,
|
216 |
+
"Ios18.reshape" : 12,
|
217 |
+
"Ios18.mul" : 57
|
218 |
+
}
|
219 |
+
},
|
220 |
+
{
|
221 |
+
"inputSchema" : [
|
222 |
+
{
|
223 |
+
"hasShapeFlexibility" : "0",
|
224 |
+
"isOptional" : "0",
|
225 |
+
"dataType" : "Float16",
|
226 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 1)",
|
227 |
+
"shortDescription" : "",
|
228 |
+
"shape" : "[1, 4096, 1, 1]",
|
229 |
+
"name" : "x",
|
230 |
+
"type" : "MultiArray"
|
231 |
+
},
|
232 |
+
{
|
233 |
+
"hasShapeFlexibility" : "0",
|
234 |
+
"isOptional" : "0",
|
235 |
+
"dataType" : "Float16",
|
236 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
237 |
+
"shortDescription" : "",
|
238 |
+
"shape" : "[128, 1]",
|
239 |
+
"name" : "cos",
|
240 |
+
"type" : "MultiArray"
|
241 |
+
},
|
242 |
+
{
|
243 |
+
"hasShapeFlexibility" : "0",
|
244 |
+
"isOptional" : "0",
|
245 |
+
"dataType" : "Float16",
|
246 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
247 |
+
"shortDescription" : "",
|
248 |
+
"shape" : "[128, 1]",
|
249 |
+
"name" : "sin",
|
250 |
+
"type" : "MultiArray"
|
251 |
+
},
|
252 |
+
{
|
253 |
+
"hasShapeFlexibility" : "0",
|
254 |
+
"isOptional" : "0",
|
255 |
+
"dataType" : "Float16",
|
256 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 1 × 512)",
|
257 |
+
"shortDescription" : "",
|
258 |
+
"shape" : "[1, 1, 1, 512]",
|
259 |
+
"name" : "mask",
|
260 |
+
"type" : "MultiArray"
|
261 |
+
},
|
262 |
+
{
|
263 |
+
"hasShapeFlexibility" : "0",
|
264 |
+
"isOptional" : "1",
|
265 |
+
"dataType" : "Float16",
|
266 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
267 |
+
"shortDescription" : "",
|
268 |
+
"shape" : "[1, 32, 128, 511]",
|
269 |
+
"name" : "k_cache_0",
|
270 |
+
"type" : "MultiArray"
|
271 |
+
},
|
272 |
+
{
|
273 |
+
"hasShapeFlexibility" : "0",
|
274 |
+
"isOptional" : "1",
|
275 |
+
"dataType" : "Float16",
|
276 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
277 |
+
"shortDescription" : "",
|
278 |
+
"shape" : "[1, 32, 128, 511]",
|
279 |
+
"name" : "v_cache_0",
|
280 |
+
"type" : "MultiArray"
|
281 |
+
},
|
282 |
+
{
|
283 |
+
"hasShapeFlexibility" : "0",
|
284 |
+
"isOptional" : "1",
|
285 |
+
"dataType" : "Float16",
|
286 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
287 |
+
"shortDescription" : "",
|
288 |
+
"shape" : "[1, 32, 128, 511]",
|
289 |
+
"name" : "k_cache_1",
|
290 |
+
"type" : "MultiArray"
|
291 |
+
},
|
292 |
+
{
|
293 |
+
"hasShapeFlexibility" : "0",
|
294 |
+
"isOptional" : "1",
|
295 |
+
"dataType" : "Float16",
|
296 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
297 |
+
"shortDescription" : "",
|
298 |
+
"shape" : "[1, 32, 128, 511]",
|
299 |
+
"name" : "v_cache_1",
|
300 |
+
"type" : "MultiArray"
|
301 |
+
},
|
302 |
+
{
|
303 |
+
"hasShapeFlexibility" : "0",
|
304 |
+
"isOptional" : "1",
|
305 |
+
"dataType" : "Float16",
|
306 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
307 |
+
"shortDescription" : "",
|
308 |
+
"shape" : "[1, 32, 128, 511]",
|
309 |
+
"name" : "k_cache_2",
|
310 |
+
"type" : "MultiArray"
|
311 |
+
},
|
312 |
+
{
|
313 |
+
"hasShapeFlexibility" : "0",
|
314 |
+
"isOptional" : "1",
|
315 |
+
"dataType" : "Float16",
|
316 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
317 |
+
"shortDescription" : "",
|
318 |
+
"shape" : "[1, 32, 128, 511]",
|
319 |
+
"name" : "v_cache_2",
|
320 |
+
"type" : "MultiArray"
|
321 |
+
}
|
322 |
+
],
|
323 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
324 |
+
"storagePrecision" : "Float16",
|
325 |
+
"stateSchema" : [
|
326 |
+
|
327 |
+
],
|
328 |
+
"outputSchema" : [
|
329 |
+
{
|
330 |
+
"hasShapeFlexibility" : "0",
|
331 |
+
"isOptional" : "0",
|
332 |
+
"dataType" : "Float16",
|
333 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 1)",
|
334 |
+
"shortDescription" : "",
|
335 |
+
"shape" : "[1, 4096, 1, 1]",
|
336 |
+
"name" : "new_x",
|
337 |
+
"type" : "MultiArray"
|
338 |
+
},
|
339 |
+
{
|
340 |
+
"hasShapeFlexibility" : "0",
|
341 |
+
"isOptional" : "0",
|
342 |
+
"dataType" : "Float16",
|
343 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
344 |
+
"shortDescription" : "",
|
345 |
+
"shape" : "[1, 32, 128, 511]",
|
346 |
+
"name" : "new_k_cache_0",
|
347 |
+
"type" : "MultiArray"
|
348 |
+
},
|
349 |
+
{
|
350 |
+
"hasShapeFlexibility" : "0",
|
351 |
+
"isOptional" : "0",
|
352 |
+
"dataType" : "Float16",
|
353 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
354 |
+
"shortDescription" : "",
|
355 |
+
"shape" : "[1, 32, 128, 511]",
|
356 |
+
"name" : "new_k_cache_1",
|
357 |
+
"type" : "MultiArray"
|
358 |
+
},
|
359 |
+
{
|
360 |
+
"hasShapeFlexibility" : "0",
|
361 |
+
"isOptional" : "0",
|
362 |
+
"dataType" : "Float16",
|
363 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
364 |
+
"shortDescription" : "",
|
365 |
+
"shape" : "[1, 32, 128, 511]",
|
366 |
+
"name" : "new_k_cache_2",
|
367 |
+
"type" : "MultiArray"
|
368 |
+
},
|
369 |
+
{
|
370 |
+
"hasShapeFlexibility" : "0",
|
371 |
+
"isOptional" : "0",
|
372 |
+
"dataType" : "Float16",
|
373 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
374 |
+
"shortDescription" : "",
|
375 |
+
"shape" : "[1, 32, 128, 511]",
|
376 |
+
"name" : "new_v_cache_0",
|
377 |
+
"type" : "MultiArray"
|
378 |
+
},
|
379 |
+
{
|
380 |
+
"hasShapeFlexibility" : "0",
|
381 |
+
"isOptional" : "0",
|
382 |
+
"dataType" : "Float16",
|
383 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
384 |
+
"shortDescription" : "",
|
385 |
+
"shape" : "[1, 32, 128, 511]",
|
386 |
+
"name" : "new_v_cache_1",
|
387 |
+
"type" : "MultiArray"
|
388 |
+
},
|
389 |
+
{
|
390 |
+
"hasShapeFlexibility" : "0",
|
391 |
+
"isOptional" : "0",
|
392 |
+
"dataType" : "Float16",
|
393 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
394 |
+
"shortDescription" : "",
|
395 |
+
"shape" : "[1, 32, 128, 511]",
|
396 |
+
"name" : "new_v_cache_2",
|
397 |
+
"type" : "MultiArray"
|
398 |
+
}
|
399 |
+
],
|
400 |
+
"name" : "input_1_context_512",
|
401 |
+
"mlProgramOperationTypeHistogram" : {
|
402 |
+
"Ios18.constexprLutToDense" : 21,
|
403 |
+
"Ios18.conv" : 21,
|
404 |
+
"Ios18.matmul" : 6,
|
405 |
+
"Ios18.expandDims" : 6,
|
406 |
+
"Ios18.concat" : 18,
|
407 |
+
"Ios18.add" : 15,
|
408 |
+
"Ios18.realDiv" : 6,
|
409 |
+
"Ios18.silu" : 3,
|
410 |
+
"Ios18.softmax" : 3,
|
411 |
+
"Ios18.sliceByIndex" : 18,
|
412 |
+
"Ios16.reduceL2Norm" : 6,
|
413 |
+
"Ios18.squeeze" : 6,
|
414 |
+
"Ios18.reshape" : 12,
|
415 |
+
"Ios18.mul" : 57
|
416 |
+
}
|
417 |
+
}
|
418 |
+
],
|
419 |
+
"mlProgramOperationTypeHistogram" : {
|
420 |
+
"Ios18.constexprLutToDense" : 21,
|
421 |
+
"Ios18.conv" : 21,
|
422 |
+
"Ios18.matmul" : 6,
|
423 |
+
"Ios18.expandDims" : 6,
|
424 |
+
"Ios18.concat" : 18,
|
425 |
+
"Ios18.add" : 15,
|
426 |
+
"Ios18.realDiv" : 6,
|
427 |
+
"Ios18.silu" : 3,
|
428 |
+
"Ios18.softmax" : 3,
|
429 |
+
"Ios18.sliceByIndex" : 18,
|
430 |
+
"Ios16.reduceL2Norm" : 6,
|
431 |
+
"Ios18.squeeze" : 6,
|
432 |
+
"Ios18.reshape" : 12,
|
433 |
+
"Ios18.mul" : 57
|
434 |
+
},
|
435 |
+
"isUpdatable" : "0",
|
436 |
+
"stateSchema" : [
|
437 |
+
|
438 |
+
],
|
439 |
+
"availability" : {
|
440 |
+
"macOS" : "15.0",
|
441 |
+
"tvOS" : "18.0",
|
442 |
+
"visionOS" : "2.0",
|
443 |
+
"watchOS" : "11.0",
|
444 |
+
"iOS" : "18.0",
|
445 |
+
"macCatalyst" : "18.0"
|
446 |
+
},
|
447 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
448 |
+
"modelType" : {
|
449 |
+
"name" : "MLModelType_mlProgram"
|
450 |
+
},
|
451 |
+
"inputSchema" : [
|
452 |
+
{
|
453 |
+
"hasShapeFlexibility" : "0",
|
454 |
+
"isOptional" : "0",
|
455 |
+
"dataType" : "Float16",
|
456 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
457 |
+
"shortDescription" : "",
|
458 |
+
"shape" : "[1, 4096, 1, 512]",
|
459 |
+
"name" : "x",
|
460 |
+
"type" : "MultiArray"
|
461 |
+
},
|
462 |
+
{
|
463 |
+
"hasShapeFlexibility" : "0",
|
464 |
+
"isOptional" : "0",
|
465 |
+
"dataType" : "Float16",
|
466 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
467 |
+
"shortDescription" : "",
|
468 |
+
"shape" : "[128, 512]",
|
469 |
+
"name" : "cos",
|
470 |
+
"type" : "MultiArray"
|
471 |
+
},
|
472 |
+
{
|
473 |
+
"hasShapeFlexibility" : "0",
|
474 |
+
"isOptional" : "0",
|
475 |
+
"dataType" : "Float16",
|
476 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
477 |
+
"shortDescription" : "",
|
478 |
+
"shape" : "[128, 512]",
|
479 |
+
"name" : "sin",
|
480 |
+
"type" : "MultiArray"
|
481 |
+
},
|
482 |
+
{
|
483 |
+
"hasShapeFlexibility" : "0",
|
484 |
+
"isOptional" : "0",
|
485 |
+
"dataType" : "Float16",
|
486 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
487 |
+
"shortDescription" : "",
|
488 |
+
"shape" : "[1, 1, 512, 512]",
|
489 |
+
"name" : "mask",
|
490 |
+
"type" : "MultiArray"
|
491 |
+
}
|
492 |
+
],
|
493 |
+
"defaultFunctionName" : "input_512_context_512",
|
494 |
+
"generatedClassName" : "Llama_2_7b_hf_2024_07_02_20_36_17_merged_chunk10",
|
495 |
+
"userDefinedMetadata" : {
|
496 |
+
|
497 |
+
},
|
498 |
+
"method" : "predict"
|
499 |
+
}
|
500 |
+
]
|
sequoia/Llama-2-7b-hf_chunk10.mlmodelc/model.mil
ADDED
The diff for this file is too large to render.
See raw diff
|
|
sequoia/Llama-2-7b-hf_chunk10.mlmodelc/weights/weight.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:db77bfdf59178a243b216dccd0f4cf6d4f6cf150fec93ddb40c943bdbb97e07b
|
3 |
+
size 303876448
|
sequoia/Llama-2-7b-hf_chunk11.mlmodelc/analytics/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:84e317a82cdf4e96f808f63e77f10098844d47ad522545181edfac4d287c9c92
|
3 |
+
size 243
|
sequoia/Llama-2-7b-hf_chunk11.mlmodelc/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e430d0795ff5c384187174f5718a2c13d0070f5d6a811831e18862497865a86d
|
3 |
+
size 1037
|
sequoia/Llama-2-7b-hf_chunk11.mlmodelc/metadata.json
ADDED
@@ -0,0 +1,500 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 × 4096 × 1 × 512)",
|
11 |
+
"shortDescription" : "",
|
12 |
+
"shape" : "[1, 4096, 1, 512]",
|
13 |
+
"name" : "new_x",
|
14 |
+
"type" : "MultiArray"
|
15 |
+
},
|
16 |
+
{
|
17 |
+
"hasShapeFlexibility" : "0",
|
18 |
+
"isOptional" : "0",
|
19 |
+
"dataType" : "Float16",
|
20 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
21 |
+
"shortDescription" : "",
|
22 |
+
"shape" : "[1, 32, 128, 511]",
|
23 |
+
"name" : "new_k_cache_0",
|
24 |
+
"type" : "MultiArray"
|
25 |
+
},
|
26 |
+
{
|
27 |
+
"hasShapeFlexibility" : "0",
|
28 |
+
"isOptional" : "0",
|
29 |
+
"dataType" : "Float16",
|
30 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
31 |
+
"shortDescription" : "",
|
32 |
+
"shape" : "[1, 32, 128, 511]",
|
33 |
+
"name" : "new_k_cache_1",
|
34 |
+
"type" : "MultiArray"
|
35 |
+
},
|
36 |
+
{
|
37 |
+
"hasShapeFlexibility" : "0",
|
38 |
+
"isOptional" : "0",
|
39 |
+
"dataType" : "Float16",
|
40 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
41 |
+
"shortDescription" : "",
|
42 |
+
"shape" : "[1, 32, 128, 511]",
|
43 |
+
"name" : "new_k_cache_2",
|
44 |
+
"type" : "MultiArray"
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"hasShapeFlexibility" : "0",
|
48 |
+
"isOptional" : "0",
|
49 |
+
"dataType" : "Float16",
|
50 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
51 |
+
"shortDescription" : "",
|
52 |
+
"shape" : "[1, 32, 128, 511]",
|
53 |
+
"name" : "new_v_cache_0",
|
54 |
+
"type" : "MultiArray"
|
55 |
+
},
|
56 |
+
{
|
57 |
+
"hasShapeFlexibility" : "0",
|
58 |
+
"isOptional" : "0",
|
59 |
+
"dataType" : "Float16",
|
60 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
61 |
+
"shortDescription" : "",
|
62 |
+
"shape" : "[1, 32, 128, 511]",
|
63 |
+
"name" : "new_v_cache_1",
|
64 |
+
"type" : "MultiArray"
|
65 |
+
},
|
66 |
+
{
|
67 |
+
"hasShapeFlexibility" : "0",
|
68 |
+
"isOptional" : "0",
|
69 |
+
"dataType" : "Float16",
|
70 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
71 |
+
"shortDescription" : "",
|
72 |
+
"shape" : "[1, 32, 128, 511]",
|
73 |
+
"name" : "new_v_cache_2",
|
74 |
+
"type" : "MultiArray"
|
75 |
+
}
|
76 |
+
],
|
77 |
+
"modelParameters" : [
|
78 |
+
|
79 |
+
],
|
80 |
+
"specificationVersion" : 9,
|
81 |
+
"functions" : [
|
82 |
+
{
|
83 |
+
"inputSchema" : [
|
84 |
+
{
|
85 |
+
"hasShapeFlexibility" : "0",
|
86 |
+
"isOptional" : "0",
|
87 |
+
"dataType" : "Float16",
|
88 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
89 |
+
"shortDescription" : "",
|
90 |
+
"shape" : "[1, 4096, 1, 512]",
|
91 |
+
"name" : "x",
|
92 |
+
"type" : "MultiArray"
|
93 |
+
},
|
94 |
+
{
|
95 |
+
"hasShapeFlexibility" : "0",
|
96 |
+
"isOptional" : "0",
|
97 |
+
"dataType" : "Float16",
|
98 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
99 |
+
"shortDescription" : "",
|
100 |
+
"shape" : "[128, 512]",
|
101 |
+
"name" : "cos",
|
102 |
+
"type" : "MultiArray"
|
103 |
+
},
|
104 |
+
{
|
105 |
+
"hasShapeFlexibility" : "0",
|
106 |
+
"isOptional" : "0",
|
107 |
+
"dataType" : "Float16",
|
108 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
109 |
+
"shortDescription" : "",
|
110 |
+
"shape" : "[128, 512]",
|
111 |
+
"name" : "sin",
|
112 |
+
"type" : "MultiArray"
|
113 |
+
},
|
114 |
+
{
|
115 |
+
"hasShapeFlexibility" : "0",
|
116 |
+
"isOptional" : "0",
|
117 |
+
"dataType" : "Float16",
|
118 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
119 |
+
"shortDescription" : "",
|
120 |
+
"shape" : "[1, 1, 512, 512]",
|
121 |
+
"name" : "mask",
|
122 |
+
"type" : "MultiArray"
|
123 |
+
}
|
124 |
+
],
|
125 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
126 |
+
"storagePrecision" : "Float16",
|
127 |
+
"stateSchema" : [
|
128 |
+
|
129 |
+
],
|
130 |
+
"outputSchema" : [
|
131 |
+
{
|
132 |
+
"hasShapeFlexibility" : "0",
|
133 |
+
"isOptional" : "0",
|
134 |
+
"dataType" : "Float16",
|
135 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
136 |
+
"shortDescription" : "",
|
137 |
+
"shape" : "[1, 4096, 1, 512]",
|
138 |
+
"name" : "new_x",
|
139 |
+
"type" : "MultiArray"
|
140 |
+
},
|
141 |
+
{
|
142 |
+
"hasShapeFlexibility" : "0",
|
143 |
+
"isOptional" : "0",
|
144 |
+
"dataType" : "Float16",
|
145 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
146 |
+
"shortDescription" : "",
|
147 |
+
"shape" : "[1, 32, 128, 511]",
|
148 |
+
"name" : "new_k_cache_0",
|
149 |
+
"type" : "MultiArray"
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"hasShapeFlexibility" : "0",
|
153 |
+
"isOptional" : "0",
|
154 |
+
"dataType" : "Float16",
|
155 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
156 |
+
"shortDescription" : "",
|
157 |
+
"shape" : "[1, 32, 128, 511]",
|
158 |
+
"name" : "new_k_cache_1",
|
159 |
+
"type" : "MultiArray"
|
160 |
+
},
|
161 |
+
{
|
162 |
+
"hasShapeFlexibility" : "0",
|
163 |
+
"isOptional" : "0",
|
164 |
+
"dataType" : "Float16",
|
165 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
166 |
+
"shortDescription" : "",
|
167 |
+
"shape" : "[1, 32, 128, 511]",
|
168 |
+
"name" : "new_k_cache_2",
|
169 |
+
"type" : "MultiArray"
|
170 |
+
},
|
171 |
+
{
|
172 |
+
"hasShapeFlexibility" : "0",
|
173 |
+
"isOptional" : "0",
|
174 |
+
"dataType" : "Float16",
|
175 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
176 |
+
"shortDescription" : "",
|
177 |
+
"shape" : "[1, 32, 128, 511]",
|
178 |
+
"name" : "new_v_cache_0",
|
179 |
+
"type" : "MultiArray"
|
180 |
+
},
|
181 |
+
{
|
182 |
+
"hasShapeFlexibility" : "0",
|
183 |
+
"isOptional" : "0",
|
184 |
+
"dataType" : "Float16",
|
185 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
186 |
+
"shortDescription" : "",
|
187 |
+
"shape" : "[1, 32, 128, 511]",
|
188 |
+
"name" : "new_v_cache_1",
|
189 |
+
"type" : "MultiArray"
|
190 |
+
},
|
191 |
+
{
|
192 |
+
"hasShapeFlexibility" : "0",
|
193 |
+
"isOptional" : "0",
|
194 |
+
"dataType" : "Float16",
|
195 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
196 |
+
"shortDescription" : "",
|
197 |
+
"shape" : "[1, 32, 128, 511]",
|
198 |
+
"name" : "new_v_cache_2",
|
199 |
+
"type" : "MultiArray"
|
200 |
+
}
|
201 |
+
],
|
202 |
+
"name" : "input_512_context_512",
|
203 |
+
"mlProgramOperationTypeHistogram" : {
|
204 |
+
"Ios18.constexprLutToDense" : 21,
|
205 |
+
"Ios18.conv" : 21,
|
206 |
+
"Ios18.matmul" : 6,
|
207 |
+
"Ios18.expandDims" : 6,
|
208 |
+
"Ios18.concat" : 18,
|
209 |
+
"Ios18.add" : 15,
|
210 |
+
"Ios18.realDiv" : 6,
|
211 |
+
"Ios18.silu" : 3,
|
212 |
+
"Ios18.softmax" : 3,
|
213 |
+
"Ios18.sliceByIndex" : 18,
|
214 |
+
"Ios16.reduceL2Norm" : 6,
|
215 |
+
"Ios18.squeeze" : 6,
|
216 |
+
"Ios18.reshape" : 12,
|
217 |
+
"Ios18.mul" : 57
|
218 |
+
}
|
219 |
+
},
|
220 |
+
{
|
221 |
+
"inputSchema" : [
|
222 |
+
{
|
223 |
+
"hasShapeFlexibility" : "0",
|
224 |
+
"isOptional" : "0",
|
225 |
+
"dataType" : "Float16",
|
226 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 1)",
|
227 |
+
"shortDescription" : "",
|
228 |
+
"shape" : "[1, 4096, 1, 1]",
|
229 |
+
"name" : "x",
|
230 |
+
"type" : "MultiArray"
|
231 |
+
},
|
232 |
+
{
|
233 |
+
"hasShapeFlexibility" : "0",
|
234 |
+
"isOptional" : "0",
|
235 |
+
"dataType" : "Float16",
|
236 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
237 |
+
"shortDescription" : "",
|
238 |
+
"shape" : "[128, 1]",
|
239 |
+
"name" : "cos",
|
240 |
+
"type" : "MultiArray"
|
241 |
+
},
|
242 |
+
{
|
243 |
+
"hasShapeFlexibility" : "0",
|
244 |
+
"isOptional" : "0",
|
245 |
+
"dataType" : "Float16",
|
246 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
247 |
+
"shortDescription" : "",
|
248 |
+
"shape" : "[128, 1]",
|
249 |
+
"name" : "sin",
|
250 |
+
"type" : "MultiArray"
|
251 |
+
},
|
252 |
+
{
|
253 |
+
"hasShapeFlexibility" : "0",
|
254 |
+
"isOptional" : "0",
|
255 |
+
"dataType" : "Float16",
|
256 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 1 × 512)",
|
257 |
+
"shortDescription" : "",
|
258 |
+
"shape" : "[1, 1, 1, 512]",
|
259 |
+
"name" : "mask",
|
260 |
+
"type" : "MultiArray"
|
261 |
+
},
|
262 |
+
{
|
263 |
+
"hasShapeFlexibility" : "0",
|
264 |
+
"isOptional" : "1",
|
265 |
+
"dataType" : "Float16",
|
266 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
267 |
+
"shortDescription" : "",
|
268 |
+
"shape" : "[1, 32, 128, 511]",
|
269 |
+
"name" : "k_cache_0",
|
270 |
+
"type" : "MultiArray"
|
271 |
+
},
|
272 |
+
{
|
273 |
+
"hasShapeFlexibility" : "0",
|
274 |
+
"isOptional" : "1",
|
275 |
+
"dataType" : "Float16",
|
276 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
277 |
+
"shortDescription" : "",
|
278 |
+
"shape" : "[1, 32, 128, 511]",
|
279 |
+
"name" : "v_cache_0",
|
280 |
+
"type" : "MultiArray"
|
281 |
+
},
|
282 |
+
{
|
283 |
+
"hasShapeFlexibility" : "0",
|
284 |
+
"isOptional" : "1",
|
285 |
+
"dataType" : "Float16",
|
286 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
287 |
+
"shortDescription" : "",
|
288 |
+
"shape" : "[1, 32, 128, 511]",
|
289 |
+
"name" : "k_cache_1",
|
290 |
+
"type" : "MultiArray"
|
291 |
+
},
|
292 |
+
{
|
293 |
+
"hasShapeFlexibility" : "0",
|
294 |
+
"isOptional" : "1",
|
295 |
+
"dataType" : "Float16",
|
296 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
297 |
+
"shortDescription" : "",
|
298 |
+
"shape" : "[1, 32, 128, 511]",
|
299 |
+
"name" : "v_cache_1",
|
300 |
+
"type" : "MultiArray"
|
301 |
+
},
|
302 |
+
{
|
303 |
+
"hasShapeFlexibility" : "0",
|
304 |
+
"isOptional" : "1",
|
305 |
+
"dataType" : "Float16",
|
306 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
307 |
+
"shortDescription" : "",
|
308 |
+
"shape" : "[1, 32, 128, 511]",
|
309 |
+
"name" : "k_cache_2",
|
310 |
+
"type" : "MultiArray"
|
311 |
+
},
|
312 |
+
{
|
313 |
+
"hasShapeFlexibility" : "0",
|
314 |
+
"isOptional" : "1",
|
315 |
+
"dataType" : "Float16",
|
316 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
317 |
+
"shortDescription" : "",
|
318 |
+
"shape" : "[1, 32, 128, 511]",
|
319 |
+
"name" : "v_cache_2",
|
320 |
+
"type" : "MultiArray"
|
321 |
+
}
|
322 |
+
],
|
323 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
324 |
+
"storagePrecision" : "Float16",
|
325 |
+
"stateSchema" : [
|
326 |
+
|
327 |
+
],
|
328 |
+
"outputSchema" : [
|
329 |
+
{
|
330 |
+
"hasShapeFlexibility" : "0",
|
331 |
+
"isOptional" : "0",
|
332 |
+
"dataType" : "Float16",
|
333 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 1)",
|
334 |
+
"shortDescription" : "",
|
335 |
+
"shape" : "[1, 4096, 1, 1]",
|
336 |
+
"name" : "new_x",
|
337 |
+
"type" : "MultiArray"
|
338 |
+
},
|
339 |
+
{
|
340 |
+
"hasShapeFlexibility" : "0",
|
341 |
+
"isOptional" : "0",
|
342 |
+
"dataType" : "Float16",
|
343 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
344 |
+
"shortDescription" : "",
|
345 |
+
"shape" : "[1, 32, 128, 511]",
|
346 |
+
"name" : "new_k_cache_0",
|
347 |
+
"type" : "MultiArray"
|
348 |
+
},
|
349 |
+
{
|
350 |
+
"hasShapeFlexibility" : "0",
|
351 |
+
"isOptional" : "0",
|
352 |
+
"dataType" : "Float16",
|
353 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
354 |
+
"shortDescription" : "",
|
355 |
+
"shape" : "[1, 32, 128, 511]",
|
356 |
+
"name" : "new_k_cache_1",
|
357 |
+
"type" : "MultiArray"
|
358 |
+
},
|
359 |
+
{
|
360 |
+
"hasShapeFlexibility" : "0",
|
361 |
+
"isOptional" : "0",
|
362 |
+
"dataType" : "Float16",
|
363 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
364 |
+
"shortDescription" : "",
|
365 |
+
"shape" : "[1, 32, 128, 511]",
|
366 |
+
"name" : "new_k_cache_2",
|
367 |
+
"type" : "MultiArray"
|
368 |
+
},
|
369 |
+
{
|
370 |
+
"hasShapeFlexibility" : "0",
|
371 |
+
"isOptional" : "0",
|
372 |
+
"dataType" : "Float16",
|
373 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
374 |
+
"shortDescription" : "",
|
375 |
+
"shape" : "[1, 32, 128, 511]",
|
376 |
+
"name" : "new_v_cache_0",
|
377 |
+
"type" : "MultiArray"
|
378 |
+
},
|
379 |
+
{
|
380 |
+
"hasShapeFlexibility" : "0",
|
381 |
+
"isOptional" : "0",
|
382 |
+
"dataType" : "Float16",
|
383 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
384 |
+
"shortDescription" : "",
|
385 |
+
"shape" : "[1, 32, 128, 511]",
|
386 |
+
"name" : "new_v_cache_1",
|
387 |
+
"type" : "MultiArray"
|
388 |
+
},
|
389 |
+
{
|
390 |
+
"hasShapeFlexibility" : "0",
|
391 |
+
"isOptional" : "0",
|
392 |
+
"dataType" : "Float16",
|
393 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
394 |
+
"shortDescription" : "",
|
395 |
+
"shape" : "[1, 32, 128, 511]",
|
396 |
+
"name" : "new_v_cache_2",
|
397 |
+
"type" : "MultiArray"
|
398 |
+
}
|
399 |
+
],
|
400 |
+
"name" : "input_1_context_512",
|
401 |
+
"mlProgramOperationTypeHistogram" : {
|
402 |
+
"Ios18.constexprLutToDense" : 21,
|
403 |
+
"Ios18.conv" : 21,
|
404 |
+
"Ios18.matmul" : 6,
|
405 |
+
"Ios18.expandDims" : 6,
|
406 |
+
"Ios18.concat" : 18,
|
407 |
+
"Ios18.add" : 15,
|
408 |
+
"Ios18.realDiv" : 6,
|
409 |
+
"Ios18.silu" : 3,
|
410 |
+
"Ios18.softmax" : 3,
|
411 |
+
"Ios18.sliceByIndex" : 18,
|
412 |
+
"Ios16.reduceL2Norm" : 6,
|
413 |
+
"Ios18.squeeze" : 6,
|
414 |
+
"Ios18.reshape" : 12,
|
415 |
+
"Ios18.mul" : 57
|
416 |
+
}
|
417 |
+
}
|
418 |
+
],
|
419 |
+
"mlProgramOperationTypeHistogram" : {
|
420 |
+
"Ios18.constexprLutToDense" : 21,
|
421 |
+
"Ios18.conv" : 21,
|
422 |
+
"Ios18.matmul" : 6,
|
423 |
+
"Ios18.expandDims" : 6,
|
424 |
+
"Ios18.concat" : 18,
|
425 |
+
"Ios18.add" : 15,
|
426 |
+
"Ios18.realDiv" : 6,
|
427 |
+
"Ios18.silu" : 3,
|
428 |
+
"Ios18.softmax" : 3,
|
429 |
+
"Ios18.sliceByIndex" : 18,
|
430 |
+
"Ios16.reduceL2Norm" : 6,
|
431 |
+
"Ios18.squeeze" : 6,
|
432 |
+
"Ios18.reshape" : 12,
|
433 |
+
"Ios18.mul" : 57
|
434 |
+
},
|
435 |
+
"isUpdatable" : "0",
|
436 |
+
"stateSchema" : [
|
437 |
+
|
438 |
+
],
|
439 |
+
"availability" : {
|
440 |
+
"macOS" : "15.0",
|
441 |
+
"tvOS" : "18.0",
|
442 |
+
"visionOS" : "2.0",
|
443 |
+
"watchOS" : "11.0",
|
444 |
+
"iOS" : "18.0",
|
445 |
+
"macCatalyst" : "18.0"
|
446 |
+
},
|
447 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
448 |
+
"modelType" : {
|
449 |
+
"name" : "MLModelType_mlProgram"
|
450 |
+
},
|
451 |
+
"inputSchema" : [
|
452 |
+
{
|
453 |
+
"hasShapeFlexibility" : "0",
|
454 |
+
"isOptional" : "0",
|
455 |
+
"dataType" : "Float16",
|
456 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
457 |
+
"shortDescription" : "",
|
458 |
+
"shape" : "[1, 4096, 1, 512]",
|
459 |
+
"name" : "x",
|
460 |
+
"type" : "MultiArray"
|
461 |
+
},
|
462 |
+
{
|
463 |
+
"hasShapeFlexibility" : "0",
|
464 |
+
"isOptional" : "0",
|
465 |
+
"dataType" : "Float16",
|
466 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
467 |
+
"shortDescription" : "",
|
468 |
+
"shape" : "[128, 512]",
|
469 |
+
"name" : "cos",
|
470 |
+
"type" : "MultiArray"
|
471 |
+
},
|
472 |
+
{
|
473 |
+
"hasShapeFlexibility" : "0",
|
474 |
+
"isOptional" : "0",
|
475 |
+
"dataType" : "Float16",
|
476 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
477 |
+
"shortDescription" : "",
|
478 |
+
"shape" : "[128, 512]",
|
479 |
+
"name" : "sin",
|
480 |
+
"type" : "MultiArray"
|
481 |
+
},
|
482 |
+
{
|
483 |
+
"hasShapeFlexibility" : "0",
|
484 |
+
"isOptional" : "0",
|
485 |
+
"dataType" : "Float16",
|
486 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
487 |
+
"shortDescription" : "",
|
488 |
+
"shape" : "[1, 1, 512, 512]",
|
489 |
+
"name" : "mask",
|
490 |
+
"type" : "MultiArray"
|
491 |
+
}
|
492 |
+
],
|
493 |
+
"defaultFunctionName" : "input_512_context_512",
|
494 |
+
"generatedClassName" : "Llama_2_7b_hf_2024_07_02_20_36_17_merged_chunk11",
|
495 |
+
"userDefinedMetadata" : {
|
496 |
+
|
497 |
+
},
|
498 |
+
"method" : "predict"
|
499 |
+
}
|
500 |
+
]
|
sequoia/Llama-2-7b-hf_chunk11.mlmodelc/model.mil
ADDED
The diff for this file is too large to render.
See raw diff
|
|
sequoia/Llama-2-7b-hf_chunk11.mlmodelc/weights/weight.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:09f7120b0957def3b75cd81f0fc797a1bdbb8be6c0f0834e3e514bc60a2198bf
|
3 |
+
size 303876448
|
sequoia/Llama-2-7b-hf_chunk12.mlmodelc/analytics/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:877129a9d42c3d4d9b1b793d51e152d6fed08881a973bbb5ed4a001571623eb0
|
3 |
+
size 243
|
sequoia/Llama-2-7b-hf_chunk12.mlmodelc/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7e4186acc6251c3785f2b0af36e33eacfe6b4f78971ae86bda2e885776607d79
|
3 |
+
size 831
|
sequoia/Llama-2-7b-hf_chunk12.mlmodelc/metadata.json
ADDED
@@ -0,0 +1,423 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 × 512 × 32000)",
|
11 |
+
"shortDescription" : "",
|
12 |
+
"shape" : "[1, 512, 32000]",
|
13 |
+
"name" : "logits",
|
14 |
+
"type" : "MultiArray"
|
15 |
+
},
|
16 |
+
{
|
17 |
+
"hasShapeFlexibility" : "0",
|
18 |
+
"isOptional" : "0",
|
19 |
+
"dataType" : "Float16",
|
20 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
21 |
+
"shortDescription" : "",
|
22 |
+
"shape" : "[1, 32, 128, 511]",
|
23 |
+
"name" : "new_k_cache_0",
|
24 |
+
"type" : "MultiArray"
|
25 |
+
},
|
26 |
+
{
|
27 |
+
"hasShapeFlexibility" : "0",
|
28 |
+
"isOptional" : "0",
|
29 |
+
"dataType" : "Float16",
|
30 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
31 |
+
"shortDescription" : "",
|
32 |
+
"shape" : "[1, 32, 128, 511]",
|
33 |
+
"name" : "new_k_cache_1",
|
34 |
+
"type" : "MultiArray"
|
35 |
+
},
|
36 |
+
{
|
37 |
+
"hasShapeFlexibility" : "0",
|
38 |
+
"isOptional" : "0",
|
39 |
+
"dataType" : "Float16",
|
40 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
41 |
+
"shortDescription" : "",
|
42 |
+
"shape" : "[1, 32, 128, 511]",
|
43 |
+
"name" : "new_v_cache_0",
|
44 |
+
"type" : "MultiArray"
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"hasShapeFlexibility" : "0",
|
48 |
+
"isOptional" : "0",
|
49 |
+
"dataType" : "Float16",
|
50 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
51 |
+
"shortDescription" : "",
|
52 |
+
"shape" : "[1, 32, 128, 511]",
|
53 |
+
"name" : "new_v_cache_1",
|
54 |
+
"type" : "MultiArray"
|
55 |
+
}
|
56 |
+
],
|
57 |
+
"modelParameters" : [
|
58 |
+
|
59 |
+
],
|
60 |
+
"specificationVersion" : 9,
|
61 |
+
"functions" : [
|
62 |
+
{
|
63 |
+
"inputSchema" : [
|
64 |
+
{
|
65 |
+
"hasShapeFlexibility" : "0",
|
66 |
+
"isOptional" : "0",
|
67 |
+
"dataType" : "Float16",
|
68 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
69 |
+
"shortDescription" : "",
|
70 |
+
"shape" : "[1, 4096, 1, 512]",
|
71 |
+
"name" : "x",
|
72 |
+
"type" : "MultiArray"
|
73 |
+
},
|
74 |
+
{
|
75 |
+
"hasShapeFlexibility" : "0",
|
76 |
+
"isOptional" : "0",
|
77 |
+
"dataType" : "Float16",
|
78 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
79 |
+
"shortDescription" : "",
|
80 |
+
"shape" : "[128, 512]",
|
81 |
+
"name" : "cos",
|
82 |
+
"type" : "MultiArray"
|
83 |
+
},
|
84 |
+
{
|
85 |
+
"hasShapeFlexibility" : "0",
|
86 |
+
"isOptional" : "0",
|
87 |
+
"dataType" : "Float16",
|
88 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
89 |
+
"shortDescription" : "",
|
90 |
+
"shape" : "[128, 512]",
|
91 |
+
"name" : "sin",
|
92 |
+
"type" : "MultiArray"
|
93 |
+
},
|
94 |
+
{
|
95 |
+
"hasShapeFlexibility" : "0",
|
96 |
+
"isOptional" : "0",
|
97 |
+
"dataType" : "Float16",
|
98 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
99 |
+
"shortDescription" : "",
|
100 |
+
"shape" : "[1, 1, 512, 512]",
|
101 |
+
"name" : "mask",
|
102 |
+
"type" : "MultiArray"
|
103 |
+
}
|
104 |
+
],
|
105 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
106 |
+
"storagePrecision" : "Float16",
|
107 |
+
"stateSchema" : [
|
108 |
+
|
109 |
+
],
|
110 |
+
"outputSchema" : [
|
111 |
+
{
|
112 |
+
"hasShapeFlexibility" : "0",
|
113 |
+
"isOptional" : "0",
|
114 |
+
"dataType" : "Float16",
|
115 |
+
"formattedType" : "MultiArray (Float16 1 × 512 × 32000)",
|
116 |
+
"shortDescription" : "",
|
117 |
+
"shape" : "[1, 512, 32000]",
|
118 |
+
"name" : "logits",
|
119 |
+
"type" : "MultiArray"
|
120 |
+
},
|
121 |
+
{
|
122 |
+
"hasShapeFlexibility" : "0",
|
123 |
+
"isOptional" : "0",
|
124 |
+
"dataType" : "Float16",
|
125 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
126 |
+
"shortDescription" : "",
|
127 |
+
"shape" : "[1, 32, 128, 511]",
|
128 |
+
"name" : "new_k_cache_0",
|
129 |
+
"type" : "MultiArray"
|
130 |
+
},
|
131 |
+
{
|
132 |
+
"hasShapeFlexibility" : "0",
|
133 |
+
"isOptional" : "0",
|
134 |
+
"dataType" : "Float16",
|
135 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
136 |
+
"shortDescription" : "",
|
137 |
+
"shape" : "[1, 32, 128, 511]",
|
138 |
+
"name" : "new_k_cache_1",
|
139 |
+
"type" : "MultiArray"
|
140 |
+
},
|
141 |
+
{
|
142 |
+
"hasShapeFlexibility" : "0",
|
143 |
+
"isOptional" : "0",
|
144 |
+
"dataType" : "Float16",
|
145 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
146 |
+
"shortDescription" : "",
|
147 |
+
"shape" : "[1, 32, 128, 511]",
|
148 |
+
"name" : "new_v_cache_0",
|
149 |
+
"type" : "MultiArray"
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"hasShapeFlexibility" : "0",
|
153 |
+
"isOptional" : "0",
|
154 |
+
"dataType" : "Float16",
|
155 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
156 |
+
"shortDescription" : "",
|
157 |
+
"shape" : "[1, 32, 128, 511]",
|
158 |
+
"name" : "new_v_cache_1",
|
159 |
+
"type" : "MultiArray"
|
160 |
+
}
|
161 |
+
],
|
162 |
+
"name" : "input_512_context_512",
|
163 |
+
"mlProgramOperationTypeHistogram" : {
|
164 |
+
"Ios18.constexprLutToDense" : 14,
|
165 |
+
"Ios18.conv" : 14,
|
166 |
+
"Ios18.matmul" : 6,
|
167 |
+
"Ios18.expandDims" : 5,
|
168 |
+
"Ios18.concat" : 14,
|
169 |
+
"Ios18.add" : 10,
|
170 |
+
"Ios18.realDiv" : 5,
|
171 |
+
"Ios18.silu" : 2,
|
172 |
+
"Ios18.softmax" : 2,
|
173 |
+
"Ios18.sliceByIndex" : 12,
|
174 |
+
"Ios18.transpose" : 1,
|
175 |
+
"Ios16.reduceL2Norm" : 5,
|
176 |
+
"Ios18.squeeze" : 6,
|
177 |
+
"Ios18.reshape" : 11,
|
178 |
+
"Ios18.mul" : 40
|
179 |
+
}
|
180 |
+
},
|
181 |
+
{
|
182 |
+
"inputSchema" : [
|
183 |
+
{
|
184 |
+
"hasShapeFlexibility" : "0",
|
185 |
+
"isOptional" : "0",
|
186 |
+
"dataType" : "Float16",
|
187 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 1)",
|
188 |
+
"shortDescription" : "",
|
189 |
+
"shape" : "[1, 4096, 1, 1]",
|
190 |
+
"name" : "x",
|
191 |
+
"type" : "MultiArray"
|
192 |
+
},
|
193 |
+
{
|
194 |
+
"hasShapeFlexibility" : "0",
|
195 |
+
"isOptional" : "0",
|
196 |
+
"dataType" : "Float16",
|
197 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
198 |
+
"shortDescription" : "",
|
199 |
+
"shape" : "[128, 1]",
|
200 |
+
"name" : "cos",
|
201 |
+
"type" : "MultiArray"
|
202 |
+
},
|
203 |
+
{
|
204 |
+
"hasShapeFlexibility" : "0",
|
205 |
+
"isOptional" : "0",
|
206 |
+
"dataType" : "Float16",
|
207 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
208 |
+
"shortDescription" : "",
|
209 |
+
"shape" : "[128, 1]",
|
210 |
+
"name" : "sin",
|
211 |
+
"type" : "MultiArray"
|
212 |
+
},
|
213 |
+
{
|
214 |
+
"hasShapeFlexibility" : "0",
|
215 |
+
"isOptional" : "0",
|
216 |
+
"dataType" : "Float16",
|
217 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 1 × 512)",
|
218 |
+
"shortDescription" : "",
|
219 |
+
"shape" : "[1, 1, 1, 512]",
|
220 |
+
"name" : "mask",
|
221 |
+
"type" : "MultiArray"
|
222 |
+
},
|
223 |
+
{
|
224 |
+
"hasShapeFlexibility" : "0",
|
225 |
+
"isOptional" : "1",
|
226 |
+
"dataType" : "Float16",
|
227 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
228 |
+
"shortDescription" : "",
|
229 |
+
"shape" : "[1, 32, 128, 511]",
|
230 |
+
"name" : "k_cache_0",
|
231 |
+
"type" : "MultiArray"
|
232 |
+
},
|
233 |
+
{
|
234 |
+
"hasShapeFlexibility" : "0",
|
235 |
+
"isOptional" : "1",
|
236 |
+
"dataType" : "Float16",
|
237 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
238 |
+
"shortDescription" : "",
|
239 |
+
"shape" : "[1, 32, 128, 511]",
|
240 |
+
"name" : "v_cache_0",
|
241 |
+
"type" : "MultiArray"
|
242 |
+
},
|
243 |
+
{
|
244 |
+
"hasShapeFlexibility" : "0",
|
245 |
+
"isOptional" : "1",
|
246 |
+
"dataType" : "Float16",
|
247 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
248 |
+
"shortDescription" : "",
|
249 |
+
"shape" : "[1, 32, 128, 511]",
|
250 |
+
"name" : "k_cache_1",
|
251 |
+
"type" : "MultiArray"
|
252 |
+
},
|
253 |
+
{
|
254 |
+
"hasShapeFlexibility" : "0",
|
255 |
+
"isOptional" : "1",
|
256 |
+
"dataType" : "Float16",
|
257 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
258 |
+
"shortDescription" : "",
|
259 |
+
"shape" : "[1, 32, 128, 511]",
|
260 |
+
"name" : "v_cache_1",
|
261 |
+
"type" : "MultiArray"
|
262 |
+
}
|
263 |
+
],
|
264 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
265 |
+
"storagePrecision" : "Float16",
|
266 |
+
"stateSchema" : [
|
267 |
+
|
268 |
+
],
|
269 |
+
"outputSchema" : [
|
270 |
+
{
|
271 |
+
"hasShapeFlexibility" : "0",
|
272 |
+
"isOptional" : "0",
|
273 |
+
"dataType" : "Float16",
|
274 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 32000)",
|
275 |
+
"shortDescription" : "",
|
276 |
+
"shape" : "[1, 1, 32000]",
|
277 |
+
"name" : "logits",
|
278 |
+
"type" : "MultiArray"
|
279 |
+
},
|
280 |
+
{
|
281 |
+
"hasShapeFlexibility" : "0",
|
282 |
+
"isOptional" : "0",
|
283 |
+
"dataType" : "Float16",
|
284 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
285 |
+
"shortDescription" : "",
|
286 |
+
"shape" : "[1, 32, 128, 511]",
|
287 |
+
"name" : "new_k_cache_0",
|
288 |
+
"type" : "MultiArray"
|
289 |
+
},
|
290 |
+
{
|
291 |
+
"hasShapeFlexibility" : "0",
|
292 |
+
"isOptional" : "0",
|
293 |
+
"dataType" : "Float16",
|
294 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
295 |
+
"shortDescription" : "",
|
296 |
+
"shape" : "[1, 32, 128, 511]",
|
297 |
+
"name" : "new_k_cache_1",
|
298 |
+
"type" : "MultiArray"
|
299 |
+
},
|
300 |
+
{
|
301 |
+
"hasShapeFlexibility" : "0",
|
302 |
+
"isOptional" : "0",
|
303 |
+
"dataType" : "Float16",
|
304 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
305 |
+
"shortDescription" : "",
|
306 |
+
"shape" : "[1, 32, 128, 511]",
|
307 |
+
"name" : "new_v_cache_0",
|
308 |
+
"type" : "MultiArray"
|
309 |
+
},
|
310 |
+
{
|
311 |
+
"hasShapeFlexibility" : "0",
|
312 |
+
"isOptional" : "0",
|
313 |
+
"dataType" : "Float16",
|
314 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
315 |
+
"shortDescription" : "",
|
316 |
+
"shape" : "[1, 32, 128, 511]",
|
317 |
+
"name" : "new_v_cache_1",
|
318 |
+
"type" : "MultiArray"
|
319 |
+
}
|
320 |
+
],
|
321 |
+
"name" : "input_1_context_512",
|
322 |
+
"mlProgramOperationTypeHistogram" : {
|
323 |
+
"Ios18.constexprLutToDense" : 14,
|
324 |
+
"Ios18.conv" : 14,
|
325 |
+
"Ios18.matmul" : 6,
|
326 |
+
"Ios18.expandDims" : 5,
|
327 |
+
"Ios18.concat" : 14,
|
328 |
+
"Ios18.add" : 10,
|
329 |
+
"Ios18.realDiv" : 5,
|
330 |
+
"Ios18.silu" : 2,
|
331 |
+
"Ios18.softmax" : 2,
|
332 |
+
"Ios18.sliceByIndex" : 12,
|
333 |
+
"Ios18.transpose" : 1,
|
334 |
+
"Ios16.reduceL2Norm" : 5,
|
335 |
+
"Ios18.squeeze" : 6,
|
336 |
+
"Ios18.reshape" : 11,
|
337 |
+
"Ios18.mul" : 40
|
338 |
+
}
|
339 |
+
}
|
340 |
+
],
|
341 |
+
"mlProgramOperationTypeHistogram" : {
|
342 |
+
"Ios18.constexprLutToDense" : 14,
|
343 |
+
"Ios18.conv" : 14,
|
344 |
+
"Ios18.matmul" : 6,
|
345 |
+
"Ios18.expandDims" : 5,
|
346 |
+
"Ios18.concat" : 14,
|
347 |
+
"Ios18.add" : 10,
|
348 |
+
"Ios18.realDiv" : 5,
|
349 |
+
"Ios18.silu" : 2,
|
350 |
+
"Ios18.softmax" : 2,
|
351 |
+
"Ios18.sliceByIndex" : 12,
|
352 |
+
"Ios18.transpose" : 1,
|
353 |
+
"Ios16.reduceL2Norm" : 5,
|
354 |
+
"Ios18.squeeze" : 6,
|
355 |
+
"Ios18.reshape" : 11,
|
356 |
+
"Ios18.mul" : 40
|
357 |
+
},
|
358 |
+
"isUpdatable" : "0",
|
359 |
+
"stateSchema" : [
|
360 |
+
|
361 |
+
],
|
362 |
+
"availability" : {
|
363 |
+
"macOS" : "15.0",
|
364 |
+
"tvOS" : "18.0",
|
365 |
+
"visionOS" : "2.0",
|
366 |
+
"watchOS" : "11.0",
|
367 |
+
"iOS" : "18.0",
|
368 |
+
"macCatalyst" : "18.0"
|
369 |
+
},
|
370 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
371 |
+
"modelType" : {
|
372 |
+
"name" : "MLModelType_mlProgram"
|
373 |
+
},
|
374 |
+
"inputSchema" : [
|
375 |
+
{
|
376 |
+
"hasShapeFlexibility" : "0",
|
377 |
+
"isOptional" : "0",
|
378 |
+
"dataType" : "Float16",
|
379 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
380 |
+
"shortDescription" : "",
|
381 |
+
"shape" : "[1, 4096, 1, 512]",
|
382 |
+
"name" : "x",
|
383 |
+
"type" : "MultiArray"
|
384 |
+
},
|
385 |
+
{
|
386 |
+
"hasShapeFlexibility" : "0",
|
387 |
+
"isOptional" : "0",
|
388 |
+
"dataType" : "Float16",
|
389 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
390 |
+
"shortDescription" : "",
|
391 |
+
"shape" : "[128, 512]",
|
392 |
+
"name" : "cos",
|
393 |
+
"type" : "MultiArray"
|
394 |
+
},
|
395 |
+
{
|
396 |
+
"hasShapeFlexibility" : "0",
|
397 |
+
"isOptional" : "0",
|
398 |
+
"dataType" : "Float16",
|
399 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
400 |
+
"shortDescription" : "",
|
401 |
+
"shape" : "[128, 512]",
|
402 |
+
"name" : "sin",
|
403 |
+
"type" : "MultiArray"
|
404 |
+
},
|
405 |
+
{
|
406 |
+
"hasShapeFlexibility" : "0",
|
407 |
+
"isOptional" : "0",
|
408 |
+
"dataType" : "Float16",
|
409 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
410 |
+
"shortDescription" : "",
|
411 |
+
"shape" : "[1, 1, 512, 512]",
|
412 |
+
"name" : "mask",
|
413 |
+
"type" : "MultiArray"
|
414 |
+
}
|
415 |
+
],
|
416 |
+
"defaultFunctionName" : "input_512_context_512",
|
417 |
+
"generatedClassName" : "Llama_2_7b_hf_2024_07_02_20_36_17_merged_chunk12",
|
418 |
+
"userDefinedMetadata" : {
|
419 |
+
|
420 |
+
},
|
421 |
+
"method" : "predict"
|
422 |
+
}
|
423 |
+
]
|
sequoia/Llama-2-7b-hf_chunk12.mlmodelc/model.mil
ADDED
The diff for this file is too large to render.
See raw diff
|
|
sequoia/Llama-2-7b-hf_chunk12.mlmodelc/weights/weight.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:71cb014a89f49d984f26bda21c23a7fe591df5d08a82fd42fc81a071a333a802
|
3 |
+
size 464737056
|
sequoia/Llama-2-7b-hf_chunk2.mlmodelc/analytics/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:84e317a82cdf4e96f808f63e77f10098844d47ad522545181edfac4d287c9c92
|
3 |
+
size 243
|
sequoia/Llama-2-7b-hf_chunk2.mlmodelc/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e430d0795ff5c384187174f5718a2c13d0070f5d6a811831e18862497865a86d
|
3 |
+
size 1037
|
sequoia/Llama-2-7b-hf_chunk2.mlmodelc/metadata.json
ADDED
@@ -0,0 +1,500 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 × 4096 × 1 × 512)",
|
11 |
+
"shortDescription" : "",
|
12 |
+
"shape" : "[1, 4096, 1, 512]",
|
13 |
+
"name" : "new_x",
|
14 |
+
"type" : "MultiArray"
|
15 |
+
},
|
16 |
+
{
|
17 |
+
"hasShapeFlexibility" : "0",
|
18 |
+
"isOptional" : "0",
|
19 |
+
"dataType" : "Float16",
|
20 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
21 |
+
"shortDescription" : "",
|
22 |
+
"shape" : "[1, 32, 128, 511]",
|
23 |
+
"name" : "new_k_cache_0",
|
24 |
+
"type" : "MultiArray"
|
25 |
+
},
|
26 |
+
{
|
27 |
+
"hasShapeFlexibility" : "0",
|
28 |
+
"isOptional" : "0",
|
29 |
+
"dataType" : "Float16",
|
30 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
31 |
+
"shortDescription" : "",
|
32 |
+
"shape" : "[1, 32, 128, 511]",
|
33 |
+
"name" : "new_k_cache_1",
|
34 |
+
"type" : "MultiArray"
|
35 |
+
},
|
36 |
+
{
|
37 |
+
"hasShapeFlexibility" : "0",
|
38 |
+
"isOptional" : "0",
|
39 |
+
"dataType" : "Float16",
|
40 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
41 |
+
"shortDescription" : "",
|
42 |
+
"shape" : "[1, 32, 128, 511]",
|
43 |
+
"name" : "new_k_cache_2",
|
44 |
+
"type" : "MultiArray"
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"hasShapeFlexibility" : "0",
|
48 |
+
"isOptional" : "0",
|
49 |
+
"dataType" : "Float16",
|
50 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
51 |
+
"shortDescription" : "",
|
52 |
+
"shape" : "[1, 32, 128, 511]",
|
53 |
+
"name" : "new_v_cache_0",
|
54 |
+
"type" : "MultiArray"
|
55 |
+
},
|
56 |
+
{
|
57 |
+
"hasShapeFlexibility" : "0",
|
58 |
+
"isOptional" : "0",
|
59 |
+
"dataType" : "Float16",
|
60 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
61 |
+
"shortDescription" : "",
|
62 |
+
"shape" : "[1, 32, 128, 511]",
|
63 |
+
"name" : "new_v_cache_1",
|
64 |
+
"type" : "MultiArray"
|
65 |
+
},
|
66 |
+
{
|
67 |
+
"hasShapeFlexibility" : "0",
|
68 |
+
"isOptional" : "0",
|
69 |
+
"dataType" : "Float16",
|
70 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
71 |
+
"shortDescription" : "",
|
72 |
+
"shape" : "[1, 32, 128, 511]",
|
73 |
+
"name" : "new_v_cache_2",
|
74 |
+
"type" : "MultiArray"
|
75 |
+
}
|
76 |
+
],
|
77 |
+
"modelParameters" : [
|
78 |
+
|
79 |
+
],
|
80 |
+
"specificationVersion" : 9,
|
81 |
+
"functions" : [
|
82 |
+
{
|
83 |
+
"inputSchema" : [
|
84 |
+
{
|
85 |
+
"hasShapeFlexibility" : "0",
|
86 |
+
"isOptional" : "0",
|
87 |
+
"dataType" : "Float16",
|
88 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
89 |
+
"shortDescription" : "",
|
90 |
+
"shape" : "[1, 4096, 1, 512]",
|
91 |
+
"name" : "x",
|
92 |
+
"type" : "MultiArray"
|
93 |
+
},
|
94 |
+
{
|
95 |
+
"hasShapeFlexibility" : "0",
|
96 |
+
"isOptional" : "0",
|
97 |
+
"dataType" : "Float16",
|
98 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
99 |
+
"shortDescription" : "",
|
100 |
+
"shape" : "[128, 512]",
|
101 |
+
"name" : "cos",
|
102 |
+
"type" : "MultiArray"
|
103 |
+
},
|
104 |
+
{
|
105 |
+
"hasShapeFlexibility" : "0",
|
106 |
+
"isOptional" : "0",
|
107 |
+
"dataType" : "Float16",
|
108 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
109 |
+
"shortDescription" : "",
|
110 |
+
"shape" : "[128, 512]",
|
111 |
+
"name" : "sin",
|
112 |
+
"type" : "MultiArray"
|
113 |
+
},
|
114 |
+
{
|
115 |
+
"hasShapeFlexibility" : "0",
|
116 |
+
"isOptional" : "0",
|
117 |
+
"dataType" : "Float16",
|
118 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
119 |
+
"shortDescription" : "",
|
120 |
+
"shape" : "[1, 1, 512, 512]",
|
121 |
+
"name" : "mask",
|
122 |
+
"type" : "MultiArray"
|
123 |
+
}
|
124 |
+
],
|
125 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
126 |
+
"storagePrecision" : "Float16",
|
127 |
+
"stateSchema" : [
|
128 |
+
|
129 |
+
],
|
130 |
+
"outputSchema" : [
|
131 |
+
{
|
132 |
+
"hasShapeFlexibility" : "0",
|
133 |
+
"isOptional" : "0",
|
134 |
+
"dataType" : "Float16",
|
135 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
136 |
+
"shortDescription" : "",
|
137 |
+
"shape" : "[1, 4096, 1, 512]",
|
138 |
+
"name" : "new_x",
|
139 |
+
"type" : "MultiArray"
|
140 |
+
},
|
141 |
+
{
|
142 |
+
"hasShapeFlexibility" : "0",
|
143 |
+
"isOptional" : "0",
|
144 |
+
"dataType" : "Float16",
|
145 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
146 |
+
"shortDescription" : "",
|
147 |
+
"shape" : "[1, 32, 128, 511]",
|
148 |
+
"name" : "new_k_cache_0",
|
149 |
+
"type" : "MultiArray"
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"hasShapeFlexibility" : "0",
|
153 |
+
"isOptional" : "0",
|
154 |
+
"dataType" : "Float16",
|
155 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
156 |
+
"shortDescription" : "",
|
157 |
+
"shape" : "[1, 32, 128, 511]",
|
158 |
+
"name" : "new_k_cache_1",
|
159 |
+
"type" : "MultiArray"
|
160 |
+
},
|
161 |
+
{
|
162 |
+
"hasShapeFlexibility" : "0",
|
163 |
+
"isOptional" : "0",
|
164 |
+
"dataType" : "Float16",
|
165 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
166 |
+
"shortDescription" : "",
|
167 |
+
"shape" : "[1, 32, 128, 511]",
|
168 |
+
"name" : "new_k_cache_2",
|
169 |
+
"type" : "MultiArray"
|
170 |
+
},
|
171 |
+
{
|
172 |
+
"hasShapeFlexibility" : "0",
|
173 |
+
"isOptional" : "0",
|
174 |
+
"dataType" : "Float16",
|
175 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
176 |
+
"shortDescription" : "",
|
177 |
+
"shape" : "[1, 32, 128, 511]",
|
178 |
+
"name" : "new_v_cache_0",
|
179 |
+
"type" : "MultiArray"
|
180 |
+
},
|
181 |
+
{
|
182 |
+
"hasShapeFlexibility" : "0",
|
183 |
+
"isOptional" : "0",
|
184 |
+
"dataType" : "Float16",
|
185 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
186 |
+
"shortDescription" : "",
|
187 |
+
"shape" : "[1, 32, 128, 511]",
|
188 |
+
"name" : "new_v_cache_1",
|
189 |
+
"type" : "MultiArray"
|
190 |
+
},
|
191 |
+
{
|
192 |
+
"hasShapeFlexibility" : "0",
|
193 |
+
"isOptional" : "0",
|
194 |
+
"dataType" : "Float16",
|
195 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
196 |
+
"shortDescription" : "",
|
197 |
+
"shape" : "[1, 32, 128, 511]",
|
198 |
+
"name" : "new_v_cache_2",
|
199 |
+
"type" : "MultiArray"
|
200 |
+
}
|
201 |
+
],
|
202 |
+
"name" : "input_512_context_512",
|
203 |
+
"mlProgramOperationTypeHistogram" : {
|
204 |
+
"Ios18.constexprLutToDense" : 21,
|
205 |
+
"Ios18.conv" : 21,
|
206 |
+
"Ios18.matmul" : 6,
|
207 |
+
"Ios18.expandDims" : 6,
|
208 |
+
"Ios18.concat" : 18,
|
209 |
+
"Ios18.add" : 15,
|
210 |
+
"Ios18.realDiv" : 6,
|
211 |
+
"Ios18.silu" : 3,
|
212 |
+
"Ios18.softmax" : 3,
|
213 |
+
"Ios18.sliceByIndex" : 18,
|
214 |
+
"Ios16.reduceL2Norm" : 6,
|
215 |
+
"Ios18.squeeze" : 6,
|
216 |
+
"Ios18.reshape" : 12,
|
217 |
+
"Ios18.mul" : 57
|
218 |
+
}
|
219 |
+
},
|
220 |
+
{
|
221 |
+
"inputSchema" : [
|
222 |
+
{
|
223 |
+
"hasShapeFlexibility" : "0",
|
224 |
+
"isOptional" : "0",
|
225 |
+
"dataType" : "Float16",
|
226 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 1)",
|
227 |
+
"shortDescription" : "",
|
228 |
+
"shape" : "[1, 4096, 1, 1]",
|
229 |
+
"name" : "x",
|
230 |
+
"type" : "MultiArray"
|
231 |
+
},
|
232 |
+
{
|
233 |
+
"hasShapeFlexibility" : "0",
|
234 |
+
"isOptional" : "0",
|
235 |
+
"dataType" : "Float16",
|
236 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
237 |
+
"shortDescription" : "",
|
238 |
+
"shape" : "[128, 1]",
|
239 |
+
"name" : "cos",
|
240 |
+
"type" : "MultiArray"
|
241 |
+
},
|
242 |
+
{
|
243 |
+
"hasShapeFlexibility" : "0",
|
244 |
+
"isOptional" : "0",
|
245 |
+
"dataType" : "Float16",
|
246 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
247 |
+
"shortDescription" : "",
|
248 |
+
"shape" : "[128, 1]",
|
249 |
+
"name" : "sin",
|
250 |
+
"type" : "MultiArray"
|
251 |
+
},
|
252 |
+
{
|
253 |
+
"hasShapeFlexibility" : "0",
|
254 |
+
"isOptional" : "0",
|
255 |
+
"dataType" : "Float16",
|
256 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 1 × 512)",
|
257 |
+
"shortDescription" : "",
|
258 |
+
"shape" : "[1, 1, 1, 512]",
|
259 |
+
"name" : "mask",
|
260 |
+
"type" : "MultiArray"
|
261 |
+
},
|
262 |
+
{
|
263 |
+
"hasShapeFlexibility" : "0",
|
264 |
+
"isOptional" : "1",
|
265 |
+
"dataType" : "Float16",
|
266 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
267 |
+
"shortDescription" : "",
|
268 |
+
"shape" : "[1, 32, 128, 511]",
|
269 |
+
"name" : "k_cache_0",
|
270 |
+
"type" : "MultiArray"
|
271 |
+
},
|
272 |
+
{
|
273 |
+
"hasShapeFlexibility" : "0",
|
274 |
+
"isOptional" : "1",
|
275 |
+
"dataType" : "Float16",
|
276 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
277 |
+
"shortDescription" : "",
|
278 |
+
"shape" : "[1, 32, 128, 511]",
|
279 |
+
"name" : "v_cache_0",
|
280 |
+
"type" : "MultiArray"
|
281 |
+
},
|
282 |
+
{
|
283 |
+
"hasShapeFlexibility" : "0",
|
284 |
+
"isOptional" : "1",
|
285 |
+
"dataType" : "Float16",
|
286 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
287 |
+
"shortDescription" : "",
|
288 |
+
"shape" : "[1, 32, 128, 511]",
|
289 |
+
"name" : "k_cache_1",
|
290 |
+
"type" : "MultiArray"
|
291 |
+
},
|
292 |
+
{
|
293 |
+
"hasShapeFlexibility" : "0",
|
294 |
+
"isOptional" : "1",
|
295 |
+
"dataType" : "Float16",
|
296 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
297 |
+
"shortDescription" : "",
|
298 |
+
"shape" : "[1, 32, 128, 511]",
|
299 |
+
"name" : "v_cache_1",
|
300 |
+
"type" : "MultiArray"
|
301 |
+
},
|
302 |
+
{
|
303 |
+
"hasShapeFlexibility" : "0",
|
304 |
+
"isOptional" : "1",
|
305 |
+
"dataType" : "Float16",
|
306 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
307 |
+
"shortDescription" : "",
|
308 |
+
"shape" : "[1, 32, 128, 511]",
|
309 |
+
"name" : "k_cache_2",
|
310 |
+
"type" : "MultiArray"
|
311 |
+
},
|
312 |
+
{
|
313 |
+
"hasShapeFlexibility" : "0",
|
314 |
+
"isOptional" : "1",
|
315 |
+
"dataType" : "Float16",
|
316 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
317 |
+
"shortDescription" : "",
|
318 |
+
"shape" : "[1, 32, 128, 511]",
|
319 |
+
"name" : "v_cache_2",
|
320 |
+
"type" : "MultiArray"
|
321 |
+
}
|
322 |
+
],
|
323 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
324 |
+
"storagePrecision" : "Float16",
|
325 |
+
"stateSchema" : [
|
326 |
+
|
327 |
+
],
|
328 |
+
"outputSchema" : [
|
329 |
+
{
|
330 |
+
"hasShapeFlexibility" : "0",
|
331 |
+
"isOptional" : "0",
|
332 |
+
"dataType" : "Float16",
|
333 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 1)",
|
334 |
+
"shortDescription" : "",
|
335 |
+
"shape" : "[1, 4096, 1, 1]",
|
336 |
+
"name" : "new_x",
|
337 |
+
"type" : "MultiArray"
|
338 |
+
},
|
339 |
+
{
|
340 |
+
"hasShapeFlexibility" : "0",
|
341 |
+
"isOptional" : "0",
|
342 |
+
"dataType" : "Float16",
|
343 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
344 |
+
"shortDescription" : "",
|
345 |
+
"shape" : "[1, 32, 128, 511]",
|
346 |
+
"name" : "new_k_cache_0",
|
347 |
+
"type" : "MultiArray"
|
348 |
+
},
|
349 |
+
{
|
350 |
+
"hasShapeFlexibility" : "0",
|
351 |
+
"isOptional" : "0",
|
352 |
+
"dataType" : "Float16",
|
353 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
354 |
+
"shortDescription" : "",
|
355 |
+
"shape" : "[1, 32, 128, 511]",
|
356 |
+
"name" : "new_k_cache_1",
|
357 |
+
"type" : "MultiArray"
|
358 |
+
},
|
359 |
+
{
|
360 |
+
"hasShapeFlexibility" : "0",
|
361 |
+
"isOptional" : "0",
|
362 |
+
"dataType" : "Float16",
|
363 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
364 |
+
"shortDescription" : "",
|
365 |
+
"shape" : "[1, 32, 128, 511]",
|
366 |
+
"name" : "new_k_cache_2",
|
367 |
+
"type" : "MultiArray"
|
368 |
+
},
|
369 |
+
{
|
370 |
+
"hasShapeFlexibility" : "0",
|
371 |
+
"isOptional" : "0",
|
372 |
+
"dataType" : "Float16",
|
373 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
374 |
+
"shortDescription" : "",
|
375 |
+
"shape" : "[1, 32, 128, 511]",
|
376 |
+
"name" : "new_v_cache_0",
|
377 |
+
"type" : "MultiArray"
|
378 |
+
},
|
379 |
+
{
|
380 |
+
"hasShapeFlexibility" : "0",
|
381 |
+
"isOptional" : "0",
|
382 |
+
"dataType" : "Float16",
|
383 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
384 |
+
"shortDescription" : "",
|
385 |
+
"shape" : "[1, 32, 128, 511]",
|
386 |
+
"name" : "new_v_cache_1",
|
387 |
+
"type" : "MultiArray"
|
388 |
+
},
|
389 |
+
{
|
390 |
+
"hasShapeFlexibility" : "0",
|
391 |
+
"isOptional" : "0",
|
392 |
+
"dataType" : "Float16",
|
393 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
394 |
+
"shortDescription" : "",
|
395 |
+
"shape" : "[1, 32, 128, 511]",
|
396 |
+
"name" : "new_v_cache_2",
|
397 |
+
"type" : "MultiArray"
|
398 |
+
}
|
399 |
+
],
|
400 |
+
"name" : "input_1_context_512",
|
401 |
+
"mlProgramOperationTypeHistogram" : {
|
402 |
+
"Ios18.constexprLutToDense" : 21,
|
403 |
+
"Ios18.conv" : 21,
|
404 |
+
"Ios18.matmul" : 6,
|
405 |
+
"Ios18.expandDims" : 6,
|
406 |
+
"Ios18.concat" : 18,
|
407 |
+
"Ios18.add" : 15,
|
408 |
+
"Ios18.realDiv" : 6,
|
409 |
+
"Ios18.silu" : 3,
|
410 |
+
"Ios18.softmax" : 3,
|
411 |
+
"Ios18.sliceByIndex" : 18,
|
412 |
+
"Ios16.reduceL2Norm" : 6,
|
413 |
+
"Ios18.squeeze" : 6,
|
414 |
+
"Ios18.reshape" : 12,
|
415 |
+
"Ios18.mul" : 57
|
416 |
+
}
|
417 |
+
}
|
418 |
+
],
|
419 |
+
"mlProgramOperationTypeHistogram" : {
|
420 |
+
"Ios18.constexprLutToDense" : 21,
|
421 |
+
"Ios18.conv" : 21,
|
422 |
+
"Ios18.matmul" : 6,
|
423 |
+
"Ios18.expandDims" : 6,
|
424 |
+
"Ios18.concat" : 18,
|
425 |
+
"Ios18.add" : 15,
|
426 |
+
"Ios18.realDiv" : 6,
|
427 |
+
"Ios18.silu" : 3,
|
428 |
+
"Ios18.softmax" : 3,
|
429 |
+
"Ios18.sliceByIndex" : 18,
|
430 |
+
"Ios16.reduceL2Norm" : 6,
|
431 |
+
"Ios18.squeeze" : 6,
|
432 |
+
"Ios18.reshape" : 12,
|
433 |
+
"Ios18.mul" : 57
|
434 |
+
},
|
435 |
+
"isUpdatable" : "0",
|
436 |
+
"stateSchema" : [
|
437 |
+
|
438 |
+
],
|
439 |
+
"availability" : {
|
440 |
+
"macOS" : "15.0",
|
441 |
+
"tvOS" : "18.0",
|
442 |
+
"visionOS" : "2.0",
|
443 |
+
"watchOS" : "11.0",
|
444 |
+
"iOS" : "18.0",
|
445 |
+
"macCatalyst" : "18.0"
|
446 |
+
},
|
447 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
448 |
+
"modelType" : {
|
449 |
+
"name" : "MLModelType_mlProgram"
|
450 |
+
},
|
451 |
+
"inputSchema" : [
|
452 |
+
{
|
453 |
+
"hasShapeFlexibility" : "0",
|
454 |
+
"isOptional" : "0",
|
455 |
+
"dataType" : "Float16",
|
456 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
457 |
+
"shortDescription" : "",
|
458 |
+
"shape" : "[1, 4096, 1, 512]",
|
459 |
+
"name" : "x",
|
460 |
+
"type" : "MultiArray"
|
461 |
+
},
|
462 |
+
{
|
463 |
+
"hasShapeFlexibility" : "0",
|
464 |
+
"isOptional" : "0",
|
465 |
+
"dataType" : "Float16",
|
466 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
467 |
+
"shortDescription" : "",
|
468 |
+
"shape" : "[128, 512]",
|
469 |
+
"name" : "cos",
|
470 |
+
"type" : "MultiArray"
|
471 |
+
},
|
472 |
+
{
|
473 |
+
"hasShapeFlexibility" : "0",
|
474 |
+
"isOptional" : "0",
|
475 |
+
"dataType" : "Float16",
|
476 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
477 |
+
"shortDescription" : "",
|
478 |
+
"shape" : "[128, 512]",
|
479 |
+
"name" : "sin",
|
480 |
+
"type" : "MultiArray"
|
481 |
+
},
|
482 |
+
{
|
483 |
+
"hasShapeFlexibility" : "0",
|
484 |
+
"isOptional" : "0",
|
485 |
+
"dataType" : "Float16",
|
486 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
487 |
+
"shortDescription" : "",
|
488 |
+
"shape" : "[1, 1, 512, 512]",
|
489 |
+
"name" : "mask",
|
490 |
+
"type" : "MultiArray"
|
491 |
+
}
|
492 |
+
],
|
493 |
+
"defaultFunctionName" : "input_512_context_512",
|
494 |
+
"generatedClassName" : "Llama_2_7b_hf_2024_07_02_20_36_17_merged_chunk2",
|
495 |
+
"userDefinedMetadata" : {
|
496 |
+
|
497 |
+
},
|
498 |
+
"method" : "predict"
|
499 |
+
}
|
500 |
+
]
|
sequoia/Llama-2-7b-hf_chunk2.mlmodelc/model.mil
ADDED
The diff for this file is too large to render.
See raw diff
|
|
sequoia/Llama-2-7b-hf_chunk2.mlmodelc/weights/weight.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5d1e5b2e712dd0e457b1f1f358ae4edf653ab539814a891fec6631982b60b055
|
3 |
+
size 303876448
|
sequoia/Llama-2-7b-hf_chunk3.mlmodelc/analytics/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:84e317a82cdf4e96f808f63e77f10098844d47ad522545181edfac4d287c9c92
|
3 |
+
size 243
|
sequoia/Llama-2-7b-hf_chunk3.mlmodelc/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e430d0795ff5c384187174f5718a2c13d0070f5d6a811831e18862497865a86d
|
3 |
+
size 1037
|
sequoia/Llama-2-7b-hf_chunk3.mlmodelc/metadata.json
ADDED
@@ -0,0 +1,500 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 × 4096 × 1 × 512)",
|
11 |
+
"shortDescription" : "",
|
12 |
+
"shape" : "[1, 4096, 1, 512]",
|
13 |
+
"name" : "new_x",
|
14 |
+
"type" : "MultiArray"
|
15 |
+
},
|
16 |
+
{
|
17 |
+
"hasShapeFlexibility" : "0",
|
18 |
+
"isOptional" : "0",
|
19 |
+
"dataType" : "Float16",
|
20 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
21 |
+
"shortDescription" : "",
|
22 |
+
"shape" : "[1, 32, 128, 511]",
|
23 |
+
"name" : "new_k_cache_0",
|
24 |
+
"type" : "MultiArray"
|
25 |
+
},
|
26 |
+
{
|
27 |
+
"hasShapeFlexibility" : "0",
|
28 |
+
"isOptional" : "0",
|
29 |
+
"dataType" : "Float16",
|
30 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
31 |
+
"shortDescription" : "",
|
32 |
+
"shape" : "[1, 32, 128, 511]",
|
33 |
+
"name" : "new_k_cache_1",
|
34 |
+
"type" : "MultiArray"
|
35 |
+
},
|
36 |
+
{
|
37 |
+
"hasShapeFlexibility" : "0",
|
38 |
+
"isOptional" : "0",
|
39 |
+
"dataType" : "Float16",
|
40 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
41 |
+
"shortDescription" : "",
|
42 |
+
"shape" : "[1, 32, 128, 511]",
|
43 |
+
"name" : "new_k_cache_2",
|
44 |
+
"type" : "MultiArray"
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"hasShapeFlexibility" : "0",
|
48 |
+
"isOptional" : "0",
|
49 |
+
"dataType" : "Float16",
|
50 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
51 |
+
"shortDescription" : "",
|
52 |
+
"shape" : "[1, 32, 128, 511]",
|
53 |
+
"name" : "new_v_cache_0",
|
54 |
+
"type" : "MultiArray"
|
55 |
+
},
|
56 |
+
{
|
57 |
+
"hasShapeFlexibility" : "0",
|
58 |
+
"isOptional" : "0",
|
59 |
+
"dataType" : "Float16",
|
60 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
61 |
+
"shortDescription" : "",
|
62 |
+
"shape" : "[1, 32, 128, 511]",
|
63 |
+
"name" : "new_v_cache_1",
|
64 |
+
"type" : "MultiArray"
|
65 |
+
},
|
66 |
+
{
|
67 |
+
"hasShapeFlexibility" : "0",
|
68 |
+
"isOptional" : "0",
|
69 |
+
"dataType" : "Float16",
|
70 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
71 |
+
"shortDescription" : "",
|
72 |
+
"shape" : "[1, 32, 128, 511]",
|
73 |
+
"name" : "new_v_cache_2",
|
74 |
+
"type" : "MultiArray"
|
75 |
+
}
|
76 |
+
],
|
77 |
+
"modelParameters" : [
|
78 |
+
|
79 |
+
],
|
80 |
+
"specificationVersion" : 9,
|
81 |
+
"functions" : [
|
82 |
+
{
|
83 |
+
"inputSchema" : [
|
84 |
+
{
|
85 |
+
"hasShapeFlexibility" : "0",
|
86 |
+
"isOptional" : "0",
|
87 |
+
"dataType" : "Float16",
|
88 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
89 |
+
"shortDescription" : "",
|
90 |
+
"shape" : "[1, 4096, 1, 512]",
|
91 |
+
"name" : "x",
|
92 |
+
"type" : "MultiArray"
|
93 |
+
},
|
94 |
+
{
|
95 |
+
"hasShapeFlexibility" : "0",
|
96 |
+
"isOptional" : "0",
|
97 |
+
"dataType" : "Float16",
|
98 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
99 |
+
"shortDescription" : "",
|
100 |
+
"shape" : "[128, 512]",
|
101 |
+
"name" : "cos",
|
102 |
+
"type" : "MultiArray"
|
103 |
+
},
|
104 |
+
{
|
105 |
+
"hasShapeFlexibility" : "0",
|
106 |
+
"isOptional" : "0",
|
107 |
+
"dataType" : "Float16",
|
108 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
109 |
+
"shortDescription" : "",
|
110 |
+
"shape" : "[128, 512]",
|
111 |
+
"name" : "sin",
|
112 |
+
"type" : "MultiArray"
|
113 |
+
},
|
114 |
+
{
|
115 |
+
"hasShapeFlexibility" : "0",
|
116 |
+
"isOptional" : "0",
|
117 |
+
"dataType" : "Float16",
|
118 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
119 |
+
"shortDescription" : "",
|
120 |
+
"shape" : "[1, 1, 512, 512]",
|
121 |
+
"name" : "mask",
|
122 |
+
"type" : "MultiArray"
|
123 |
+
}
|
124 |
+
],
|
125 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
126 |
+
"storagePrecision" : "Float16",
|
127 |
+
"stateSchema" : [
|
128 |
+
|
129 |
+
],
|
130 |
+
"outputSchema" : [
|
131 |
+
{
|
132 |
+
"hasShapeFlexibility" : "0",
|
133 |
+
"isOptional" : "0",
|
134 |
+
"dataType" : "Float16",
|
135 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
136 |
+
"shortDescription" : "",
|
137 |
+
"shape" : "[1, 4096, 1, 512]",
|
138 |
+
"name" : "new_x",
|
139 |
+
"type" : "MultiArray"
|
140 |
+
},
|
141 |
+
{
|
142 |
+
"hasShapeFlexibility" : "0",
|
143 |
+
"isOptional" : "0",
|
144 |
+
"dataType" : "Float16",
|
145 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
146 |
+
"shortDescription" : "",
|
147 |
+
"shape" : "[1, 32, 128, 511]",
|
148 |
+
"name" : "new_k_cache_0",
|
149 |
+
"type" : "MultiArray"
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"hasShapeFlexibility" : "0",
|
153 |
+
"isOptional" : "0",
|
154 |
+
"dataType" : "Float16",
|
155 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
156 |
+
"shortDescription" : "",
|
157 |
+
"shape" : "[1, 32, 128, 511]",
|
158 |
+
"name" : "new_k_cache_1",
|
159 |
+
"type" : "MultiArray"
|
160 |
+
},
|
161 |
+
{
|
162 |
+
"hasShapeFlexibility" : "0",
|
163 |
+
"isOptional" : "0",
|
164 |
+
"dataType" : "Float16",
|
165 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
166 |
+
"shortDescription" : "",
|
167 |
+
"shape" : "[1, 32, 128, 511]",
|
168 |
+
"name" : "new_k_cache_2",
|
169 |
+
"type" : "MultiArray"
|
170 |
+
},
|
171 |
+
{
|
172 |
+
"hasShapeFlexibility" : "0",
|
173 |
+
"isOptional" : "0",
|
174 |
+
"dataType" : "Float16",
|
175 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
176 |
+
"shortDescription" : "",
|
177 |
+
"shape" : "[1, 32, 128, 511]",
|
178 |
+
"name" : "new_v_cache_0",
|
179 |
+
"type" : "MultiArray"
|
180 |
+
},
|
181 |
+
{
|
182 |
+
"hasShapeFlexibility" : "0",
|
183 |
+
"isOptional" : "0",
|
184 |
+
"dataType" : "Float16",
|
185 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
186 |
+
"shortDescription" : "",
|
187 |
+
"shape" : "[1, 32, 128, 511]",
|
188 |
+
"name" : "new_v_cache_1",
|
189 |
+
"type" : "MultiArray"
|
190 |
+
},
|
191 |
+
{
|
192 |
+
"hasShapeFlexibility" : "0",
|
193 |
+
"isOptional" : "0",
|
194 |
+
"dataType" : "Float16",
|
195 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
196 |
+
"shortDescription" : "",
|
197 |
+
"shape" : "[1, 32, 128, 511]",
|
198 |
+
"name" : "new_v_cache_2",
|
199 |
+
"type" : "MultiArray"
|
200 |
+
}
|
201 |
+
],
|
202 |
+
"name" : "input_512_context_512",
|
203 |
+
"mlProgramOperationTypeHistogram" : {
|
204 |
+
"Ios18.constexprLutToDense" : 21,
|
205 |
+
"Ios18.conv" : 21,
|
206 |
+
"Ios18.matmul" : 6,
|
207 |
+
"Ios18.expandDims" : 6,
|
208 |
+
"Ios18.concat" : 18,
|
209 |
+
"Ios18.add" : 15,
|
210 |
+
"Ios18.realDiv" : 6,
|
211 |
+
"Ios18.silu" : 3,
|
212 |
+
"Ios18.softmax" : 3,
|
213 |
+
"Ios18.sliceByIndex" : 18,
|
214 |
+
"Ios16.reduceL2Norm" : 6,
|
215 |
+
"Ios18.squeeze" : 6,
|
216 |
+
"Ios18.reshape" : 12,
|
217 |
+
"Ios18.mul" : 57
|
218 |
+
}
|
219 |
+
},
|
220 |
+
{
|
221 |
+
"inputSchema" : [
|
222 |
+
{
|
223 |
+
"hasShapeFlexibility" : "0",
|
224 |
+
"isOptional" : "0",
|
225 |
+
"dataType" : "Float16",
|
226 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 1)",
|
227 |
+
"shortDescription" : "",
|
228 |
+
"shape" : "[1, 4096, 1, 1]",
|
229 |
+
"name" : "x",
|
230 |
+
"type" : "MultiArray"
|
231 |
+
},
|
232 |
+
{
|
233 |
+
"hasShapeFlexibility" : "0",
|
234 |
+
"isOptional" : "0",
|
235 |
+
"dataType" : "Float16",
|
236 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
237 |
+
"shortDescription" : "",
|
238 |
+
"shape" : "[128, 1]",
|
239 |
+
"name" : "cos",
|
240 |
+
"type" : "MultiArray"
|
241 |
+
},
|
242 |
+
{
|
243 |
+
"hasShapeFlexibility" : "0",
|
244 |
+
"isOptional" : "0",
|
245 |
+
"dataType" : "Float16",
|
246 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
247 |
+
"shortDescription" : "",
|
248 |
+
"shape" : "[128, 1]",
|
249 |
+
"name" : "sin",
|
250 |
+
"type" : "MultiArray"
|
251 |
+
},
|
252 |
+
{
|
253 |
+
"hasShapeFlexibility" : "0",
|
254 |
+
"isOptional" : "0",
|
255 |
+
"dataType" : "Float16",
|
256 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 1 × 512)",
|
257 |
+
"shortDescription" : "",
|
258 |
+
"shape" : "[1, 1, 1, 512]",
|
259 |
+
"name" : "mask",
|
260 |
+
"type" : "MultiArray"
|
261 |
+
},
|
262 |
+
{
|
263 |
+
"hasShapeFlexibility" : "0",
|
264 |
+
"isOptional" : "1",
|
265 |
+
"dataType" : "Float16",
|
266 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
267 |
+
"shortDescription" : "",
|
268 |
+
"shape" : "[1, 32, 128, 511]",
|
269 |
+
"name" : "k_cache_0",
|
270 |
+
"type" : "MultiArray"
|
271 |
+
},
|
272 |
+
{
|
273 |
+
"hasShapeFlexibility" : "0",
|
274 |
+
"isOptional" : "1",
|
275 |
+
"dataType" : "Float16",
|
276 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
277 |
+
"shortDescription" : "",
|
278 |
+
"shape" : "[1, 32, 128, 511]",
|
279 |
+
"name" : "v_cache_0",
|
280 |
+
"type" : "MultiArray"
|
281 |
+
},
|
282 |
+
{
|
283 |
+
"hasShapeFlexibility" : "0",
|
284 |
+
"isOptional" : "1",
|
285 |
+
"dataType" : "Float16",
|
286 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
287 |
+
"shortDescription" : "",
|
288 |
+
"shape" : "[1, 32, 128, 511]",
|
289 |
+
"name" : "k_cache_1",
|
290 |
+
"type" : "MultiArray"
|
291 |
+
},
|
292 |
+
{
|
293 |
+
"hasShapeFlexibility" : "0",
|
294 |
+
"isOptional" : "1",
|
295 |
+
"dataType" : "Float16",
|
296 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
297 |
+
"shortDescription" : "",
|
298 |
+
"shape" : "[1, 32, 128, 511]",
|
299 |
+
"name" : "v_cache_1",
|
300 |
+
"type" : "MultiArray"
|
301 |
+
},
|
302 |
+
{
|
303 |
+
"hasShapeFlexibility" : "0",
|
304 |
+
"isOptional" : "1",
|
305 |
+
"dataType" : "Float16",
|
306 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
307 |
+
"shortDescription" : "",
|
308 |
+
"shape" : "[1, 32, 128, 511]",
|
309 |
+
"name" : "k_cache_2",
|
310 |
+
"type" : "MultiArray"
|
311 |
+
},
|
312 |
+
{
|
313 |
+
"hasShapeFlexibility" : "0",
|
314 |
+
"isOptional" : "1",
|
315 |
+
"dataType" : "Float16",
|
316 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
317 |
+
"shortDescription" : "",
|
318 |
+
"shape" : "[1, 32, 128, 511]",
|
319 |
+
"name" : "v_cache_2",
|
320 |
+
"type" : "MultiArray"
|
321 |
+
}
|
322 |
+
],
|
323 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
324 |
+
"storagePrecision" : "Float16",
|
325 |
+
"stateSchema" : [
|
326 |
+
|
327 |
+
],
|
328 |
+
"outputSchema" : [
|
329 |
+
{
|
330 |
+
"hasShapeFlexibility" : "0",
|
331 |
+
"isOptional" : "0",
|
332 |
+
"dataType" : "Float16",
|
333 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 1)",
|
334 |
+
"shortDescription" : "",
|
335 |
+
"shape" : "[1, 4096, 1, 1]",
|
336 |
+
"name" : "new_x",
|
337 |
+
"type" : "MultiArray"
|
338 |
+
},
|
339 |
+
{
|
340 |
+
"hasShapeFlexibility" : "0",
|
341 |
+
"isOptional" : "0",
|
342 |
+
"dataType" : "Float16",
|
343 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
344 |
+
"shortDescription" : "",
|
345 |
+
"shape" : "[1, 32, 128, 511]",
|
346 |
+
"name" : "new_k_cache_0",
|
347 |
+
"type" : "MultiArray"
|
348 |
+
},
|
349 |
+
{
|
350 |
+
"hasShapeFlexibility" : "0",
|
351 |
+
"isOptional" : "0",
|
352 |
+
"dataType" : "Float16",
|
353 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
354 |
+
"shortDescription" : "",
|
355 |
+
"shape" : "[1, 32, 128, 511]",
|
356 |
+
"name" : "new_k_cache_1",
|
357 |
+
"type" : "MultiArray"
|
358 |
+
},
|
359 |
+
{
|
360 |
+
"hasShapeFlexibility" : "0",
|
361 |
+
"isOptional" : "0",
|
362 |
+
"dataType" : "Float16",
|
363 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
364 |
+
"shortDescription" : "",
|
365 |
+
"shape" : "[1, 32, 128, 511]",
|
366 |
+
"name" : "new_k_cache_2",
|
367 |
+
"type" : "MultiArray"
|
368 |
+
},
|
369 |
+
{
|
370 |
+
"hasShapeFlexibility" : "0",
|
371 |
+
"isOptional" : "0",
|
372 |
+
"dataType" : "Float16",
|
373 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
374 |
+
"shortDescription" : "",
|
375 |
+
"shape" : "[1, 32, 128, 511]",
|
376 |
+
"name" : "new_v_cache_0",
|
377 |
+
"type" : "MultiArray"
|
378 |
+
},
|
379 |
+
{
|
380 |
+
"hasShapeFlexibility" : "0",
|
381 |
+
"isOptional" : "0",
|
382 |
+
"dataType" : "Float16",
|
383 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
384 |
+
"shortDescription" : "",
|
385 |
+
"shape" : "[1, 32, 128, 511]",
|
386 |
+
"name" : "new_v_cache_1",
|
387 |
+
"type" : "MultiArray"
|
388 |
+
},
|
389 |
+
{
|
390 |
+
"hasShapeFlexibility" : "0",
|
391 |
+
"isOptional" : "0",
|
392 |
+
"dataType" : "Float16",
|
393 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
394 |
+
"shortDescription" : "",
|
395 |
+
"shape" : "[1, 32, 128, 511]",
|
396 |
+
"name" : "new_v_cache_2",
|
397 |
+
"type" : "MultiArray"
|
398 |
+
}
|
399 |
+
],
|
400 |
+
"name" : "input_1_context_512",
|
401 |
+
"mlProgramOperationTypeHistogram" : {
|
402 |
+
"Ios18.constexprLutToDense" : 21,
|
403 |
+
"Ios18.conv" : 21,
|
404 |
+
"Ios18.matmul" : 6,
|
405 |
+
"Ios18.expandDims" : 6,
|
406 |
+
"Ios18.concat" : 18,
|
407 |
+
"Ios18.add" : 15,
|
408 |
+
"Ios18.realDiv" : 6,
|
409 |
+
"Ios18.silu" : 3,
|
410 |
+
"Ios18.softmax" : 3,
|
411 |
+
"Ios18.sliceByIndex" : 18,
|
412 |
+
"Ios16.reduceL2Norm" : 6,
|
413 |
+
"Ios18.squeeze" : 6,
|
414 |
+
"Ios18.reshape" : 12,
|
415 |
+
"Ios18.mul" : 57
|
416 |
+
}
|
417 |
+
}
|
418 |
+
],
|
419 |
+
"mlProgramOperationTypeHistogram" : {
|
420 |
+
"Ios18.constexprLutToDense" : 21,
|
421 |
+
"Ios18.conv" : 21,
|
422 |
+
"Ios18.matmul" : 6,
|
423 |
+
"Ios18.expandDims" : 6,
|
424 |
+
"Ios18.concat" : 18,
|
425 |
+
"Ios18.add" : 15,
|
426 |
+
"Ios18.realDiv" : 6,
|
427 |
+
"Ios18.silu" : 3,
|
428 |
+
"Ios18.softmax" : 3,
|
429 |
+
"Ios18.sliceByIndex" : 18,
|
430 |
+
"Ios16.reduceL2Norm" : 6,
|
431 |
+
"Ios18.squeeze" : 6,
|
432 |
+
"Ios18.reshape" : 12,
|
433 |
+
"Ios18.mul" : 57
|
434 |
+
},
|
435 |
+
"isUpdatable" : "0",
|
436 |
+
"stateSchema" : [
|
437 |
+
|
438 |
+
],
|
439 |
+
"availability" : {
|
440 |
+
"macOS" : "15.0",
|
441 |
+
"tvOS" : "18.0",
|
442 |
+
"visionOS" : "2.0",
|
443 |
+
"watchOS" : "11.0",
|
444 |
+
"iOS" : "18.0",
|
445 |
+
"macCatalyst" : "18.0"
|
446 |
+
},
|
447 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
448 |
+
"modelType" : {
|
449 |
+
"name" : "MLModelType_mlProgram"
|
450 |
+
},
|
451 |
+
"inputSchema" : [
|
452 |
+
{
|
453 |
+
"hasShapeFlexibility" : "0",
|
454 |
+
"isOptional" : "0",
|
455 |
+
"dataType" : "Float16",
|
456 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
457 |
+
"shortDescription" : "",
|
458 |
+
"shape" : "[1, 4096, 1, 512]",
|
459 |
+
"name" : "x",
|
460 |
+
"type" : "MultiArray"
|
461 |
+
},
|
462 |
+
{
|
463 |
+
"hasShapeFlexibility" : "0",
|
464 |
+
"isOptional" : "0",
|
465 |
+
"dataType" : "Float16",
|
466 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
467 |
+
"shortDescription" : "",
|
468 |
+
"shape" : "[128, 512]",
|
469 |
+
"name" : "cos",
|
470 |
+
"type" : "MultiArray"
|
471 |
+
},
|
472 |
+
{
|
473 |
+
"hasShapeFlexibility" : "0",
|
474 |
+
"isOptional" : "0",
|
475 |
+
"dataType" : "Float16",
|
476 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
477 |
+
"shortDescription" : "",
|
478 |
+
"shape" : "[128, 512]",
|
479 |
+
"name" : "sin",
|
480 |
+
"type" : "MultiArray"
|
481 |
+
},
|
482 |
+
{
|
483 |
+
"hasShapeFlexibility" : "0",
|
484 |
+
"isOptional" : "0",
|
485 |
+
"dataType" : "Float16",
|
486 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
487 |
+
"shortDescription" : "",
|
488 |
+
"shape" : "[1, 1, 512, 512]",
|
489 |
+
"name" : "mask",
|
490 |
+
"type" : "MultiArray"
|
491 |
+
}
|
492 |
+
],
|
493 |
+
"defaultFunctionName" : "input_512_context_512",
|
494 |
+
"generatedClassName" : "Llama_2_7b_hf_2024_07_02_20_36_17_merged_chunk3",
|
495 |
+
"userDefinedMetadata" : {
|
496 |
+
|
497 |
+
},
|
498 |
+
"method" : "predict"
|
499 |
+
}
|
500 |
+
]
|
sequoia/Llama-2-7b-hf_chunk3.mlmodelc/model.mil
ADDED
The diff for this file is too large to render.
See raw diff
|
|
sequoia/Llama-2-7b-hf_chunk3.mlmodelc/weights/weight.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6ac04155a78ee2dc7d4ba01a2bac2eedcd17fd33957a8bd396223a60f033204f
|
3 |
+
size 303876448
|
sequoia/Llama-2-7b-hf_chunk4.mlmodelc/analytics/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:84e317a82cdf4e96f808f63e77f10098844d47ad522545181edfac4d287c9c92
|
3 |
+
size 243
|
sequoia/Llama-2-7b-hf_chunk4.mlmodelc/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e430d0795ff5c384187174f5718a2c13d0070f5d6a811831e18862497865a86d
|
3 |
+
size 1037
|
sequoia/Llama-2-7b-hf_chunk4.mlmodelc/metadata.json
ADDED
@@ -0,0 +1,500 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 × 4096 × 1 × 512)",
|
11 |
+
"shortDescription" : "",
|
12 |
+
"shape" : "[1, 4096, 1, 512]",
|
13 |
+
"name" : "new_x",
|
14 |
+
"type" : "MultiArray"
|
15 |
+
},
|
16 |
+
{
|
17 |
+
"hasShapeFlexibility" : "0",
|
18 |
+
"isOptional" : "0",
|
19 |
+
"dataType" : "Float16",
|
20 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
21 |
+
"shortDescription" : "",
|
22 |
+
"shape" : "[1, 32, 128, 511]",
|
23 |
+
"name" : "new_k_cache_0",
|
24 |
+
"type" : "MultiArray"
|
25 |
+
},
|
26 |
+
{
|
27 |
+
"hasShapeFlexibility" : "0",
|
28 |
+
"isOptional" : "0",
|
29 |
+
"dataType" : "Float16",
|
30 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
31 |
+
"shortDescription" : "",
|
32 |
+
"shape" : "[1, 32, 128, 511]",
|
33 |
+
"name" : "new_k_cache_1",
|
34 |
+
"type" : "MultiArray"
|
35 |
+
},
|
36 |
+
{
|
37 |
+
"hasShapeFlexibility" : "0",
|
38 |
+
"isOptional" : "0",
|
39 |
+
"dataType" : "Float16",
|
40 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
41 |
+
"shortDescription" : "",
|
42 |
+
"shape" : "[1, 32, 128, 511]",
|
43 |
+
"name" : "new_k_cache_2",
|
44 |
+
"type" : "MultiArray"
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"hasShapeFlexibility" : "0",
|
48 |
+
"isOptional" : "0",
|
49 |
+
"dataType" : "Float16",
|
50 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
51 |
+
"shortDescription" : "",
|
52 |
+
"shape" : "[1, 32, 128, 511]",
|
53 |
+
"name" : "new_v_cache_0",
|
54 |
+
"type" : "MultiArray"
|
55 |
+
},
|
56 |
+
{
|
57 |
+
"hasShapeFlexibility" : "0",
|
58 |
+
"isOptional" : "0",
|
59 |
+
"dataType" : "Float16",
|
60 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
61 |
+
"shortDescription" : "",
|
62 |
+
"shape" : "[1, 32, 128, 511]",
|
63 |
+
"name" : "new_v_cache_1",
|
64 |
+
"type" : "MultiArray"
|
65 |
+
},
|
66 |
+
{
|
67 |
+
"hasShapeFlexibility" : "0",
|
68 |
+
"isOptional" : "0",
|
69 |
+
"dataType" : "Float16",
|
70 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
71 |
+
"shortDescription" : "",
|
72 |
+
"shape" : "[1, 32, 128, 511]",
|
73 |
+
"name" : "new_v_cache_2",
|
74 |
+
"type" : "MultiArray"
|
75 |
+
}
|
76 |
+
],
|
77 |
+
"modelParameters" : [
|
78 |
+
|
79 |
+
],
|
80 |
+
"specificationVersion" : 9,
|
81 |
+
"functions" : [
|
82 |
+
{
|
83 |
+
"inputSchema" : [
|
84 |
+
{
|
85 |
+
"hasShapeFlexibility" : "0",
|
86 |
+
"isOptional" : "0",
|
87 |
+
"dataType" : "Float16",
|
88 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
89 |
+
"shortDescription" : "",
|
90 |
+
"shape" : "[1, 4096, 1, 512]",
|
91 |
+
"name" : "x",
|
92 |
+
"type" : "MultiArray"
|
93 |
+
},
|
94 |
+
{
|
95 |
+
"hasShapeFlexibility" : "0",
|
96 |
+
"isOptional" : "0",
|
97 |
+
"dataType" : "Float16",
|
98 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
99 |
+
"shortDescription" : "",
|
100 |
+
"shape" : "[128, 512]",
|
101 |
+
"name" : "cos",
|
102 |
+
"type" : "MultiArray"
|
103 |
+
},
|
104 |
+
{
|
105 |
+
"hasShapeFlexibility" : "0",
|
106 |
+
"isOptional" : "0",
|
107 |
+
"dataType" : "Float16",
|
108 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
109 |
+
"shortDescription" : "",
|
110 |
+
"shape" : "[128, 512]",
|
111 |
+
"name" : "sin",
|
112 |
+
"type" : "MultiArray"
|
113 |
+
},
|
114 |
+
{
|
115 |
+
"hasShapeFlexibility" : "0",
|
116 |
+
"isOptional" : "0",
|
117 |
+
"dataType" : "Float16",
|
118 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
119 |
+
"shortDescription" : "",
|
120 |
+
"shape" : "[1, 1, 512, 512]",
|
121 |
+
"name" : "mask",
|
122 |
+
"type" : "MultiArray"
|
123 |
+
}
|
124 |
+
],
|
125 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
126 |
+
"storagePrecision" : "Float16",
|
127 |
+
"stateSchema" : [
|
128 |
+
|
129 |
+
],
|
130 |
+
"outputSchema" : [
|
131 |
+
{
|
132 |
+
"hasShapeFlexibility" : "0",
|
133 |
+
"isOptional" : "0",
|
134 |
+
"dataType" : "Float16",
|
135 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
136 |
+
"shortDescription" : "",
|
137 |
+
"shape" : "[1, 4096, 1, 512]",
|
138 |
+
"name" : "new_x",
|
139 |
+
"type" : "MultiArray"
|
140 |
+
},
|
141 |
+
{
|
142 |
+
"hasShapeFlexibility" : "0",
|
143 |
+
"isOptional" : "0",
|
144 |
+
"dataType" : "Float16",
|
145 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
146 |
+
"shortDescription" : "",
|
147 |
+
"shape" : "[1, 32, 128, 511]",
|
148 |
+
"name" : "new_k_cache_0",
|
149 |
+
"type" : "MultiArray"
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"hasShapeFlexibility" : "0",
|
153 |
+
"isOptional" : "0",
|
154 |
+
"dataType" : "Float16",
|
155 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
156 |
+
"shortDescription" : "",
|
157 |
+
"shape" : "[1, 32, 128, 511]",
|
158 |
+
"name" : "new_k_cache_1",
|
159 |
+
"type" : "MultiArray"
|
160 |
+
},
|
161 |
+
{
|
162 |
+
"hasShapeFlexibility" : "0",
|
163 |
+
"isOptional" : "0",
|
164 |
+
"dataType" : "Float16",
|
165 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
166 |
+
"shortDescription" : "",
|
167 |
+
"shape" : "[1, 32, 128, 511]",
|
168 |
+
"name" : "new_k_cache_2",
|
169 |
+
"type" : "MultiArray"
|
170 |
+
},
|
171 |
+
{
|
172 |
+
"hasShapeFlexibility" : "0",
|
173 |
+
"isOptional" : "0",
|
174 |
+
"dataType" : "Float16",
|
175 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
176 |
+
"shortDescription" : "",
|
177 |
+
"shape" : "[1, 32, 128, 511]",
|
178 |
+
"name" : "new_v_cache_0",
|
179 |
+
"type" : "MultiArray"
|
180 |
+
},
|
181 |
+
{
|
182 |
+
"hasShapeFlexibility" : "0",
|
183 |
+
"isOptional" : "0",
|
184 |
+
"dataType" : "Float16",
|
185 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
186 |
+
"shortDescription" : "",
|
187 |
+
"shape" : "[1, 32, 128, 511]",
|
188 |
+
"name" : "new_v_cache_1",
|
189 |
+
"type" : "MultiArray"
|
190 |
+
},
|
191 |
+
{
|
192 |
+
"hasShapeFlexibility" : "0",
|
193 |
+
"isOptional" : "0",
|
194 |
+
"dataType" : "Float16",
|
195 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
196 |
+
"shortDescription" : "",
|
197 |
+
"shape" : "[1, 32, 128, 511]",
|
198 |
+
"name" : "new_v_cache_2",
|
199 |
+
"type" : "MultiArray"
|
200 |
+
}
|
201 |
+
],
|
202 |
+
"name" : "input_512_context_512",
|
203 |
+
"mlProgramOperationTypeHistogram" : {
|
204 |
+
"Ios18.constexprLutToDense" : 21,
|
205 |
+
"Ios18.conv" : 21,
|
206 |
+
"Ios18.matmul" : 6,
|
207 |
+
"Ios18.expandDims" : 6,
|
208 |
+
"Ios18.concat" : 18,
|
209 |
+
"Ios18.add" : 15,
|
210 |
+
"Ios18.realDiv" : 6,
|
211 |
+
"Ios18.silu" : 3,
|
212 |
+
"Ios18.softmax" : 3,
|
213 |
+
"Ios18.sliceByIndex" : 18,
|
214 |
+
"Ios16.reduceL2Norm" : 6,
|
215 |
+
"Ios18.squeeze" : 6,
|
216 |
+
"Ios18.reshape" : 12,
|
217 |
+
"Ios18.mul" : 57
|
218 |
+
}
|
219 |
+
},
|
220 |
+
{
|
221 |
+
"inputSchema" : [
|
222 |
+
{
|
223 |
+
"hasShapeFlexibility" : "0",
|
224 |
+
"isOptional" : "0",
|
225 |
+
"dataType" : "Float16",
|
226 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 1)",
|
227 |
+
"shortDescription" : "",
|
228 |
+
"shape" : "[1, 4096, 1, 1]",
|
229 |
+
"name" : "x",
|
230 |
+
"type" : "MultiArray"
|
231 |
+
},
|
232 |
+
{
|
233 |
+
"hasShapeFlexibility" : "0",
|
234 |
+
"isOptional" : "0",
|
235 |
+
"dataType" : "Float16",
|
236 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
237 |
+
"shortDescription" : "",
|
238 |
+
"shape" : "[128, 1]",
|
239 |
+
"name" : "cos",
|
240 |
+
"type" : "MultiArray"
|
241 |
+
},
|
242 |
+
{
|
243 |
+
"hasShapeFlexibility" : "0",
|
244 |
+
"isOptional" : "0",
|
245 |
+
"dataType" : "Float16",
|
246 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
247 |
+
"shortDescription" : "",
|
248 |
+
"shape" : "[128, 1]",
|
249 |
+
"name" : "sin",
|
250 |
+
"type" : "MultiArray"
|
251 |
+
},
|
252 |
+
{
|
253 |
+
"hasShapeFlexibility" : "0",
|
254 |
+
"isOptional" : "0",
|
255 |
+
"dataType" : "Float16",
|
256 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 1 × 512)",
|
257 |
+
"shortDescription" : "",
|
258 |
+
"shape" : "[1, 1, 1, 512]",
|
259 |
+
"name" : "mask",
|
260 |
+
"type" : "MultiArray"
|
261 |
+
},
|
262 |
+
{
|
263 |
+
"hasShapeFlexibility" : "0",
|
264 |
+
"isOptional" : "1",
|
265 |
+
"dataType" : "Float16",
|
266 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
267 |
+
"shortDescription" : "",
|
268 |
+
"shape" : "[1, 32, 128, 511]",
|
269 |
+
"name" : "k_cache_0",
|
270 |
+
"type" : "MultiArray"
|
271 |
+
},
|
272 |
+
{
|
273 |
+
"hasShapeFlexibility" : "0",
|
274 |
+
"isOptional" : "1",
|
275 |
+
"dataType" : "Float16",
|
276 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
277 |
+
"shortDescription" : "",
|
278 |
+
"shape" : "[1, 32, 128, 511]",
|
279 |
+
"name" : "v_cache_0",
|
280 |
+
"type" : "MultiArray"
|
281 |
+
},
|
282 |
+
{
|
283 |
+
"hasShapeFlexibility" : "0",
|
284 |
+
"isOptional" : "1",
|
285 |
+
"dataType" : "Float16",
|
286 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
287 |
+
"shortDescription" : "",
|
288 |
+
"shape" : "[1, 32, 128, 511]",
|
289 |
+
"name" : "k_cache_1",
|
290 |
+
"type" : "MultiArray"
|
291 |
+
},
|
292 |
+
{
|
293 |
+
"hasShapeFlexibility" : "0",
|
294 |
+
"isOptional" : "1",
|
295 |
+
"dataType" : "Float16",
|
296 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
297 |
+
"shortDescription" : "",
|
298 |
+
"shape" : "[1, 32, 128, 511]",
|
299 |
+
"name" : "v_cache_1",
|
300 |
+
"type" : "MultiArray"
|
301 |
+
},
|
302 |
+
{
|
303 |
+
"hasShapeFlexibility" : "0",
|
304 |
+
"isOptional" : "1",
|
305 |
+
"dataType" : "Float16",
|
306 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
307 |
+
"shortDescription" : "",
|
308 |
+
"shape" : "[1, 32, 128, 511]",
|
309 |
+
"name" : "k_cache_2",
|
310 |
+
"type" : "MultiArray"
|
311 |
+
},
|
312 |
+
{
|
313 |
+
"hasShapeFlexibility" : "0",
|
314 |
+
"isOptional" : "1",
|
315 |
+
"dataType" : "Float16",
|
316 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
317 |
+
"shortDescription" : "",
|
318 |
+
"shape" : "[1, 32, 128, 511]",
|
319 |
+
"name" : "v_cache_2",
|
320 |
+
"type" : "MultiArray"
|
321 |
+
}
|
322 |
+
],
|
323 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
324 |
+
"storagePrecision" : "Float16",
|
325 |
+
"stateSchema" : [
|
326 |
+
|
327 |
+
],
|
328 |
+
"outputSchema" : [
|
329 |
+
{
|
330 |
+
"hasShapeFlexibility" : "0",
|
331 |
+
"isOptional" : "0",
|
332 |
+
"dataType" : "Float16",
|
333 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 1)",
|
334 |
+
"shortDescription" : "",
|
335 |
+
"shape" : "[1, 4096, 1, 1]",
|
336 |
+
"name" : "new_x",
|
337 |
+
"type" : "MultiArray"
|
338 |
+
},
|
339 |
+
{
|
340 |
+
"hasShapeFlexibility" : "0",
|
341 |
+
"isOptional" : "0",
|
342 |
+
"dataType" : "Float16",
|
343 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
344 |
+
"shortDescription" : "",
|
345 |
+
"shape" : "[1, 32, 128, 511]",
|
346 |
+
"name" : "new_k_cache_0",
|
347 |
+
"type" : "MultiArray"
|
348 |
+
},
|
349 |
+
{
|
350 |
+
"hasShapeFlexibility" : "0",
|
351 |
+
"isOptional" : "0",
|
352 |
+
"dataType" : "Float16",
|
353 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
354 |
+
"shortDescription" : "",
|
355 |
+
"shape" : "[1, 32, 128, 511]",
|
356 |
+
"name" : "new_k_cache_1",
|
357 |
+
"type" : "MultiArray"
|
358 |
+
},
|
359 |
+
{
|
360 |
+
"hasShapeFlexibility" : "0",
|
361 |
+
"isOptional" : "0",
|
362 |
+
"dataType" : "Float16",
|
363 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
364 |
+
"shortDescription" : "",
|
365 |
+
"shape" : "[1, 32, 128, 511]",
|
366 |
+
"name" : "new_k_cache_2",
|
367 |
+
"type" : "MultiArray"
|
368 |
+
},
|
369 |
+
{
|
370 |
+
"hasShapeFlexibility" : "0",
|
371 |
+
"isOptional" : "0",
|
372 |
+
"dataType" : "Float16",
|
373 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
374 |
+
"shortDescription" : "",
|
375 |
+
"shape" : "[1, 32, 128, 511]",
|
376 |
+
"name" : "new_v_cache_0",
|
377 |
+
"type" : "MultiArray"
|
378 |
+
},
|
379 |
+
{
|
380 |
+
"hasShapeFlexibility" : "0",
|
381 |
+
"isOptional" : "0",
|
382 |
+
"dataType" : "Float16",
|
383 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
384 |
+
"shortDescription" : "",
|
385 |
+
"shape" : "[1, 32, 128, 511]",
|
386 |
+
"name" : "new_v_cache_1",
|
387 |
+
"type" : "MultiArray"
|
388 |
+
},
|
389 |
+
{
|
390 |
+
"hasShapeFlexibility" : "0",
|
391 |
+
"isOptional" : "0",
|
392 |
+
"dataType" : "Float16",
|
393 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
394 |
+
"shortDescription" : "",
|
395 |
+
"shape" : "[1, 32, 128, 511]",
|
396 |
+
"name" : "new_v_cache_2",
|
397 |
+
"type" : "MultiArray"
|
398 |
+
}
|
399 |
+
],
|
400 |
+
"name" : "input_1_context_512",
|
401 |
+
"mlProgramOperationTypeHistogram" : {
|
402 |
+
"Ios18.constexprLutToDense" : 21,
|
403 |
+
"Ios18.conv" : 21,
|
404 |
+
"Ios18.matmul" : 6,
|
405 |
+
"Ios18.expandDims" : 6,
|
406 |
+
"Ios18.concat" : 18,
|
407 |
+
"Ios18.add" : 15,
|
408 |
+
"Ios18.realDiv" : 6,
|
409 |
+
"Ios18.silu" : 3,
|
410 |
+
"Ios18.softmax" : 3,
|
411 |
+
"Ios18.sliceByIndex" : 18,
|
412 |
+
"Ios16.reduceL2Norm" : 6,
|
413 |
+
"Ios18.squeeze" : 6,
|
414 |
+
"Ios18.reshape" : 12,
|
415 |
+
"Ios18.mul" : 57
|
416 |
+
}
|
417 |
+
}
|
418 |
+
],
|
419 |
+
"mlProgramOperationTypeHistogram" : {
|
420 |
+
"Ios18.constexprLutToDense" : 21,
|
421 |
+
"Ios18.conv" : 21,
|
422 |
+
"Ios18.matmul" : 6,
|
423 |
+
"Ios18.expandDims" : 6,
|
424 |
+
"Ios18.concat" : 18,
|
425 |
+
"Ios18.add" : 15,
|
426 |
+
"Ios18.realDiv" : 6,
|
427 |
+
"Ios18.silu" : 3,
|
428 |
+
"Ios18.softmax" : 3,
|
429 |
+
"Ios18.sliceByIndex" : 18,
|
430 |
+
"Ios16.reduceL2Norm" : 6,
|
431 |
+
"Ios18.squeeze" : 6,
|
432 |
+
"Ios18.reshape" : 12,
|
433 |
+
"Ios18.mul" : 57
|
434 |
+
},
|
435 |
+
"isUpdatable" : "0",
|
436 |
+
"stateSchema" : [
|
437 |
+
|
438 |
+
],
|
439 |
+
"availability" : {
|
440 |
+
"macOS" : "15.0",
|
441 |
+
"tvOS" : "18.0",
|
442 |
+
"visionOS" : "2.0",
|
443 |
+
"watchOS" : "11.0",
|
444 |
+
"iOS" : "18.0",
|
445 |
+
"macCatalyst" : "18.0"
|
446 |
+
},
|
447 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
448 |
+
"modelType" : {
|
449 |
+
"name" : "MLModelType_mlProgram"
|
450 |
+
},
|
451 |
+
"inputSchema" : [
|
452 |
+
{
|
453 |
+
"hasShapeFlexibility" : "0",
|
454 |
+
"isOptional" : "0",
|
455 |
+
"dataType" : "Float16",
|
456 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
457 |
+
"shortDescription" : "",
|
458 |
+
"shape" : "[1, 4096, 1, 512]",
|
459 |
+
"name" : "x",
|
460 |
+
"type" : "MultiArray"
|
461 |
+
},
|
462 |
+
{
|
463 |
+
"hasShapeFlexibility" : "0",
|
464 |
+
"isOptional" : "0",
|
465 |
+
"dataType" : "Float16",
|
466 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
467 |
+
"shortDescription" : "",
|
468 |
+
"shape" : "[128, 512]",
|
469 |
+
"name" : "cos",
|
470 |
+
"type" : "MultiArray"
|
471 |
+
},
|
472 |
+
{
|
473 |
+
"hasShapeFlexibility" : "0",
|
474 |
+
"isOptional" : "0",
|
475 |
+
"dataType" : "Float16",
|
476 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
477 |
+
"shortDescription" : "",
|
478 |
+
"shape" : "[128, 512]",
|
479 |
+
"name" : "sin",
|
480 |
+
"type" : "MultiArray"
|
481 |
+
},
|
482 |
+
{
|
483 |
+
"hasShapeFlexibility" : "0",
|
484 |
+
"isOptional" : "0",
|
485 |
+
"dataType" : "Float16",
|
486 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
487 |
+
"shortDescription" : "",
|
488 |
+
"shape" : "[1, 1, 512, 512]",
|
489 |
+
"name" : "mask",
|
490 |
+
"type" : "MultiArray"
|
491 |
+
}
|
492 |
+
],
|
493 |
+
"defaultFunctionName" : "input_512_context_512",
|
494 |
+
"generatedClassName" : "Llama_2_7b_hf_2024_07_02_20_36_17_merged_chunk4",
|
495 |
+
"userDefinedMetadata" : {
|
496 |
+
|
497 |
+
},
|
498 |
+
"method" : "predict"
|
499 |
+
}
|
500 |
+
]
|
sequoia/Llama-2-7b-hf_chunk4.mlmodelc/model.mil
ADDED
The diff for this file is too large to render.
See raw diff
|
|
sequoia/Llama-2-7b-hf_chunk4.mlmodelc/weights/weight.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f3b99e2f973b8638ce47daa04e5c014c160cb53c015a6125b8c0f70bed5cfb8d
|
3 |
+
size 303876448
|
sequoia/Llama-2-7b-hf_chunk5.mlmodelc/analytics/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:84e317a82cdf4e96f808f63e77f10098844d47ad522545181edfac4d287c9c92
|
3 |
+
size 243
|
sequoia/Llama-2-7b-hf_chunk5.mlmodelc/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e430d0795ff5c384187174f5718a2c13d0070f5d6a811831e18862497865a86d
|
3 |
+
size 1037
|
sequoia/Llama-2-7b-hf_chunk5.mlmodelc/metadata.json
ADDED
@@ -0,0 +1,500 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 × 4096 × 1 × 512)",
|
11 |
+
"shortDescription" : "",
|
12 |
+
"shape" : "[1, 4096, 1, 512]",
|
13 |
+
"name" : "new_x",
|
14 |
+
"type" : "MultiArray"
|
15 |
+
},
|
16 |
+
{
|
17 |
+
"hasShapeFlexibility" : "0",
|
18 |
+
"isOptional" : "0",
|
19 |
+
"dataType" : "Float16",
|
20 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
21 |
+
"shortDescription" : "",
|
22 |
+
"shape" : "[1, 32, 128, 511]",
|
23 |
+
"name" : "new_k_cache_0",
|
24 |
+
"type" : "MultiArray"
|
25 |
+
},
|
26 |
+
{
|
27 |
+
"hasShapeFlexibility" : "0",
|
28 |
+
"isOptional" : "0",
|
29 |
+
"dataType" : "Float16",
|
30 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
31 |
+
"shortDescription" : "",
|
32 |
+
"shape" : "[1, 32, 128, 511]",
|
33 |
+
"name" : "new_k_cache_1",
|
34 |
+
"type" : "MultiArray"
|
35 |
+
},
|
36 |
+
{
|
37 |
+
"hasShapeFlexibility" : "0",
|
38 |
+
"isOptional" : "0",
|
39 |
+
"dataType" : "Float16",
|
40 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
41 |
+
"shortDescription" : "",
|
42 |
+
"shape" : "[1, 32, 128, 511]",
|
43 |
+
"name" : "new_k_cache_2",
|
44 |
+
"type" : "MultiArray"
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"hasShapeFlexibility" : "0",
|
48 |
+
"isOptional" : "0",
|
49 |
+
"dataType" : "Float16",
|
50 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
51 |
+
"shortDescription" : "",
|
52 |
+
"shape" : "[1, 32, 128, 511]",
|
53 |
+
"name" : "new_v_cache_0",
|
54 |
+
"type" : "MultiArray"
|
55 |
+
},
|
56 |
+
{
|
57 |
+
"hasShapeFlexibility" : "0",
|
58 |
+
"isOptional" : "0",
|
59 |
+
"dataType" : "Float16",
|
60 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
61 |
+
"shortDescription" : "",
|
62 |
+
"shape" : "[1, 32, 128, 511]",
|
63 |
+
"name" : "new_v_cache_1",
|
64 |
+
"type" : "MultiArray"
|
65 |
+
},
|
66 |
+
{
|
67 |
+
"hasShapeFlexibility" : "0",
|
68 |
+
"isOptional" : "0",
|
69 |
+
"dataType" : "Float16",
|
70 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
71 |
+
"shortDescription" : "",
|
72 |
+
"shape" : "[1, 32, 128, 511]",
|
73 |
+
"name" : "new_v_cache_2",
|
74 |
+
"type" : "MultiArray"
|
75 |
+
}
|
76 |
+
],
|
77 |
+
"modelParameters" : [
|
78 |
+
|
79 |
+
],
|
80 |
+
"specificationVersion" : 9,
|
81 |
+
"functions" : [
|
82 |
+
{
|
83 |
+
"inputSchema" : [
|
84 |
+
{
|
85 |
+
"hasShapeFlexibility" : "0",
|
86 |
+
"isOptional" : "0",
|
87 |
+
"dataType" : "Float16",
|
88 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
89 |
+
"shortDescription" : "",
|
90 |
+
"shape" : "[1, 4096, 1, 512]",
|
91 |
+
"name" : "x",
|
92 |
+
"type" : "MultiArray"
|
93 |
+
},
|
94 |
+
{
|
95 |
+
"hasShapeFlexibility" : "0",
|
96 |
+
"isOptional" : "0",
|
97 |
+
"dataType" : "Float16",
|
98 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
99 |
+
"shortDescription" : "",
|
100 |
+
"shape" : "[128, 512]",
|
101 |
+
"name" : "cos",
|
102 |
+
"type" : "MultiArray"
|
103 |
+
},
|
104 |
+
{
|
105 |
+
"hasShapeFlexibility" : "0",
|
106 |
+
"isOptional" : "0",
|
107 |
+
"dataType" : "Float16",
|
108 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
109 |
+
"shortDescription" : "",
|
110 |
+
"shape" : "[128, 512]",
|
111 |
+
"name" : "sin",
|
112 |
+
"type" : "MultiArray"
|
113 |
+
},
|
114 |
+
{
|
115 |
+
"hasShapeFlexibility" : "0",
|
116 |
+
"isOptional" : "0",
|
117 |
+
"dataType" : "Float16",
|
118 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
119 |
+
"shortDescription" : "",
|
120 |
+
"shape" : "[1, 1, 512, 512]",
|
121 |
+
"name" : "mask",
|
122 |
+
"type" : "MultiArray"
|
123 |
+
}
|
124 |
+
],
|
125 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
126 |
+
"storagePrecision" : "Float16",
|
127 |
+
"stateSchema" : [
|
128 |
+
|
129 |
+
],
|
130 |
+
"outputSchema" : [
|
131 |
+
{
|
132 |
+
"hasShapeFlexibility" : "0",
|
133 |
+
"isOptional" : "0",
|
134 |
+
"dataType" : "Float16",
|
135 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
136 |
+
"shortDescription" : "",
|
137 |
+
"shape" : "[1, 4096, 1, 512]",
|
138 |
+
"name" : "new_x",
|
139 |
+
"type" : "MultiArray"
|
140 |
+
},
|
141 |
+
{
|
142 |
+
"hasShapeFlexibility" : "0",
|
143 |
+
"isOptional" : "0",
|
144 |
+
"dataType" : "Float16",
|
145 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
146 |
+
"shortDescription" : "",
|
147 |
+
"shape" : "[1, 32, 128, 511]",
|
148 |
+
"name" : "new_k_cache_0",
|
149 |
+
"type" : "MultiArray"
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"hasShapeFlexibility" : "0",
|
153 |
+
"isOptional" : "0",
|
154 |
+
"dataType" : "Float16",
|
155 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
156 |
+
"shortDescription" : "",
|
157 |
+
"shape" : "[1, 32, 128, 511]",
|
158 |
+
"name" : "new_k_cache_1",
|
159 |
+
"type" : "MultiArray"
|
160 |
+
},
|
161 |
+
{
|
162 |
+
"hasShapeFlexibility" : "0",
|
163 |
+
"isOptional" : "0",
|
164 |
+
"dataType" : "Float16",
|
165 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
166 |
+
"shortDescription" : "",
|
167 |
+
"shape" : "[1, 32, 128, 511]",
|
168 |
+
"name" : "new_k_cache_2",
|
169 |
+
"type" : "MultiArray"
|
170 |
+
},
|
171 |
+
{
|
172 |
+
"hasShapeFlexibility" : "0",
|
173 |
+
"isOptional" : "0",
|
174 |
+
"dataType" : "Float16",
|
175 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
176 |
+
"shortDescription" : "",
|
177 |
+
"shape" : "[1, 32, 128, 511]",
|
178 |
+
"name" : "new_v_cache_0",
|
179 |
+
"type" : "MultiArray"
|
180 |
+
},
|
181 |
+
{
|
182 |
+
"hasShapeFlexibility" : "0",
|
183 |
+
"isOptional" : "0",
|
184 |
+
"dataType" : "Float16",
|
185 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
186 |
+
"shortDescription" : "",
|
187 |
+
"shape" : "[1, 32, 128, 511]",
|
188 |
+
"name" : "new_v_cache_1",
|
189 |
+
"type" : "MultiArray"
|
190 |
+
},
|
191 |
+
{
|
192 |
+
"hasShapeFlexibility" : "0",
|
193 |
+
"isOptional" : "0",
|
194 |
+
"dataType" : "Float16",
|
195 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
196 |
+
"shortDescription" : "",
|
197 |
+
"shape" : "[1, 32, 128, 511]",
|
198 |
+
"name" : "new_v_cache_2",
|
199 |
+
"type" : "MultiArray"
|
200 |
+
}
|
201 |
+
],
|
202 |
+
"name" : "input_512_context_512",
|
203 |
+
"mlProgramOperationTypeHistogram" : {
|
204 |
+
"Ios18.constexprLutToDense" : 21,
|
205 |
+
"Ios18.conv" : 21,
|
206 |
+
"Ios18.matmul" : 6,
|
207 |
+
"Ios18.expandDims" : 6,
|
208 |
+
"Ios18.concat" : 18,
|
209 |
+
"Ios18.add" : 15,
|
210 |
+
"Ios18.realDiv" : 6,
|
211 |
+
"Ios18.silu" : 3,
|
212 |
+
"Ios18.softmax" : 3,
|
213 |
+
"Ios18.sliceByIndex" : 18,
|
214 |
+
"Ios16.reduceL2Norm" : 6,
|
215 |
+
"Ios18.squeeze" : 6,
|
216 |
+
"Ios18.reshape" : 12,
|
217 |
+
"Ios18.mul" : 57
|
218 |
+
}
|
219 |
+
},
|
220 |
+
{
|
221 |
+
"inputSchema" : [
|
222 |
+
{
|
223 |
+
"hasShapeFlexibility" : "0",
|
224 |
+
"isOptional" : "0",
|
225 |
+
"dataType" : "Float16",
|
226 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 1)",
|
227 |
+
"shortDescription" : "",
|
228 |
+
"shape" : "[1, 4096, 1, 1]",
|
229 |
+
"name" : "x",
|
230 |
+
"type" : "MultiArray"
|
231 |
+
},
|
232 |
+
{
|
233 |
+
"hasShapeFlexibility" : "0",
|
234 |
+
"isOptional" : "0",
|
235 |
+
"dataType" : "Float16",
|
236 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
237 |
+
"shortDescription" : "",
|
238 |
+
"shape" : "[128, 1]",
|
239 |
+
"name" : "cos",
|
240 |
+
"type" : "MultiArray"
|
241 |
+
},
|
242 |
+
{
|
243 |
+
"hasShapeFlexibility" : "0",
|
244 |
+
"isOptional" : "0",
|
245 |
+
"dataType" : "Float16",
|
246 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
247 |
+
"shortDescription" : "",
|
248 |
+
"shape" : "[128, 1]",
|
249 |
+
"name" : "sin",
|
250 |
+
"type" : "MultiArray"
|
251 |
+
},
|
252 |
+
{
|
253 |
+
"hasShapeFlexibility" : "0",
|
254 |
+
"isOptional" : "0",
|
255 |
+
"dataType" : "Float16",
|
256 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 1 × 512)",
|
257 |
+
"shortDescription" : "",
|
258 |
+
"shape" : "[1, 1, 1, 512]",
|
259 |
+
"name" : "mask",
|
260 |
+
"type" : "MultiArray"
|
261 |
+
},
|
262 |
+
{
|
263 |
+
"hasShapeFlexibility" : "0",
|
264 |
+
"isOptional" : "1",
|
265 |
+
"dataType" : "Float16",
|
266 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
267 |
+
"shortDescription" : "",
|
268 |
+
"shape" : "[1, 32, 128, 511]",
|
269 |
+
"name" : "k_cache_0",
|
270 |
+
"type" : "MultiArray"
|
271 |
+
},
|
272 |
+
{
|
273 |
+
"hasShapeFlexibility" : "0",
|
274 |
+
"isOptional" : "1",
|
275 |
+
"dataType" : "Float16",
|
276 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
277 |
+
"shortDescription" : "",
|
278 |
+
"shape" : "[1, 32, 128, 511]",
|
279 |
+
"name" : "v_cache_0",
|
280 |
+
"type" : "MultiArray"
|
281 |
+
},
|
282 |
+
{
|
283 |
+
"hasShapeFlexibility" : "0",
|
284 |
+
"isOptional" : "1",
|
285 |
+
"dataType" : "Float16",
|
286 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
287 |
+
"shortDescription" : "",
|
288 |
+
"shape" : "[1, 32, 128, 511]",
|
289 |
+
"name" : "k_cache_1",
|
290 |
+
"type" : "MultiArray"
|
291 |
+
},
|
292 |
+
{
|
293 |
+
"hasShapeFlexibility" : "0",
|
294 |
+
"isOptional" : "1",
|
295 |
+
"dataType" : "Float16",
|
296 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
297 |
+
"shortDescription" : "",
|
298 |
+
"shape" : "[1, 32, 128, 511]",
|
299 |
+
"name" : "v_cache_1",
|
300 |
+
"type" : "MultiArray"
|
301 |
+
},
|
302 |
+
{
|
303 |
+
"hasShapeFlexibility" : "0",
|
304 |
+
"isOptional" : "1",
|
305 |
+
"dataType" : "Float16",
|
306 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
307 |
+
"shortDescription" : "",
|
308 |
+
"shape" : "[1, 32, 128, 511]",
|
309 |
+
"name" : "k_cache_2",
|
310 |
+
"type" : "MultiArray"
|
311 |
+
},
|
312 |
+
{
|
313 |
+
"hasShapeFlexibility" : "0",
|
314 |
+
"isOptional" : "1",
|
315 |
+
"dataType" : "Float16",
|
316 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
317 |
+
"shortDescription" : "",
|
318 |
+
"shape" : "[1, 32, 128, 511]",
|
319 |
+
"name" : "v_cache_2",
|
320 |
+
"type" : "MultiArray"
|
321 |
+
}
|
322 |
+
],
|
323 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
324 |
+
"storagePrecision" : "Float16",
|
325 |
+
"stateSchema" : [
|
326 |
+
|
327 |
+
],
|
328 |
+
"outputSchema" : [
|
329 |
+
{
|
330 |
+
"hasShapeFlexibility" : "0",
|
331 |
+
"isOptional" : "0",
|
332 |
+
"dataType" : "Float16",
|
333 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 1)",
|
334 |
+
"shortDescription" : "",
|
335 |
+
"shape" : "[1, 4096, 1, 1]",
|
336 |
+
"name" : "new_x",
|
337 |
+
"type" : "MultiArray"
|
338 |
+
},
|
339 |
+
{
|
340 |
+
"hasShapeFlexibility" : "0",
|
341 |
+
"isOptional" : "0",
|
342 |
+
"dataType" : "Float16",
|
343 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
344 |
+
"shortDescription" : "",
|
345 |
+
"shape" : "[1, 32, 128, 511]",
|
346 |
+
"name" : "new_k_cache_0",
|
347 |
+
"type" : "MultiArray"
|
348 |
+
},
|
349 |
+
{
|
350 |
+
"hasShapeFlexibility" : "0",
|
351 |
+
"isOptional" : "0",
|
352 |
+
"dataType" : "Float16",
|
353 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
354 |
+
"shortDescription" : "",
|
355 |
+
"shape" : "[1, 32, 128, 511]",
|
356 |
+
"name" : "new_k_cache_1",
|
357 |
+
"type" : "MultiArray"
|
358 |
+
},
|
359 |
+
{
|
360 |
+
"hasShapeFlexibility" : "0",
|
361 |
+
"isOptional" : "0",
|
362 |
+
"dataType" : "Float16",
|
363 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
364 |
+
"shortDescription" : "",
|
365 |
+
"shape" : "[1, 32, 128, 511]",
|
366 |
+
"name" : "new_k_cache_2",
|
367 |
+
"type" : "MultiArray"
|
368 |
+
},
|
369 |
+
{
|
370 |
+
"hasShapeFlexibility" : "0",
|
371 |
+
"isOptional" : "0",
|
372 |
+
"dataType" : "Float16",
|
373 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
374 |
+
"shortDescription" : "",
|
375 |
+
"shape" : "[1, 32, 128, 511]",
|
376 |
+
"name" : "new_v_cache_0",
|
377 |
+
"type" : "MultiArray"
|
378 |
+
},
|
379 |
+
{
|
380 |
+
"hasShapeFlexibility" : "0",
|
381 |
+
"isOptional" : "0",
|
382 |
+
"dataType" : "Float16",
|
383 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
384 |
+
"shortDescription" : "",
|
385 |
+
"shape" : "[1, 32, 128, 511]",
|
386 |
+
"name" : "new_v_cache_1",
|
387 |
+
"type" : "MultiArray"
|
388 |
+
},
|
389 |
+
{
|
390 |
+
"hasShapeFlexibility" : "0",
|
391 |
+
"isOptional" : "0",
|
392 |
+
"dataType" : "Float16",
|
393 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
394 |
+
"shortDescription" : "",
|
395 |
+
"shape" : "[1, 32, 128, 511]",
|
396 |
+
"name" : "new_v_cache_2",
|
397 |
+
"type" : "MultiArray"
|
398 |
+
}
|
399 |
+
],
|
400 |
+
"name" : "input_1_context_512",
|
401 |
+
"mlProgramOperationTypeHistogram" : {
|
402 |
+
"Ios18.constexprLutToDense" : 21,
|
403 |
+
"Ios18.conv" : 21,
|
404 |
+
"Ios18.matmul" : 6,
|
405 |
+
"Ios18.expandDims" : 6,
|
406 |
+
"Ios18.concat" : 18,
|
407 |
+
"Ios18.add" : 15,
|
408 |
+
"Ios18.realDiv" : 6,
|
409 |
+
"Ios18.silu" : 3,
|
410 |
+
"Ios18.softmax" : 3,
|
411 |
+
"Ios18.sliceByIndex" : 18,
|
412 |
+
"Ios16.reduceL2Norm" : 6,
|
413 |
+
"Ios18.squeeze" : 6,
|
414 |
+
"Ios18.reshape" : 12,
|
415 |
+
"Ios18.mul" : 57
|
416 |
+
}
|
417 |
+
}
|
418 |
+
],
|
419 |
+
"mlProgramOperationTypeHistogram" : {
|
420 |
+
"Ios18.constexprLutToDense" : 21,
|
421 |
+
"Ios18.conv" : 21,
|
422 |
+
"Ios18.matmul" : 6,
|
423 |
+
"Ios18.expandDims" : 6,
|
424 |
+
"Ios18.concat" : 18,
|
425 |
+
"Ios18.add" : 15,
|
426 |
+
"Ios18.realDiv" : 6,
|
427 |
+
"Ios18.silu" : 3,
|
428 |
+
"Ios18.softmax" : 3,
|
429 |
+
"Ios18.sliceByIndex" : 18,
|
430 |
+
"Ios16.reduceL2Norm" : 6,
|
431 |
+
"Ios18.squeeze" : 6,
|
432 |
+
"Ios18.reshape" : 12,
|
433 |
+
"Ios18.mul" : 57
|
434 |
+
},
|
435 |
+
"isUpdatable" : "0",
|
436 |
+
"stateSchema" : [
|
437 |
+
|
438 |
+
],
|
439 |
+
"availability" : {
|
440 |
+
"macOS" : "15.0",
|
441 |
+
"tvOS" : "18.0",
|
442 |
+
"visionOS" : "2.0",
|
443 |
+
"watchOS" : "11.0",
|
444 |
+
"iOS" : "18.0",
|
445 |
+
"macCatalyst" : "18.0"
|
446 |
+
},
|
447 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
448 |
+
"modelType" : {
|
449 |
+
"name" : "MLModelType_mlProgram"
|
450 |
+
},
|
451 |
+
"inputSchema" : [
|
452 |
+
{
|
453 |
+
"hasShapeFlexibility" : "0",
|
454 |
+
"isOptional" : "0",
|
455 |
+
"dataType" : "Float16",
|
456 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
457 |
+
"shortDescription" : "",
|
458 |
+
"shape" : "[1, 4096, 1, 512]",
|
459 |
+
"name" : "x",
|
460 |
+
"type" : "MultiArray"
|
461 |
+
},
|
462 |
+
{
|
463 |
+
"hasShapeFlexibility" : "0",
|
464 |
+
"isOptional" : "0",
|
465 |
+
"dataType" : "Float16",
|
466 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
467 |
+
"shortDescription" : "",
|
468 |
+
"shape" : "[128, 512]",
|
469 |
+
"name" : "cos",
|
470 |
+
"type" : "MultiArray"
|
471 |
+
},
|
472 |
+
{
|
473 |
+
"hasShapeFlexibility" : "0",
|
474 |
+
"isOptional" : "0",
|
475 |
+
"dataType" : "Float16",
|
476 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
477 |
+
"shortDescription" : "",
|
478 |
+
"shape" : "[128, 512]",
|
479 |
+
"name" : "sin",
|
480 |
+
"type" : "MultiArray"
|
481 |
+
},
|
482 |
+
{
|
483 |
+
"hasShapeFlexibility" : "0",
|
484 |
+
"isOptional" : "0",
|
485 |
+
"dataType" : "Float16",
|
486 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
487 |
+
"shortDescription" : "",
|
488 |
+
"shape" : "[1, 1, 512, 512]",
|
489 |
+
"name" : "mask",
|
490 |
+
"type" : "MultiArray"
|
491 |
+
}
|
492 |
+
],
|
493 |
+
"defaultFunctionName" : "input_512_context_512",
|
494 |
+
"generatedClassName" : "Llama_2_7b_hf_2024_07_02_20_36_17_merged_chunk5",
|
495 |
+
"userDefinedMetadata" : {
|
496 |
+
|
497 |
+
},
|
498 |
+
"method" : "predict"
|
499 |
+
}
|
500 |
+
]
|
sequoia/Llama-2-7b-hf_chunk5.mlmodelc/model.mil
ADDED
The diff for this file is too large to render.
See raw diff
|
|
sequoia/Llama-2-7b-hf_chunk5.mlmodelc/weights/weight.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:52221781d6f35b15afe3960bcb020cc08c0fcfa936c1903497deed92e008e870
|
3 |
+
size 303876448
|
sequoia/Llama-2-7b-hf_chunk6.mlmodelc/analytics/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:84e317a82cdf4e96f808f63e77f10098844d47ad522545181edfac4d287c9c92
|
3 |
+
size 243
|
sequoia/Llama-2-7b-hf_chunk6.mlmodelc/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e430d0795ff5c384187174f5718a2c13d0070f5d6a811831e18862497865a86d
|
3 |
+
size 1037
|
sequoia/Llama-2-7b-hf_chunk6.mlmodelc/metadata.json
ADDED
@@ -0,0 +1,500 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 × 4096 × 1 × 512)",
|
11 |
+
"shortDescription" : "",
|
12 |
+
"shape" : "[1, 4096, 1, 512]",
|
13 |
+
"name" : "new_x",
|
14 |
+
"type" : "MultiArray"
|
15 |
+
},
|
16 |
+
{
|
17 |
+
"hasShapeFlexibility" : "0",
|
18 |
+
"isOptional" : "0",
|
19 |
+
"dataType" : "Float16",
|
20 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
21 |
+
"shortDescription" : "",
|
22 |
+
"shape" : "[1, 32, 128, 511]",
|
23 |
+
"name" : "new_k_cache_0",
|
24 |
+
"type" : "MultiArray"
|
25 |
+
},
|
26 |
+
{
|
27 |
+
"hasShapeFlexibility" : "0",
|
28 |
+
"isOptional" : "0",
|
29 |
+
"dataType" : "Float16",
|
30 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
31 |
+
"shortDescription" : "",
|
32 |
+
"shape" : "[1, 32, 128, 511]",
|
33 |
+
"name" : "new_k_cache_1",
|
34 |
+
"type" : "MultiArray"
|
35 |
+
},
|
36 |
+
{
|
37 |
+
"hasShapeFlexibility" : "0",
|
38 |
+
"isOptional" : "0",
|
39 |
+
"dataType" : "Float16",
|
40 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
41 |
+
"shortDescription" : "",
|
42 |
+
"shape" : "[1, 32, 128, 511]",
|
43 |
+
"name" : "new_k_cache_2",
|
44 |
+
"type" : "MultiArray"
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"hasShapeFlexibility" : "0",
|
48 |
+
"isOptional" : "0",
|
49 |
+
"dataType" : "Float16",
|
50 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
51 |
+
"shortDescription" : "",
|
52 |
+
"shape" : "[1, 32, 128, 511]",
|
53 |
+
"name" : "new_v_cache_0",
|
54 |
+
"type" : "MultiArray"
|
55 |
+
},
|
56 |
+
{
|
57 |
+
"hasShapeFlexibility" : "0",
|
58 |
+
"isOptional" : "0",
|
59 |
+
"dataType" : "Float16",
|
60 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
61 |
+
"shortDescription" : "",
|
62 |
+
"shape" : "[1, 32, 128, 511]",
|
63 |
+
"name" : "new_v_cache_1",
|
64 |
+
"type" : "MultiArray"
|
65 |
+
},
|
66 |
+
{
|
67 |
+
"hasShapeFlexibility" : "0",
|
68 |
+
"isOptional" : "0",
|
69 |
+
"dataType" : "Float16",
|
70 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
71 |
+
"shortDescription" : "",
|
72 |
+
"shape" : "[1, 32, 128, 511]",
|
73 |
+
"name" : "new_v_cache_2",
|
74 |
+
"type" : "MultiArray"
|
75 |
+
}
|
76 |
+
],
|
77 |
+
"modelParameters" : [
|
78 |
+
|
79 |
+
],
|
80 |
+
"specificationVersion" : 9,
|
81 |
+
"functions" : [
|
82 |
+
{
|
83 |
+
"inputSchema" : [
|
84 |
+
{
|
85 |
+
"hasShapeFlexibility" : "0",
|
86 |
+
"isOptional" : "0",
|
87 |
+
"dataType" : "Float16",
|
88 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
89 |
+
"shortDescription" : "",
|
90 |
+
"shape" : "[1, 4096, 1, 512]",
|
91 |
+
"name" : "x",
|
92 |
+
"type" : "MultiArray"
|
93 |
+
},
|
94 |
+
{
|
95 |
+
"hasShapeFlexibility" : "0",
|
96 |
+
"isOptional" : "0",
|
97 |
+
"dataType" : "Float16",
|
98 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
99 |
+
"shortDescription" : "",
|
100 |
+
"shape" : "[128, 512]",
|
101 |
+
"name" : "cos",
|
102 |
+
"type" : "MultiArray"
|
103 |
+
},
|
104 |
+
{
|
105 |
+
"hasShapeFlexibility" : "0",
|
106 |
+
"isOptional" : "0",
|
107 |
+
"dataType" : "Float16",
|
108 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
109 |
+
"shortDescription" : "",
|
110 |
+
"shape" : "[128, 512]",
|
111 |
+
"name" : "sin",
|
112 |
+
"type" : "MultiArray"
|
113 |
+
},
|
114 |
+
{
|
115 |
+
"hasShapeFlexibility" : "0",
|
116 |
+
"isOptional" : "0",
|
117 |
+
"dataType" : "Float16",
|
118 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
119 |
+
"shortDescription" : "",
|
120 |
+
"shape" : "[1, 1, 512, 512]",
|
121 |
+
"name" : "mask",
|
122 |
+
"type" : "MultiArray"
|
123 |
+
}
|
124 |
+
],
|
125 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
126 |
+
"storagePrecision" : "Float16",
|
127 |
+
"stateSchema" : [
|
128 |
+
|
129 |
+
],
|
130 |
+
"outputSchema" : [
|
131 |
+
{
|
132 |
+
"hasShapeFlexibility" : "0",
|
133 |
+
"isOptional" : "0",
|
134 |
+
"dataType" : "Float16",
|
135 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
136 |
+
"shortDescription" : "",
|
137 |
+
"shape" : "[1, 4096, 1, 512]",
|
138 |
+
"name" : "new_x",
|
139 |
+
"type" : "MultiArray"
|
140 |
+
},
|
141 |
+
{
|
142 |
+
"hasShapeFlexibility" : "0",
|
143 |
+
"isOptional" : "0",
|
144 |
+
"dataType" : "Float16",
|
145 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
146 |
+
"shortDescription" : "",
|
147 |
+
"shape" : "[1, 32, 128, 511]",
|
148 |
+
"name" : "new_k_cache_0",
|
149 |
+
"type" : "MultiArray"
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"hasShapeFlexibility" : "0",
|
153 |
+
"isOptional" : "0",
|
154 |
+
"dataType" : "Float16",
|
155 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
156 |
+
"shortDescription" : "",
|
157 |
+
"shape" : "[1, 32, 128, 511]",
|
158 |
+
"name" : "new_k_cache_1",
|
159 |
+
"type" : "MultiArray"
|
160 |
+
},
|
161 |
+
{
|
162 |
+
"hasShapeFlexibility" : "0",
|
163 |
+
"isOptional" : "0",
|
164 |
+
"dataType" : "Float16",
|
165 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
166 |
+
"shortDescription" : "",
|
167 |
+
"shape" : "[1, 32, 128, 511]",
|
168 |
+
"name" : "new_k_cache_2",
|
169 |
+
"type" : "MultiArray"
|
170 |
+
},
|
171 |
+
{
|
172 |
+
"hasShapeFlexibility" : "0",
|
173 |
+
"isOptional" : "0",
|
174 |
+
"dataType" : "Float16",
|
175 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
176 |
+
"shortDescription" : "",
|
177 |
+
"shape" : "[1, 32, 128, 511]",
|
178 |
+
"name" : "new_v_cache_0",
|
179 |
+
"type" : "MultiArray"
|
180 |
+
},
|
181 |
+
{
|
182 |
+
"hasShapeFlexibility" : "0",
|
183 |
+
"isOptional" : "0",
|
184 |
+
"dataType" : "Float16",
|
185 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
186 |
+
"shortDescription" : "",
|
187 |
+
"shape" : "[1, 32, 128, 511]",
|
188 |
+
"name" : "new_v_cache_1",
|
189 |
+
"type" : "MultiArray"
|
190 |
+
},
|
191 |
+
{
|
192 |
+
"hasShapeFlexibility" : "0",
|
193 |
+
"isOptional" : "0",
|
194 |
+
"dataType" : "Float16",
|
195 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
196 |
+
"shortDescription" : "",
|
197 |
+
"shape" : "[1, 32, 128, 511]",
|
198 |
+
"name" : "new_v_cache_2",
|
199 |
+
"type" : "MultiArray"
|
200 |
+
}
|
201 |
+
],
|
202 |
+
"name" : "input_512_context_512",
|
203 |
+
"mlProgramOperationTypeHistogram" : {
|
204 |
+
"Ios18.constexprLutToDense" : 21,
|
205 |
+
"Ios18.conv" : 21,
|
206 |
+
"Ios18.matmul" : 6,
|
207 |
+
"Ios18.expandDims" : 6,
|
208 |
+
"Ios18.concat" : 18,
|
209 |
+
"Ios18.add" : 15,
|
210 |
+
"Ios18.realDiv" : 6,
|
211 |
+
"Ios18.silu" : 3,
|
212 |
+
"Ios18.softmax" : 3,
|
213 |
+
"Ios18.sliceByIndex" : 18,
|
214 |
+
"Ios16.reduceL2Norm" : 6,
|
215 |
+
"Ios18.squeeze" : 6,
|
216 |
+
"Ios18.reshape" : 12,
|
217 |
+
"Ios18.mul" : 57
|
218 |
+
}
|
219 |
+
},
|
220 |
+
{
|
221 |
+
"inputSchema" : [
|
222 |
+
{
|
223 |
+
"hasShapeFlexibility" : "0",
|
224 |
+
"isOptional" : "0",
|
225 |
+
"dataType" : "Float16",
|
226 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 1)",
|
227 |
+
"shortDescription" : "",
|
228 |
+
"shape" : "[1, 4096, 1, 1]",
|
229 |
+
"name" : "x",
|
230 |
+
"type" : "MultiArray"
|
231 |
+
},
|
232 |
+
{
|
233 |
+
"hasShapeFlexibility" : "0",
|
234 |
+
"isOptional" : "0",
|
235 |
+
"dataType" : "Float16",
|
236 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
237 |
+
"shortDescription" : "",
|
238 |
+
"shape" : "[128, 1]",
|
239 |
+
"name" : "cos",
|
240 |
+
"type" : "MultiArray"
|
241 |
+
},
|
242 |
+
{
|
243 |
+
"hasShapeFlexibility" : "0",
|
244 |
+
"isOptional" : "0",
|
245 |
+
"dataType" : "Float16",
|
246 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
247 |
+
"shortDescription" : "",
|
248 |
+
"shape" : "[128, 1]",
|
249 |
+
"name" : "sin",
|
250 |
+
"type" : "MultiArray"
|
251 |
+
},
|
252 |
+
{
|
253 |
+
"hasShapeFlexibility" : "0",
|
254 |
+
"isOptional" : "0",
|
255 |
+
"dataType" : "Float16",
|
256 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 1 × 512)",
|
257 |
+
"shortDescription" : "",
|
258 |
+
"shape" : "[1, 1, 1, 512]",
|
259 |
+
"name" : "mask",
|
260 |
+
"type" : "MultiArray"
|
261 |
+
},
|
262 |
+
{
|
263 |
+
"hasShapeFlexibility" : "0",
|
264 |
+
"isOptional" : "1",
|
265 |
+
"dataType" : "Float16",
|
266 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
267 |
+
"shortDescription" : "",
|
268 |
+
"shape" : "[1, 32, 128, 511]",
|
269 |
+
"name" : "k_cache_0",
|
270 |
+
"type" : "MultiArray"
|
271 |
+
},
|
272 |
+
{
|
273 |
+
"hasShapeFlexibility" : "0",
|
274 |
+
"isOptional" : "1",
|
275 |
+
"dataType" : "Float16",
|
276 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
277 |
+
"shortDescription" : "",
|
278 |
+
"shape" : "[1, 32, 128, 511]",
|
279 |
+
"name" : "v_cache_0",
|
280 |
+
"type" : "MultiArray"
|
281 |
+
},
|
282 |
+
{
|
283 |
+
"hasShapeFlexibility" : "0",
|
284 |
+
"isOptional" : "1",
|
285 |
+
"dataType" : "Float16",
|
286 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
287 |
+
"shortDescription" : "",
|
288 |
+
"shape" : "[1, 32, 128, 511]",
|
289 |
+
"name" : "k_cache_1",
|
290 |
+
"type" : "MultiArray"
|
291 |
+
},
|
292 |
+
{
|
293 |
+
"hasShapeFlexibility" : "0",
|
294 |
+
"isOptional" : "1",
|
295 |
+
"dataType" : "Float16",
|
296 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
297 |
+
"shortDescription" : "",
|
298 |
+
"shape" : "[1, 32, 128, 511]",
|
299 |
+
"name" : "v_cache_1",
|
300 |
+
"type" : "MultiArray"
|
301 |
+
},
|
302 |
+
{
|
303 |
+
"hasShapeFlexibility" : "0",
|
304 |
+
"isOptional" : "1",
|
305 |
+
"dataType" : "Float16",
|
306 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
307 |
+
"shortDescription" : "",
|
308 |
+
"shape" : "[1, 32, 128, 511]",
|
309 |
+
"name" : "k_cache_2",
|
310 |
+
"type" : "MultiArray"
|
311 |
+
},
|
312 |
+
{
|
313 |
+
"hasShapeFlexibility" : "0",
|
314 |
+
"isOptional" : "1",
|
315 |
+
"dataType" : "Float16",
|
316 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
317 |
+
"shortDescription" : "",
|
318 |
+
"shape" : "[1, 32, 128, 511]",
|
319 |
+
"name" : "v_cache_2",
|
320 |
+
"type" : "MultiArray"
|
321 |
+
}
|
322 |
+
],
|
323 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
324 |
+
"storagePrecision" : "Float16",
|
325 |
+
"stateSchema" : [
|
326 |
+
|
327 |
+
],
|
328 |
+
"outputSchema" : [
|
329 |
+
{
|
330 |
+
"hasShapeFlexibility" : "0",
|
331 |
+
"isOptional" : "0",
|
332 |
+
"dataType" : "Float16",
|
333 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 1)",
|
334 |
+
"shortDescription" : "",
|
335 |
+
"shape" : "[1, 4096, 1, 1]",
|
336 |
+
"name" : "new_x",
|
337 |
+
"type" : "MultiArray"
|
338 |
+
},
|
339 |
+
{
|
340 |
+
"hasShapeFlexibility" : "0",
|
341 |
+
"isOptional" : "0",
|
342 |
+
"dataType" : "Float16",
|
343 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
344 |
+
"shortDescription" : "",
|
345 |
+
"shape" : "[1, 32, 128, 511]",
|
346 |
+
"name" : "new_k_cache_0",
|
347 |
+
"type" : "MultiArray"
|
348 |
+
},
|
349 |
+
{
|
350 |
+
"hasShapeFlexibility" : "0",
|
351 |
+
"isOptional" : "0",
|
352 |
+
"dataType" : "Float16",
|
353 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
354 |
+
"shortDescription" : "",
|
355 |
+
"shape" : "[1, 32, 128, 511]",
|
356 |
+
"name" : "new_k_cache_1",
|
357 |
+
"type" : "MultiArray"
|
358 |
+
},
|
359 |
+
{
|
360 |
+
"hasShapeFlexibility" : "0",
|
361 |
+
"isOptional" : "0",
|
362 |
+
"dataType" : "Float16",
|
363 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
364 |
+
"shortDescription" : "",
|
365 |
+
"shape" : "[1, 32, 128, 511]",
|
366 |
+
"name" : "new_k_cache_2",
|
367 |
+
"type" : "MultiArray"
|
368 |
+
},
|
369 |
+
{
|
370 |
+
"hasShapeFlexibility" : "0",
|
371 |
+
"isOptional" : "0",
|
372 |
+
"dataType" : "Float16",
|
373 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
374 |
+
"shortDescription" : "",
|
375 |
+
"shape" : "[1, 32, 128, 511]",
|
376 |
+
"name" : "new_v_cache_0",
|
377 |
+
"type" : "MultiArray"
|
378 |
+
},
|
379 |
+
{
|
380 |
+
"hasShapeFlexibility" : "0",
|
381 |
+
"isOptional" : "0",
|
382 |
+
"dataType" : "Float16",
|
383 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
384 |
+
"shortDescription" : "",
|
385 |
+
"shape" : "[1, 32, 128, 511]",
|
386 |
+
"name" : "new_v_cache_1",
|
387 |
+
"type" : "MultiArray"
|
388 |
+
},
|
389 |
+
{
|
390 |
+
"hasShapeFlexibility" : "0",
|
391 |
+
"isOptional" : "0",
|
392 |
+
"dataType" : "Float16",
|
393 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
394 |
+
"shortDescription" : "",
|
395 |
+
"shape" : "[1, 32, 128, 511]",
|
396 |
+
"name" : "new_v_cache_2",
|
397 |
+
"type" : "MultiArray"
|
398 |
+
}
|
399 |
+
],
|
400 |
+
"name" : "input_1_context_512",
|
401 |
+
"mlProgramOperationTypeHistogram" : {
|
402 |
+
"Ios18.constexprLutToDense" : 21,
|
403 |
+
"Ios18.conv" : 21,
|
404 |
+
"Ios18.matmul" : 6,
|
405 |
+
"Ios18.expandDims" : 6,
|
406 |
+
"Ios18.concat" : 18,
|
407 |
+
"Ios18.add" : 15,
|
408 |
+
"Ios18.realDiv" : 6,
|
409 |
+
"Ios18.silu" : 3,
|
410 |
+
"Ios18.softmax" : 3,
|
411 |
+
"Ios18.sliceByIndex" : 18,
|
412 |
+
"Ios16.reduceL2Norm" : 6,
|
413 |
+
"Ios18.squeeze" : 6,
|
414 |
+
"Ios18.reshape" : 12,
|
415 |
+
"Ios18.mul" : 57
|
416 |
+
}
|
417 |
+
}
|
418 |
+
],
|
419 |
+
"mlProgramOperationTypeHistogram" : {
|
420 |
+
"Ios18.constexprLutToDense" : 21,
|
421 |
+
"Ios18.conv" : 21,
|
422 |
+
"Ios18.matmul" : 6,
|
423 |
+
"Ios18.expandDims" : 6,
|
424 |
+
"Ios18.concat" : 18,
|
425 |
+
"Ios18.add" : 15,
|
426 |
+
"Ios18.realDiv" : 6,
|
427 |
+
"Ios18.silu" : 3,
|
428 |
+
"Ios18.softmax" : 3,
|
429 |
+
"Ios18.sliceByIndex" : 18,
|
430 |
+
"Ios16.reduceL2Norm" : 6,
|
431 |
+
"Ios18.squeeze" : 6,
|
432 |
+
"Ios18.reshape" : 12,
|
433 |
+
"Ios18.mul" : 57
|
434 |
+
},
|
435 |
+
"isUpdatable" : "0",
|
436 |
+
"stateSchema" : [
|
437 |
+
|
438 |
+
],
|
439 |
+
"availability" : {
|
440 |
+
"macOS" : "15.0",
|
441 |
+
"tvOS" : "18.0",
|
442 |
+
"visionOS" : "2.0",
|
443 |
+
"watchOS" : "11.0",
|
444 |
+
"iOS" : "18.0",
|
445 |
+
"macCatalyst" : "18.0"
|
446 |
+
},
|
447 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
448 |
+
"modelType" : {
|
449 |
+
"name" : "MLModelType_mlProgram"
|
450 |
+
},
|
451 |
+
"inputSchema" : [
|
452 |
+
{
|
453 |
+
"hasShapeFlexibility" : "0",
|
454 |
+
"isOptional" : "0",
|
455 |
+
"dataType" : "Float16",
|
456 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
457 |
+
"shortDescription" : "",
|
458 |
+
"shape" : "[1, 4096, 1, 512]",
|
459 |
+
"name" : "x",
|
460 |
+
"type" : "MultiArray"
|
461 |
+
},
|
462 |
+
{
|
463 |
+
"hasShapeFlexibility" : "0",
|
464 |
+
"isOptional" : "0",
|
465 |
+
"dataType" : "Float16",
|
466 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
467 |
+
"shortDescription" : "",
|
468 |
+
"shape" : "[128, 512]",
|
469 |
+
"name" : "cos",
|
470 |
+
"type" : "MultiArray"
|
471 |
+
},
|
472 |
+
{
|
473 |
+
"hasShapeFlexibility" : "0",
|
474 |
+
"isOptional" : "0",
|
475 |
+
"dataType" : "Float16",
|
476 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
477 |
+
"shortDescription" : "",
|
478 |
+
"shape" : "[128, 512]",
|
479 |
+
"name" : "sin",
|
480 |
+
"type" : "MultiArray"
|
481 |
+
},
|
482 |
+
{
|
483 |
+
"hasShapeFlexibility" : "0",
|
484 |
+
"isOptional" : "0",
|
485 |
+
"dataType" : "Float16",
|
486 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
487 |
+
"shortDescription" : "",
|
488 |
+
"shape" : "[1, 1, 512, 512]",
|
489 |
+
"name" : "mask",
|
490 |
+
"type" : "MultiArray"
|
491 |
+
}
|
492 |
+
],
|
493 |
+
"defaultFunctionName" : "input_512_context_512",
|
494 |
+
"generatedClassName" : "Llama_2_7b_hf_2024_07_02_20_36_17_merged_chunk6",
|
495 |
+
"userDefinedMetadata" : {
|
496 |
+
|
497 |
+
},
|
498 |
+
"method" : "predict"
|
499 |
+
}
|
500 |
+
]
|
sequoia/Llama-2-7b-hf_chunk6.mlmodelc/model.mil
ADDED
The diff for this file is too large to render.
See raw diff
|
|
sequoia/Llama-2-7b-hf_chunk6.mlmodelc/weights/weight.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ae7f63a044b6c50c7a3edb1985bcbac9cf19fe13ca1664c0f4815406387d7aa4
|
3 |
+
size 303876448
|
sequoia/Llama-2-7b-hf_chunk7.mlmodelc/analytics/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:84e317a82cdf4e96f808f63e77f10098844d47ad522545181edfac4d287c9c92
|
3 |
+
size 243
|
sequoia/Llama-2-7b-hf_chunk7.mlmodelc/coremldata.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e430d0795ff5c384187174f5718a2c13d0070f5d6a811831e18862497865a86d
|
3 |
+
size 1037
|
sequoia/Llama-2-7b-hf_chunk7.mlmodelc/metadata.json
ADDED
@@ -0,0 +1,500 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 × 4096 × 1 × 512)",
|
11 |
+
"shortDescription" : "",
|
12 |
+
"shape" : "[1, 4096, 1, 512]",
|
13 |
+
"name" : "new_x",
|
14 |
+
"type" : "MultiArray"
|
15 |
+
},
|
16 |
+
{
|
17 |
+
"hasShapeFlexibility" : "0",
|
18 |
+
"isOptional" : "0",
|
19 |
+
"dataType" : "Float16",
|
20 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
21 |
+
"shortDescription" : "",
|
22 |
+
"shape" : "[1, 32, 128, 511]",
|
23 |
+
"name" : "new_k_cache_0",
|
24 |
+
"type" : "MultiArray"
|
25 |
+
},
|
26 |
+
{
|
27 |
+
"hasShapeFlexibility" : "0",
|
28 |
+
"isOptional" : "0",
|
29 |
+
"dataType" : "Float16",
|
30 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
31 |
+
"shortDescription" : "",
|
32 |
+
"shape" : "[1, 32, 128, 511]",
|
33 |
+
"name" : "new_k_cache_1",
|
34 |
+
"type" : "MultiArray"
|
35 |
+
},
|
36 |
+
{
|
37 |
+
"hasShapeFlexibility" : "0",
|
38 |
+
"isOptional" : "0",
|
39 |
+
"dataType" : "Float16",
|
40 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
41 |
+
"shortDescription" : "",
|
42 |
+
"shape" : "[1, 32, 128, 511]",
|
43 |
+
"name" : "new_k_cache_2",
|
44 |
+
"type" : "MultiArray"
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"hasShapeFlexibility" : "0",
|
48 |
+
"isOptional" : "0",
|
49 |
+
"dataType" : "Float16",
|
50 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
51 |
+
"shortDescription" : "",
|
52 |
+
"shape" : "[1, 32, 128, 511]",
|
53 |
+
"name" : "new_v_cache_0",
|
54 |
+
"type" : "MultiArray"
|
55 |
+
},
|
56 |
+
{
|
57 |
+
"hasShapeFlexibility" : "0",
|
58 |
+
"isOptional" : "0",
|
59 |
+
"dataType" : "Float16",
|
60 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
61 |
+
"shortDescription" : "",
|
62 |
+
"shape" : "[1, 32, 128, 511]",
|
63 |
+
"name" : "new_v_cache_1",
|
64 |
+
"type" : "MultiArray"
|
65 |
+
},
|
66 |
+
{
|
67 |
+
"hasShapeFlexibility" : "0",
|
68 |
+
"isOptional" : "0",
|
69 |
+
"dataType" : "Float16",
|
70 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
71 |
+
"shortDescription" : "",
|
72 |
+
"shape" : "[1, 32, 128, 511]",
|
73 |
+
"name" : "new_v_cache_2",
|
74 |
+
"type" : "MultiArray"
|
75 |
+
}
|
76 |
+
],
|
77 |
+
"modelParameters" : [
|
78 |
+
|
79 |
+
],
|
80 |
+
"specificationVersion" : 9,
|
81 |
+
"functions" : [
|
82 |
+
{
|
83 |
+
"inputSchema" : [
|
84 |
+
{
|
85 |
+
"hasShapeFlexibility" : "0",
|
86 |
+
"isOptional" : "0",
|
87 |
+
"dataType" : "Float16",
|
88 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
89 |
+
"shortDescription" : "",
|
90 |
+
"shape" : "[1, 4096, 1, 512]",
|
91 |
+
"name" : "x",
|
92 |
+
"type" : "MultiArray"
|
93 |
+
},
|
94 |
+
{
|
95 |
+
"hasShapeFlexibility" : "0",
|
96 |
+
"isOptional" : "0",
|
97 |
+
"dataType" : "Float16",
|
98 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
99 |
+
"shortDescription" : "",
|
100 |
+
"shape" : "[128, 512]",
|
101 |
+
"name" : "cos",
|
102 |
+
"type" : "MultiArray"
|
103 |
+
},
|
104 |
+
{
|
105 |
+
"hasShapeFlexibility" : "0",
|
106 |
+
"isOptional" : "0",
|
107 |
+
"dataType" : "Float16",
|
108 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
109 |
+
"shortDescription" : "",
|
110 |
+
"shape" : "[128, 512]",
|
111 |
+
"name" : "sin",
|
112 |
+
"type" : "MultiArray"
|
113 |
+
},
|
114 |
+
{
|
115 |
+
"hasShapeFlexibility" : "0",
|
116 |
+
"isOptional" : "0",
|
117 |
+
"dataType" : "Float16",
|
118 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
119 |
+
"shortDescription" : "",
|
120 |
+
"shape" : "[1, 1, 512, 512]",
|
121 |
+
"name" : "mask",
|
122 |
+
"type" : "MultiArray"
|
123 |
+
}
|
124 |
+
],
|
125 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
126 |
+
"storagePrecision" : "Float16",
|
127 |
+
"stateSchema" : [
|
128 |
+
|
129 |
+
],
|
130 |
+
"outputSchema" : [
|
131 |
+
{
|
132 |
+
"hasShapeFlexibility" : "0",
|
133 |
+
"isOptional" : "0",
|
134 |
+
"dataType" : "Float16",
|
135 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
136 |
+
"shortDescription" : "",
|
137 |
+
"shape" : "[1, 4096, 1, 512]",
|
138 |
+
"name" : "new_x",
|
139 |
+
"type" : "MultiArray"
|
140 |
+
},
|
141 |
+
{
|
142 |
+
"hasShapeFlexibility" : "0",
|
143 |
+
"isOptional" : "0",
|
144 |
+
"dataType" : "Float16",
|
145 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
146 |
+
"shortDescription" : "",
|
147 |
+
"shape" : "[1, 32, 128, 511]",
|
148 |
+
"name" : "new_k_cache_0",
|
149 |
+
"type" : "MultiArray"
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"hasShapeFlexibility" : "0",
|
153 |
+
"isOptional" : "0",
|
154 |
+
"dataType" : "Float16",
|
155 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
156 |
+
"shortDescription" : "",
|
157 |
+
"shape" : "[1, 32, 128, 511]",
|
158 |
+
"name" : "new_k_cache_1",
|
159 |
+
"type" : "MultiArray"
|
160 |
+
},
|
161 |
+
{
|
162 |
+
"hasShapeFlexibility" : "0",
|
163 |
+
"isOptional" : "0",
|
164 |
+
"dataType" : "Float16",
|
165 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
166 |
+
"shortDescription" : "",
|
167 |
+
"shape" : "[1, 32, 128, 511]",
|
168 |
+
"name" : "new_k_cache_2",
|
169 |
+
"type" : "MultiArray"
|
170 |
+
},
|
171 |
+
{
|
172 |
+
"hasShapeFlexibility" : "0",
|
173 |
+
"isOptional" : "0",
|
174 |
+
"dataType" : "Float16",
|
175 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
176 |
+
"shortDescription" : "",
|
177 |
+
"shape" : "[1, 32, 128, 511]",
|
178 |
+
"name" : "new_v_cache_0",
|
179 |
+
"type" : "MultiArray"
|
180 |
+
},
|
181 |
+
{
|
182 |
+
"hasShapeFlexibility" : "0",
|
183 |
+
"isOptional" : "0",
|
184 |
+
"dataType" : "Float16",
|
185 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
186 |
+
"shortDescription" : "",
|
187 |
+
"shape" : "[1, 32, 128, 511]",
|
188 |
+
"name" : "new_v_cache_1",
|
189 |
+
"type" : "MultiArray"
|
190 |
+
},
|
191 |
+
{
|
192 |
+
"hasShapeFlexibility" : "0",
|
193 |
+
"isOptional" : "0",
|
194 |
+
"dataType" : "Float16",
|
195 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
196 |
+
"shortDescription" : "",
|
197 |
+
"shape" : "[1, 32, 128, 511]",
|
198 |
+
"name" : "new_v_cache_2",
|
199 |
+
"type" : "MultiArray"
|
200 |
+
}
|
201 |
+
],
|
202 |
+
"name" : "input_512_context_512",
|
203 |
+
"mlProgramOperationTypeHistogram" : {
|
204 |
+
"Ios18.constexprLutToDense" : 21,
|
205 |
+
"Ios18.conv" : 21,
|
206 |
+
"Ios18.matmul" : 6,
|
207 |
+
"Ios18.expandDims" : 6,
|
208 |
+
"Ios18.concat" : 18,
|
209 |
+
"Ios18.add" : 15,
|
210 |
+
"Ios18.realDiv" : 6,
|
211 |
+
"Ios18.silu" : 3,
|
212 |
+
"Ios18.softmax" : 3,
|
213 |
+
"Ios18.sliceByIndex" : 18,
|
214 |
+
"Ios16.reduceL2Norm" : 6,
|
215 |
+
"Ios18.squeeze" : 6,
|
216 |
+
"Ios18.reshape" : 12,
|
217 |
+
"Ios18.mul" : 57
|
218 |
+
}
|
219 |
+
},
|
220 |
+
{
|
221 |
+
"inputSchema" : [
|
222 |
+
{
|
223 |
+
"hasShapeFlexibility" : "0",
|
224 |
+
"isOptional" : "0",
|
225 |
+
"dataType" : "Float16",
|
226 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 1)",
|
227 |
+
"shortDescription" : "",
|
228 |
+
"shape" : "[1, 4096, 1, 1]",
|
229 |
+
"name" : "x",
|
230 |
+
"type" : "MultiArray"
|
231 |
+
},
|
232 |
+
{
|
233 |
+
"hasShapeFlexibility" : "0",
|
234 |
+
"isOptional" : "0",
|
235 |
+
"dataType" : "Float16",
|
236 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
237 |
+
"shortDescription" : "",
|
238 |
+
"shape" : "[128, 1]",
|
239 |
+
"name" : "cos",
|
240 |
+
"type" : "MultiArray"
|
241 |
+
},
|
242 |
+
{
|
243 |
+
"hasShapeFlexibility" : "0",
|
244 |
+
"isOptional" : "0",
|
245 |
+
"dataType" : "Float16",
|
246 |
+
"formattedType" : "MultiArray (Float16 128 × 1)",
|
247 |
+
"shortDescription" : "",
|
248 |
+
"shape" : "[128, 1]",
|
249 |
+
"name" : "sin",
|
250 |
+
"type" : "MultiArray"
|
251 |
+
},
|
252 |
+
{
|
253 |
+
"hasShapeFlexibility" : "0",
|
254 |
+
"isOptional" : "0",
|
255 |
+
"dataType" : "Float16",
|
256 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 1 × 512)",
|
257 |
+
"shortDescription" : "",
|
258 |
+
"shape" : "[1, 1, 1, 512]",
|
259 |
+
"name" : "mask",
|
260 |
+
"type" : "MultiArray"
|
261 |
+
},
|
262 |
+
{
|
263 |
+
"hasShapeFlexibility" : "0",
|
264 |
+
"isOptional" : "1",
|
265 |
+
"dataType" : "Float16",
|
266 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
267 |
+
"shortDescription" : "",
|
268 |
+
"shape" : "[1, 32, 128, 511]",
|
269 |
+
"name" : "k_cache_0",
|
270 |
+
"type" : "MultiArray"
|
271 |
+
},
|
272 |
+
{
|
273 |
+
"hasShapeFlexibility" : "0",
|
274 |
+
"isOptional" : "1",
|
275 |
+
"dataType" : "Float16",
|
276 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
277 |
+
"shortDescription" : "",
|
278 |
+
"shape" : "[1, 32, 128, 511]",
|
279 |
+
"name" : "v_cache_0",
|
280 |
+
"type" : "MultiArray"
|
281 |
+
},
|
282 |
+
{
|
283 |
+
"hasShapeFlexibility" : "0",
|
284 |
+
"isOptional" : "1",
|
285 |
+
"dataType" : "Float16",
|
286 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
287 |
+
"shortDescription" : "",
|
288 |
+
"shape" : "[1, 32, 128, 511]",
|
289 |
+
"name" : "k_cache_1",
|
290 |
+
"type" : "MultiArray"
|
291 |
+
},
|
292 |
+
{
|
293 |
+
"hasShapeFlexibility" : "0",
|
294 |
+
"isOptional" : "1",
|
295 |
+
"dataType" : "Float16",
|
296 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
297 |
+
"shortDescription" : "",
|
298 |
+
"shape" : "[1, 32, 128, 511]",
|
299 |
+
"name" : "v_cache_1",
|
300 |
+
"type" : "MultiArray"
|
301 |
+
},
|
302 |
+
{
|
303 |
+
"hasShapeFlexibility" : "0",
|
304 |
+
"isOptional" : "1",
|
305 |
+
"dataType" : "Float16",
|
306 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
307 |
+
"shortDescription" : "",
|
308 |
+
"shape" : "[1, 32, 128, 511]",
|
309 |
+
"name" : "k_cache_2",
|
310 |
+
"type" : "MultiArray"
|
311 |
+
},
|
312 |
+
{
|
313 |
+
"hasShapeFlexibility" : "0",
|
314 |
+
"isOptional" : "1",
|
315 |
+
"dataType" : "Float16",
|
316 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)?",
|
317 |
+
"shortDescription" : "",
|
318 |
+
"shape" : "[1, 32, 128, 511]",
|
319 |
+
"name" : "v_cache_2",
|
320 |
+
"type" : "MultiArray"
|
321 |
+
}
|
322 |
+
],
|
323 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
324 |
+
"storagePrecision" : "Float16",
|
325 |
+
"stateSchema" : [
|
326 |
+
|
327 |
+
],
|
328 |
+
"outputSchema" : [
|
329 |
+
{
|
330 |
+
"hasShapeFlexibility" : "0",
|
331 |
+
"isOptional" : "0",
|
332 |
+
"dataType" : "Float16",
|
333 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 1)",
|
334 |
+
"shortDescription" : "",
|
335 |
+
"shape" : "[1, 4096, 1, 1]",
|
336 |
+
"name" : "new_x",
|
337 |
+
"type" : "MultiArray"
|
338 |
+
},
|
339 |
+
{
|
340 |
+
"hasShapeFlexibility" : "0",
|
341 |
+
"isOptional" : "0",
|
342 |
+
"dataType" : "Float16",
|
343 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
344 |
+
"shortDescription" : "",
|
345 |
+
"shape" : "[1, 32, 128, 511]",
|
346 |
+
"name" : "new_k_cache_0",
|
347 |
+
"type" : "MultiArray"
|
348 |
+
},
|
349 |
+
{
|
350 |
+
"hasShapeFlexibility" : "0",
|
351 |
+
"isOptional" : "0",
|
352 |
+
"dataType" : "Float16",
|
353 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
354 |
+
"shortDescription" : "",
|
355 |
+
"shape" : "[1, 32, 128, 511]",
|
356 |
+
"name" : "new_k_cache_1",
|
357 |
+
"type" : "MultiArray"
|
358 |
+
},
|
359 |
+
{
|
360 |
+
"hasShapeFlexibility" : "0",
|
361 |
+
"isOptional" : "0",
|
362 |
+
"dataType" : "Float16",
|
363 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
364 |
+
"shortDescription" : "",
|
365 |
+
"shape" : "[1, 32, 128, 511]",
|
366 |
+
"name" : "new_k_cache_2",
|
367 |
+
"type" : "MultiArray"
|
368 |
+
},
|
369 |
+
{
|
370 |
+
"hasShapeFlexibility" : "0",
|
371 |
+
"isOptional" : "0",
|
372 |
+
"dataType" : "Float16",
|
373 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
374 |
+
"shortDescription" : "",
|
375 |
+
"shape" : "[1, 32, 128, 511]",
|
376 |
+
"name" : "new_v_cache_0",
|
377 |
+
"type" : "MultiArray"
|
378 |
+
},
|
379 |
+
{
|
380 |
+
"hasShapeFlexibility" : "0",
|
381 |
+
"isOptional" : "0",
|
382 |
+
"dataType" : "Float16",
|
383 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
384 |
+
"shortDescription" : "",
|
385 |
+
"shape" : "[1, 32, 128, 511]",
|
386 |
+
"name" : "new_v_cache_1",
|
387 |
+
"type" : "MultiArray"
|
388 |
+
},
|
389 |
+
{
|
390 |
+
"hasShapeFlexibility" : "0",
|
391 |
+
"isOptional" : "0",
|
392 |
+
"dataType" : "Float16",
|
393 |
+
"formattedType" : "MultiArray (Float16 1 × 32 × 128 × 511)",
|
394 |
+
"shortDescription" : "",
|
395 |
+
"shape" : "[1, 32, 128, 511]",
|
396 |
+
"name" : "new_v_cache_2",
|
397 |
+
"type" : "MultiArray"
|
398 |
+
}
|
399 |
+
],
|
400 |
+
"name" : "input_1_context_512",
|
401 |
+
"mlProgramOperationTypeHistogram" : {
|
402 |
+
"Ios18.constexprLutToDense" : 21,
|
403 |
+
"Ios18.conv" : 21,
|
404 |
+
"Ios18.matmul" : 6,
|
405 |
+
"Ios18.expandDims" : 6,
|
406 |
+
"Ios18.concat" : 18,
|
407 |
+
"Ios18.add" : 15,
|
408 |
+
"Ios18.realDiv" : 6,
|
409 |
+
"Ios18.silu" : 3,
|
410 |
+
"Ios18.softmax" : 3,
|
411 |
+
"Ios18.sliceByIndex" : 18,
|
412 |
+
"Ios16.reduceL2Norm" : 6,
|
413 |
+
"Ios18.squeeze" : 6,
|
414 |
+
"Ios18.reshape" : 12,
|
415 |
+
"Ios18.mul" : 57
|
416 |
+
}
|
417 |
+
}
|
418 |
+
],
|
419 |
+
"mlProgramOperationTypeHistogram" : {
|
420 |
+
"Ios18.constexprLutToDense" : 21,
|
421 |
+
"Ios18.conv" : 21,
|
422 |
+
"Ios18.matmul" : 6,
|
423 |
+
"Ios18.expandDims" : 6,
|
424 |
+
"Ios18.concat" : 18,
|
425 |
+
"Ios18.add" : 15,
|
426 |
+
"Ios18.realDiv" : 6,
|
427 |
+
"Ios18.silu" : 3,
|
428 |
+
"Ios18.softmax" : 3,
|
429 |
+
"Ios18.sliceByIndex" : 18,
|
430 |
+
"Ios16.reduceL2Norm" : 6,
|
431 |
+
"Ios18.squeeze" : 6,
|
432 |
+
"Ios18.reshape" : 12,
|
433 |
+
"Ios18.mul" : 57
|
434 |
+
},
|
435 |
+
"isUpdatable" : "0",
|
436 |
+
"stateSchema" : [
|
437 |
+
|
438 |
+
],
|
439 |
+
"availability" : {
|
440 |
+
"macOS" : "15.0",
|
441 |
+
"tvOS" : "18.0",
|
442 |
+
"visionOS" : "2.0",
|
443 |
+
"watchOS" : "11.0",
|
444 |
+
"iOS" : "18.0",
|
445 |
+
"macCatalyst" : "18.0"
|
446 |
+
},
|
447 |
+
"computePrecision" : "Mixed (Float16, Int32)",
|
448 |
+
"modelType" : {
|
449 |
+
"name" : "MLModelType_mlProgram"
|
450 |
+
},
|
451 |
+
"inputSchema" : [
|
452 |
+
{
|
453 |
+
"hasShapeFlexibility" : "0",
|
454 |
+
"isOptional" : "0",
|
455 |
+
"dataType" : "Float16",
|
456 |
+
"formattedType" : "MultiArray (Float16 1 × 4096 × 1 × 512)",
|
457 |
+
"shortDescription" : "",
|
458 |
+
"shape" : "[1, 4096, 1, 512]",
|
459 |
+
"name" : "x",
|
460 |
+
"type" : "MultiArray"
|
461 |
+
},
|
462 |
+
{
|
463 |
+
"hasShapeFlexibility" : "0",
|
464 |
+
"isOptional" : "0",
|
465 |
+
"dataType" : "Float16",
|
466 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
467 |
+
"shortDescription" : "",
|
468 |
+
"shape" : "[128, 512]",
|
469 |
+
"name" : "cos",
|
470 |
+
"type" : "MultiArray"
|
471 |
+
},
|
472 |
+
{
|
473 |
+
"hasShapeFlexibility" : "0",
|
474 |
+
"isOptional" : "0",
|
475 |
+
"dataType" : "Float16",
|
476 |
+
"formattedType" : "MultiArray (Float16 128 × 512)",
|
477 |
+
"shortDescription" : "",
|
478 |
+
"shape" : "[128, 512]",
|
479 |
+
"name" : "sin",
|
480 |
+
"type" : "MultiArray"
|
481 |
+
},
|
482 |
+
{
|
483 |
+
"hasShapeFlexibility" : "0",
|
484 |
+
"isOptional" : "0",
|
485 |
+
"dataType" : "Float16",
|
486 |
+
"formattedType" : "MultiArray (Float16 1 × 1 × 512 × 512)",
|
487 |
+
"shortDescription" : "",
|
488 |
+
"shape" : "[1, 1, 512, 512]",
|
489 |
+
"name" : "mask",
|
490 |
+
"type" : "MultiArray"
|
491 |
+
}
|
492 |
+
],
|
493 |
+
"defaultFunctionName" : "input_512_context_512",
|
494 |
+
"generatedClassName" : "Llama_2_7b_hf_2024_07_02_20_36_17_merged_chunk7",
|
495 |
+
"userDefinedMetadata" : {
|
496 |
+
|
497 |
+
},
|
498 |
+
"method" : "predict"
|
499 |
+
}
|
500 |
+
]
|
sequoia/Llama-2-7b-hf_chunk7.mlmodelc/model.mil
ADDED
The diff for this file is too large to render.
See raw diff
|
|
sequoia/Llama-2-7b-hf_chunk7.mlmodelc/weights/weight.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ecc8fb381ae8b90255908ed5bfd9a12f6fe951ea2bb7fbce84b65b3acaf64135
|
3 |
+
size 303876448
|