Skip to content

Make MultiDiffusion work with a Generator #450

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.

Sign up for GitHub

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

Jump to bottom
Merged
merged 1 commit into from
Sep 26, 2024
Merged

Make MultiDiffusion work with a Generator #450

merged 1 commit into from
Sep 26, 2024

Conversation

Laurent2916
Copy link
Contributor

  • made MultiUpscaler.diffuse_targets "stateless" (i.e. it doesn't call torch.randn to sample noise)
  • added a generator argument to MultiUpscaler.upscale, and use it when sampling the noise argument used in MultiUpscaler.diffuse_targets
  • modified test_multi_upscaler accordingly

@Laurent2916 Laurent2916 requested a review from catwell September 26, 2024 09:17
@Laurent2916 Laurent2916 self-assigned this Sep 26, 2024
@Laurent2916 Laurent2916 merged commit 5a92285 into main Sep 26, 2024
3 checks passed
@Laurent2916 Laurent2916 deleted the pr/multidiffusion_random_generator branch September 26, 2024 09:31
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@catwell catwell catwell approved these changes

Assignees

@Laurent2916 Laurent2916

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@Laurent2916 @catwell