Add `scale_factor` to vae config.

Update unet/config.json

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

Create README.md

flax

initial commit