Update requirements.txt
Browse files- requirements.txt +3 -1
requirements.txt
CHANGED
|
@@ -1,3 +1,5 @@
|
|
|
|
|
|
|
|
| 1 |
transformers==4.42.3
|
| 2 |
timm==1.0.7
|
| 3 |
einops==0.8.0
|
|
@@ -18,5 +20,5 @@ sacrebleu
|
|
| 18 |
scikit-learn
|
| 19 |
gdown
|
| 20 |
omegaconf==2.3.0
|
| 21 |
-
mmcv
|
| 22 |
mmdet==3.3.0
|
|
|
|
| 1 |
+
torch==2.1.0
|
| 2 |
+
torchvision==0.16.1
|
| 3 |
transformers==4.42.3
|
| 4 |
timm==1.0.7
|
| 5 |
einops==0.8.0
|
|
|
|
| 20 |
scikit-learn
|
| 21 |
gdown
|
| 22 |
omegaconf==2.3.0
|
| 23 |
+
mmcv==2.0.1
|
| 24 |
mmdet==3.3.0
|