riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity
5a2b77a6c1
transformers/tests
History
Gerald Cuder 5a2b77a6c1
Fix error in mixed precision training of TFCvtModel (#22267) 
* Make sure CVT can be trained using mixed precision

* Add test for keras-fit with mixed-precision

* Update tests/models/cvt/test_modeling_tf_cvt.py

Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>

---------

Co-authored-by: gcuder <Gerald.Cuder@iacapps.com>
Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>
2023-03-21 12:12:57 +00:00
..
benchmark [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
deepspeed [deepspeed] offload + non-cpuadam optimizer exception (#22043) 2023-03-09 08:12:57 -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 🔥py38 + torch 2 🔥🔥🔥🚀 (#22204) 2023-03-16 22:59:23 +01:00
mixed_int8 [bnb] fix bnb decoders bug (#21688) 2023-02-20 12:21:58 +00:00
models Fix error in mixed precision training of TFCvtModel (#22267) 2023-03-21 12:12:57 +00:00
onnx 🔥py38 + torch 2 🔥🔥🔥🚀 (#22204) 2023-03-16 22:59:23 +01:00
optimization Make schedulers picklable by making lr_lambda fns global (#21768) 2023-03-02 12:08:43 -05:00
pipelines 🔥py38 + torch 2 🔥🔥🔥🚀 (#22204) 2023-03-16 22:59:23 +01:00
repo_utils Add an utility file to get information from test files (#21856) 2023-03-01 17:53:29 +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 🔥py38 + torch 2 🔥🔥🔥🚀 (#22204) 2023-03-16 22:59:23 +01:00
utils Add Blip and Blip2 for pipeline tests (#21904) 2023-03-02 18:20:34 +01:00
__init__.py
test_configuration_common.py Remove set_access_token usage + fail tests if FutureWarning (#22051) 2023-03-09 09:23:48 -05:00
test_feature_extraction_common.py Remove set_access_token usage + fail tests if FutureWarning (#22051) 2023-03-09 09:23:48 -05:00
test_image_processing_common.py Remove set_access_token usage + fail tests if FutureWarning (#22051) 2023-03-09 09:23:48 -05:00
test_image_transforms.py to_pil - don't rescale if int and in range 0-255 (#22158) 2023-03-14 15:43:44 +00:00
test_modeling_common.py [Safetensors] Add explicit flag to from pretrained (#22083) 2023-03-13 21:39:06 +01:00
test_modeling_flax_common.py Remove set_access_token usage + fail tests if FutureWarning (#22051) 2023-03-09 09:23:48 -05:00
test_modeling_tf_common.py Remove set_access_token usage + fail tests if FutureWarning (#22051) 2023-03-09 09:23:48 -05:00
test_pipeline_mixin.py Move is_pipeline_test_to_skip to specific model test classes (#21999) 2023-03-14 10:03:02 +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 Remove set_access_token usage + fail tests if FutureWarning (#22051) 2023-03-09 09:23:48 -05:00