diff --git a/requirements.txt b/requirements.txt
index 3a7c943..094d90f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1 +1,2 @@
 antlr4-python3-runtime>=4.11.0
+torch