skytnt commited on
Commit
fc457c0
1 Parent(s): 86f7f0a

torch cuda

Browse files
Files changed (1) hide show
  1. 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