Update setup.py

MarcusLoppe · Jun 18, 2024 · 5f6d670 · 5f6d670
1 parent c7a1cbd
commit 5f6d670
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -33,6 +33,7 @@
     'jaxtyping',
     'local-attention>=1.9.0',
     'numpy',
+    'matplotlib',
     'pytorch-custom-utils>=0.0.9',
     'taylor-series-linear-attention>=0.1.6',
     'torch>=2.1',