Dionyssos commited on
Commit
989ad3f
·
1 Parent(s): ac6157a

HACK pulse noise at End of sentence

Browse files
Files changed (2) hide show
  1. models.py +1 -1
  2. msinference.py +1 -2
models.py CHANGED
@@ -299,7 +299,7 @@ class AdaIN1d(nn.Module):
299
  h = h.view(h.size(0), h.size(1), 1)
300
  gamma, beta = torch.chunk(h, chunks=2, dim=1)
301
  # affine (1 + lin(x)) * inst(x) + lin(x) is this a skip connection where the weight is a lin of itself
302
- return (1 + gamma) * self.norm(x) + beta
303
 
304
  class UpSample1d(nn.Module):
305
  def __init__(self, layer_type):
 
299
  h = h.view(h.size(0), h.size(1), 1)
300
  gamma, beta = torch.chunk(h, chunks=2, dim=1)
301
  # affine (1 + lin(x)) * inst(x) + lin(x) is this a skip connection where the weight is a lin of itself
302
+ return (1 + gamma) * self.norm(x) + beta # norm(x) = PLBERT has norm / beta&gamma = style has no norm()
303
 
304
  class UpSample1d(nn.Module):
305
  def __init__(self, layer_type):
msinference.py CHANGED
@@ -240,8 +240,7 @@ def inference(text,
240
  x = model.decoder(asr,
241
  F0_pred, N_pred, ref.squeeze().unsqueeze(0))
242
 
243
-
244
- x = x.squeeze().cpu().numpy()[..., :-74] # weird pulse at the end of the model
245
 
246
  x /= np.abs(x).max() + 1e-7
247
 
 
240
  x = model.decoder(asr,
241
  F0_pred, N_pred, ref.squeeze().unsqueeze(0))
242
 
243
+ x = x.squeeze().cpu().numpy()[..., :-2504] # weird pulse at the end of sentences
 
244
 
245
  x /= np.abs(x).max() + 1e-7
246