From 927731f3b49be17d99134dddd85351d8335767c4 Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 22 Mar 2022 11:41:32 -0600 Subject: [PATCH] tts9: fix position embeddings snafu --- codes/models/audio/tts/unet_diffusion_tts9.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/codes/models/audio/tts/unet_diffusion_tts9.py b/codes/models/audio/tts/unet_diffusion_tts9.py index 672882f0..278fde70 100644 --- a/codes/models/audio/tts/unet_diffusion_tts9.py +++ b/codes/models/audio/tts/unet_diffusion_tts9.py @@ -219,7 +219,7 @@ class DiffusionTts(nn.Module): attn_dropout=dropout, use_rmsnorm=True, ff_glu=True, - rotary_emb_dim=True, + rotary_pos_emb=True, ) )) self.latent_converter = nn.Conv1d(in_latent_channels, conditioning_dim, 1) @@ -238,7 +238,7 @@ class DiffusionTts(nn.Module): attn_dropout=dropout, use_rmsnorm=True, ff_glu=True, - rotary_emb_dim=True, + rotary_pos_emb=True, ) )) else: