This commit is contained in:
James Betker 2022-06-14 21:52:22 -06:00
parent 804b365d5f
commit fae05229ec

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