diff --git a/README.md b/README.md index 1771588..7f9be97 100644 --- a/README.md +++ b/README.md @@ -2,12 +2,12 @@ BertViz

- Visualize Attention in Transformer Models (BERT, GPT2, etc.) + Visualize Attention in NLP Models

Quick TourGetting Started • - Colab Tutorial • + Colab TutorialBlogPaperCitation @@ -27,7 +27,7 @@ For updates on BertViz and related projects, feel free to follow me on [Twitter] The *head view* visualizes attention for one or more attention heads in the same layer. It is based on the excellent [Tensor2Tensor visualization tool](https://github.com/tensorflow/tensor2tensor/tree/master/tensor2tensor/visualization) by [Llion Jones](https://medium.com/@llionj). -🕹 Try out the head view in the [Interactive Colab Tutorial](https://colab.research.google.com/drive/1MV7u8hdMgpwUd9nIlONQp-EBo8Fsj7CJ?usp=sharing) (all visualizations pre-loaded). +🕹 Try out the head view in the [Interactive Colab Tutorial](https://colab.research.google.com/drive/1hXIQ77A4TYS4y3UthWF-Ci7V7vVUoxmQ?usp=sharing) (all visualizations pre-loaded).

@@ -37,14 +37,14 @@ The *head view* visualizes attention for one or more attention heads in the same The *model view* shows a bird's-eye view of attention across all layers and heads. -🕹 Try out the model view in the [Interactive Colab Tutorial](https://colab.research.google.com/drive/1MV7u8hdMgpwUd9nIlONQp-EBo8Fsj7CJ?usp=sharing) (all visualizations pre-loaded). +🕹 Try out the model view in the [Interactive Colab Tutorial](https://colab.research.google.com/drive/1hXIQ77A4TYS4y3UthWF-Ci7V7vVUoxmQ?usp=sharing) (all visualizations pre-loaded). ![model view](images/model-view-noscroll.gif) ### Neuron View The *neuron view* visualizes individual neurons in the query and key vectors and shows how they are used to compute attention. -🕹 Try out the neuron view in the [Interactive Colab Tutorial](https://colab.research.google.com/drive/1MV7u8hdMgpwUd9nIlONQp-EBo8Fsj7CJ?usp=sharing) (all visualizations pre-loaded). +🕹 Try out the neuron view in the [Interactive Colab Tutorial](https://colab.research.google.com/drive/1hXIQ77A4TYS4y3UthWF-Ci7V7vVUoxmQ?usp=sharing) (all visualizations pre-loaded). ![neuron view](images/neuron-view-dark.gif) @@ -117,10 +117,10 @@ jupyter notebook ``` ## 🕹 Interactive Tutorial -Check out the [Interactive Colab Tutorial](https://colab.research.google.com/drive/1MV7u8hdMgpwUd9nIlONQp-EBo8Fsj7CJ?usp=sharing) +Check out the [Interactive Colab Tutorial](https://colab.research.google.com/drive/1hXIQ77A4TYS4y3UthWF-Ci7V7vVUoxmQ?usp=sharing) to learn more about BertViz and try out the tool. Note: all visualizations are pre-loaded, so there is no need to execute any cells. -[![Tutorial](images/tutorial-screenshots.jpg)](https://colab.research.google.com/drive/1MV7u8hdMgpwUd9nIlONQp-EBo8Fsj7CJ?usp=sharing) +[![Tutorial](images/tutorial-screenshots.jpg)](https://colab.research.google.com/drive/1hXIQ77A4TYS4y3UthWF-Ci7V7vVUoxmQ?usp=sharing) ## 📖 Documentation diff --git a/setup.py b/setup.py index a2b7711..e55de52 100644 --- a/setup.py +++ b/setup.py @@ -10,7 +10,7 @@ # This call to setup() does all the work setup( name="bertviz", - version="1.2.0", + version="1.3.0", description="Attention visualization tool for NLP Transformer models.", long_description=README, long_description_content_type="text/markdown",