wulewule / configs /model_cfg.yaml
zhiyun.xu
update demo
d573b56
raw
history blame
766 Bytes
########### 主配置文件加载优先级,先rag_cfg.yaml,再本文件覆盖重复配置 ##########
defaults:
- rag_cfg
- _self_
llm_default_dir: ./models
llm_model: ${llm_default_dir}/wulewule_v1_1_8b-w4a16-4bit # wulewule model path, w4a16-4bit must turn on "use_lmdepoly"
# llm_model: ${repo_root}/models/wulewule_v1_1_8b # wulewule model path
use_lmdepoly: True
use_rag: True ## 是否开启rag
stream_response: True # whether use streaming output
#######################################################################
# lmdeploy 4bit used #
#######################################################################
cache_max_entry_count: 0.2 #lmdeploy 4bit, k/v cache内存占比调整为总显存的 20%