text-generation-inference/server/text_generation_server/models/custom_modeling
Adam Kowalski 7f9072228a
fix(server): Adding logger import to t5_modeling.py (#585)
Logger is referenced during the apex importing but is not imported,
causing a NameError
2023-07-12 10:40:32 +02:00
..
__init__.py feat(server): flash santacoder (#153) 2023-04-03 19:06:42 +02:00
bloom_modeling.py feat: better errors for warmup and TP (#575) 2023-07-10 14:47:15 +02:00
flash_llama_modeling.py feat: better errors for warmup and TP (#575) 2023-07-10 14:47:15 +02:00
flash_neox_modeling.py feat: better errors for warmup and TP (#575) 2023-07-10 14:47:15 +02:00
flash_rw_modeling.py feat: better errors for warmup and TP (#575) 2023-07-10 14:47:15 +02:00
flash_santacoder_modeling.py feat(server): Support for env value for GPTQ_BITS and GPTQ_GROUPSIZE. (#580) 2023-07-12 10:00:02 +02:00
mpt_modeling.py feat: better errors for warmup and TP (#575) 2023-07-10 14:47:15 +02:00
neox_modeling.py feat: better errors for warmup and TP (#575) 2023-07-10 14:47:15 +02:00
opt_modeling.py feat: better errors for warmup and TP (#575) 2023-07-10 14:47:15 +02:00
t5_modeling.py fix(server): Adding logger import to t5_modeling.py (#585) 2023-07-12 10:40:32 +02:00