From d7bc8848f4462ee142106cf67e2a3ae17641ee8b Mon Sep 17 00:00:00 2001 From: Mengni Wang Date: Mon, 10 Jul 2023 09:04:24 +0800 Subject: [PATCH] update requirements Signed-off-by: Mengni Wang --- requirements-dev.txt | 1 - tools/ci_build/requirements.txt | 3 +++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/requirements-dev.txt b/requirements-dev.txt index 6390d08d0bab8..73e04e6b37c0b 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -18,4 +18,3 @@ scipy sympy wheel setuptools>=41.4.0 -neural-compressor diff --git a/tools/ci_build/requirements.txt b/tools/ci_build/requirements.txt index 5d8e56a70045c..00d17ee81a628 100644 --- a/tools/ci_build/requirements.txt +++ b/tools/ci_build/requirements.txt @@ -5,3 +5,6 @@ numpy==1.24.0 coloredlogs==15.0 transformers==4.24.0 psutil + +# package used by smooth quant test +neural-compressor>=2.2