hf_text-generation-inference/server/text_generation_server/models
OlivierDehaene 83e442ca9a feat(server): use encoding to get prefill tokens 2023-06-20 18:29:55 +02:00
..
custom_modeling feat(router): add ngrok integration (#453) 2023-06-16 16:25:11 +02:00
__init__.py fix(server): fix warpers on CPU (#472) 2023-06-20 11:06:10 +02:00
bloom.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
causal_lm.py feat(server): use encoding to get prefill tokens 2023-06-20 18:29:55 +02:00
flash_causal_lm.py feat(server): use encoding to get prefill tokens 2023-06-20 18:29:55 +02:00
flash_llama.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
flash_neox.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
flash_rw.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
flash_santacoder.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
galactica.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
gpt_neox.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
model.py fix(server): fix has_position_ids (#395) 2023-06-01 11:41:35 +02:00
opt.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
rw.py feat(server): add retry on download (#384) 2023-05-31 10:57:53 +02:00
santacoder.py feat(server): support trust_remote_code (#363) 2023-05-23 20:40:39 +02:00
seq2seq_lm.py feat(server): only compute prefill logprobs when asked (#406) 2023-06-02 17:12:30 +02:00
t5.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
types.py feat(server): use encoding to get prefill tokens 2023-06-20 18:29:55 +02:00