diff --git a/requirements.txt b/requirements.txt
index 126764b..7de91f9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,3 +2,4 @@ protobuf>=3.19.5,<3.20.1
 transformers>=4.26.1
 icetk
 cpm_kernels
+torch>=1.10