I got this error below while training a VQGAN model using mps acceleration and it threw this error while running the training script
