diff --git a/README.md b/README.md index 9f3ec7a..226a68b 100644 --- a/README.md +++ b/README.md @@ -42,7 +42,7 @@ 🚩 本项目未涉及微调、训练过程，但可利用微调或训练对本项目效果进行优化。 -🌐 [AutoDL 镜像](https://www.codewithgpu.com/i/imClumsyPanda/langchain-ChatGLM/langchain-ChatGLM) 中 `v5` 版本所使用代码已更新至本项目 `0.2.0` 版本。 +🌐 [AutoDL 镜像](https://www.codewithgpu.com/i/imClumsyPanda/langchain-ChatGLM/Langchain-Chatchat) 中 `v5` 版本所使用代码已更新至本项目 `0.2.0` 版本。 🐳 [Docker 镜像](registry.cn-beijing.aliyuncs.com/chatchat/chatchat:0.2.0) @@ -214,6 +214,7 @@ embedding_model_dict = { ```shell $ python init_database.py ``` + - 如果您是第一次运行本项目，知识库尚未建立，或者配置文件中的知识库类型、嵌入模型发生变化，需要以下命令初始化或重建知识库： ```shell @@ -242,8 +243,7 @@ embedding_model_dict = { $ python server/llm_api.py ``` -项目支持多卡加载，需在 llm_api.py 中修改 create_model_worker_app 函数中如下三个参数: - +项目支持多卡加载，需在 llm_api.py 中修改 create_model_worker_app 函数中，修改如下三个参数: ```python gpus=None, num_gpus=1, @@ -258,14 +258,12 @@ max_gpu_memory="20GiB" ##### 5.1.2 基于命令行脚本 llm_api_launch.py 启动 LLM 服务 -**!!!注意:** +⚠️ **注意:** **1.llm_api_launch.py脚本仅适用于linux和mac设备,win平台请使用wls;** **2.加载非默认模型需要用命令行参数--model-path-address指定指定模型，不会读取model_config.py配置;** -**!!!** - 在项目根目录下，执行 [server/llm_api_launch.py](server/llm_api.py) 脚本启动 **LLM 模型**服务： ```shell @@ -284,7 +282,7 @@ $ python server/llm_api_launch.py --model-path-addresss model1@host1@port1 model $ python server/llm_api_launch.py --gpus 0,1 --num-gpus 2 --max-gpu-memory 10GiB ``` -注：以如上方式启动LLM服务会以nohup命令在后台运行 fastchat 服务，如需停止服务，可以运行如下命令,但该脚本**仅适用于linux和mac平台,win平台请使用wsl**： +注：以如上方式启动LLM服务会以nohup命令在后台运行 FastChat 服务，如需停止服务，可以运行如下命令： ```shell $ python server/llm_api_shutdown.py --serve all @@ -352,6 +350,7 @@ $ streamlit run webui.py --server.port 666 - Web UI 对话界面： ![](img/webui_0813_0.png) + - Web UI 知识库管理页面： ![](img/webui_0813_1.png) @@ -422,7 +421,7 @@ $ python webui_allinone.py --model-path-address model1@host1@port1 model2@host2@ $ python webui_alline.py --model-path-address model@host@port --num-gpus 2 --gpus 0,1 --max-gpu-memory 10GiB ``` -其他参数详见各脚本及fastchat服务说明。 +其他参数详见各脚本及 Fastchat 服务说明。 上述两个一键启动脚本会后台运行多个服务，如要停止所有服务，可使用 `shutdown_all.sh` 脚本： @@ -472,6 +471,6 @@ bash shutdown_all.sh ## 项目交流群 -二维码 +二维码 🎉 langchain-ChatGLM 项目微信交流群，如果你也对本项目感兴趣，欢迎加入群聊参与讨论交流。 diff --git a/img/qr_code_52.jpg b/img/qr_code_52.jpg new file mode 100644 index 0000000..18793d5 Binary files /dev/null and b/img/qr_code_52.jpg differ diff --git a/server/api_allinone.py b/server/api_allinone.py index 4379d7d..3be8581 100644 --- a/server/api_allinone.py +++ b/server/api_allinone.py @@ -11,10 +11,11 @@ python server/api_allinone.py --model-path-address model@host@port --num-gpus 2 """ import sys import os + sys.path.append(os.path.dirname(__file__)) sys.path.append(os.path.dirname(os.path.dirname(__file__))) -from llm_api_launch import launch_all,parser,controller_args,worker_args,server_args +from llm_api_launch import launch_all, parser, controller_args, worker_args, server_args from api import create_app import uvicorn @@ -23,8 +24,8 @@ parser.add_argument("--api-port", type=int, default=7861) parser.add_argument("--ssl_keyfile", type=str) parser.add_argument("--ssl_certfile", type=str) +api_args = ["api-host", "api-port", "ssl_keyfile", "ssl_certfile"] -api_args = ["api-host","api-port","ssl_keyfile","ssl_certfile"] def run_api(host, port, **kwargs): app = create_app() @@ -38,18 +39,19 @@ def run_api(host, port, **kwargs): else: uvicorn.run(app, host=host, port=port) + if __name__ == "__main__": print("Luanching api_allinone，it would take a while, please be patient...") print("正在启动api_allinone，LLM服务启动约3-10分钟，请耐心等待...") # 初始化消息 args = parser.parse_args() args_dict = vars(args) - launch_all(args=args,controller_args=controller_args,worker_args=worker_args,server_args=server_args) + launch_all(args=args, controller_args=controller_args, worker_args=worker_args, server_args=server_args) run_api( - host=args.api_host, - port=args.api_port, - ssl_keyfile=args.ssl_keyfile, - ssl_certfile=args.ssl_certfile, - ) + host=args.api_host, + port=args.api_port, + ssl_keyfile=args.ssl_keyfile, + ssl_certfile=args.ssl_certfile, + ) print("Luanching api_allinone done.") print("api_allinone启动完毕.") diff --git a/server/llm_api_launch.py b/server/llm_api_launch.py index 044bab6..0f7710a 100644 --- a/server/llm_api_launch.py +++ b/server/llm_api_launch.py @@ -132,7 +132,7 @@ worker_args = [ "gptq-ckpt", "gptq-wbits", "gptq-groupsize", "gptq-act-order", "model-names", "limit-worker-concurrency", "stream-interval", "no-register", - "controller-address","worker-address" + "controller-address", "worker-address" ] # -----------------openai server--------------------------- @@ -159,8 +159,6 @@ server_args = ["server-host", "server-port", "allow-credentials", "api-keys", "controller-address" ] - - # 0,controller, model_worker, openai_api_server # 1, 命令行选项 # 2,LOG_PATH @@ -201,7 +199,7 @@ def string_args(args, args_list): return args_str -def launch_worker(item,args,worker_args=worker_args): +def launch_worker(item, args, worker_args=worker_args): log_name = item.split("/")[-1].split("\\")[-1].replace("-", "_").replace("@", "_").replace(".", "_") # 先分割model-path-address,在传到string_args中分析参数 args.model_path, args.worker_host, args.worker_port = item.split("@") @@ -230,11 +228,11 @@ def launch_all(args, subprocess.run(controller_check_sh, shell=True, check=True) print(f"worker启动时间视设备不同而不同，约需3-10分钟，请耐心等待...") if isinstance(args.model_path_address, str): - launch_worker(args.model_path_address,args=args,worker_args=worker_args) + launch_worker(args.model_path_address, args=args, worker_args=worker_args) else: for idx, item in enumerate(args.model_path_address): print(f"开始加载第{idx}个模型:{item}") - launch_worker(item,args=args,worker_args=worker_args) + launch_worker(item, args=args, worker_args=worker_args) server_str_args = string_args(args, server_args) server_sh = base_launch_sh.format("openai_api_server", server_str_args, LOG_PATH, "openai_api_server") @@ -244,11 +242,12 @@ def launch_all(args, print("Launching LLM service done!") print("LLM服务启动完毕。") + if __name__ == "__main__": args = parser.parse_args() # 必须要加http//:，否则InvalidSchema: No connection adapters were found args = argparse.Namespace(**vars(args), - **{"controller-address": f"http://{args.controller_host}:{str(args.controller_port)}"}) + **{"controller-address": f"http://{args.controller_host}:{str(args.controller_port)}"}) if args.gpus: if len(args.gpus.split(",")) < args.num_gpus: