transformers/tests/deepspeed
Joao Gante beb2a09687
DeepSpeed: hardcode `torch.arange` dtype on `float` usage to avoid incorrect initialization (#28760)
2024-01-31 14:39:07 +00: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 DeepSpeed: hardcode `torch.arange` dtype on `float` usage to avoid incorrect initialization (#28760) 2024-01-31 14:39:07 +00:00
test_model_zoo.py Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
vit_feature_extractor.json missing file (#17164) 2022-05-10 10:19:50 -07:00