diff --git a/test/apis/pytorch/text-generator/requirements.txt b/test/apis/pytorch/text-generator/requirements.txt
index 1447500abe..4d131781b1 100644
--- a/test/apis/pytorch/text-generator/requirements.txt
+++ b/test/apis/pytorch/text-generator/requirements.txt
@@ -1,2 +1,2 @@
-torch
+torch==1.7.*
 transformers==3.0.*