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

Update spkr.py

Browse files
Files changed (1) hide show
  1. spkr.py +2 -1
spkr.py CHANGED
@@ -16,7 +16,8 @@ class SpeakerEmbedding:
16
 
17
  @torch.no_grad()
18
  def __call__(self, wav, sr):
19
- wav = torch.tensor(wav, device=self.device)
 
20
  if sr != self.sample_rate:
21
  wav = resample(wav, sr, self.sample_rate)
22
  sr = self.sample_rate
 
16
 
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