Add StyleTTS2 inference

This commit is contained in:
Justin John
2024-01-10 21:15:52 +05:30
parent 92b4323427
commit 09a69050ee
57 changed files with 164061 additions and 0 deletions
+14
View File
@@ -0,0 +1,14 @@
def init_weights(m, mean=0.0, std=0.01):
classname = m.__class__.__name__
if classname.find("Conv") != -1:
m.weight.data.normal_(mean, std)
def apply_weight_norm(m):
classname = m.__class__.__name__
if classname.find("Conv") != -1:
weight_norm(m)
def get_padding(kernel_size, dilation=1):
return int((kernel_size * dilation - dilation) / 2)