File tree Expand file tree Collapse file tree 4 files changed +8
-7
lines changed Expand file tree Collapse file tree 4 files changed +8
-7
lines changed Original file line number Diff line number Diff 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 .
Original file line number Diff line number Diff line change 1- evalscope[all]>=0.5.2
1+ evalscope[opencompass]
2+ evalscope[vlmeval]
Original file line number Diff line number Diff line change 11# please use python=3.10, cuda12.*
22# sh requirements/install_all.sh
33pip 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
66pip install auto_gptq optimum bitsandbytes -U
77pip install git+https://github.com/modelscope/ms-swift.git#egg=ms-swift[all]
88pip install timm -U
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments