Correctly transfer tensors to device for transformers_api#22

victorjoos wants to merge 1 commit intomainTrackingLaboratory/tracklab:mainfrom

fix-gpu-transformers-apiTrackingLaboratory/tracklab:fix-gpu-transformers-apiCopy head branch name to clipboard