kxngluca/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

tts9: fix position embeddings snafu

Browse Source
This commit is contained in:
James Betker 2022-03-22 11:41:32 -06:00
parent 536511fc4b
commit 927731f3b4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/models/audio/tts/unet_diffusion_tts9.py
View File

@ -219,7 +219,7 @@ class DiffusionTts(nn.Module):
attn_dropout=dropout, attn_dropout=dropout,
use_rmsnorm=True, use_rmsnorm=True,
ff_glu=True, ff_glu=True,
rotary_emb_dim=True, rotary_pos_emb=True,
) )
)) ))
self.latent_converter = nn.Conv1d(in_latent_channels, conditioning_dim, 1) self.latent_converter = nn.Conv1d(in_latent_channels, conditioning_dim, 1)
@ -238,7 +238,7 @@ class DiffusionTts(nn.Module):
attn_dropout=dropout, attn_dropout=dropout,
use_rmsnorm=True, use_rmsnorm=True,
ff_glu=True, ff_glu=True,
rotary_emb_dim=True, rotary_pos_emb=True,
) )
)) ))
else: else: