hf_text-generation-inference/server/text_generation_server/models/custom_modeling
OlivierDehaene f59fb8b630
feat(router): add ngrok integration (#453)
2023-06-16 16:25:11 +02:00
..
__init__.py feat(server): flash santacoder (#153) 2023-04-03 19:06:42 +02:00
bloom_modeling.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
flash_llama_modeling.py feat(server): pre-allocate past key values for flash causal LM (#412) 2023-06-12 18:30:29 +02:00
flash_neox_modeling.py feat(server): pre-allocate past key values for flash causal LM (#412) 2023-06-12 18:30:29 +02:00
flash_rw_modeling.py feat(server): pre-allocate past key values for flash causal LM (#412) 2023-06-12 18:30:29 +02:00
flash_santacoder_modeling.py feat(router): add ngrok integration (#453) 2023-06-16 16:25:11 +02:00
neox_modeling.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
opt_modeling.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
t5_modeling.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00