diff --git a/workflows/compression_aware_training/requirements.txt b/workflows/compression_aware_training/requirements.txt index 40bc25c1adf..11b4db7fbd9 100644 --- a/workflows/compression_aware_training/requirements.txt +++ b/workflows/compression_aware_training/requirements.txt @@ -4,5 +4,5 @@ datasets==2.11.0 intel-extension-for-transformers==1.1 intel_extension_for_pytorch==1.13.100 neural-compressor==2.1 -torch==1.13.1 +torch==2.2.0 transformers