for backward compatibility, only fix if legacy=True is explicitly specified

Author: pesser

taming/modules/vqvae/quantize.py

@@ -70,7 +70,7 @@ def forward(self, z):
         # ......\end......... (TODO)
 
         # compute loss for embedding
-        loss = self.beta * torch.mean((z_q.detach()-z)**2) + \
+        loss = torch.mean((z_q.detach()-z)**2) + self.beta * \
             torch.mean((z_q - z.detach()) ** 2)
 
         # preserve gradients