File tree 2 files changed +3
-3
lines changed
.azure-pipelines/scripts/ut
neural_compressor/adaptor
2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -11,8 +11,7 @@ bash /intel-extension-for-transformers/.github/workflows/script/prepare_env.sh
11
11
bash /intel-extension-for-transformers/.github/workflows/script/install_binary.sh
12
12
13
13
# prepare test env
14
- # tmp install transformers for incompatible issue
15
- pip install transformers
14
+ sed -i ' /neural-compressor.git/d' /intel-extension-for-transformers/tests/requirements.txt
16
15
pip install -r /intel-extension-for-transformers/tests/requirements.txt
17
16
LOG_DIR=/neural-compressor/log_dir
18
17
mkdir -p ${LOG_DIR}
Original file line number Diff line number Diff line change @@ -1813,7 +1813,8 @@ def smooth_quant(
1813
1813
kwargs ["percentile" ] = percentile
1814
1814
if scales_per_op is not None :
1815
1815
kwargs ["scales_per_op" ] = scales_per_op
1816
- auto_alpha_args ["init_alpha" ] = default_alpha
1816
+ if alpha == "auto" :
1817
+ auto_alpha_args ["init_alpha" ] = default_alpha
1817
1818
model ._model = self .sq .transform (
1818
1819
alpha = alpha ,
1819
1820
folding = folding ,
You can’t perform that action at this time.
0 commit comments