From fae05229ec51bb1446608414e9457b24aa8154db Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 14 Jun 2022 21:52:22 -0600 Subject: [PATCH] asdf --- codes/models/audio/music/transformer_diffusion12.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/audio/music/transformer_diffusion12.py b/codes/models/audio/music/transformer_diffusion12.py index bd5cd0b5..f82a267d 100644 --- a/codes/models/audio/music/transformer_diffusion12.py +++ b/codes/models/audio/music/transformer_diffusion12.py @@ -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