Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fine tuning #8

Open
tekmen0 opened this issue Mar 29, 2024 · 5 comments
Open

Fine tuning #8

tekmen0 opened this issue Mar 29, 2024 · 5 comments

Comments

@tekmen0
Copy link

tekmen0 commented Mar 29, 2024

Thank you for such a great work, I need to train models for image translation tasks (no text conditioning needed). Do you plan to release how to fine-tune, or can I integrate lroa or dreamboot methods into your code? Do you have any plans to release source codes?

@IDKiro
Copy link
Owner

IDKiro commented Mar 29, 2024

Releasing the source code requires communication with my superior. It is also feasible to finetune the model using conventional SD training methods, and the model will still retain its ability to sample in a few steps (for example, 4 steps) after a short period of training. We are currently researching how to distill the existing LoRA into SDXS, and if successful, the training code will be released together (if the company permits me to release the code).

@IDKiro
Copy link
Owner

IDKiro commented Apr 11, 2024

I've already submitted the open source application to the company, maybe within a week.

@IDKiro
Copy link
Owner

IDKiro commented Apr 18, 2024

I am very sorry that the open source application was rejected by the company. I will close the issue for the time being and reopen it if there is further progress.Again, I apologize.

@IDKiro IDKiro closed this as completed Apr 18, 2024
@IDKiro IDKiro reopened this May 27, 2024
@kmpartner
Copy link

I tried to finetune of sdxs-512-0.9 using diffusers text-to-image training docs file.
As mentioned, finetued model generate image at 4 steps after short period of finetuning. However, 1 step image generation disappeared in finetuned model and more than 4 steps does not improve image quality so much (in 8 or 10 steps).

I feel that it seems to train other model during finetunig because 1 step generation disappeared. Is it the case?

Is it possible to generate images in 4 steps with similar qualities compared to sdxs-512-0.9 by continuing finetuning or do I need to change training settings such as noise scheduler, batch size, etc... to enhance quality?

@IDKiro
Copy link
Owner

IDKiro commented Jul 1, 2024

Try this https://github.com/tianweiy/DMD2

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants