Add Flax stable diffusion img2img pipeline (#1355)
* add flax img2img pipeline
* update pipeline
* black format file
* remove argg from get_timesteps
* update get_timesteps
* fix bug: make use of timesteps for for_loop
* black file
* black, isort, flake8
* update docstring
* update readme
* update flax img2img readme
* update sd pipeline init
* Update src/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion_img2img.py
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
* Update src/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion_img2img.py
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
* update inits
* revert change
* update var name to image, typo
* update readme
* return new t_start instead of modified timestep
* black format
* isort files
* update docs
* fix-copies
* update prng_seed typing
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
