Squeezeformer
Squeezeformer copied to clipboard
Masking Attention
How can I pass mask argument to use local attention instead of global attention in MHA module? Thank Sir!