Skip to content

Commit d41dca4

Browse files
committed
Minor README change.
1 parent 155ebcd commit d41dca4

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

README.md

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,10 @@
11
# BertViz
22

3-
BertViz is a tool for visualizing attention in the Transformer model, supporting most models from the [transformers](https://github.com/huggingface/transformers) library (BERT, GPT-2, XLNet, RoBERTa, XLM, CTRL, etc.). It extends the [Tensor2Tensor visualization tool](https://github.com/tensorflow/tensor2tensor/tree/master/tensor2tensor/visualization) by [Llion Jones](https://medium.com/@llionj) and the [transformers](https://github.com/huggingface/transformers) library from [HuggingFace](https://github.com/huggingface).
3+
BertViz is a tool for visualizing attention in the Transformer model, supporting most models from the
4+
[transformers](https://github.com/huggingface/transformers) library (BERT, GPT-2, XLNet, RoBERTa, XLM, CTRL, MarianMT,
5+
etc.). It extends the
6+
[Tensor2Tensor visualization tool](https://github.com/tensorflow/tensor2tensor/tree/master/tensor2tensor/visualization)
7+
by [Llion Jones](https://medium.com/@llionj) and the [transformers](https://github.com/huggingface/transformers) library from [HuggingFace](https://github.com/huggingface).
48

59
## Resources
610

@@ -120,10 +124,6 @@ git clone --depth 1 git@github.com:jessevig/bertviz.git
120124
cd bertviz/notebooks
121125
jupyter notebook
122126
```
123-
124-
Specifying `--depth 1`, as shown above, excludes older revisions thus providing a faster download. Note that the sample notebooks do not cover
125-
all Huggingface models, but the code should be similar for those not included.
126-
127127

128128
## Detailed Instructions
129129

0 commit comments

Comments
 (0)