ValueError: Keras .predict seems to require different input than .train_on_batch

Question

I have trained the following GAN with this model: https://github.com/neka-nat/image_completion_tf2/blob/master/model.py (I figured it would be a little chaotic to add the en