Quantization method of diffusion model

Abstract

A quantization method of a diffusion model comprises performing post-training quantization on a first diffusion model, to obtain a second diffusion model; dividing time steps of the second diffusion model into a plurality of time step groups; adjusting a weight of the second diffusion model for each of the plurality of time step groups, to obtain a third diffusion model; training the third diffusion model using an output of the first diffusion model, to adjust adjustment parameters of the third diffusion model, wherein the adjustment parameters of the third diffusion model reflect a change of a weight of the third diffusion model with respect to a weight of the second diffusion model; and quantizing the adjusted adjustment parameters of the third diffusion model to obtain a quantized diffusion model.

Images