fix torch v

Optum · Aug 2, 2024 · b85427e · b85427e
1 parent 61b151c
commit b85427e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/models/llama/requirements.txt b/models/llama/requirements.txt
@@ -4,4 +4,4 @@ numpy
 peft
 transformers==4.35.1
 sentencepiece==0.1.99
-torch==1.13.1
+torch
diff --git a/models/long_roberta/requirements.txt b/models/long_roberta/requirements.txt
@@ -7,7 +7,7 @@ pandas
 datasets
 protobuf
 matplotlib
-torch==1.13.1
+torch
 accelerate
 pyarrow
-pytest
+pytest