Control image brightness The Stable Diffusion pipeline is mediocre at generating images that are either very bright or dark as explained in the Common Diffusion Noise Schedules and Sample Steps are Flawed paper. The solutions proposed in the paper are currently implemented in the DDIMScheduler which you can use to improve the lighting in your images. 💡 Take a look at the paper linked above for more details about the proposed solutions! One of the solutions is to train a model with v prediction and v loss. Add the following flag to the train_text_to_image.py or train_text_to_image_lora.py scripts to enable v_prediction:   Copied --prediction_type="v_prediction" For example, let’s use the ptx0/pseudo-journey-v2 checkpoint which has been finetuned with v_prediction. Next, configure the following parameters in the DDIMScheduler: rescale_betas_zero_snr=True, rescales the noise schedule to zero terminal signal-to-noise ratio (SNR) timestep_spacing="trailing", starts sampling from the last timestep   Copied from diffusers import DiffusionPipeline, DDIMScheduler