Skip to content

Commit 24ed452

Browse files
committed
Update README to remove neuron view single instance limitation.
1 parent 9c94349 commit 24ed452

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,7 @@ GPT-2 ([Notebook](notebooks/neuron_view_gpt2.ipynb),
203203
RoBERTa
204204
([Notebook](notebooks/neuron_view_roberta.ipynb))
205205

206-
Note that only one instance of the Neuron View may be displayed within a notebook. For full API, please refer to the [source](bertviz/neuron_view.py).
206+
For full API, please refer to the [source](bertviz/neuron_view.py).
207207

208208
### Encoder-decoder models (BART, T5, etc.)
209209

@@ -406,7 +406,7 @@ returned from Huggingface models). In some case, Tensorflow checkpoints may be l
406406
* When running on Colab, some of the visualizations will fail (runtime disconnection) when the input text is long. To mitigate this, you may wish to filter the layers displayed by setting the **`include_layers`** parameter, as described [above](#filtering-layers).
407407
* The *neuron view* only supports the custom BERT, GPT-2, and RoBERTa models included with the tool. This view needs access to the query and key vectors,
408408
which required modifying the model code (see `transformers_neuron_view` directory), which has only been done for these three models.
409-
Also, only one neuron view may be included per notebook.
409+
410410

411411
### Attention as "explanation"?
412412
* Visualizing attention weights illuminates one type of architecture within the model but does not

0 commit comments

Comments
 (0)