Merge pull request #28 from Optum/pvs

fix torch v
Optum · Aug 2, 2024 · 5ac6c6d · 5ac6c6d
2 parents 61b151c + b85427e
commit 5ac6c6d
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