Add attention visualization code

#1

This adds attention visualization code and a few optimizations. Don't be alarmed about the models being deleted, they're downloaded at runtime now. My instances weren't building successfully until I deleted the model checkpoints from the repo.

drhead changed pull request status to open
cdnuts changed pull request status to merged

Sign up or log in to comment