Spaces:
Sleeping
Sleeping
fix: Add torch CPU to requirements
Browse files- requirements.txt +3 -0
requirements.txt
CHANGED
|
@@ -20,6 +20,9 @@ streamlit==1.35.0
|
|
| 20 |
# torchvision==0.18.0
|
| 21 |
# torchaudio==2.3.0
|
| 22 |
|
|
|
|
|
|
|
|
|
|
| 23 |
# Transformers / HuggingFace
|
| 24 |
transformers==4.43.0
|
| 25 |
huggingface_hub==0.23.4
|
|
|
|
| 20 |
# torchvision==0.18.0
|
| 21 |
# torchaudio==2.3.0
|
| 22 |
|
| 23 |
+
--extra-index-url https://download.pytorch.org/whl/cpu
|
| 24 |
+
torch
|
| 25 |
+
|
| 26 |
# Transformers / HuggingFace
|
| 27 |
transformers==4.43.0
|
| 28 |
huggingface_hub==0.23.4
|