search

ThisisBillhe / tiny-stable-diffusion

Tiny optimized Stable-diffusion that can run on GPUs with just 1GB of VRAM. (Beta)
Other
154 stars 13 forks source link

training scripts #5

Open zmy-08 opened 2 months ago

zmy-08 commented 2 months ago

It's really impressive work! Could you please let me know if there are training scripts available? I'd like to train it on my own dataset.

ThisisBillhe commented 2 months ago

I do not train diffusion models from scratch. But you can follow my paper PTQD and EfficientDM to convert a well-trained full-precision diffusion model to a low-bit one.