Redo LeViT attention bias caching in a way that works with both torchscript and DataParallel
Redo LeViT attention bias caching in a way that works with both torchscript and DataParallel

rwightman/pytorch-image-models,rwightman/pytorch-image-models