Tool to help Visualize flex attention mod_maps.
APACHE-2.0 License
A Streamlit app to visualize the attention mask of the Flex-Attention model.
Install the requirements
$ pip install -r requirements.txt
Run the app
$ streamlit run streamlit_app.py