openai_whisper-large-v3-v20240930_547MB/TextDecoder.mlmodelc/metadata.json
4.8 KB · 166 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 × 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" : 2,
54 "Squeeze" : 1,
55 "Ios16.reduceMean" : 1,
56 "Ios16.softmax" : 8,
57 "Split" : 2,
58 "Ios16.linear" : 1,
59 "Ios16.add" : 65,
60 "Concat" : 3,
61 "ExpandDims" : 6,
62 "Ios16.sub" : 1,
63 "Ios16.conv" : 120,
64 "Ios16.gelu" : 4,
65 "Ios16.constexprLutToDense" : 40,
66 "Ios16.matmul" : 16,
67 "Ios16.layerNorm" : 13,
68 "Ios16.batchNorm" : 13,
69 "Ios16.reshape" : 32,
70 "SliceByIndex" : 12,
71 "Ios16.mul" : 24
72 },
73 "computePrecision" : "Mixed (Float16, Int32)",
74 "isUpdatable" : "0",
75 "availability" : {
76 "macOS" : "13.0",
77 "tvOS" : "16.0",
78 "visionOS" : "1.0",
79 "watchOS" : "9.0",
80 "iOS" : "16.0",
81 "macCatalyst" : "16.0"
82 },
83 "modelType" : {
84 "name" : "MLModelType_mlProgram"
85 },
86 "userDefinedMetadata" : {
87 "com.github.apple.coremltools.source_dialect" : "TorchScript",
88 "com.github.apple.coremltools.version" : "8.0",
89 "com.github.apple.coremltools.source" : "torch==2.4.1"
90 },
91 "inputSchema" : [
92 {
93 "hasShapeFlexibility" : "0",
94 "isOptional" : "0",
95 "dataType" : "Int32",
96 "formattedType" : "MultiArray (Int32 1)",
97 "shortDescription" : "",
98 "shape" : "[1]",
99 "name" : "input_ids",
100 "type" : "MultiArray"
101 },
102 {
103 "hasShapeFlexibility" : "0",
104 "isOptional" : "0",
105 "dataType" : "Int32",
106 "formattedType" : "MultiArray (Int32 1)",
107 "shortDescription" : "",
108 "shape" : "[1]",
109 "name" : "cache_length",
110 "type" : "MultiArray"
111 },
112 {
113 "hasShapeFlexibility" : "0",
114 "isOptional" : "0",
115 "dataType" : "Float16",
116 "formattedType" : "MultiArray (Float16 1 × 5120 × 1 × 448)",
117 "shortDescription" : "",
118 "shape" : "[1, 5120, 1, 448]",
119 "name" : "key_cache",
120 "type" : "MultiArray"
121 },
122 {
123 "hasShapeFlexibility" : "0",
124 "isOptional" : "0",
125 "dataType" : "Float16",
126 "formattedType" : "MultiArray (Float16 1 × 5120 × 1 × 448)",
127 "shortDescription" : "",
128 "shape" : "[1, 5120, 1, 448]",
129 "name" : "value_cache",
130 "type" : "MultiArray"
131 },
132 {
133 "hasShapeFlexibility" : "0",
134 "isOptional" : "0",
135 "dataType" : "Float16",
136 "formattedType" : "MultiArray (Float16 1 × 448)",
137 "shortDescription" : "",
138 "shape" : "[1, 448]",
139 "name" : "kv_cache_update_mask",
140 "type" : "MultiArray"
141 },
142 {
143 "hasShapeFlexibility" : "0",
144 "isOptional" : "0",
145 "dataType" : "Float16",
146 "formattedType" : "MultiArray (Float16 1 × 1280 × 1 × 1500)",
147 "shortDescription" : "",
148 "shape" : "[1, 1280, 1, 1500]",
149 "name" : "encoder_output_embeds",
150 "type" : "MultiArray"
151 },
152 {
153 "hasShapeFlexibility" : "0",
154 "isOptional" : "0",
155 "dataType" : "Float16",
156 "formattedType" : "MultiArray (Float16 1 × 448)",
157 "shortDescription" : "",
158 "shape" : "[1, 448]",
159 "name" : "decoder_key_padding_mask",
160 "type" : "MultiArray"
161 }
162 ],
163 "generatedClassName" : "TextDecoder_mixedBitPalettized_4_1_bit",
164 "method" : "predict"
165 }
166 ]