Skip to content

Add the K-LMS scheduler to the inpainting pipeline + tests#587

Merged
anton-l merged 2 commits intomainfrom
lms-inpainting
Sep 20, 2022
Merged

Add the K-LMS scheduler to the inpainting pipeline + tests#587
anton-l merged 2 commits intomainfrom
lms-inpainting

Conversation

@anton-l
Copy link
Copy Markdown
Member

@anton-l anton-l commented Sep 20, 2022

It's been a long time :)
The changes are mostly ported from the img2img pipeline.

Closes #341
Closes #272
Closes #562
Closes #582

@HuggingFaceDocBuilderDev
Copy link
Copy Markdown

HuggingFaceDocBuilderDev commented Sep 20, 2022

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

Copy link
Copy Markdown
Contributor

@patil-suraj patil-suraj left a comment

Choose a reason for hiding this comment

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

Very cool!

Comment on lines -268 to -269
latent_model_input = latent_model_input.to(self.unet.dtype)
t = t.to(self.unet.dtype)
Copy link
Copy Markdown
Member Author

Choose a reason for hiding this comment

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

These extra type casts are no longer needed, even under autocast. Not sure what was causing these problems before, so I'll keep an eye on the issues.

@anton-l anton-l merged commit 8a6833b into main Sep 20, 2022
@kashif kashif deleted the lms-inpainting branch September 22, 2022 15:05
yoonseokjin pushed a commit to yoonseokjin/diffusers that referenced this pull request Dec 25, 2023
…ce#587)

* Add the K-LMS scheduler to the inpainting pipeline + tests

* Remove redundant casts
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

3 participants