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

get rid of unused param

Browse Source
This commit is contained in:
James Betker 2022-06-15 09:14:06 -06:00
parent 3757ff9526
commit 6fc86bbbe7
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
codes/trainer/injectors/audio_injectors.py
View File

@ -191,8 +191,7 @@ class GptVoiceLatentInjector(Injector):
codes = self.dvae.get_codebook_indices(mel_inputs)
latents = self.gpt(mel_conds, state[self.text_input_key],
state[self.text_lengths_key], codes, state[self.input_lengths_key],
text_first=True, raw_mels=None, return_attentions=False, return_latent=True,
clip_inputs=False)
text_first=True, raw_mels=None, return_attentions=False, return_latent=True)
assert latents.shape[1] == codes.shape[1]
return {self.output: latents}