From 7feb0e110095df37bc0c2ec2ba6e8b6c4b5cee12 Mon Sep 17 00:00:00 2001 From: "Sun, Xuehao" Date: Fri, 18 Oct 2024 17:42:54 +0800 Subject: [PATCH] remove autoround limit Signed-off-by: Sun, Xuehao --- .azure-pipelines/scripts/ut/env_setup.sh | 2 +- test/3x/torch/requirements.txt | 2 +- test/requirements.txt | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.azure-pipelines/scripts/ut/env_setup.sh b/.azure-pipelines/scripts/ut/env_setup.sh index fadd60240da..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@5dd16fc34a974a8c2f5a4288ce72e61ec3b1410f + pip install auto-round fi # test deps diff --git a/test/3x/torch/requirements.txt b/test/3x/torch/requirements.txt index 17c25a59be7..5b97060f9f8 100644 --- a/test/3x/torch/requirements.txt +++ b/test/3x/torch/requirements.txt @@ -1,4 +1,4 @@ -auto_round @ git+https://github.com/intel/auto-round.git@5dd16fc34a974a8c2f5a4288ce72e61ec3b1410f +auto_round deepspeed @ git+https://github.com/HabanaAI/DeepSpeed.git@1.18.0 expecttest intel_extension_for_pytorch diff --git a/test/requirements.txt b/test/requirements.txt index f2bbf5d9e71..cddb5c2b390 100644 --- a/test/requirements.txt +++ b/test/requirements.txt @@ -1,6 +1,6 @@ --find-links https://download.pytorch.org/whl/torch_stable.html accelerate -auto-round @ git+https://github.com/intel/auto-round.git@5dd16fc34a974a8c2f5a4288ce72e61ec3b1410f +auto-round dynast==1.6.0rc1 horovod intel-extension-for-pytorch