bart_lfqa / quantize_config.json
uzaaft's picture
Upload folder using huggingface_hub
3cdd73c
{
"per_channel": true,
"reduce_range": true,
"per_model_config": {
"decoder_with_past_model": {
"op_types": [
"Div",
"MatMul",
"Transpose",
"Constant",
"Unsqueeze",
"Softmax",
"Where",
"Sqrt",
"ReduceMean",
"Expand",
"Reshape",
"Erf",
"Concat",
"Gather",
"Equal",
"Sub",
"Cast",
"Range",
"Shape",
"Mul",
"Pow",
"Add",
"ConstantOfShape"
],
"weight_type": "QInt8"
},
"decoder_model": {
"op_types": [
"Div",
"MatMul",
"Transpose",
"Constant",
"Slice",
"Unsqueeze",
"Softmax",
"Where",
"ReduceMean",
"Sqrt",
"Less",
"Expand",
"Reshape",
"Erf",
"Concat",
"Squeeze",
"Gather",
"Equal",
"Sub",
"Cast",
"Range",
"Shape",
"Mul",
"Pow",
"Add",
"ConstantOfShape"
],
"weight_type": "QInt8"
},
"encoder_model": {
"op_types": [
"Div",
"MatMul",
"Transpose",
"Constant",
"Unsqueeze",
"Softmax",
"Where",
"Sqrt",
"ReduceMean",
"Expand",
"Reshape",
"Erf",
"Concat",
"Gather",
"Equal",
"Sub",
"Cast",
"Range",
"Shape",
"Mul",
"Pow",
"Add",
"ConstantOfShape"
],
"weight_type": "QInt8"
},
"decoder_model_merged": {
"op_types": [
"Div",
"MatMul",
"Transpose",
"Constant",
"Slice",
"Unsqueeze",
"Softmax",
"Where",
"ReduceMean",
"Sqrt",
"Less",
"Expand",
"Reshape",
"Erf",
"Concat",
"Squeeze",
"Gather",
"Equal",
"Sub",
"Cast",
"If",
"Range",
"Shape",
"Mul",
"Pow",
"Add",
"ConstantOfShape"
],
"weight_type": "QInt8"
}
}
}