Update requirements.txt
Browse files- requirements.txt +5 -3
requirements.txt
CHANGED
@@ -1,8 +1,10 @@
|
|
1 |
# core UI / runtime
|
2 |
gradio
|
3 |
-
|
4 |
-
torch
|
5 |
-
|
|
|
|
|
6 |
transformers>=4.40.0
|
7 |
sentence-transformers>=2.6.0
|
8 |
accelerate>=0.28.0
|
|
|
1 |
# core UI / runtime
|
2 |
gradio
|
3 |
+
vllm‑cu121>=0.7.4 # or the CUDA version that matches the HF Space image
|
4 |
+
torch==2.2.*+cu121 # pre‑built CUDA wheel
|
5 |
+
txagent>=0.1.2
|
6 |
+
tooluniverse>=0.4.1
|
7 |
+
|
8 |
transformers>=4.40.0
|
9 |
sentence-transformers>=2.6.0
|
10 |
accelerate>=0.28.0
|