Temporarily lock torch to 1.7 for pytorch text-generator gpu tests (#1943)

vishalbollu · web-flow · commit b0e7412c51ef · 2021-03-08T18:36:23.000-05:00
diff --git 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.*
