voz / decoder.mlmodelc /metadata.json
SolsticeAI's picture finnvoorhees's picture
Duplicate from desert-ant-labs/voz
e69441b
Raw History Blame Contribute Delete
18.5 kB
[
{
"metadataOutputVersion" : "3.0",
"storagePrecision" : "Mixed (Float16, Int8)",
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 16 × 8198 × 1 × 8)",
"shortDescription" : "",
"shape" : "[16, 8198, 1, 8]",
"name" : "logits",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 16 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[16, 1280, 1, 1]",
"name" : "h_out",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 16 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[16, 1280, 1, 1]",
"name" : "c_out",
"type" : "MultiArray"
}
],
"modelParameters" : [
],
"specificationVersion" : 9,
"functions" : [
{
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 16 × 640 × 1 × 1)",
"shortDescription" : "",
"shape" : "[16, 640, 1, 1]",
"name" : "embed",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 16 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[16, 1280, 1, 1]",
"name" : "h_in",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 16 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[16, 1280, 1, 1]",
"name" : "c_in",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 16 × 640 × 1 × 8)",
"shortDescription" : "",
"shape" : "[16, 640, 1, 8]",
"name" : "enc_step",
"type" : "MultiArray"
}
],
"computePrecision" : "Mixed (Float16, Int32)",
"storagePrecision" : "Mixed (Float16, Int8)",
"stateSchema" : [
],
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 16 × 8198 × 1 × 8)",
"shortDescription" : "",
"shape" : "[16, 8198, 1, 8]",
"name" : "logits",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 16 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[16, 1280, 1, 1]",
"name" : "h_out",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 16 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[16, 1280, 1, 1]",
"name" : "c_out",
"type" : "MultiArray"
}
],
"name" : "main",
"mlProgramOperationTypeHistogram" : {
"Ios18.tanh" : 4,
"Ios18.relu" : 1,
"Ios18.sigmoid" : 6,
"Ios18.constexprBlockwiseShiftScale" : 6,
"Ios18.concat" : 2,
"Ios18.sliceByIndex" : 12,
"Ios18.conv" : 6,
"Ios18.mul" : 6,
"Ios18.add" : 5
}
},
{
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 8 × 640 × 1 × 1)",
"shortDescription" : "",
"shape" : "[8, 640, 1, 1]",
"name" : "embed",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 8 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[8, 1280, 1, 1]",
"name" : "h_in",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 8 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[8, 1280, 1, 1]",
"name" : "c_in",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 8 × 640 × 1 × 8)",
"shortDescription" : "",
"shape" : "[8, 640, 1, 8]",
"name" : "enc_step",
"type" : "MultiArray"
}
],
"computePrecision" : "Mixed (Float16, Int32)",
"storagePrecision" : "Mixed (Float16, Int8)",
"stateSchema" : [
],
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 8 × 8198 × 1 × 8)",
"shortDescription" : "",
"shape" : "[8, 8198, 1, 8]",
"name" : "logits",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 8 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[8, 1280, 1, 1]",
"name" : "h_out",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 8 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[8, 1280, 1, 1]",
"name" : "c_out",
"type" : "MultiArray"
}
],
"name" : "decoder_8",
"mlProgramOperationTypeHistogram" : {
"Ios18.tanh" : 4,
"Ios18.relu" : 1,
"Ios18.sigmoid" : 6,
"Ios18.constexprBlockwiseShiftScale" : 6,
"Ios18.concat" : 2,
"Ios18.sliceByIndex" : 12,
"Ios18.conv" : 6,
"Ios18.mul" : 6,
"Ios18.add" : 5
}
},
{
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 4 × 640 × 1 × 1)",
"shortDescription" : "",
"shape" : "[4, 640, 1, 1]",
"name" : "embed",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 4 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[4, 1280, 1, 1]",
"name" : "h_in",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 4 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[4, 1280, 1, 1]",
"name" : "c_in",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 4 × 640 × 1 × 8)",
"shortDescription" : "",
"shape" : "[4, 640, 1, 8]",
"name" : "enc_step",
"type" : "MultiArray"
}
],
"computePrecision" : "Mixed (Float16, Int32)",
"storagePrecision" : "Mixed (Float16, Int8)",
"stateSchema" : [
],
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 4 × 8198 × 1 × 8)",
"shortDescription" : "",
"shape" : "[4, 8198, 1, 8]",
"name" : "logits",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 4 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[4, 1280, 1, 1]",
"name" : "h_out",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 4 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[4, 1280, 1, 1]",
"name" : "c_out",
"type" : "MultiArray"
}
],
"name" : "decoder_4",
"mlProgramOperationTypeHistogram" : {
"Ios18.tanh" : 4,
"Ios18.relu" : 1,
"Ios18.sigmoid" : 6,
"Ios18.constexprBlockwiseShiftScale" : 6,
"Ios18.concat" : 2,
"Ios18.sliceByIndex" : 12,
"Ios18.conv" : 6,
"Ios18.mul" : 6,
"Ios18.add" : 5
}
},
{
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 2 × 640 × 1 × 1)",
"shortDescription" : "",
"shape" : "[2, 640, 1, 1]",
"name" : "embed",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 2 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[2, 1280, 1, 1]",
"name" : "h_in",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 2 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[2, 1280, 1, 1]",
"name" : "c_in",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 2 × 640 × 1 × 8)",
"shortDescription" : "",
"shape" : "[2, 640, 1, 8]",
"name" : "enc_step",
"type" : "MultiArray"
}
],
"computePrecision" : "Mixed (Float16, Int32)",
"storagePrecision" : "Mixed (Float16, Int8)",
"stateSchema" : [
],
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 2 × 8198 × 1 × 8)",
"shortDescription" : "",
"shape" : "[2, 8198, 1, 8]",
"name" : "logits",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 2 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[2, 1280, 1, 1]",
"name" : "h_out",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 2 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[2, 1280, 1, 1]",
"name" : "c_out",
"type" : "MultiArray"
}
],
"name" : "decoder_2",
"mlProgramOperationTypeHistogram" : {
"Ios18.tanh" : 4,
"Ios18.relu" : 1,
"Ios18.sigmoid" : 6,
"Ios18.constexprBlockwiseShiftScale" : 6,
"Ios18.concat" : 2,
"Ios18.sliceByIndex" : 12,
"Ios18.conv" : 6,
"Ios18.mul" : 6,
"Ios18.add" : 5
}
},
{
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 1 × 640 × 1 × 1)",
"shortDescription" : "",
"shape" : "[1, 640, 1, 1]",
"name" : "embed",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 1 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[1, 1280, 1, 1]",
"name" : "h_in",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 1 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[1, 1280, 1, 1]",
"name" : "c_in",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 1 × 640 × 1 × 8)",
"shortDescription" : "",
"shape" : "[1, 640, 1, 8]",
"name" : "enc_step",
"type" : "MultiArray"
}
],
"computePrecision" : "Mixed (Float16, Int32)",
"storagePrecision" : "Mixed (Float16, Int8)",
"stateSchema" : [
],
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 1 × 8198 × 1 × 8)",
"shortDescription" : "",
"shape" : "[1, 8198, 1, 8]",
"name" : "logits",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 1 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[1, 1280, 1, 1]",
"name" : "h_out",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 1 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[1, 1280, 1, 1]",
"name" : "c_out",
"type" : "MultiArray"
}
],
"name" : "decoder_1",
"mlProgramOperationTypeHistogram" : {
"Ios18.tanh" : 4,
"Ios18.relu" : 1,
"Ios18.sigmoid" : 6,
"Ios18.constexprBlockwiseShiftScale" : 6,
"Ios18.concat" : 2,
"Ios18.sliceByIndex" : 12,
"Ios18.conv" : 6,
"Ios18.mul" : 6,
"Ios18.add" : 5
}
}
],
"mlProgramOperationTypeHistogram" : {
"Ios18.tanh" : 4,
"Ios18.relu" : 1,
"Ios18.sigmoid" : 6,
"Ios18.constexprBlockwiseShiftScale" : 6,
"Ios18.concat" : 2,
"Ios18.sliceByIndex" : 12,
"Ios18.conv" : 6,
"Ios18.mul" : 6,
"Ios18.add" : 5
},
"isUpdatable" : "0",
"stateSchema" : [
],
"availability" : {
"macOS" : "15.0",
"tvOS" : "18.0",
"visionOS" : "2.0",
"watchOS" : "11.0",
"iOS" : "18.0",
"macCatalyst" : "18.0"
},
"computePrecision" : "Mixed (Float16, Int32)",
"modelType" : {
"name" : "MLModelType_mlProgram"
},
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 16 × 640 × 1 × 1)",
"shortDescription" : "",
"shape" : "[16, 640, 1, 1]",
"name" : "embed",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 16 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[16, 1280, 1, 1]",
"name" : "h_in",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 16 × 1280 × 1 × 1)",
"shortDescription" : "",
"shape" : "[16, 1280, 1, 1]",
"name" : "c_in",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float16",
"formattedType" : "MultiArray (Float16 16 × 640 × 1 × 8)",
"shortDescription" : "",
"shape" : "[16, 640, 1, 8]",
"name" : "enc_step",
"type" : "MultiArray"
}
],
"defaultFunctionName" : "main",
"generatedClassName" : "ParakeetDecodeStep",
"userDefinedMetadata" : {
},
"method" : "predict"
}
]