Add attention visualization code

#1
by drhead - opened

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
Ready to merge
This branch is ready to get merged automatically.
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment