basujindal / stable-diffusion

Optimized Stable Diffusion modified to run on lower GPU VRAM
Other
3.14k stars 467 forks source link

can this code be used for finetuning? #218

Open KyonP opened 1 year ago

KyonP commented 1 year ago

I am trying to train stable-diffusion models for my custom text2img dataset.

Still struggling to fit into my 3090 24GB VRAM; maybe this code is the solution.

Example config YAML files for reference would be helpful.

basujindal commented 1 year ago

Hi, this code was specifically modified for faster inference, removing the possibility of using it for fine-tuning. Thanks.