update to latest vq library version

Author: lucidrains

setup.py

@@ -8,7 +8,7 @@
           'stylegan2_pytorch = stylegan2_pytorch.cli:main',
       ],
   },
-  version = '1.2.3',
+  version = '1.2.4',
   license='GPLv3+',
   description = 'StyleGan2 in Pytorch',
   author = 'Phil Wang',
@@ -28,7 +28,7 @@
       'torchvision',
       'pillow',
       'pytorch-fid',
-      'vector-quantize-pytorch'
+      'vector-quantize-pytorch>=0.1.0'
   ],
   classifiers=[
       'Development Status :: 4 - Beta',

stylegan2_pytorch/stylegan2_pytorch.py

@@ -605,7 +605,7 @@ def forward(self, x):
                 x = attn_block(x)
 
             if exists(q_block):
-                x, loss = q_block(x)
+                x, _, loss = q_block(x)
                 quantize_loss += loss
 
         x = self.final_conv(x)