Fix torch pin to 2.1.2 (xformers==0.0.23.post1 requires torch==2.1.2 exactly, conflicting with the 2.1.0 I pinned previously)
Browse files- requirements.txt +3 -3
requirements.txt
CHANGED
|
@@ -1,8 +1,8 @@
|
|
| 1 |
spaces
|
| 2 |
setuptools<82
|
| 3 |
-
torch==2.1.
|
| 4 |
-
torchvision==0.16.
|
| 5 |
-
torchaudio==2.1.
|
| 6 |
huggingface_hub==0.25.2
|
| 7 |
absl-py==2.1.0
|
| 8 |
accelerate==1.1.1
|
|
|
|
| 1 |
spaces
|
| 2 |
setuptools<82
|
| 3 |
+
torch==2.1.2 --index-url https://download.pytorch.org/whl/cu121
|
| 4 |
+
torchvision==0.16.2 --index-url https://download.pytorch.org/whl/cu121
|
| 5 |
+
torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu121
|
| 6 |
huggingface_hub==0.25.2
|
| 7 |
absl-py==2.1.0
|
| 8 |
accelerate==1.1.1
|