Blaizzy/model-activation-visualiser

model activation visualiser

if uv is installed:

uv run --with git+https://github.com/attentionmech/mav mav --model gpt2 --prompt "hello mello"

otherwise:

0. venv setup and activate
1. pip install git+https://github.com/attentionmech/mav
2. mav --model gpt2 --prompt "hello mello"

(can replace with other hf models like meta-llama/Llama-3.2-1B or HuggingFaceTB/SmolLM-135M)