search

Xiuyu-Li / q-diffusion

[ICCV 2023] Q-Diffusion: Quantizing Diffusion Models.
https://xiuyuli.com/qdiffusion/
MIT License
301 stars 21 forks source link

Code for model calibration #10

Closed Cheeun closed 9 months ago

Cheeun commented 1 year ago

Hello, I am curious if there are any plans to share your code for model calibration. (Code for obtaining the quantized model checkpoints) Btw, thanks for sharing your code! Your work is cool!

Xiuyu-Li commented 1 year ago

Hi, Thank you for your interest in our work! We're currently prioritizing the development of our end-to-end quantization implementation, and thus the refinement and release of the model calibration code will follow accordingly. At this point, we can't specify a timeline for its release, but we will try to get it done in the following few weeks. We appreciate your understanding and patience.

Just a side note, since our calibration process is quite simple with the key idea being sampling the whole time steps distribution uniformly, you can also try replicating this part by creating the calibration dataset based on our algorithm 1 in the paper and referencing this codebase.

Cheeun commented 1 year ago

Thanks for your detailed answer! While waiting for the release, I will try reproducing it with your reference!

arthursunbao commented 1 year ago

+1, looking forward for the release of model calibration code :)

hanhanpp commented 1 year ago

waiting for the new release of calibraiton methods !

Xiuyu-Li commented 9 months ago

Hi all, the calibration code has been added by #20.

Closing the issue for now. Let me know if there are any further questions.