hf_text-generation-inference/server/text_generation_server/models
Nicolas Patry 9ecfa16b12
Speculative (#1308)
2023-12-11 12:46:30 +01:00
..
custom_modeling Add RoCm support (#1243) 2023-11-27 14:08:12 +01:00
__init__.py Speculative (#1308) 2023-12-11 12:46:30 +01:00
bloom.py Handling bloom prefix. (#1090) 2023-10-03 11:55:10 +02:00
cache_manager.py feat: add mistral model (#1071) 2023-09-28 09:55:47 +02:00
causal_lm.py Speculative (#1308) 2023-12-11 12:46:30 +01:00
flash_causal_lm.py Speculative (#1308) 2023-12-11 12:46:30 +01:00
flash_llama.py Speculative (#1308) 2023-12-11 12:46:30 +01:00
flash_mistral.py Speculative (#1308) 2023-12-11 12:46:30 +01:00
flash_neox.py feat(server): Using `quantize_config.json` instead of GPTQ_BITS env variables. (#671) 2023-07-25 13:00:27 +02:00
flash_rw.py Fix Falcon weight mapping for H2O.ai checkpoints (#953) 2023-08-31 21:15:14 +02:00
flash_santacoder.py feat(server): Using `quantize_config.json` instead of GPTQ_BITS env variables. (#671) 2023-07-25 13:00:27 +02:00
galactica.py Fix missing arguments in Galactica's from_pb (#1022) 2023-09-21 08:15:59 +02:00
gpt_neox.py enable bfloat16 for cpu (#1034) 2023-09-19 17:19:28 +02:00
idefics.py enable bfloat16 for cpu (#1034) 2023-09-19 17:19:28 +02:00
idefics_causal_lm.py Speculative (#1308) 2023-12-11 12:46:30 +01:00
model.py Speculative (#1308) 2023-12-11 12:46:30 +01:00
mpt.py enable bfloat16 for cpu (#1034) 2023-09-19 17:19:28 +02:00
opt.py enable bfloat16 for cpu (#1034) 2023-09-19 17:19:28 +02:00
rw.py enable bfloat16 for cpu (#1034) 2023-09-19 17:19:28 +02:00
santacoder.py enable bfloat16 for cpu (#1034) 2023-09-19 17:19:28 +02:00
seq2seq_lm.py Speculative (#1308) 2023-12-11 12:46:30 +01:00
t5.py enable bfloat16 for cpu (#1034) 2023-09-19 17:19:28 +02:00
types.py Speculative (#1308) 2023-12-11 12:46:30 +01:00