diff --git a/distilgpt2/requirements.txt b/distilgpt2/requirements.txt
index 973b15c..3244b33 100644
--- a/distilgpt2/requirements.txt
+++ b/distilgpt2/requirements.txt
@@ -1,2 +1,3 @@
 gradio==3.21.0
 transformers==4.27.1
+torch==2.0.0