openai_whisper-large-v3-v20240930_turbo_632MB/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), Sparse)",
5 "outputSchema" : [
6 {
7 "hasShapeFlexibility" : "0",
8 "isOptional" : "0",
9 "dataType" : "Float16",
10 "formattedType" : "MultiArray (Float16 1 × 1 × 51866)",
11 "shortDescription" : "",
12 "shape" : "[1, 1, 51866]",
13 "name" : "logits",
14 "type" : "MultiArray"
15 },
16 {
17 "hasShapeFlexibility" : "0",
18 "isOptional" : "0",
19 "dataType" : "Float16",
20 "formattedType" : "MultiArray (Float16 1 × 5120 × 1 × 1)",
21 "shortDescription" : "",
22 "shape" : "[1, 5120, 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 × 5120 × 1 × 1)",
31 "shortDescription" : "",
32 "shape" : "[1, 5120, 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 "Ios16.gather" : 3,
54 "Squeeze" : 1,
55 "Ios16.reduceMean" : 1,
56 "Ios16.softmax" : 8,
57 "Split" : 2,
58 "Ios16.linear" : 1,
59 "Ios16.add" : 66,
60 "Concat" : 3,
61 "ExpandDims" : 6,
62 "Ios16.sub" : 1,
63 "Ios16.conv" : 80,
64 "Ios16.gelu" : 4,
65 "Ios16.constexprLutToDense" : 40,
66 "Ios16.constexprSparseToDense" : 41,
67 "Ios16.layerNorm" : 13,
68 "Ios16.batchNorm" : 13,
69 "Ios16.matmul" : 16,
70 "Ios16.reshape" : 32,
71 "SliceByIndex" : 12,
72 "Ios16.mul" : 24
73 },
74 "computePrecision" : "Mixed (Float16, Int32)",
75 "isUpdatable" : "0",
76 "availability" : {
77 "macOS" : "13.0",
78 "tvOS" : "16.0",
79 "visionOS" : "1.0",
80 "watchOS" : "9.0",
81 "iOS" : "16.0",
82 "macCatalyst" : "16.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.4.1",
90 "com.github.apple.coremltools.version" : "8.0"
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 × 5120 × 1 × 224)",
118 "shortDescription" : "",
119 "shape" : "[1, 5120, 1, 224]",
120 "name" : "key_cache",
121 "type" : "MultiArray"
122 },
123 {
124 "hasShapeFlexibility" : "0",
125 "isOptional" : "0",
126 "dataType" : "Float16",
127 "formattedType" : "MultiArray (Float16 1 × 5120 × 1 × 224)",
128 "shortDescription" : "",
129 "shape" : "[1, 5120, 1, 224]",
130 "name" : "value_cache",
131 "type" : "MultiArray"
132 },
133 {
134 "hasShapeFlexibility" : "0",
135 "isOptional" : "0",
136 "dataType" : "Float16",
137 "formattedType" : "MultiArray (Float16 1 × 224)",
138 "shortDescription" : "",
139 "shape" : "[1, 224]",
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 × 224)",
158 "shortDescription" : "",
159 "shape" : "[1, 224]",
160 "name" : "decoder_key_padding_mask",
161 "type" : "MultiArray"
162 }
163 ],
164 "generatedClassName" : "TextDecoder_mixedBitPalettized_4_1_bit",
165 "method" : "predict"
166 }
167 ]