pytorch-labs / attention-gym

Helpful tools and examples for working with flex-attention

BSD 3-Clause "New" or "Revised" License

490 stars 24 forks source link

Fix a typo #13

Closed sangyeon-k closed 3 months ago

sangyeon-k commented 3 months ago

Summary

Fixes a typo in attn_gym/masks/sliding_window.py