a module to visualise attention layer activations from transformer based models from huggingface
Project description
attention-visualiser
| Status | Coverage |
|---|---|
 |
 |
A module to visualise attention layer activations from transformer based models from huggingface
installation
pip install attention-visualiser
usage
from attention_visualiser import AttentionVisualiser
from transformers import AutoModel, AutoTokenizer
# visualising activations from gpt
model_name = "openai-community/openai-gpt"
model = AutoModel.from_pretrained(model_name)
model.eval()
tokenizer = AutoTokenizer.from_pretrained(model_name)
text = "Look on my Works, ye Mighty, and despair!"
encoded_inputs = tokenizer.encode_plus(text, truncation=True, return_tensors="pt")
visualiser = AttentionVisualiser(model, tokenizer)
# visualise from the first attn layer
visualiser.visualise_attn_layer(0, encoded_inputs)
local dev
# env setup
uv sync
source .venv/bin/activate
# tests
uv run pytest
# tests with coverage
uv run pytest --cov --cov-report=xml
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
attention_visualiser-1.0.3.tar.gz
(110.7 kB
view details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file attention_visualiser-1.0.3.tar.gz.
File metadata
- Download URL: attention_visualiser-1.0.3.tar.gz
- Upload date:
- Size: 110.7 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.11.6 {"installer":{"name":"uv","version":"0.11.6","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"macOS","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
0e57f22fc6ce9456ef62fd44c99b4ff3fae5e0e78f505a6c4c2153c2548ec478
|
|
| MD5 |
ae564bc5c068258bdb4b353f049b5d12
|
|
| BLAKE2b-256 |
036cd55800e80b6b25e919a6da19b0dad9c187b6eb19a41448002e955c455e60
|
File details
Details for the file attention_visualiser-1.0.3-py3-none-any.whl.
File metadata
- Download URL: attention_visualiser-1.0.3-py3-none-any.whl
- Upload date:
- Size: 3.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.11.6 {"installer":{"name":"uv","version":"0.11.6","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"macOS","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
55a832dab779c9b3b23b5c77ba0fd2a8d2a68a9d5c3ee30fe004fc4356417b2d
|
|
| MD5 |
b8e16ff36e54d15e03e8478d4f9df28d
|
|
| BLAKE2b-256 |
3a452319fc18823a818eedc7baa957cffb8d665d8327aa31fb85738dbc7c48b6
|
