
🔥ChatLLM 基于知识库🔥
Click to Star⭐️免费领取tokens
Install
pip install -U chatllm
Usages
计划推出国内 oneapi,支持各种主流大模型,兼容openai客户端生态。
from chatllm.applications import ChatBase
qa = ChatBase()
qa.load_llm(model_name_or_path="THUDM/chatglm-6b")
for i in qa(query='周杰伦是谁', knowledge_base='周杰伦是傻子'):
print(i, end='')
OpenaiEcosystem
Click to 无缝对接openai生态
# 服务端
pip install "chatllm[openai]" && chatllm-run openai <本地模型地址>
import openai
openai.api_base = 'http://0.0.0.0:8000/v1'
openai.api_key = 'chatllm'
prompt = "你好"
completion = openai.Completion.create(prompt=prompt, stream=True, model="text-davinci-003")
for c in completion:
print(c.choices[0].text, end='')
ChatOCR
Click to ChatOCR
from meutils.pipe import *
from chatllm.llmchain.applications import ChatOCR
llm = ChatOCR()
file_path = "data/invoice.jpg"
llm.display(file_path, 700)
llm.chat('识别编号,公司名称,开票日期,开票人,收款人,复核人,金额', file_path=file_path) | xprint

ChatMind
Click to ChatMind
pip install "chatllm" && chatllm-run webui --name chatmind

ChatPDF
Click to ChatPDF
pip install "chatllm[pdf]" && chatllm-run webui --name chatpdf
from chatllm.applications.chatpdf import ChatPDF
qa = ChatPDF(encode_model='nghuyong/ernie-3.0-nano-zh')
qa.load_llm(model_name_or_path="THUDM/chatglm-6b")
qa.create_index('财报.pdf')
for i in qa(query='东北证券主营业务'):
print(i, end='')
- 支持召回结果查看

Deploy
Click to Deploy
量化等级 | 最低 GPU 显存(推理) | 最低 GPU 显存(高效参数微调) |
---|
FP16(无量化) | 13 GB | 14 GB |
INT8 | 8 GB | 9 GB |
INT4 | 6 GB | 7 GB |
TODO
Click to TODO
若二维码失效加微信拉群 313303303