hf_text-generation-inference/server/text_generation_server/models
Victor SANH 2bc287bfcd
small fix on idefics (#954)
transposing the fixes from
https://github.com/huggingface/transformers/pull/25787
2023-09-01 18:44:34 +02:00
..
custom_modeling small fix on idefics (#954) 2023-09-01 18:44:34 +02:00
__init__.py Adding Idefics multi modal model. (#842) 2023-08-17 14:38:49 +02:00
bloom.py feat(server): Using `quantize_config.json` instead of GPTQ_BITS env variables. (#671) 2023-07-25 13:00:27 +02:00
causal_lm.py Rebased #617 (#868) 2023-08-28 11:43:47 +02:00
flash_causal_lm.py Rebased #617 (#868) 2023-08-28 11:43:47 +02:00
flash_llama.py fix: LlamaTokenizerFast to AutoTokenizer at flash_llama.py (#619) 2023-08-14 14:20:18 +02: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 feat(server): Using `quantize_config.json` instead of GPTQ_BITS env variables. (#671) 2023-07-25 13:00:27 +02:00
gpt_neox.py feat(server): Using `quantize_config.json` instead of GPTQ_BITS env variables. (#671) 2023-07-25 13:00:27 +02:00
idefics.py Adding Idefics multi modal model. (#842) 2023-08-17 14:38:49 +02:00
idefics_causal_lm.py Rebased #617 (#868) 2023-08-28 11:43:47 +02:00
model.py Fix typing in `Model.generate_token` (#733) 2023-07-31 14:35:14 +02:00
mpt.py feat(server): Using `quantize_config.json` instead of GPTQ_BITS env variables. (#671) 2023-07-25 13:00:27 +02:00
opt.py feat(server): Using `quantize_config.json` instead of GPTQ_BITS env variables. (#671) 2023-07-25 13:00:27 +02:00
rw.py "Fix" for rw-1b. (#860) 2023-08-17 09:05:41 +02:00
santacoder.py Directly load GPTBigCode to specified device (#618) 2023-07-21 11:27:31 +02:00
seq2seq_lm.py Rebased #617 (#868) 2023-08-28 11:43:47 +02:00
t5.py fix(server): T5 weights names. (#582) 2023-07-12 10:01:42 +02:00
types.py Rebased #617 (#868) 2023-08-28 11:43:47 +02:00