huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-23 07:52:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
44ce098c10
text-generation-inference/server/text_generation/models
History
OlivierDehaene 44ce098c10
feat(server): pre-allocate max attention mask (#75)
2023-02-24 12:49:21 +01:00
..
__init__.py feat: add safetensors conversion (#63) 2023-02-14 13:02:16 +01:00
bloom.py feat: add safetensors conversion (#63) 2023-02-14 13:02:16 +01:00
causal_lm.py feat(server): pre-allocate max attention mask (#75) 2023-02-24 12:49:21 +01:00
galactica.py feat(server): pre-allocate max attention mask (#75) 2023-02-24 12:49:21 +01:00
gpt_neox.py feat: add safetensors conversion (#63) 2023-02-14 13:02:16 +01:00
model.py fix(server): Minor refactorization using new_zeros (#24) 2023-01-17 09:10:22 +01:00
santacoder.py feat: add safetensors conversion (#63) 2023-02-14 13:02:16 +01:00
seq2seq_lm.py feat(server): pre-allocate max attention mask (#75) 2023-02-24 12:49:21 +01:00
t5.py feat(server): pre-allocate max attention mask (#75) 2023-02-24 12:49:21 +01:00
types.py feat(router): refactor API and add openAPI schemas (#53) 2023-02-03 12:43:37 +01:00