kartynnik commited on
Commit
947703f
1 Parent(s): 2d9add8

Remove another `.cuda()`

Browse files
Files changed (1) hide show
  1. hierspeechpp_speechsynthesizer.py +1 -1
hierspeechpp_speechsynthesizer.py CHANGED
@@ -629,7 +629,7 @@ class SynthesizerTrn(nn.Module):
629
 
630
  self.emb = torch.nn.Embedding(1, 256)
631
  torch.nn.init.normal_(self.emb.weight, 0.0, 256 ** -0.5)
632
- self.null = torch.LongTensor([0]).cuda()
633
  self.uncond_ratio = uncond_ratio
634
  self.cfg = cfg
635
  @torch.no_grad()
 
629
 
630
  self.emb = torch.nn.Embedding(1, 256)
631
  torch.nn.init.normal_(self.emb.weight, 0.0, 256 ** -0.5)
632
+ self.null = torch.LongTensor([0])
633
  self.uncond_ratio = uncond_ratio
634
  self.cfg = cfg
635
  @torch.no_grad()