Getting Error while training the VAE model on the personal system

Question

Getting error while calling .fit function when training VAE using OpenAI gym\'s "CarRacing-v0" environment on my system. The same model trains well on google Colab.