transformers/tests
Younes Belkada 9b25c164bd
[core / Quantization] Fix for 8bit serialization tests (#27234)
* fix for 8bit serialization

* added regression tests.

* fixup
2023-11-02 12:03:51 +01:00
..
benchmark
bettertransformer
deepspeed
extended Device agnostic trainer testing (#27131) 2023-10-30 18:16:40 +00:00
fixtures
fsdp device agnostic fsdp testing (#27120) 2023-11-01 07:17:06 +01:00
generation [WhisperForCausalLM] Add WhisperForCausalLM for speculative decoding (#27195) 2023-11-01 16:01:53 +01:00
models [Whisper, Bart, MBart] Add Flash Attention 2 (#27203) 2023-11-01 21:03:01 +01:00
optimization
peft_integration
pipelines Backward compatibility fix for the Conversation class (#27176) 2023-10-31 15:12:06 +00:00
quantization [core / Quantization] Fix for 8bit serialization tests (#27234) 2023-11-02 12:03:51 +01:00
repo_utils
sagemaker
tokenization
tools
trainer Safetensors serialization by default (#27064) 2023-10-31 19:16:49 +01:00
utils Safetensors serialization by default (#27064) 2023-10-31 19:16:49 +01:00
__init__.py
test_backbone_common.py
test_configuration_common.py
test_configuration_utils.py
test_feature_extraction_common.py
test_feature_extraction_utils.py
test_image_processing_common.py
test_image_processing_utils.py
test_image_transforms.py
test_modeling_common.py [Whisper, Bart, MBart] Add Flash Attention 2 (#27203) 2023-11-01 21:03:01 +01:00
test_modeling_flax_common.py
test_modeling_flax_utils.py Safetensors serialization by default (#27064) 2023-10-31 19:16:49 +01:00
test_modeling_tf_common.py
test_modeling_tf_utils.py Safetensors serialization by default (#27064) 2023-10-31 19:16:49 +01:00
test_modeling_utils.py Safetensors serialization by default (#27064) 2023-10-31 19:16:49 +01:00
test_pipeline_mixin.py Shorten the conversation tests for speed + fixing position overflows (#26960) 2023-10-31 14:20:04 +00:00
test_sequence_feature_extraction_common.py
test_tokenization_common.py
test_tokenization_utils.py