[training] fixes to the quantization training script and add AdEMAMix optimizer as an option#9806

sayakpaul merged 4 commits intomainhuggingface/diffusers:mainfrom

fix-quantization-traininghuggingface/diffusers:fix-quantization-trainingCopy head branch name to clipboard