asdf

2022-03-01 21:41:31 -07:00 · 2022-03-01 21:41:31 -07:00 · beb7c8a39d
commit beb7c8a39d
parent 70fa780edb
1 changed files with 1 additions and 0 deletions
--- a/codes/models/gpt_voice/unet_diffusion_tts7.py
+++ b/codes/models/gpt_voice/unet_diffusion_tts7.py
@ -417,6 +417,7 @@ class DiffusionTts(nn.Module):
        }
        if self.enable_unaligned_inputs:
            groups['unaligned_encoder'] = list(self.unaligned_encoder.parameters())
+        return groups


    def forward(self, x, timesteps, tokens=None, conditioning_input=None, lr_input=None, unaligned_input=None, conditioning_free=False):