distilvit / quantize_config.json
tarekziade's picture
Upload 15 files
690919f verified
raw
history blame
1.56 kB
{
"per_channel": false,
"reduce_range": false,
"per_model_config": {
"decoder_model": {
"op_types": [
"Gemm",
"Slice",
"Transpose",
"Reshape",
"Where",
"Sqrt",
"ReduceMean",
"Pow",
"Constant",
"Tanh",
"Softmax",
"Concat",
"Sub",
"MatMul",
"Mul",
"Range",
"Gather",
"Shape",
"ConstantOfShape",
"Squeeze",
"Div",
"Unsqueeze",
"Cast",
"Add",
"Split"
],
"weight_type": "QInt8"
},
"encoder_model": {
"op_types": [
"Slice",
"Transpose",
"Reshape",
"Where",
"Sqrt",
"ReduceMean",
"Pow",
"Constant",
"Softmax",
"Concat",
"Sub",
"MatMul",
"Mul",
"Equal",
"Gather",
"Conv",
"Shape",
"ConstantOfShape",
"Div",
"Unsqueeze",
"Erf",
"Add",
"Expand"
],
"weight_type": "QUInt8"
}
}
}