Merge pull request #6 from Wheest/main

Added sentencepiece dependency
lucidrains · May 26, 2022 · 9519314 · 9519314
2 parents 94407a9 + 98a563f
commit 9519314
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -23,6 +23,7 @@
     'numpy',
     'pydantic',
     'resize-right',
+    'sentencepiece',
     'torch>=1.6',
     'torchvision',
     'transformers',