mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

asdf

Browse Source
This commit is contained in:
James Betker 2022-03-01 21:41:31 -07:00
parent 70fa780edb
commit beb7c8a39d
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
codes/models/gpt_voice/unet_diffusion_tts7.py
View File

@ -417,6 +417,7 @@ class DiffusionTts(nn.Module):
} }
if self.enable_unaligned_inputs: if self.enable_unaligned_inputs:
groups['unaligned_encoder'] = list(self.unaligned_encoder.parameters()) 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): def forward(self, x, timesteps, tokens=None, conditioning_input=None, lr_input=None, unaligned_input=None, conditioning_free=False):