From 5c1bef252102771b39e2549049fd29e2d87c50c2 Mon Sep 17 00:00:00 2001
From: lucidrains <lucidrains@gmail.com>
Date: Fri, 5 Jul 2024 17:27:31 -0700
Subject: [PATCH] fix tests

---
 setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.py b/setup.py
index 89c69eb..094a202 100644
--- 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',