sleepychord / image-local-attention Goto Github PK
View Code? Open in Web Editor NEWThis project forked from zzd1992/image-local-attention
A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.