Merge branch 'main' into Agent_Hub_Dev

# Conflicts:
#	pilot/model/proxy/llms/chatgpt.py
#	pilot/scene/base_chat.py
#	setup.py
This commit is contained in:
yhjun1026
2023-11-20 16:26:25 +08:00
64 changed files with 5245 additions and 540 deletions

View File

@@ -317,9 +317,12 @@ def core_requires():
# TODO move transformers to default
"transformers>=4.31.0",
"alembic==1.12.0",
# for excel
"openpyxl==3.1.2",
"chardet==5.1.0",
"xlrd==2.0.1",
# for cache, TODO pympler has not been updated for a long time and needs to find a new toolkit.
"pympler",
]
@@ -364,6 +367,8 @@ def quantization_requires():
)
pkgs = [f"bitsandbytes @ {local_pkg}"]
print(pkgs)
# For chatglm2-6b-int4
pkgs += ["cpm_kernels"]
setup_spec.extras["quantization"] = pkgs
@@ -409,6 +414,13 @@ def vllm_requires():
setup_spec.extras["vllm"] = ["vllm"]
def cache_requires():
"""
pip install "db-gpt[cache]"
"""
setup_spec.extras["cache"] = ["rocksdict", "msgpack"]
# def chat_scene():
# setup_spec.extras["chat"] = [
# ""
@@ -459,6 +471,7 @@ all_datasource_requires()
openai_requires()
gpt4all_requires()
vllm_requires()
cache_requires()
# must be last
default_requires()