OSError: stabilityai/stable-diffusion-2-1 does not appear to have a file named flax_model.msgpack or pytorch_model.bin

Trying to run train_text_to_image_flax as described here

Getting an error: OSError: stabilityai/stable-diffusion-2-1 does not appear to have a file named flax_model.msgpack or pytorch_model.bin.

Switching between versions does not help.