Skip to content

Support EDM-style training in DreamBooth LoRA SDXL script #7126

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.

Sign up for GitHub

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

Merged
merged 29 commits into from
Mar 3, 2024
Merged

Support EDM-style training in DreamBooth LoRA SDXL script #7126

Changes from 2 commits
Commits
Show all changes
29 commits
Select commit Hold shift + click to select a range
f3013e2
add: dreambooth lora script for Playground v2.5
sayakpaul Feb 28, 2024
cb5f0d7
fix: kwarg
sayakpaul Feb 28, 2024
84047c2
address suraj's comments.
sayakpaul Feb 28, 2024
746b625
Apply suggestions from code review
sayakpaul Feb 28, 2024
dcf36b4
Merge branch 'main' into playground-dreambooth-lora
sayakpaul Feb 28, 2024
0d1547f
apply suraj's suggestion
sayakpaul Feb 28, 2024
c9dd1d0
incorporate changes in the canonical script./
sayakpaul Feb 28, 2024
e2b7144
tracker naming
sayakpaul Feb 28, 2024
db025fc
fix: schedule determination
sayakpaul Feb 28, 2024
80ef425
add: two simple tests
sayakpaul Feb 28, 2024
3ae0e28
remove playground script
sayakpaul Feb 28, 2024
2f49630
note about edm-style training
sayakpaul Feb 28, 2024
c8ed8af
address pedro's comments.
sayakpaul Feb 28, 2024
e02e6f6
address part of Suraj's comments.
sayakpaul Feb 28, 2024
a8bea31
Apply suggestions from code review
sayakpaul Feb 28, 2024
60bf1e6
remove guidance_scale.
sayakpaul Feb 28, 2024
00156a3
use mse_loss.
sayakpaul Feb 28, 2024
3621e18
add comments for preconditioning.
sayakpaul Feb 28, 2024
206f2c7
quality
sayakpaul Feb 28, 2024
f7fc1f6
Update examples/dreambooth/train_dreambooth_lora_sdxl.py
sayakpaul Feb 29, 2024
d96d8ea
Merge branch 'main' into playground-dreambooth-lora
sayakpaul Feb 29, 2024
dde7595
tackle v-pred.
sayakpaul Feb 29, 2024
128b877
Empty-Commit
sayakpaul Feb 29, 2024
e052fe1
Merge branch 'main' into playground-dreambooth-lora
sayakpaul Feb 29, 2024
65c382c
support edm for sdxl too.
sayakpaul Feb 29, 2024
0f046d8
Merge branch 'main' into playground-dreambooth-lora
sayakpaul Mar 1, 2024
10c55bb
address suraj's comments.
sayakpaul Mar 1, 2024
e3210fc
Merge branch 'main' into playground-dreambooth-lora
sayakpaul Mar 2, 2024
fe75b46
Empty-Commit
sayakpaul Mar 2, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading