def prepare_bias(cls, module: QuantWBIOL): if module.bias is not None and not module.is_bias_quant_enabled: bias = module.bias.detach() elif module.bias is not None and module.is_bias_quant_enabled: bias = module.quant_bias() else: bias = module.bias return bias
def prepare_bias(cls, module: QuantWBIOL): if module.bias is not None and not module.is_bias_quant_enabled: bias = module.bias.detach() elif module.bias is not None and module.is_bias_quant_enabled: # export the dequantized value bias = module.quant_bias().value else: bias = module.bias return bias