force conversion to rgb or rgba depending on settings

Author: lucidrains

setup.py

@@ -8,7 +8,7 @@
           'stylegan2_pytorch = stylegan2_pytorch.cli:main',
       ],
   },
-  version = '1.2.1',
+  version = '1.2.2',
   license='GPLv3+',
   description = 'StyleGan2 in Pytorch',
   author = 'Phil Wang',

stylegan2_pytorch/stylegan2_pytorch.py

@@ -245,12 +245,12 @@ def slerp(val, low, high):
 # dataset
 
 def convert_rgb_to_transparent(image):
-    if image.mode == 'RGB':
+    if image.mode != 'RGBA':
         return image.convert('RGBA')
     return image
 
 def convert_transparent_to_rgb(image):
-    if image.mode == 'RGBA':
+    if image.mode != 'RGB':
         return image.convert('RGB')
     return image