text-generation-inference/server/text_generation_server/models/custom_modeling
Nicolas Patry edc9ce9beb Cleanup.
2023-05-15 15:21:49 +02:00
..
__init__.py feat(server): flash santacoder (#153) 2023-04-03 19:06:42 +02:00
flash_llama_modeling.py Cleanup. 2023-05-15 15:21:49 +02:00
flash_neox_modeling.py Cleanup. 2023-05-15 15:21:49 +02:00
flash_santacoder_modeling.py Single place for TP layers + Dropout Layer Norm + FastLinear 2023-05-15 15:02:19 +02:00