Skip to content

Conversation

@patil-suraj
Copy link
Contributor

This PR fixes the add_noise in DDPM and DDIM schedulers. It makes sure that the alphas are on the same device as the original sample.

@kashif
Copy link
Contributor

kashif commented Sep 27, 2022

👍🏽

Comment on lines -390 to -391
timesteps = timesteps.to(self.alphas_cumprod.device)

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Not needed as we already put it on device above

@HuggingFaceDocBuilderDev
Copy link

HuggingFaceDocBuilderDev commented Sep 27, 2022

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

Copy link
Contributor

@patrickvonplaten patrickvonplaten left a comment

Choose a reason for hiding this comment

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

Thanks!

@patil-suraj patil-suraj merged commit 3304538 into main Sep 27, 2022
@patil-suraj patil-suraj deleted the fix-add-noise branch September 27, 2022 15:32
prathikr pushed a commit to prathikr/diffusers that referenced this pull request Oct 26, 2022
yoonseokjin pushed a commit to yoonseokjin/diffusers that referenced this pull request Dec 25, 2023
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.

5 participants