deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

asdf

Browse Source
This commit is contained in:
James Betker 2022-06-14 21:52:22 -06:00
parent 804b365d5f
commit fae05229ec
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/audio/music/transformer_diffusion12.py
View File

@ -407,7 +407,7 @@ class TransformerDiffusionWithPretrainedVqvae(nn.Module):
class TransformerDiffusionWithMultiPretrainedVqvae(nn.Module):
def __init__(self, num_vaes=4, vqargs, **kwargs):
def __init__(self, num_vaes=4, vqargs={}, **kwargs):
super().__init__()
self.internal_step = 0