kxngluca/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests 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

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

@ -76,7 +76,6 @@ 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_emb = self.text_emb(text)