Go to file
Michael Feil ff703cb867
Adding ctranslate2 quantization and inference: moving the contribution (#1)
* rebaseing the commit on preemo fork.

* reformatting and changes.

* update dockerfile

* update changes for dockerfile

* adapt path

* rebaseing the commit on preemo fork.

* reformatting and changes.

* update dockerfile

* update changes for dockerfile

* adapt path

---------

Co-authored-by: michaelfeil <me@michaelfeil.eu>
2023-10-02 11:12:49 -07:00
.github update PR template 2023-08-01 18:18:28 +02:00
assets feat(benchmark): tui based benchmarking tool (#149) 2023-03-30 15:26:27 +02:00
benchmark Compilation fix: Correct method argument types in generation.rs and validation.rs (#10) 2023-08-23 13:52:49 -07:00
clients/python feat(server): only compute prefill logprobs when asked (#406) 2023-06-02 17:12:30 +02:00
docs Wrapping completions and chat/completions endpoint (#2) 2023-09-27 08:58:07 -07:00
integration-tests feat: add cuda memory fraction (#659) 2023-07-24 11:43:58 +02:00
launcher Adding ctranslate2 quantization and inference: moving the contribution (#1) 2023-10-02 11:12:49 -07:00
load_tests feat: add nightly load testing (#358) 2023-05-23 17:42:19 +02:00
proto feat(server): auto max_batch_total_tokens for flash att models (#630) 2023-07-19 09:31:25 +02:00
router Wrapping completions and chat/completions endpoint (#2) 2023-09-27 08:58:07 -07:00
server Adding ctranslate2 quantization and inference: moving the contribution (#1) 2023-10-02 11:12:49 -07:00
.dockerignore chore: add `flash-attention` to docker ignore (#287) 2023-05-05 17:52:09 +02:00
.gitignore feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
Cargo.lock v0.9.4 (#713) 2023-07-27 19:25:15 +02:00
Cargo.toml v0.9.4 (#713) 2023-07-27 19:25:15 +02:00
Dockerfile Adding ctranslate2 quantization and inference: moving the contribution (#1) 2023-10-02 11:12:49 -07:00
LICENSE Claim copyright (#7) 2023-08-02 17:23:54 -07:00
Makefile docs(README): update readme 2023-07-25 19:45:25 +02:00
README-HuggingFace.md Add a new README (#3) 2023-08-01 12:22:07 -07:00
README.md Update README.md 2023-08-03 23:23:02 +02:00
rust-toolchain.toml v0.9.0 (#525) 2023-07-01 19:25:41 +02:00
sagemaker-entrypoint.sh feat(sagemaker): add trust remote code to entrypoint (#394) 2023-06-02 09:51:06 +02:00

README.md

Text Generation Inference

This is Preemo's fork of text-generation-inference, originally developed by Hugging Face. The original README is at README-HuggingFace.md. Since Hugging Face's text-generation-inference is no longer open-source, we have forked it and will continue to develop it here.

Our goal is to create an open-source text generation inference server that is modularized to allow for easy add state-of-the-art models, functionalities and optimizations. Functionalities and optimizations should be composable, so that users can easily combine them to create a custom inference server that fits their needs.

our plan

We at Preemo are currently busy working on our first release of our other product, so we expect to be able to start open-source development on this repository in September 2023. We will be working on the following, to ease the external contributions:

  • Adding a public visible CI/CD pipeline that runs tests and builds docker images
  • Unifying the build tools
  • Modularizing the codebase, introducing a plugin system

Our long-term goal is to grow the community around this repository, as a playground for trying out new ideas and optimizations in LLM inference. We at Preemo will implement features that interest us, but we also welcome contributions from the community, as long as they are modularized and composable.

Extra features in comparison to Hugging Face text-generation-inference v0.9.4

4bit quantization

4bit quantization is available using the NF4 and FP4 data types from bitsandbytes. It can be enabled by providing --quantize bitsandbytes-nf4 or --quantize bitsandbytes-fp4 as a command line argument to text-generation-launcher.