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