From 8a2c68c1923938f6a153e64fefce7db026d0cd65 Mon Sep 17 00:00:00 2001 From: lucidrains Date: Tue, 31 Oct 2023 10:13:46 -0700 Subject: [PATCH] fix --- magvit2_pytorch/magvit2_pytorch.py | 2 +- magvit2_pytorch/version.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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'