mirror of
https://github.com/wassname/activation_store.git
synced 2026-06-27 16:44:07 +08:00
f0311c022fc4c4d12e7723383e1efa0f923a7a44
activation_store
Utility library to persistently store transformer activations on disk.
These activations can be quite large (layers x batch x sequence x hidden_size), so generating them to disk helps avoid out of memory errors.
Install using pip install git+https://github.com/wassname/activation_store.git.
Development
git clone https//github.com/wassname/activation_store.git
uv sync
see exampes in nbs folder.
Description
Languages
Jupyter Notebook
56.1%
Python
43.9%