huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-24 16:32:12 +00:00
Code Issues Packages Projects Releases Wiki Activity
edfbfdfb3f
text-generation-inference/server/text_generation_server/models
History
Félix Marty edfbfdfb3f Merge branch 'main' into gptq-cuda-kernels
2023-07-19 16:58:54 +02:00
..
custom_modeling Merge branch 'main' into gptq-cuda-kernels 2023-07-19 16:58:54 +02:00
__init__.py feat(server): flash attention v2 (#624) 2023-07-18 16:21:18 +02:00
bloom.py feat: Add the option to force another dtype than f16. (#513) 2023-06-30 20:30:09 +02:00
causal_lm.py support all, test llama 2023-07-13 15:41:57 +00:00
flash_causal_lm.py Merge branch 'main' into gptq-cuda-kernels 2023-07-19 16:58:54 +02:00
flash_llama.py Merge branch 'main' into gptq-cuda-kernels 2023-07-19 16:58:54 +02:00
flash_neox.py support all, test llama 2023-07-13 15:41:57 +00:00
flash_rw.py support all, test llama 2023-07-13 15:41:57 +00:00
flash_santacoder.py support all, test llama 2023-07-13 15:41:57 +00:00
galactica.py support all, test llama 2023-07-13 15:41:57 +00:00
gpt_neox.py support all, test llama 2023-07-13 15:41:57 +00:00
model.py Merge branch 'main' into gptq-cuda-kernels 2023-07-19 16:58:54 +02:00
mpt.py support all, test llama 2023-07-13 15:41:57 +00:00
opt.py support all, test llama 2023-07-13 15:41:57 +00:00
rw.py support all, test llama 2023-07-13 15:41:57 +00:00
santacoder.py support all, test llama 2023-07-13 15:41:57 +00:00
seq2seq_lm.py support all, test llama 2023-07-13 15:41:57 +00:00
t5.py support all, test llama 2023-07-13 15:41:57 +00:00
types.py feat(server): support vectorized warpers in flash causal lm (#317) 2023-05-26 12:30:27 +02:00