search

PixArt-alpha / PixArt-sigma

PixArt-Σ: Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generation
https://pixart-alpha.github.io/PixArt-sigma-project/
GNU Affero General Public License v3.0
1.44k stars 67 forks source link

DMD for own model #105

Closed hzhupku closed 6 days ago

hzhupku commented 1 month ago

Hi,

Thanks for the great code! I am wondering if I can use dmd code to distill my own model instead of loading a pretrained one from diffuser. For example, I fine-tuned a pixelart-sigma model and want to distill a particular checkpoint.

lawrence-cj commented 1 week ago

Of course. The training code is ok for training your own model.