transformers/tests/deepspeed
Arthur b912f5ee43
use torch.testing.assertclose instead to get more details about error in cis (#35659)
* use torch.testing.assertclose instead to get more details about error in cis

* fix

* style

* test_all

* revert for I bert

* fixes and updates

* more image processing fixes

* more image processors

* fix mamba and co

* style

* less strick

* ok I won't be strict

* skip and be done

* up
2025-01-24 16:55:28 +01:00
..
ds_config_zero2.json [Deepspeed] add support for bf16 mode (#14569) 2022-03-11 17:53:53 -08:00
ds_config_zero3.json Update ds_config_zero3.json (#30829) 2024-05-15 10:02:31 -04:00
test_deepspeed.py use torch.testing.assertclose instead to get more details about error in cis (#35659) 2025-01-24 16:55:28 +01:00
test_model_zoo.py Pass datasets trust_remote_code (#31406) 2024-06-17 17:29:13 +01:00
vit_feature_extractor.json missing file (#17164) 2022-05-10 10:19:50 -07:00