transformers/tests
Arthur c51dc4f927
[torch] remove deprecated uint8 in favor of bool (#21384)
* uint8 -> bool

* fix copies

* style

* update test modeling commen when checking attention buffers

* style

* use logical not on random mask instead of subtraction with 1

* remove torch uint8

* quality

* remove modified modeling utils

* Update based on review

Co-authored-by: sgugger <sylvain.gugger@gmail.com>

---------

Co-authored-by: sgugger <sylvain.gugger@gmail.com>
2023-02-27 11:46:02 +01:00
..
benchmark [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
deepspeed [deepspeed tests] fix issues introduced by #21700 (#21769) 2023-02-23 13:22:25 -08:00
extended Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
fixtures [WIP] add SpeechT5 model (#18922) 2023-02-03 12:43:46 -05:00
generation Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
mixed_int8 [bnb] fix bnb decoders bug (#21688) 2023-02-20 12:21:58 +00:00
models [SpeechT5] Fix HiFiGAN tests (#21788) 2023-02-24 16:55:38 +01:00
onnx Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
optimization Add inverse sqrt learning rate scheduler (#21495) 2023-02-07 15:00:50 -05:00
pipelines [Pipeline] Add zero shot audio classificatoin pipeline (#21600) 2023-02-27 11:43:44 +01:00
repo_utils Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
sagemaker Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
tokenization Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
trainer Skip test_log_level for now 2023-02-23 12:11:20 +01:00
utils Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01: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 Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_feature_extraction_common.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_image_processing_common.py Make ImageProcessorMixin compatible with subfolder kwarg (#21725) 2023-02-23 09:28:18 +01:00
test_image_transforms.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_modeling_common.py [torch] remove deprecated uint8 in favor of bool (#21384) 2023-02-27 11:46:02 +01:00
test_modeling_flax_common.py [Flax] adding support for batch norm layers (#21581) 2023-02-24 08:47:33 +01:00
test_modeling_tf_common.py Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
test_sequence_feature_extraction_common.py Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
test_tokenization_common.py Cleanup quality (#21493) 2023-02-07 12:27:31 -05:00