fix

hpcaitech · Jul 21, 2023 · 5342fff · 5342fff
1 parent b4359af
commit 5342fff
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/requirements/requirements-test.txt b/requirements/requirements-test.txt
@@ -16,3 +16,4 @@ triton==2.0.0.dev20221202
 git+https://github.com/HazyResearch/flash-attention.git@c422fee3776eb3ea24e011ef641fd5fbeb212623#egg=flash_attn
 requests==2.27.1 # downgrade to avoid huggingface error https://github.com/huggingface/transformers/issues/17611
 SentencePiece
+datasets