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