StormblessedKal commited on
Commit
96979b3
1 Parent(s): 49535ec

change device for whisper

Browse files
Files changed (1) hide show
  1. builder/fetch_models.py +2 -2
builder/fetch_models.py CHANGED
@@ -1,7 +1,7 @@
1
  import se_extractor as se
2
 
3
  _ = se.generate_voice_segments('openai_source_output.mp3',vad=False)
4
- _ = se.generate_voice_segments('openai_source_output.mp3',vad=True)
5
 
6
  from resemble_enhance.enhancer.inference import denoise, enhance
7
  import torchaudio
@@ -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, 'cpu', nfe=32, solver='midpoint', lambd=0.9, tau=0.5)
 
1
  import se_extractor as se
2
 
3
  _ = se.generate_voice_segments('openai_source_output.mp3',vad=False)
4
+ _ = se.load_model()
5
 
6
  from resemble_enhance.enhancer.inference import denoise, enhance
7
  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)