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
..
gptq	feat(server): Add inference support for GPTQ (llama + falcon tested) + Quantization script (#438)	2023-06-26 12:27:01 +02:00
__init__.py	feat(server): Rework model loading (#344)	2023-06-08 14:51:52 +02:00
convert.py	fix(server): harden the weights choice to save on disk. (#561)	2023-07-07 14:50:12 +02:00
dist.py	fix(server): Do not init process group if already initialized (#388)	2023-06-26 12:32:54 +02:00
hub.py	feat(server): Adding new ignore_rule for conversion. (#485)	2023-06-23 12:41:13 +02:00
layers.py	feat(server): use latest flash attention commit (#543)	2023-07-04 20:23:55 +02:00
logits_process.py	fix(server): avoid errors for very small top_p values (#544)	2023-07-04 20:11:33 +02: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): add paged attention to flash models (#516)	2023-06-30 19:09:59 +02:00