diffusers
907c0c2c - Implement _dequantize for TorchAO quantizer (#13538)

Commit
3 days ago
Implement _dequantize for TorchAO quantizer (#13538) * Implement _dequantize for TorchAO quantizer - Add _dequantize() method in TorchAoHfQuantizer that dequantizes TorchAOBaseTensor weights back to standard nn.Parameter - Fix _verify_if_layer_quantized to check isinstance(weight, TorchAOBaseTensor) so dequantized layers are correctly detected as non-quantized * enable dequantize for TorchAO tester mixin Signed-off-by: jiqing-feng <jiqing.feng@intel.com> * check dequantize Signed-off-by: jiqing-feng <jiqing.feng@intel.com> * fix dequantize: clear is_quantized flag and cast dtype after dequantize * fix Signed-off-by: jiqing-feng <jiqing.feng@intel.com> * fix error report Signed-off-by: jiqing-feng <jiqing.feng@intel.com> --------- Signed-off-by: jiqing-feng <jiqing.feng@intel.com> Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
Author
Parents
Loading