[SDXL DreamBooth LoRA] add support for text encoder fine-tuning #4097
Allow low precision sd xl
65a5c452
finish
43f842cb
finish
7171d423
Merge branch 'main' into allow_low_precision_vae_sd_xl
97f69a75
feat: initial draft for supporting text encoder lora finetuning for S…
93375357
Merge remote-tracking branch 'origin/allow_low_precision_vae_sd_xl' i…
69e9bfa5
fix: variable assignments.
b487dfc9
add: autocast block.
c51e5598
add debugging
9d23e30c
vae dtype hell
ea285db0
fix: vae dtype hell.
b0ed1b66
fix: vae dtype hell 3.
20a9186b
clean up
9d3e606e
lora text encoder loader.
35b8daec
fix: unwrapping models.
9c305b11
add: tests.
4afb7939
docs.
42fb4336
handle unexpected keys.
0f958871
fix vae dtype in the final inference.
ede8ca27
fix scope problem.
34b536c8
fix: save_model_card args.
ad261744
sayakpaul
marked this pull request as ready for review 2 years ago
initialize: prefix to None.
c5a95d66
fix: dtype issues.
63f62b41
apply gixes.
2d815d2a
Merge branch 'main' into feat/sdxl-dreambooth-returns
0df9af29
debgging.
3bb3d4fa
debugging
6b496d5e
debugging
d4a68a92
debugging
b70b0036
debugging
fe9ca145
debugging
3d968bbb
add: fast tests.
bcac0322
pre-tokenize.
ff3f27f5
address: will's comments.
0c335661
fix: loader and tests.
43d70464
fix: dataloader.
8d5d5b15
simplify dataloader.
0d77b539
Merge branch 'main' into feat/sdxl-dreambooth-returns
c80915ee
length.
91b0c3ac
simplification.
52eef75d
make style && make quality
ef501c88
simplify state_dict munging
07a45c88
fix: tests.
6ca45f37
fix: state_dict packing.
5f4e0899
Apply suggestions from code review
989e54d0
sayakpaul
merged
365e8461
into main 2 years ago
sayakpaul
deleted the feat/sdxl-dreambooth-returns branch 2 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub