-
Notifications
You must be signed in to change notification settings - Fork 6k
[SDXL DreamBooth LoRA] add support for text encoder fine-tuning #4097
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
Merged
Merged
Changes from 36 commits
Commits
Show all changes
45 commits
Select commit
Hold shift + click to select a range
65a5c45
Allow low precision sd xl
patrickvonplaten 43f842c
finish
patrickvonplaten 7171d42
finish
patrickvonplaten 97f69a7
Merge branch 'main' into allow_low_precision_vae_sd_xl
patrickvonplaten 9337535
feat: initial draft for supporting text encoder lora finetuning for S…
sayakpaul 69e9bfa
Merge remote-tracking branch 'origin/allow_low_precision_vae_sd_xl' i…
sayakpaul b487dfc
fix: variable assignments.
sayakpaul c51e559
add: autocast block.
sayakpaul 9d23e30
add debugging
sayakpaul ea285db
vae dtype hell
sayakpaul b0ed1b6
fix: vae dtype hell.
sayakpaul 20a9186
fix: vae dtype hell 3.
sayakpaul 9d3e606
clean up
sayakpaul 35b8dae
lora text encoder loader.
sayakpaul 9c305b1
fix: unwrapping models.
sayakpaul 4afb793
add: tests.
sayakpaul 42fb433
docs.
sayakpaul 0f95887
handle unexpected keys.
sayakpaul ede8ca2
fix vae dtype in the final inference.
sayakpaul 34b536c
fix scope problem.
sayakpaul ad26174
fix: save_model_card args.
sayakpaul c5a95d6
initialize: prefix to None.
sayakpaul 63f62b4
fix: dtype issues.
sayakpaul 2d815d2
apply gixes.
sayakpaul 0df9af2
Merge branch 'main' into feat/sdxl-dreambooth-returns
sayakpaul 3bb3d4f
debgging.
sayakpaul 6b496d5
debugging
sayakpaul d4a68a9
debugging
sayakpaul b70b003
debugging
sayakpaul fe9ca14
debugging
sayakpaul 3d968bb
debugging
sayakpaul bcac032
add: fast tests.
sayakpaul ff3f27f
pre-tokenize.
sayakpaul 0c33566
address: will's comments.
sayakpaul 43d7046
fix: loader and tests.
sayakpaul 8d5d5b1
fix: dataloader.
sayakpaul 0d77b53
simplify dataloader.
sayakpaul c80915e
Merge branch 'main' into feat/sdxl-dreambooth-returns
sayakpaul 91b0c3a
length.
sayakpaul 52eef75
simplification.
sayakpaul ef501c8
make style && make quality
sayakpaul 07a45c8
simplify state_dict munging
sayakpaul 6ca45f3
fix: tests.
sayakpaul 5f4e089
fix: state_dict packing.
sayakpaul 989e54d
Apply suggestions from code review
sayakpaul File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Uh oh!
There was an error while loading. Please reload this page.