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

Use non-uniform noise in diffusion_tts6

Browse Source
This commit is contained in:
James Betker 2022-02-08 07:27:41 -07:00
parent f44b064c5e
commit ff35d13b99
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/unet_diffusion_tts6.py
View File

@ -381,7 +381,7 @@ class DiffusionTts(nn.Module):
assert lr_input is not None assert lr_input is not None
if self.super_sampling_max_noising_factor > 0: if self.super_sampling_max_noising_factor > 0:
noising_factor = random.uniform(0,self.super_sampling_max_noising_factor) noising_factor = random.uniform(0,self.super_sampling_max_noising_factor)
lr_input = torch.rand_like(lr_input) * noising_factor + lr_input lr_input = torch.randn_like(lr_input) * noising_factor + lr_input
lr_input = F.interpolate(lr_input, size=(x.shape[-1],), mode='nearest') lr_input = F.interpolate(lr_input, size=(x.shape[-1],), mode='nearest')
x = torch.cat([x, lr_input], dim=1) x = torch.cat([x, lr_input], dim=1)