Skip to content

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.

4 participants