StormblessedKal commited on
Commit
035ae93
1 Parent(s): 1bbd03b

enhance on cpu

Browse files
Files changed (1) hide show
  1. builder/fetch_models.py +1 -1
builder/fetch_models.py CHANGED
@@ -10,4 +10,4 @@ import torchaudio
10
  dwav, sr = torchaudio.load('openai_source_output.mp3')
11
  dwav = dwav.mean(dim=0)
12
 
13
- wav1, new_sr = enhance(dwav, sr, 'cuda', nfe=32, solver='midpoint', lambd=0.9, tau=0.5)
 
10
  dwav, sr = torchaudio.load('openai_source_output.mp3')
11
  dwav = dwav.mean(dim=0)
12
 
13
+ wav1, new_sr = enhance(dwav, sr, 'cpu', nfe=32, solver='midpoint', lambd=0.9, tau=0.5)