OlivierDehaene 3b71c38558 feat(server): flash attention v2 (#624)		2023-07-18 16:21:18 +02:00
..
gptq	feat(server): Reworking the quantization script so it's still universal (not llama specific) (#587)	2023-07-18 12:19:05 +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	fix(server): Do not init process group if already initialized (#388)	2023-06-26 12:32:54 +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	feat(server): Implements sharding for non divisible `vocab_size`. (#583)	2023-07-12 16:43:31 +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	GPTQ Env vars: catch correct type of error (#596)	2023-07-12 19:57:46 +02:00