Correct `sample_size` of Stable Diffusion 2's unet to have correct width and height default

Create README.md

flax

initial commit