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

fix text-voice-clip bug

Browse Source
This commit is contained in:
James Betker 2022-01-08 08:55:00 -07:00
parent 34774f9948
commit 438dd9ed33
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/models/gpt_voice/text_voice_clip.py
View File

@ -76,8 +76,7 @@ class VoiceCLIP(nn.Module):
speech_tokens = speech_tokens[:, :max_mel_len]
b, device = text.shape[0], text.device
if self.text_mask_percentage > 0:
text_mask = torch.rand_like(text.float()) > self.text_mask_percentage
text_mask = torch.rand_like(text.float()) > self.text_mask_percentage
text_emb = self.text_emb(text)
text_emb += self.text_pos_emb(torch.arange(text.shape[1], device=device))