[Flux LoRA] fix for prior preservation and mixed precision sampling, follow up on #11873 (#12264)
* propagate fixes from https://github.com/huggingface/diffusers/pull/11873/ to flux script
* propagate fixes from https://github.com/huggingface/diffusers/pull/11873/ to flux script
* propagate fixes from https://github.com/huggingface/diffusers/pull/11873/ to flux script
* Apply style fixes
---------
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>