hf_text-generation-inference/server/text_generation_server/models/custom_modeling
OlivierDehaene 895c5f1562
feat(server): only compute prefill logprobs when asked (#406)
Close #288
2023-06-02 17:12:30 +02:00
..
__init__.py feat(server): flash santacoder (#153) 2023-04-03 19:06:42 +02:00
flash_llama_modeling.py feat(server): only compute prefill logprobs when asked (#406) 2023-06-02 17:12:30 +02:00
flash_neox_modeling.py feat(server): only compute prefill logprobs when asked (#406) 2023-06-02 17:12:30 +02:00
flash_rw_modeling.py feat(server): only compute prefill logprobs when asked (#406) 2023-06-02 17:12:30 +02:00
flash_santacoder_modeling.py feat(server): only compute prefill logprobs when asked (#406) 2023-06-02 17:12:30 +02:00