text-generation-inference/server
Nicolas Patry e943a294bc
fix(server): harden the weights choice to save on disk. (#561)
- Look at `transformers` base class to check for
  `_key_to_ignore_on_load_missing` or `_tied_weights` which are the
  standard attributes to select the keys to NOT save on disk (since they
  are ignored)

- Modified safetensors code (to be reflected in safetensors even if it's
  an internal function).
  
- Will not work for trust_remote_code=True repos (like santacoder).

Should help with :
https://github.com/huggingface/text-generation-inference/issues/555
and : https://github.com/huggingface/text-generation-inference/pull/501
and https://github.com/huggingface/text-generation-inference/issues/556
and
https://github.com/huggingface/text-generation-inference/issues/482#issuecomment-1623713593
2023-07-07 14:50:12 +02:00
..
custom_kernels feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
tests fix(server): harden the weights choice to save on disk. (#561) 2023-07-07 14:50:12 +02:00
text_generation_server fix(server): harden the weights choice to save on disk. (#561) 2023-07-07 14:50:12 +02:00
.gitignore feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00
Makefile fix: Update server/Makefile to include Makefile-vllm (#520) 2023-07-04 09:39:25 +02:00
Makefile-flash-att feat(server): use latest flash attention commit (#543) 2023-07-04 20:23:55 +02:00
Makefile-vllm feat(server): add paged attention to flash models (#516) 2023-06-30 19:09:59 +02:00
poetry.lock feat(server): use latest flash attention commit (#543) 2023-07-04 20:23:55 +02:00
pyproject.toml v0.9.1 (#558) 2023-07-06 16:05:42 +02:00
README.md feat(router): refactor API and add openAPI schemas (#53) 2023-02-03 12:43:37 +01:00
requirements.txt feat(server): use latest flash attention commit (#543) 2023-07-04 20:23:55 +02:00

Text Generation Inference Python gRPC Server

A Python gRPC server for Text Generation Inference

Install

make install

Run

make run-dev