Skip to content
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

Add Min-SNR weighting strategy to lora script #3084

Closed
wants to merge 1 commit into from
Closed

Add Min-SNR weighting strategy to lora script #3084

wants to merge 1 commit into from

Conversation

wfng92
Copy link
Contributor

@wfng92 wfng92 commented Apr 13, 2023

PR #2899 added the Min-SNR weighting strategy to the base train_text_to_image.py script, but left out the train_text_to_image_lora.py script. This PR apply the same code snippets to train_text_to_image_lora.py since the underlying code structure is similar between both scripts.

@HuggingFaceDocBuilderDev
Copy link

HuggingFaceDocBuilderDev commented Apr 13, 2023

The documentation is not available anymore as the PR was closed or merged.

@patrickvonplaten
Copy link
Contributor

cc @sayakpaul could you check here?

Copy link
Member

@sayakpaul sayakpaul left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks a lot!

Did you run any experiments to see how using MinSNR here impacts the overall performance?

Also, we need to ensure the CI is green before we can merge.

@wfng92
Copy link
Contributor Author

wfng92 commented Apr 27, 2023

Based on a local experiment, the benefits of min-snr strategy depends on the datasets used. Closing this PR for now.

@wfng92 wfng92 closed this Apr 27, 2023
@wfng92 wfng92 deleted the patch-1 branch April 27, 2023 01:57
@sayakpaul
Copy link
Member

@wfng92 not sure why you closed the PR. It was perfectly fine for us to merge.

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.

None yet

4 participants