diff --git a/VERSION b/VERSION
index 26ca594..4cda8f1 100644
--- a/VERSION
+++ b/VERSION
@@ -1 +1 @@
-1.5.1
+1.5.2
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]