Skip to content

Fix dreambooth loss type with prior_preservation and fp16 - #826

Merged
patil-suraj merged 1 commit into
mainfrom
dreambooth-loss
Oct 13, 2022
Merged

Fix dreambooth loss type with prior_preservation and fp16#826
patil-suraj merged 1 commit into
mainfrom
dreambooth-loss

Conversation

@anton-l

@anton-l anton-l commented Oct 13, 2022

Copy link
Copy Markdown
Member

This will make the float16 loss terms compatible with accelerate.
Fixes #817

@HuggingFaceDocBuilderDev

HuggingFaceDocBuilderDev commented Oct 13, 2022

Copy link
Copy Markdown

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

@patil-suraj patil-suraj left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Looks good, thanks a lot!

@patil-suraj
patil-suraj merged commit e001fed into main Oct 13, 2022
@patil-suraj
patil-suraj deleted the dreambooth-loss branch October 13, 2022 13:41
prathikr pushed a commit to prathikr/diffusers that referenced this pull request Oct 26, 2022
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.

Mixed precision is not working on dreambooth example

3 participants