diffusers icon indicating copy to clipboard operation
diffusers copied to clipboard

[WIP] Add inpainting training script

Open patil-suraj opened this issue 2 years ago • 6 comments

What does this PR do?

Add in-painting training script.

patil-suraj avatar Jan 16 '24 06:01 patil-suraj

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

Gave this script a try it seems like it cannot be used for fine-tuning an existing sdxl inpaint model and can only be used as creating an inpainting model from base sdxl.

The error says that the number of input channels in the unet required should be 9 (base sdxl) but is 19 (inpainting sdxl). I don't know if it is possible to scale the input tensor channel from 9 to 19 since the input already contains noisy + mask + mask_image latents.

KaustubhPatange avatar Jan 25 '24 06:01 KaustubhPatange

This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the contributing guidelines are likely to be ignored.

github-actions[bot] avatar Feb 19 '24 15:02 github-actions[bot]

Hey @patil-suraj, thanks for sharing this training code.

Could you please share with us the exact training args (without dataset sharing) you used to train inpainting model including number of GPUs used (to get total batch size)? Also the dataset size would help a lot. Also would appreciate sharing wandb run if possible 🤗

AlekseyKorshuk avatar Feb 26 '24 22:02 AlekseyKorshuk

This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the contributing guidelines are likely to be ignored.

github-actions[bot] avatar Mar 31 '24 15:03 github-actions[bot]

This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the contributing guidelines are likely to be ignored.

github-actions[bot] avatar Apr 25 '24 15:04 github-actions[bot]