clap-htsat-unfused / quantize_config.json
Xenova's picture
Xenova HF staff
Update quantize_config.json
63f34b6
{
"per_channel": true,
"reduce_range": true,
"per_model_config": {
"model": {
"op_types": [
"Expand",
"ScatterND",
"Pad",
"Abs",
"Unsqueeze",
"ReduceSum",
"Not",
"CumSum",
"Constant",
"Exp",
"Sub",
"MatMul",
"Cast",
"Reshape",
"Flatten",
"Resize",
"Conv",
"ConstantOfShape",
"Gather",
"Relu",
"Div",
"Mul",
"GlobalAveragePool",
"Range",
"Erf",
"Where",
"ReduceMean",
"Pow",
"Shape",
"Concat",
"Slice",
"Softmax",
"Tanh",
"Sqrt",
"BatchNormalization",
"Add",
"Transpose",
"Gemm",
"Equal"
],
"weight_type": "QUInt8"
},
"text_model": {
"op_types": [
"ReduceMean",
"Reshape",
"Softmax",
"Pow",
"Erf",
"Tanh",
"Concat",
"Sub",
"Not",
"Expand",
"Mul",
"Transpose",
"Div",
"Constant",
"Equal",
"Unsqueeze",
"Slice",
"MatMul",
"Gather",
"ConstantOfShape",
"Shape",
"Cast",
"Where",
"Sqrt",
"Add",
"Gemm",
"CumSum",
"Relu"
],
"weight_type": "QInt8"
},
"audio_model": {
"op_types": [
"BatchNormalization",
"ScatterND",
"ReduceMean",
"Reshape",
"Softmax",
"Pow",
"Erf",
"GlobalAveragePool",
"Concat",
"Sub",
"Not",
"Expand",
"Mul",
"Transpose",
"Div",
"Constant",
"Equal",
"Unsqueeze",
"Pad",
"Slice",
"Resize",
"Range",
"MatMul",
"Gather",
"ConstantOfShape",
"Shape",
"Cast",
"Sqrt",
"Where",
"Add",
"Conv",
"Flatten",
"Gemm",
"Relu"
],
"weight_type": "QUInt8"
}
}
}