openai_whisper-large-v2_949MB/TextDecoder.mlmodelc/metadata.json
4.8 KB · 167 lines · json Raw
1 [
2 {
3 "metadataOutputVersion" : "3.0",
4 "storagePrecision" : "Mixed (Float16, Palettized (4 bits), Palettized (6 bits))",
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 × 40960 × 1 × 1)",
21 "shortDescription" : "",
22 "shape" : "[1, 40960, 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 × 40960 × 1 × 1)",
31 "shortDescription" : "",
32 "shape" : "[1, 40960, 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" : 8,
51 "mlProgramOperationTypeHistogram" : {
52 "Ios17.reshape" : 256,
53 "Ios16.reduceMean" : 195,
54 "Ios16.softmax" : 64,
55 "Ios17.matmul" : 128,
56 "Ios17.transpose" : 1,
57 "Split" : 2,
58 "Ios17.expandDims" : 6,
59 "Ios17.add" : 610,
60 "Ios17.sliceByIndex" : 46,
61 "Ios17.squeeze" : 1,
62 "Ios17.gather" : 2,
63 "Ios17.batchNorm" : 97,
64 "Ios16.gelu" : 32,
65 "Ios17.cast" : 1,
66 "Ios16.constexprLutToDense" : 320,
67 "Ios17.conv" : 960,
68 "Ios17.sub" : 98,
69 "Ios17.concat" : 3,
70 "Ios17.linear" : 1,
71 "Ios17.rsqrt" : 97,
72 "Ios17.mul" : 386
73 },
74 "computePrecision" : "Mixed (Float16, Float32, Int16, Int32)",
75 "isUpdatable" : "0",
76 "availability" : {
77 "macOS" : "14.0",
78 "tvOS" : "17.0",
79 "visionOS" : "1.0",
80 "watchOS" : "10.0",
81 "iOS" : "17.0",
82 "macCatalyst" : "17.0"
83 },
84 "modelType" : {
85 "name" : "MLModelType_mlProgram"
86 },
87 "userDefinedMetadata" : {
88 "com.github.apple.coremltools.source_dialect" : "TorchScript",
89 "com.github.apple.coremltools.source" : "torch==2.2.1",
90 "com.github.apple.coremltools.version" : "7.1"
91 },
92 "inputSchema" : [
93 {
94 "hasShapeFlexibility" : "0",
95 "isOptional" : "0",
96 "dataType" : "Int32",
97 "formattedType" : "MultiArray (Int32 1)",
98 "shortDescription" : "",
99 "shape" : "[1]",
100 "name" : "input_ids",
101 "type" : "MultiArray"
102 },
103 {
104 "hasShapeFlexibility" : "0",
105 "isOptional" : "0",
106 "dataType" : "Int32",
107 "formattedType" : "MultiArray (Int32 1)",
108 "shortDescription" : "",
109 "shape" : "[1]",
110 "name" : "cache_length",
111 "type" : "MultiArray"
112 },
113 {
114 "hasShapeFlexibility" : "0",
115 "isOptional" : "0",
116 "dataType" : "Float16",
117 "formattedType" : "MultiArray (Float16 1 × 40960 × 1 × 448)",
118 "shortDescription" : "",
119 "shape" : "[1, 40960, 1, 448]",
120 "name" : "key_cache",
121 "type" : "MultiArray"
122 },
123 {
124 "hasShapeFlexibility" : "0",
125 "isOptional" : "0",
126 "dataType" : "Float16",
127 "formattedType" : "MultiArray (Float16 1 × 40960 × 1 × 448)",
128 "shortDescription" : "",
129 "shape" : "[1, 40960, 1, 448]",
130 "name" : "value_cache",
131 "type" : "MultiArray"
132 },
133 {
134 "hasShapeFlexibility" : "0",
135 "isOptional" : "0",
136 "dataType" : "Float16",
137 "formattedType" : "MultiArray (Float16 1 × 448)",
138 "shortDescription" : "",
139 "shape" : "[1, 448]",
140 "name" : "kv_cache_update_mask",
141 "type" : "MultiArray"
142 },
143 {
144 "hasShapeFlexibility" : "0",
145 "isOptional" : "0",
146 "dataType" : "Float16",
147 "formattedType" : "MultiArray (Float16 1 × 1280 × 1 × 1500)",
148 "shortDescription" : "",
149 "shape" : "[1, 1280, 1, 1500]",
150 "name" : "encoder_output_embeds",
151 "type" : "MultiArray"
152 },
153 {
154 "hasShapeFlexibility" : "0",
155 "isOptional" : "0",
156 "dataType" : "Float16",
157 "formattedType" : "MultiArray (Float16 1 × 448)",
158 "shortDescription" : "",
159 "shape" : "[1, 448]",
160 "name" : "decoder_key_padding_mask",
161 "type" : "MultiArray"
162 }
163 ],
164 "generatedClassName" : "TextDecoder_mixedBitPalettized_4_0_bit",
165 "method" : "predict"
166 }
167 ]