diff --git a/requirements.txt b/requirements.txt
index 26a2a9d..a25b30f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,9 +8,10 @@ seqeval
 rouge-score 
 openpyxl>=3.0.7,<4.0 
 xlrd>=2.0.1,<3.0
+torch>=2.0
 
 ipykernel 
 nbdev 
 gputil 
 rjieba 
-sacremoses 
\ No newline at end of file
+sacremoses