diff --git a/pyproject.toml b/pyproject.toml
index 38d80b5..fdaf49b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -55,7 +55,8 @@ dependencies = [
     "redis ==4.3.4",
     "prometheus-client ==0.15.0",
     "polib ==1.1.1",
-    "packaging ==23.1"
+    "packaging ==23.1",
+    "torch ==2.0.1"
 ]
 
 [project.scripts]