fix tests

lucidrains · Jul 6, 2024 · 5c1bef2 · 5c1bef2
1 parent df52aa1
commit 5c1bef2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -34,6 +34,7 @@
     'numpy',
     'pytorch-custom-utils>=0.0.9',
     'rotary-embedding-torch>=0.6.4',
+    'sentencepiece',
     'taylor-series-linear-attention>=0.1.6',
     'torch>=2.1',
     'torch_geometric',