diff --git a/muse_maskgit_pytorch/dataset.py b/muse_maskgit_pytorch/dataset.py index 6eb3693..34c95a0 100644 --- a/muse_maskgit_pytorch/dataset.py +++ b/muse_maskgit_pytorch/dataset.py @@ -46,7 +46,7 @@ def __init__( stream=False, using_taming=False, random_crop=False, - alpha_channel=True, + alpha_channel=False, ): super().__init__() self.dataset = dataset