transformers
1791ef8d - Adds `TRANSFORMERS_TEST_DEVICE` (#25506)

Commit
2 years ago
Adds `TRANSFORMERS_TEST_DEVICE` (#25506) * Adds `TRANSFORMERS_TEST_DEVICE` Mirrors the same API in the diffusers library. Useful in transformers too. * replace backend checking with trying `torch.device` * Adds better error message for unknown test devices * `make style` * adds documentation showing `TRANSFORMERS_TEST_DEVICE` usage.
Author
Parents
Loading