hf_text-generation-inference/server
xiaobin 4cce84301b
fit for baichuan models (#981)
As more and more people begin to use Baichuan's open-source models, the
influence of Baichuan models is growing, especially in China. Many
community members are interested in adding support for Baichuan models
to TGI. Meanwhile, Baichuan is a very open company, and in the future,
it plans to open-source more and more models, taking all this into
consideration, we would like to add support for the Baichuan model to
TGI. To do this, we need to make some changes, which we hope can be
merged into the main branch of TGI. In the future, we would be happy to
help maintain support for Baichuan models in TGI. We sincerely hope that
our pull request can be accepted. Thank you.

By the way, the changes of this time mainly for supporting Baichuan-7B.

---------

Co-authored-by: xiaoyuze <xiaoyuze@baichuan.com>
Co-authored-by: Nicolas Patry <patry.nicolas@protonmail.com>
2023-09-08 16:51:34 +02:00
..
custom_kernels feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
exllama_kernels feat: add cuda memory fraction (#659) 2023-07-24 11:43:58 +02:00
tests Rebased #617 (#868) 2023-08-28 11:43:47 +02:00
text_generation_server fit for baichuan models (#981) 2023-09-08 16:51:34 +02:00
.gitignore Version 1.0.1 (#836) 2023-08-14 11:23:11 +02:00
Makefile fix(server): fix missing datasets in quantize 2023-07-27 14:50:45 +02:00
Makefile-flash-att feat(server): use latest flash attention commit (#543) 2023-07-04 20:23:55 +02:00
Makefile-flash-att-v2 feat(server): flash attention v2 (#624) 2023-07-18 16:21:18 +02:00
Makefile-vllm Backport https://github.com/vllm-project/vllm/pull/936 (#977) 2023-09-04 15:00:19 +02:00
README.md feat(router): refactor API and add openAPI schemas (#53) 2023-02-03 12:43:37 +01:00
poetry.lock New release. (#941) 2023-08-29 14:28:22 +02:00
pyproject.toml New release. (#941) 2023-08-29 14:28:22 +02:00
requirements.txt New release. (#941) 2023-08-29 14:28:22 +02:00

README.md

Text Generation Inference Python gRPC Server

A Python gRPC server for Text Generation Inference

Install

make install

Run

make run-dev