Spaces:
Sleeping
Sleeping
Update requirements.txt
Browse files- requirements.txt +2 -0
requirements.txt
CHANGED
|
@@ -2,6 +2,8 @@
|
|
| 2 |
torch~=2.2
|
| 3 |
torchvision==0.17.1
|
| 4 |
torchaudio===2.2.1
|
|
|
|
|
|
|
| 5 |
omegaconf==2.3.0
|
| 6 |
Pillow==10.1.0
|
| 7 |
einops==0.7.0
|
|
|
|
| 2 |
torch~=2.2
|
| 3 |
torchvision==0.17.1
|
| 4 |
torchaudio===2.2.1
|
| 5 |
+
cudatoolkit==11.8
|
| 6 |
+
pytorch-cuda==11.8
|
| 7 |
omegaconf==2.3.0
|
| 8 |
Pillow==10.1.0
|
| 9 |
einops==0.7.0
|