transformers/tests/deepspeed
Bartosz Szmelczynski 78a471ff71
Fix tapas scatter (#20149)
* First draft

* Remove scatter dependency

* Add require_torch

* update vectorized sum test, add clone call

* remove artifacts

* fix style

* fix style v2

* remove "scatter" mentions from the code base

* fix isort error

Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2022-11-14 01:04:26 -05:00
..
ds_config_zero2.json [Deepspeed] add support for bf16 mode (#14569) 2022-03-11 17:53:53 -08:00
ds_config_zero3.json [Deepspeed] add support for bf16 mode (#14569) 2022-03-11 17:53:53 -08:00
test_deepspeed.py Refine Bf16 test for deepspeed (#17734) 2022-06-16 11:27:58 -04:00
test_model_zoo.py Fix tapas scatter (#20149) 2022-11-14 01:04:26 -05:00
vit_feature_extractor.json missing file (#17164) 2022-05-10 10:19:50 -07:00