diff --git a/phi2/requirements.txt b/phi2/requirements.txt
index 3e141ec3..2251ee12 100644
--- a/phi2/requirements.txt
+++ b/phi2/requirements.txt
@@ -2,3 +2,4 @@ einops
 mlx
 numpy
 transformers
+torch