diff --git a/README.md b/README.md index 2133c3d..9ce0220 100644 --- a/README.md +++ b/README.md @@ -132,10 +132,16 @@ Web UI 可以实现如下功能: - [ ] 知识图谱/图数据库接入 - [ ] Agent 实现 - [ ] 增加更多 LLM 模型支持 - - [x] THUDM/chatglm-6b - - [x] THUDM/chatglm-6b-int4 - - [x] THUDM/chatglm-6b-int4-qe - - [x] ClueAI/ChatYuan-large-v2 + - [x] [THUDM/chatglm-6b](https://huggingface.co/THUDM/chatglm-6b) + - [x] [THUDM/chatglm-6b-int8](https://huggingface.co/THUDM/chatglm-6b-int8) + - [x] [THUDM/chatglm-6b-int4](https://huggingface.co/THUDM/chatglm-6b-int4) + - [x] [THUDM/chatglm-6b-int4-qe](https://huggingface.co/THUDM/chatglm-6b-int4-qe) + - [x] [ClueAI/ChatYuan-large-v2](https://huggingface.co/ClueAI/ChatYuan-large-v2) +- [ ] 增加更多 Embedding 模型支持 + - [x] [nghuyong/ernie-3.0-nano-zh](https://huggingface.co/nghuyong/ernie-3.0-nano-zh) + - [x] [nghuyong/ernie-3.0-base-zh](https://huggingface.co/nghuyong/ernie-3.0-base-zh) + - [x] [shibing624/text2vec-base-chinese](https://huggingface.co/shibing624/text2vec-base-chinese) + - [x] [GanymedeNil/text2vec-large-chinese](https://huggingface.co/GanymedeNil/text2vec-large-chinese) - [ ] Web UI - [x] 利用 gradio 实现 Web UI DEMO - [x] 添加输出内容及错误提示 diff --git a/configs/model_config.py b/configs/model_config.py index 40de01a..b0093bc 100644 --- a/configs/model_config.py +++ b/configs/model_config.py @@ -20,6 +20,7 @@ llm_model_dict = { "chatyuan": "ClueAI/ChatYuan-large-v2", "chatglm-6b-int4-qe": "THUDM/chatglm-6b-int4-qe", "chatglm-6b-int4": "THUDM/chatglm-6b-int4", + "chatglm-6b-int8": "THUDM/chatglm-6b-int8", "chatglm-6b": "THUDM/chatglm-6b", }