mickekring commited on
Commit
232cb26
·
verified ·
1 Parent(s): 44d05dc

Add medium/TextDecoder.mlmodelc

Browse files
medium/TextDecoder.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7ac2ee1e85a208b1f9191860179e98355389cc3597af1967b7cf1ac405523d28
3
+ size 243
medium/TextDecoder.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:83a481a849a96b725646cc4390e4fa44ff0bfec5fbe3ed68436a0d3d112c50fc
3
+ size 639
medium/TextDecoder.mlmodelc/metadata.json ADDED
@@ -0,0 +1,168 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "metadataOutputVersion" : "3.0",
4
+ "storagePrecision" : "Float16",
5
+ "outputSchema" : [
6
+ {
7
+ "hasShapeFlexibility" : "0",
8
+ "isOptional" : "0",
9
+ "dataType" : "Float16",
10
+ "formattedType" : "MultiArray (Float16 1 × 1 × 51865)",
11
+ "shortDescription" : "",
12
+ "shape" : "[1, 1, 51865]",
13
+ "name" : "logits",
14
+ "type" : "MultiArray"
15
+ },
16
+ {
17
+ "hasShapeFlexibility" : "0",
18
+ "isOptional" : "0",
19
+ "dataType" : "Float16",
20
+ "formattedType" : "MultiArray (Float16 1 × 24576 × 1 × 1)",
21
+ "shortDescription" : "",
22
+ "shape" : "[1, 24576, 1, 1]",
23
+ "name" : "key_cache_updates",
24
+ "type" : "MultiArray"
25
+ },
26
+ {
27
+ "hasShapeFlexibility" : "0",
28
+ "isOptional" : "0",
29
+ "dataType" : "Float16",
30
+ "formattedType" : "MultiArray (Float16 1 × 24576 × 1 × 1)",
31
+ "shortDescription" : "",
32
+ "shape" : "[1, 24576, 1, 1]",
33
+ "name" : "value_cache_updates",
34
+ "type" : "MultiArray"
35
+ },
36
+ {
37
+ "hasShapeFlexibility" : "0",
38
+ "isOptional" : "0",
39
+ "dataType" : "Float16",
40
+ "formattedType" : "MultiArray (Float16 1 × 1500)",
41
+ "shortDescription" : "",
42
+ "shape" : "[1, 1500]",
43
+ "name" : "alignment_heads_weights",
44
+ "type" : "MultiArray"
45
+ }
46
+ ],
47
+ "modelParameters" : [
48
+
49
+ ],
50
+ "specificationVersion" : 7,
51
+ "mlProgramOperationTypeHistogram" : {
52
+ "Ios16.linear" : 1,
53
+ "Concat" : 3,
54
+ "Ios16.reduceMean" : 1,
55
+ "Ios16.mul" : 144,
56
+ "Ios16.layerNorm" : 73,
57
+ "SliceByIndex" : 12,
58
+ "Ios16.sub" : 1,
59
+ "Transpose" : 1,
60
+ "Ios16.conv" : 240,
61
+ "Ios16.add" : 145,
62
+ "Squeeze" : 1,
63
+ "Ios16.matmul" : 96,
64
+ "Ios16.softmax" : 48,
65
+ "Ios16.gelu" : 24,
66
+ "ExpandDims" : 6,
67
+ "Ios16.batchNorm" : 73,
68
+ "Split" : 2,
69
+ "Ios16.gather" : 2,
70
+ "Ios16.reshape" : 192
71
+ },
72
+ "computePrecision" : "Mixed (Float16, Int32)",
73
+ "isUpdatable" : "0",
74
+ "stateSchema" : [
75
+
76
+ ],
77
+ "availability" : {
78
+ "macOS" : "13.0",
79
+ "tvOS" : "16.0",
80
+ "visionOS" : "1.0",
81
+ "watchOS" : "9.0",
82
+ "iOS" : "16.0",
83
+ "macCatalyst" : "16.0"
84
+ },
85
+ "modelType" : {
86
+ "name" : "MLModelType_mlProgram"
87
+ },
88
+ "userDefinedMetadata" : {
89
+ "com.github.apple.coremltools.version" : "8.3.0",
90
+ "com.github.apple.coremltools.source" : "torch==2.6.0",
91
+ "com.github.apple.coremltools.source_dialect" : "TorchScript"
92
+ },
93
+ "inputSchema" : [
94
+ {
95
+ "hasShapeFlexibility" : "0",
96
+ "isOptional" : "0",
97
+ "dataType" : "Int32",
98
+ "formattedType" : "MultiArray (Int32 1)",
99
+ "shortDescription" : "",
100
+ "shape" : "[1]",
101
+ "name" : "input_ids",
102
+ "type" : "MultiArray"
103
+ },
104
+ {
105
+ "hasShapeFlexibility" : "0",
106
+ "isOptional" : "0",
107
+ "dataType" : "Int32",
108
+ "formattedType" : "MultiArray (Int32 1)",
109
+ "shortDescription" : "",
110
+ "shape" : "[1]",
111
+ "name" : "cache_length",
112
+ "type" : "MultiArray"
113
+ },
114
+ {
115
+ "hasShapeFlexibility" : "0",
116
+ "isOptional" : "0",
117
+ "dataType" : "Float16",
118
+ "formattedType" : "MultiArray (Float16 1 × 24576 × 1 × 448)",
119
+ "shortDescription" : "",
120
+ "shape" : "[1, 24576, 1, 448]",
121
+ "name" : "key_cache",
122
+ "type" : "MultiArray"
123
+ },
124
+ {
125
+ "hasShapeFlexibility" : "0",
126
+ "isOptional" : "0",
127
+ "dataType" : "Float16",
128
+ "formattedType" : "MultiArray (Float16 1 × 24576 × 1 × 448)",
129
+ "shortDescription" : "",
130
+ "shape" : "[1, 24576, 1, 448]",
131
+ "name" : "value_cache",
132
+ "type" : "MultiArray"
133
+ },
134
+ {
135
+ "hasShapeFlexibility" : "0",
136
+ "isOptional" : "0",
137
+ "dataType" : "Float16",
138
+ "formattedType" : "MultiArray (Float16 1 × 448)",
139
+ "shortDescription" : "",
140
+ "shape" : "[1, 448]",
141
+ "name" : "kv_cache_update_mask",
142
+ "type" : "MultiArray"
143
+ },
144
+ {
145
+ "hasShapeFlexibility" : "0",
146
+ "isOptional" : "0",
147
+ "dataType" : "Float16",
148
+ "formattedType" : "MultiArray (Float16 1 × 1024 × 1 × 1500)",
149
+ "shortDescription" : "",
150
+ "shape" : "[1, 1024, 1, 1500]",
151
+ "name" : "encoder_output_embeds",
152
+ "type" : "MultiArray"
153
+ },
154
+ {
155
+ "hasShapeFlexibility" : "0",
156
+ "isOptional" : "0",
157
+ "dataType" : "Float16",
158
+ "formattedType" : "MultiArray (Float16 1 × 448)",
159
+ "shortDescription" : "",
160
+ "shape" : "[1, 448]",
161
+ "name" : "decoder_key_padding_mask",
162
+ "type" : "MultiArray"
163
+ }
164
+ ],
165
+ "generatedClassName" : "TextDecoder",
166
+ "method" : "predict"
167
+ }
168
+ ]
medium/TextDecoder.mlmodelc/model.mil ADDED
The diff for this file is too large to render. See raw diff
 
medium/TextDecoder.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d8aadf0e0934ac318efef35616ee5b13b8c0804809f2b7a163eaea85d55e3d66
3
+ size 913428274