Flex-Attention Mask Visualization A Streamlit app to visualize the attention mask of the Flex-Attention model. How to run it on your own machine Install the requirements $ pip install -r requirements.txt Run the app $ streamlit run streamlit_app.py