huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-21 06:42:10 +00:00
Code Issues Packages Projects Releases Wiki Activity
b8fefa6b55
text-generation-inference/server/text_generation_server/models
History
Leo Tronchon b8fefa6b55
raise exception on invalid images (#999) 
# What does this PR do?
This PR is meant to handle cases in which the images provided are
invalid.

## Who can review?

Anyone in the community is free to review the PR once the tests have
passed. Feel free to tag
members/contributors who may be interested in your PR.

@Narsil

---------

Co-authored-by: Nicolas Patry <patry.nicolas@protonmail.com>
2023-10-03 10:26:10 +02:00
..
custom_modeling raise exception on invalid images (#999) 2023-10-03 10:26:10 +02:00
__init__.py Fixing eetq dockerfile. (#1081) 2023-09-29 11:19:06 +02:00
bloom.py enable bfloat16 for cpu (#1034) 2023-09-19 17:19:28 +02:00
cache_manager.py feat: add mistral model (#1071) 2023-09-28 09:55:47 +02:00
causal_lm.py feat: format code (#1070) 2023-09-27 12:22:09 +02:00
flash_causal_lm.py feat: add mistral model (#1071) 2023-09-28 09:55:47 +02:00
flash_llama.py Add AWQ quantization inference support (#1019) (#1054) 2023-09-25 15:31:27 +02:00
flash_mistral.py feat: add mistral model (#1071) 2023-09-28 09:55:47 +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 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_causal_lm.py feat: format code (#1070) 2023-09-27 12:22:09 +02:00
idefics.py enable bfloat16 for cpu (#1034) 2023-09-19 17:19:28 +02:00
model.py feat: add mistral model (#1071) 2023-09-28 09:55:47 +02: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 feat: format code (#1070) 2023-09-27 12:22:09 +02:00
t5.py enable bfloat16 for cpu (#1034) 2023-09-19 17:19:28 +02:00
types.py Rebased #617 (#868) 2023-08-28 11:43:47 +02:00