Update requirements.txt
Browse files- requirements.txt +2 -2
requirements.txt
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# Core ML and Computer Vision
|
| 2 |
-
|
| 3 |
-
|
| 4 |
|
| 5 |
git+https://github.com/huggingface/transformers.git
|
| 6 |
pillow
|
|
|
|
| 1 |
# Core ML and Computer Vision
|
| 2 |
+
torch # For newer Nvidia GPUs (e.g. RTX 5090) use uv pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
|
| 3 |
+
torchvision
|
| 4 |
|
| 5 |
git+https://github.com/huggingface/transformers.git
|
| 6 |
pillow
|