Restrict version of pytorch-quantization (#20137)

openvinotoolkit · Sep 28, 2023 · 64cc3a9 · 64cc3a9
1 parent f85a276
commit 64cc3a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/model_hub_tests/torch_tests/requirements.txt b/tests/model_hub_tests/torch_tests/requirements.txt
@@ -11,7 +11,7 @@ sacremoses
 sentencepiece
 datasets
 pyctcdecode
-pytorch-quantization --extra-index-url https://pypi.ngc.nvidia.com
+pytorch-quantization<=2.1.3 --extra-index-url https://pypi.ngc.nvidia.com
 protobuf
 soundfile
 pandas