Alex McKinney 9d7afd2536 Replaces calls to .cuda with .to(torch_device) in tests (#25571) ... * Replaces calls to `.cuda` with `.to(torch_device)` in tests `torch.Tensor.cuda()` is a pre-0.4 solution to changing a tensor's device. It is recommended to prefer `.to(...)` for greater flexibility and error handling. Furthermore, this makes it more consistent with other tests (that tend to use `.to(torch_device)`) and ensures the correct device backend is used (if `torch_device` is neither `cpu` or `cuda`). * addressing review comments * more formatting changes in Bloom test * `make style` * Update tests/models/bloom/test_modeling_bloom.py Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com> * fixes style failures --------- Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>		2023-08-18 12:40:40 +02:00
..
__init__.py	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
test_modeling_flax_xglm.py	CI with num_hidden_layers=2 🚀🚀🚀 (#25266)	2023-08-02 20:22:36 +02:00
test_modeling_tf_xglm.py	Speed up TF tests by reducing hidden layer counts (#24595)	2023-06-30 16:30:33 +01:00
test_modeling_xglm.py	Replaces calls to .cuda with .to(torch_device) in tests (#25571)	2023-08-18 12:40:40 +02:00
test_tokenization_xglm.py	Update quality tooling for formatting (#21480)	2023-02-06 18:10:56 -05:00