Marigold icon indicating copy to clipboard operation
Marigold copied to clipboard

Stable Diffusion fine-tuning problem

Open 838959823 opened this issue 10 months ago • 4 comments

During fine-tuning for depth estimateaion conditioned on input image, how to deal with the text prompt required in the original pre_trained text-to-image Stable Diffusion model?

838959823 avatar Mar 27 '24 07:03 838959823

As far as i understand, you can just pass the empty prompts.

saqib736 avatar Mar 27 '24 07:03 saqib736

I cannot understand the fine-tuning, i think it is a process of pre-training for getting the CKPT, and also not find the part corresponding to fine-tuning in the code. I wanna ask for your help, pls.

billie2jean avatar May 16 '24 13:05 billie2jean

I cannot understand the fine-tuning, i think it is a process of pre-training for getting the CKPT, and also not find the part corresponding to fine-tuning in the code. I wanna ask for your help, pls.

Fine-tuning means training the pre-trained stable diffusion model on your specific dataset. This repository doesn't contain the training code yet; only the testing/inference code is available.

If you want to train or fine-tune the model on your dataset or any other dataset, I have written a simple training code for Marigold. You can find it on my GitHub here: training code for Marigold.

saqib736 avatar May 17 '24 09:05 saqib736

Thx, :)

billie2jean avatar May 17 '24 12:05 billie2jean

Training code has been released.

markkua avatar May 28 '24 08:05 markkua