Nouamane Tazi b68d408f1b add ONNX support for BLOOM (#17961) ... * add onnx support for BLOOM * use TYPE_CHECKING for type annotations * fix past_shape for bloom (different from gpt2) * use logical_or instead of `+` for onnx support * bigger `atol_for_validation` for larger bloom models * copied -> taken because it's no longer an exact copy * remove "copied from" comment Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>		2022-07-01 10:44:42 -04:00
..
benchmark	[Test refactor 1/5] Per-folder tests reorganization (#15725)	2022-02-23 15:46:28 -05:00
deepspeed	deprecate is_torch_bf16_available (#17738)	2022-06-20 08:40:11 -04:00
extended	Update self-push workflow (#17177)	2022-05-13 16:28:00 +02:00
fixtures	add a warning in SpmConverter for sentencepiece's model using the byte fallback feature (#16629)	2022-04-11 11:06:10 +02:00
generation	Update expected values in constrained beam search tests (#17887)	2022-06-28 08:53:53 +02:00
models	Update expected values in CodeGen tests (#17888)	2022-07-01 15:33:36 +02:00
onnx	add ONNX support for BLOOM (#17961)	2022-07-01 10:44:42 -04:00
optimization	[Test refactor 1/5] Per-folder tests reorganization (#15725)	2022-02-23 15:46:28 -05:00
pipelines	feat: add pipeline registry abstraction (#17905)	2022-06-30 12:11:08 -04:00
sagemaker	Black preview (#17217)	2022-05-12 16:25:55 -04:00
tokenization	fix train_new_from_iterator in the case of byte-level tokenizers (#17549)	2022-06-08 15:30:41 +02:00
trainer	skip some ipex tests until it works with torch 1.12 (#17964)	2022-06-30 18:05:29 +02:00
utils	🐛 Properly raise RepoNotFoundError when not authenticated (#17651)	2022-06-10 15:41:53 +02:00
__init__.py	GPU text generation: mMoved the encoded_prompt to correct device	2020-01-06 15:11:12 +01:00
test_configuration_common.py	Prepare transformers for v0.8.0 huggingface-hub release (#17716)	2022-06-21 11:51:18 -04:00
test_feature_extraction_common.py	Compute min_resolution in prepare_image_inputs (#17915)	2022-06-29 10:30:20 +02:00
test_modeling_common.py	Offload fixes (#17810)	2022-06-22 12:23:07 -04:00
test_modeling_flax_common.py	Flax sharded (#17760)	2022-06-22 07:04:35 +02:00
test_modeling_tf_common.py	CLI: convert sharded PT models (#17959)	2022-06-30 16:51:03 +01:00
test_sequence_feature_extraction_common.py	Some tests misusing assertTrue for comparisons fix (#16771)	2022-04-19 14:44:08 +02:00
test_tokenization_common.py	Fix properties of unset special tokens in non verbose mode (#17797)	2022-06-23 14:40:13 +02:00