We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
TRANSFORMERS_TEST_BACKEND
.cuda
.to(torch_device)
TRANSFORMERS_TEST_DEVICE