diff --git a/magvit2_pytorch/magvit2_pytorch.py b/magvit2_pytorch/magvit2_pytorch.py index d48cd98..dee5845 100644 --- a/magvit2_pytorch/magvit2_pytorch.py +++ b/magvit2_pytorch/magvit2_pytorch.py @@ -889,7 +889,7 @@ def __init__( perceptual_loss_weight = 1., antialiased_downsample = True, discr_kwargs: Optional[dict] = None, - multiscale_discrs: Optional[Tuple[Module, ...]] = None, + multiscale_discrs: Tuple[Module, ...] = tuple(), use_gan = True, adversarial_loss_weight = 1., grad_penalty_loss_weight = 10., diff --git a/magvit2_pytorch/version.py b/magvit2_pytorch/version.py index 5f0b241..fa51611 100644 --- a/magvit2_pytorch/version.py +++ b/magvit2_pytorch/version.py @@ -1 +1 @@ -__version__ = '0.0.53' +__version__ = '0.0.54'