hkproj / pytorch-stable-diffusion

Stable Diffusion implemented from scratch in PyTorch
https://www.youtube.com/watch?v=ZBKpAp_6TGI
MIT License
608 stars 138 forks source link

training code #6

Closed antonyvan closed 9 months ago

antonyvan commented 10 months ago

Hello. Thank you so much for sharing your video and code . I found yours is the most easy one to follow. Do you have a plan to write about training code ? Maybe fine tuning the pretrained model with some specific domain data.
Regards

hkproj commented 9 months ago

Duplicate question, already answered here: https://github.com/hkproj/pytorch-stable-diffusion/issues/5#issuecomment-1962781941