OlivierDehaene 6abec14a7e feat(server): batch tokenization for flash causal lm (#411)		2023-06-05 16:09:41 +02:00
..
custom_modeling	feat(server): only compute prefill logprobs when asked (#406)	2023-06-02 17:12:30 +02:00
__init__.py	v0.8.2	2023-06-01 19:49:13 +02:00
bloom.py	fix(server): fix bnb quantization for CausalLM models (#385)	2023-05-31 11:48:28 +02:00
causal_lm.py	feat(server): only compute prefill logprobs when asked (#406)	2023-06-02 17:12:30 +02:00
flash_causal_lm.py	feat(server): batch tokenization for flash causal lm (#411)	2023-06-05 16:09:41 +02:00
flash_llama.py	feat(server): support trust_remote_code (#363)	2023-05-23 20:40:39 +02:00
flash_neox.py	feat(server): support trust_remote_code (#363)	2023-05-23 20:40:39 +02:00
flash_rw.py	feat(server): add retry on download (#384)	2023-05-31 10:57:53 +02:00
flash_santacoder.py	v0.8.2	2023-06-01 19:49:13 +02:00
galactica.py	fix(server): fix bnb quantization for CausalLM models (#385)	2023-05-31 11:48:28 +02:00
gpt_neox.py	fix(server): fix bnb quantization for CausalLM models (#385)	2023-05-31 11:48:28 +02:00
model.py	fix(server): fix has_position_ids (#395)	2023-06-01 11:41:35 +02:00
opt.py	fix(server): fix bnb quantization for CausalLM models (#385)	2023-05-31 11:48:28 +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	fix(server): fix bnb quantization for CausalLM models (#385)	2023-05-31 11:48:28 +02:00
types.py	feat(server): support vectorized warpers in flash causal lm (#317)	2023-05-26 12:30:27 +02:00