Support training SD V2 with Flax#1783

patil-suraj merged 2 commits intohuggingface:mainhuggingface/diffusers:mainfrom

yasyf:yasyf/support_sdv2yasyf/diffusers:yasyf/support_sdv2Copy head branch name to clipboard