Skip to content

[FLUX] support LoRA #9057

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 7 commits into from
Aug 5, 2024
Merged

[FLUX] support LoRA #9057

merged 7 commits into from
Aug 5, 2024

Conversation

sayakpaul
Copy link
Member

@sayakpaul sayakpaul commented Aug 2, 2024

Additionally, refactors the LoRA tests a bit to streamline things. Hope that is okay.

(update): This PR DOES NOT add a training script. The open FLUX models are distilled variants of the base model (closed). So, the training dynamics might be different here, and we will take our time to figure things out. So, a training script PR will come later.

add tests

fix imports

major fixes.
@HuggingFaceDocBuilderDev

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.

fixes

final fixes?
@sayakpaul sayakpaul requested review from DN6 and yiyixuxu August 2, 2024 11:06
@bghira
Copy link
Contributor

bghira commented Aug 2, 2024

where is there any lora training code available to reference?

@sayakpaul
Copy link
Member Author

Updated the PR description.

@bghira
Copy link
Contributor

bghira commented Aug 4, 2024

to train: https://github.com/bghira/SimpleTuner/blob/main/documentation/quickstart/FLUX.md

@sayakpaul sayakpaul merged commit fc6a91e into main Aug 5, 2024
18 checks passed
@sayakpaul sayakpaul deleted the lora-support-flux branch August 5, 2024 04:54
sayakpaul added a commit that referenced this pull request Dec 23, 2024
* feat: lora support for Flux.

add tests

fix imports

major fixes.

* fix

fixes

final fixes?

* fix

* remove is_peft_available.
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

Successfully merging this pull request may close these issues.

4 participants