openmav 0.0.9

MAV

---

Introduction

MAV

Getting started

If uv is installed:

uv run --with openmav mav

or

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

Without uv:

1. Set up and activate a virtual environment

2. Install the package:

   pip install openmav
   

   or

   pip install git+https://github.com/attentionmech/mav
   

3. Run:

   mav --model gpt2 --prompt "hello mello"
   

4. or Import

   from openmav.mav import MAV
   
   MAV("gpt2", "Hello")
   

Locally from scratch

1. git clone https://github.com/attentionmech/mav
2. cd mav
3. Set up and activate a virtual environment
4. Install the package:

   pip install .
   

5. Run:

   mav --model gpt2 --prompt "hello mello"
   

You can replace gpt2 with other Hugging Face models for example:

• meta-llama/Llama-3.2-1B
• HuggingFaceTB/SmolLM-135M
• gpt2-medium
• gpt2-large

UI

Examples

running MAV with a training loop with a custom model (not pretrained one)

uv run examples/vis_train_loop.py

running MAV with custom panel selection and arrangement

uv run --with git+https://github.com/attentionmech/mav mav --model gpt2 --num-grid-rows 3 --selected-panels generated_text attention_entropy top_predictions --max-bar-length 20 --refresh-rate 0 --max-new-tokens 10000

Demos

• interactive mode
• limit chars
• sample with temperature
• running with custom model
• panel selection

Note: explore it using the command line help as well, since many sampling params are exposed.