On the Out-of-distribution Generalization of Probabilistic Image Modelling Steven McDonagh 2