Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
hf_text-generation-inference/server/text_generation_server/utils
History
OlivierDehaene ece7ffa40a
feat(server): improve flash attention import errors (#465) 
@lewtun, is this enough?

Closes #458
Closes #456
 		2023-06-19 09:53:45 +02:00
..
__init__.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
convert.py feat(server): improve flash attention import errors (#465) 2023-06-19 09:53:45 +02:00
dist.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
hub.py feat(server): improve flash attention import errors (#465) 2023-06-19 09:53:45 +02:00
layers.py feat(server): optimize dist ops (#434) 2023-06-09 11:55:29 +02:00
logits_process.py feat(server): support vectorized warpers in flash causal lm (#317) 2023-05-26 12:30:27 +02:00
tokens.py feat(server): support vectorized warpers in flash causal lm (#317) 2023-05-26 12:30:27 +02:00
watermark.py fix(server): fix flash-neox scores warping (#137) 2023-03-24 18:21:41 +01:00
weights.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00