mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-27 18:54:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix typo

Browse Source
This commit is contained in:
Erwan 2024-02-26 09:35:53 +01:00
parent 9c083b428f
commit e96d91555b
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
egs/ljspeech/TTS/vits2/vits.py
View File

@ -301,7 +301,7 @@ class VITS(nn.Module):
lids=lids, lids=lids,
) )
elif forward_type == "discriminator": elif forward_type == "discriminator":
return self._forward_discrminator( return self._forward_discriminator(
text=text, text=text,
text_lengths=text_lengths, text_lengths=text_lengths,
feats=feats, feats=feats,
@ -313,7 +313,7 @@ class VITS(nn.Module):
lids=lids, lids=lids,
) )
elif forward_type == "duration_discriminator": elif forward_type == "duration_discriminator":
return self._forward_discrminator_duration( return self._forward_discriminator_duration(
text=text, text=text,
text_lengths=text_lengths, text_lengths=text_lengths,
feats=feats, feats=feats,
@ -460,7 +460,7 @@ class VITS(nn.Module):
return loss, stats return loss, stats
def _forward_discrminator( def _forward_discriminator(
self, self,
text: torch.Tensor, text: torch.Tensor,
text_lengths: torch.Tensor, text_lengths: torch.Tensor,
@ -547,7 +547,7 @@ class VITS(nn.Module):
return loss, stats return loss, stats
def _forward_discrminator_duration( def _forward_discriminator_duration(
self, self,
text: torch.Tensor, text: torch.Tensor,
text_lengths: torch.Tensor, text_lengths: torch.Tensor,