diff --git a/codes/models/gpt_voice/unet_diffusion_tts7.py b/codes/models/gpt_voice/unet_diffusion_tts7.py
index ed5d696e..5c13c7ae 100644
--- 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):