StyleGAN generates fully black images

I have been using this code to create a StyeGAN AI with keras that i found here https://keras.io/examples/generative/stylegan/. When i train with 1 step per epoch it generates images with color. But the moment that I use 10000 steps per epoch, as the documentation recommends, the results are always plain black images. I think it may be something with how the celeb_a data set is loaded in as I found another version of the code that had better results. https://github.com/soon-yau/stylegan_keras/blob/main/stylegan.ipynb. Sadly the way this code loads in the the dataset does not work anymore. Does anybody how to manually download the dataset or even if the problem of the black images has occurred before?

Thanks in advance and have a nice day.