libokj commited on
Commit
6534610
·
1 Parent(s): 3d51369

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -11,9 +11,9 @@ email-validator
11
 
12
  # deep learning
13
  # Install with conda
14
- torch >= 2.2
15
  torch_geometric
16
- lightning >= 2.2
17
  torchmetrics
18
  transformers
19
  scikit-learn
 
11
 
12
  # deep learning
13
  # Install with conda
14
+ torch==2.2.1
15
  torch_geometric
16
+ lightning>=2.2
17
  torchmetrics
18
  transformers
19
  scikit-learn