Pin torch==2.8.0 for ZeroGPU
Browse files- requirements.txt +4 -1
requirements.txt
CHANGED
|
@@ -1,8 +1,11 @@
|
|
| 1 |
# Space branch: gradio itself comes from `sdk_version` in the README front
|
| 2 |
# matter, so it is deliberately not listed here.
|
|
|
|
|
|
|
|
|
|
| 3 |
duckdb>=1.0.0
|
| 4 |
numpy
|
| 5 |
-
torch==2.
|
| 6 |
transformers==5.13.0
|
| 7 |
sentence-transformers==5.6.0
|
| 8 |
accelerate
|
|
|
|
| 1 |
# Space branch: gradio itself comes from `sdk_version` in the README front
|
| 2 |
# matter, so it is deliberately not listed here.
|
| 3 |
+
# torch is pinned to a ZeroGPU-supported version; for a LOCAL install ignore
|
| 4 |
+
# this file's torch line and follow CLAUDE.md (the RTX A6000 driver on the
|
| 5 |
+
# dev box needs a cu121 build: torch==2.5.1).
|
| 6 |
duckdb>=1.0.0
|
| 7 |
numpy
|
| 8 |
+
torch==2.8.0
|
| 9 |
transformers==5.13.0
|
| 10 |
sentence-transformers==5.6.0
|
| 11 |
accelerate
|