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.
This PR fixes the nn_pruning/tests/test_quantization.py tests. How:
symbolic_trace
import in nn_pruning/modules/quantization.py