guangyil commited on
Commit
c3c4e12
·
verified ·
1 Parent(s): 05700c4

Update spkr.py

Browse files
Files changed (1) hide show
  1. spkr.py +1 -1
spkr.py CHANGED
@@ -17,7 +17,7 @@ class SpeakerEmbedding:
17
  @torch.no_grad()
18
  def __call__(self, wav, sr):
19
  wav = torch.tensor(wav)
20
- wav = wav.to(self.device)
21
  if sr != self.sample_rate:
22
  wav = resample(wav, sr, self.sample_rate)
23
  sr = self.sample_rate
 
17
  @torch.no_grad()
18
  def __call__(self, wav, sr):
19
  wav = torch.tensor(wav)
20
+ wav = wav.to('cuda')
21
  if sr != self.sample_rate:
22
  wav = resample(wav, sr, self.sample_rate)
23
  sr = self.sample_rate