diff --git a/.azure-pipelines/scripts/ut/env_setup.sh b/.azure-pipelines/scripts/ut/env_setup.sh index b0ee2c77cf8..d5876b07cef 100644 --- a/.azure-pipelines/scripts/ut/env_setup.sh +++ b/.azure-pipelines/scripts/ut/env_setup.sh @@ -92,7 +92,7 @@ elif [[ $(echo "${test_case}" | grep -c "tf pruning") != 0 ]]; then fi if [[ $(echo "${test_case}" | grep -c "api") != 0 ]] || [[ $(echo "${test_case}" | grep -c "adaptor") != 0 ]]; then - pip install git+https://github.com/intel/auto-round.git@ecca5349981044e1278773a251b3fc5c0a11fe7b + pip install auto-round fi # test deps diff --git a/requirements_pt.txt b/requirements_pt.txt index fd468a66332..67c5371c46c 100644 --- a/requirements_pt.txt +++ b/requirements_pt.txt @@ -1,4 +1,4 @@ -auto-round @ git+https://github.com/intel/auto-round.git@c7751c49853eb3497e7df9b92dc4733a80df34a0 +auto-round intel_extension_for_pytorch peft==0.10.0 psutil diff --git a/test/requirements.txt b/test/requirements.txt index 6608c28e710..3a24001cfd2 100644 --- a/test/requirements.txt +++ b/test/requirements.txt @@ -1,6 +1,6 @@ --find-links https://download.pytorch.org/whl/torch_stable.html accelerate==0.21.0 -auto-round @ git+https://github.com/intel/auto-round.git@c7751c49853eb3497e7df9b92dc4733a80df34a0 +auto-round dynast==1.6.0rc1 horovod intel-extension-for-pytorch