Skip to content

Commit 25d93a7

Browse files
authored
update requirements (#2957)
1 parent 7f0d9d6 commit 25d93a7

File tree

4 files changed

+8
-7
lines changed

4 files changed

+8
-7
lines changed

.dev_scripts/ci_container_test.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ if [ "$MODELSCOPE_SDK_DEBUG" == "True" ]; then
2222

2323
pip install -r requirements/framework.txt -U -i https://mirrors.aliyun.com/pypi/simple/
2424
pip install diffusers decord einops -U -i https://mirrors.aliyun.com/pypi/simple/
25-
pip install autoawq!=0.2.7.post3 -U --no-deps
25+
pip install autoawq -U --no-deps
2626

2727
# test with install
2828
pip install .

requirements/eval.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,2 @@
1-
evalscope[all]>=0.5.2
1+
evalscope[opencompass]
2+
evalscope[vlmeval]

requirements/install_all.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
# please use python=3.10, cuda12.*
22
# sh requirements/install_all.sh
33
pip install "vllm>=0.5.1" -U
4-
pip install "lmdeploy>=0.5" -U --no-deps
5-
pip install autoawq!=0.2.7.post3 -U --no-deps
4+
pip install "lmdeploy>=0.5,<0.7" -U --no-deps
5+
pip install autoawq -U --no-deps
66
pip install auto_gptq optimum bitsandbytes -U
77
pip install git+https://github.com/modelscope/ms-swift.git#egg=ms-swift[all]
88
pip install timm -U

swift/llm/export/quant.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,17 +17,17 @@ class QuantEngine(ProcessorMixin):
1717

1818
def __init__(self, args: ExportArguments):
1919
self.args = args
20-
if args.output_dir:
21-
args.save_args()
2220
kwargs = {}
2321
if args.quant_method == 'awq':
2422
from awq import AutoAWQForCausalLM
2523
kwargs['automodel_class'] = AutoAWQForCausalLM
2624
self.model, self.template = prepare_model_template(args, **kwargs)
2725
self.template.set_mode('train')
2826

29-
HfConfigFactory.set_model_config_attr(self.model, 'use_cache', True)
27+
HfConfigFactory.set_model_config_attr(self.model, 'use_cache', False)
3028
self.processor = self.template.processor
29+
if args.output_dir:
30+
args.save_args()
3131

3232
def quantize(self):
3333
args = self.args

0 commit comments

Comments
 (0)