Fix minor issue (Sygil-Dev#77)

Will have to investigate further, but for some reason it always assumes transparency even on 3 channel VAE's
ShoukanLabs · Oct 1, 2023 · 952d778 · 952d778
2 parents c720de4 + 80f8a10
commit 952d778
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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