Spaces:
Runtime error
Runtime error
Update requirements.txt
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
|
@@ -3,6 +3,7 @@ torchvision==0.16.0
|
|
| 3 |
torchaudio==2.1.0
|
| 4 |
pytorch-lightning==2.1.2
|
| 5 |
einops
|
|
|
|
| 6 |
omegaconf
|
| 7 |
deepspeed
|
| 8 |
torchmetrics
|
|
|
|
| 3 |
torchaudio==2.1.0
|
| 4 |
pytorch-lightning==2.1.2
|
| 5 |
einops
|
| 6 |
+
onnxruntime-gpu
|
| 7 |
omegaconf
|
| 8 |
deepspeed
|
| 9 |
torchmetrics
|