nn_pruning icon indicating copy to clipboard operation
nn_pruning copied to clipboard
verified publisher icon huggingface

Fix quantization tests

Open LukasHedegaard opened this issue 3 years ago • 0 comments

This PR fixes the nn_pruning/tests/test_quantization.py tests. How:

  • Fixing the symbolic_trace import in nn_pruning/modules/quantization.py
  • Finding a working combination of transformers and torch versions and fixing those in setup.py. Note that this was only set for the "tests" install to not impact general library compatibility.

LukasHedegaard avatar Jun 21 '22 12:06 LukasHedegaard