Skip to content

Commit fae4c35

Browse files
authored
fix: skip shallow diffusion
1 parent fd78746 commit fae4c35

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

usr/diff/shallow_diffusion_tts.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -234,7 +234,7 @@ def forward(self, txt_tokens, mel2ph=None, spk_embed=None,
234234
ref_mels=None, f0=None, uv=None, energy=None, infer=False, **kwargs):
235235
b, *_, device = *txt_tokens.shape, txt_tokens.device
236236
ret = self.fs2(txt_tokens, mel2ph, spk_embed, ref_mels, f0, uv, energy,
237-
skip_decoder=(not infer), infer=infer, **kwargs)
237+
skip_decoder=True, infer=infer, **kwargs)
238238
cond = ret['decoder_inp'].transpose(1, 2)
239239

240240
if not infer:

0 commit comments

Comments
 (0)