Spaces:
Running
on
Zero
Running
on
Zero
torch cuda
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
Pillow
|
2 |
numpy
|
3 |
torch
|
@@ -5,4 +6,4 @@ transformers>=4.36
|
|
5 |
gradio==4.43.0
|
6 |
pyfluidsynth
|
7 |
tqdm
|
8 |
-
huggingface_hub
|
|
|
1 |
+
--extra-index-url https://download.pytorch.org/whl/cu124
|
2 |
Pillow
|
3 |
numpy
|
4 |
torch
|
|
|
6 |
gradio==4.43.0
|
7 |
pyfluidsynth
|
8 |
tqdm
|
9 |
+
huggingface_hub
|