FareedKhan-dev / create-stable-diffusion-from-scratch

Implemented a stable diffusion architecture using PyTorch.
https://levelup.gitconnected.com/building-stable-diffusion-from-scratch-using-python-f3ebc8c42da3
35 stars 4 forks source link

finetune stable diffusion on mnist #1

Open Xueting-Bi opened 1 month ago

Xueting-Bi commented 1 month ago

Hi, thank you for nice work! And I want to know do you try finetune stable diffusion (v1.5 / v2)on mnist instead of training directly on mnist, I have tried, but the result is bad, the pre-trained model maybe can't recognize digits. May I kindly ask for your suggestions?

FareedKhan-dev commented 1 month ago

Hi! Thanks for reaching out. I’d be happy to help. Could you please share the details of your results along with the resources or notebook you used for fine-tuning? This information will help me understand the issue better and provide more specific suggestions.