Spaces:
Sleeping
Sleeping
Update requirements.txt
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
|
@@ -4,6 +4,7 @@ torch==2.2.2
|
|
| 4 |
torchvision==0.17.2
|
| 5 |
torchaudio==2.2.2
|
| 6 |
onnxruntime-gpu==1.18.0
|
|
|
|
| 7 |
transformers==4.33.0
|
| 8 |
torchgeometry
|
| 9 |
pyyaml==6.0.1
|
|
|
|
| 4 |
torchvision==0.17.2
|
| 5 |
torchaudio==2.2.2
|
| 6 |
onnxruntime-gpu==1.18.0
|
| 7 |
+
tensorrt
|
| 8 |
transformers==4.33.0
|
| 9 |
torchgeometry
|
| 10 |
pyyaml==6.0.1
|