text-generation-inference/server/text_generation_server/models
2023-04-06 13:45:08 +02:00
..
custom_modeling optional rust validation 2023-04-06 13:45:08 +02:00
__init__.py feat(server): add flash attention llama 2023-04-06 13:45:08 +02:00
bloom.py feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00
causal_lm.py improve decode 2023-04-06 13:45:08 +02:00
flash_causal_lm.py fix concatenate 2023-04-06 13:45:08 +02:00
flash_llama.py optional rust validation 2023-04-06 13:45:08 +02:00
flash_neox.py optional rust validation 2023-04-06 13:45:08 +02:00
flash_santacoder.py fix(server): fix escape characters in stop sequence (#155) 2023-04-05 19:37:41 +02:00
galactica.py improve decode 2023-04-06 13:45:08 +02:00
gpt_neox.py fix(server): add position ids to neox (#126) 2023-03-15 13:12:49 +01:00
model.py improve decode 2023-04-06 13:45:08 +02:00
santacoder.py feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00
seq2seq_lm.py improve decode 2023-04-06 13:45:08 +02:00
t5.py feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00
types.py feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00