Update torch to 2.4.0 and torchvision to 0.19.0
Browse files- requirements.txt +4 -26
requirements.txt
CHANGED
|
@@ -1,32 +1,10 @@
|
|
| 1 |
-
# fastapi
|
| 2 |
-
# uvicorn[standard]
|
| 3 |
-
|
| 4 |
-
# # vision / model stuff
|
| 5 |
-
# numpy<2.0
|
| 6 |
-
# torch==2.2.2+cpu
|
| 7 |
-
# torchvision==0.17.2+cpu
|
| 8 |
-
# --extra-index-url https://download.pytorch.org/whl/cpu
|
| 9 |
-
|
| 10 |
-
# pillow
|
| 11 |
-
# matplotlib
|
| 12 |
-
|
| 13 |
-
# # Hugging Face bits
|
| 14 |
-
# transformers>=4.42.0
|
| 15 |
-
# huggingface-hub
|
| 16 |
-
# peft
|
| 17 |
-
|
| 18 |
-
# python-multipart
|
| 19 |
-
# ultralytics
|
| 20 |
-
# opencv-python-headless
|
| 21 |
-
# pyyamlscikit-learn
|
| 22 |
-
# matplotlib
|
| 23 |
fastapi
|
| 24 |
uvicorn[standard]
|
| 25 |
|
| 26 |
-
# Core
|
| 27 |
numpy<2.0
|
| 28 |
-
torch==2.
|
| 29 |
-
torchvision==0.
|
| 30 |
--extra-index-url https://download.pytorch.org/whl/cpu
|
| 31 |
|
| 32 |
# Image Processing
|
|
@@ -35,7 +13,7 @@ matplotlib
|
|
| 35 |
scikit-learn
|
| 36 |
pyyaml
|
| 37 |
|
| 38 |
-
# Computer Vision
|
| 39 |
opencv-python-headless
|
| 40 |
ultralytics
|
| 41 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
fastapi
|
| 2 |
uvicorn[standard]
|
| 3 |
|
| 4 |
+
# Core - Updated to 2.4.0 to satisfy Transformers 4.42+ requirements
|
| 5 |
numpy<2.0
|
| 6 |
+
torch==2.4.0+cpu
|
| 7 |
+
torchvision==0.19.0+cpu
|
| 8 |
--extra-index-url https://download.pytorch.org/whl/cpu
|
| 9 |
|
| 10 |
# Image Processing
|
|
|
|
| 13 |
scikit-learn
|
| 14 |
pyyaml
|
| 15 |
|
| 16 |
+
# Computer Vision
|
| 17 |
opencv-python-headless
|
| 18 |
ultralytics
|
| 19 |
|