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

Increase baseline codes recording across all dvae models

Browse Source
This commit is contained in:
James Betker 2021-09-30 08:09:07 -06:00
parent f84ccbdfb2
commit 0396a9d2ca
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/diffusion/diffusion_dvae.py
View File

@ -112,7 +112,7 @@ class DiffusionDVAE(nn.Module):
#self.quantizer = Quantize(quantize_dim, num_discrete_codes, balancing_heuristic=True) #self.quantizer = Quantize(quantize_dim, num_discrete_codes, balancing_heuristic=True)
self.quantizer = GumbelQuantizer(quantize_dim, quantize_dim, num_discrete_codes) self.quantizer = GumbelQuantizer(quantize_dim, quantize_dim, num_discrete_codes)
# For recording codebook usage. # For recording codebook usage.
self.codes = torch.zeros((131072,), dtype=torch.long) self.codes = torch.zeros((1228800,), dtype=torch.long)
self.code_ind = 0 self.code_ind = 0
self.internal_step = 0 self.internal_step = 0
decoder_channels = [model_channels * channel_mult[s-1] for s in spectrogram_conditioning_levels] decoder_channels = [model_channels * channel_mult[s-1] for s in spectrogram_conditioning_levels]

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

@ -147,7 +147,7 @@ class AttentionDVAE(nn.Module):
self.normalization = normalization self.normalization = normalization
self.record_codes = record_codes self.record_codes = record_codes
if record_codes: if record_codes:
self.codes = torch.zeros((32768,), dtype=torch.long) self.codes = torch.zeros((1228800,), dtype=torch.long)
self.code_ind = 0 self.code_ind = 0
self.internal_step = 0 self.internal_step = 0

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

@ -142,7 +142,7 @@ class DiscreteVAE(nn.Module):
self.normalization = normalization self.normalization = normalization
self.record_codes = record_codes self.record_codes = record_codes
if record_codes: if record_codes:
self.codes = torch.zeros((32768,), dtype=torch.long) self.codes = torch.zeros((1228800,), dtype=torch.long)
self.code_ind = 0 self.code_ind = 0
self.internal_step = 0 self.internal_step = 0