Spaces:
Sleeping
Sleeping
Update requirements.txt
Browse files- requirements.txt +2 -2
requirements.txt
CHANGED
@@ -11,9 +11,9 @@ email-validator
|
|
11 |
|
12 |
# deep learning
|
13 |
# Install with conda
|
14 |
-
torch
|
15 |
torch_geometric
|
16 |
-
lightning
|
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
|