mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-06 03:20:41 +00:00
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:
13
setup.py
13
setup.py
@@ -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()
|
||||
|
Reference in New Issue
Block a user