diff --git a/requirements.txt b/requirements.txt
index 83ca249..690c99e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,3 +3,4 @@ transformers
 sentencepiece
 protobuf
 towhee
+torch