preemo_text-generation-infe.../server/text_generation_server/utils
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
..
gptq fix(server): fix quantization python requirements (#708) 2023-07-27 12:28:10 +02:00
__init__.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
convert.py fix(server): blacklist local files (#609) 2023-07-13 21:54:55 +02:00
dist.py feat: add cuda memory fraction (#659) 2023-07-24 11:43:58 +02:00
flash_attn.py feat(server): flash attention v2 (#624) 2023-07-18 16:21:18 +02:00
hub.py feat(server): Adding new ignore_rule for conversion. (#485) 2023-06-23 12:41:13 +02:00
layers.py Modify exllama weight 2023-08-03 23:20:59 +02:00
logits_process.py Adding ctranslate2 quantization and inference: moving the contribution (#1) 2023-10-02 11:12:49 -07:00
tokens.py feat(server): add paged attention to flash models (#516) 2023-06-30 19:09:59 +02:00
watermark.py fix(server): fix flash-neox scores warping (#137) 2023-03-24 18:21:41 +01:00
weights.py feat(server): Using `quantize_config.json` instead of GPTQ_BITS env variables. (#671) 2023-07-25 13:00:27 +02:00