michellelychan commited on
Commit
3167341
·
1 Parent(s): 73621c9

hello world

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from scipy.io.wavfile import write as write_wav
5
  import os
6
 
7
 
8
- ### if you run on GPU use the following code: ###
9
  # device = "cuda" if torch.cuda.is_available() else "cpu"
10
  # model = BarkModel.from_pretrained("suno/bark-small", torch_dtype=torch.float16).to(device)
11
  # model.enable_cpu_offload()
 
5
  import os
6
 
7
 
8
+ ## if you run on GPU use the following code: ###
9
  # device = "cuda" if torch.cuda.is_available() else "cpu"
10
  # model = BarkModel.from_pretrained("suno/bark-small", torch_dtype=torch.float16).to(device)
11
  # model.enable_cpu_offload()