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

One last fix for gpt_asr_hf2

Browse Source
This commit is contained in:
James Betker 2021-12-02 21:19:28 -07:00
parent 68e9db12b5
commit 85542ec547
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/gpt_voice/gpt_asr_hf2.py
View File

@ -279,7 +279,7 @@ class GptAsrHf2(nn.Module):
fake_inputs[:, -cond_used:] = cond_text[:, :cond_used]
gen = self.inference_model.generate(fake_inputs, do_sample=do_sample, bos_token_id=self.START_TOKEN, pad_token_id=0, eos_token_id=0,
max_length=self.max_symbols_per_phrase+mel_emb.shape[1], temperature=temperature, num_beams=num_beams, use_cache=True)
return gen[:, self.max_mel_frames:]
return gen[:, mel_emb.shape[1]:]
@register_model