Skip to content
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.

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

Modify ControlNet inferer so that it takes in context when the diffus… #8360

Merged
merged 3 commits into from
Feb 24, 2025

Conversation

virginiafdez
Copy link
Contributor

Fixes #8344

Description

The ControlNet inferers (latent and not latent) work in such a way that, when conditioning is used, the ControlNet does not take in the conditioning. It should, in theory, exhibit the same behaviour as the diffusion model.
I've changed this behaviour, which has included modifying ControlNetDiffusionInferer and ControlNetLatentDiffusionInferer; the methods call, sample and get_likelihood.
I've also modified the tests to take this into account.

Types of changes

  • Non-breaking change (fix or new feature that would not break existing functionality).
  • New tests added to cover the changes (modified, rather than new)
  • Quick tests passed locally by running ./runtests.sh --quick --unittests --disttests.

…ion model has context. This should be the standard behavior. Modified tests accordingly.

Signed-off-by: Virginia Fernandez <virginia.fernandez@kcl.ac.uk>
Copy link
Member

@ericspod ericspod left a comment

Choose a reason for hiding this comment

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

LGTM!

@KumoLiu
Copy link
Contributor

KumoLiu commented Feb 24, 2025

/build

@KumoLiu KumoLiu enabled auto-merge (squash) February 24, 2025 02:59
@KumoLiu
Copy link
Contributor

KumoLiu commented Feb 24, 2025

/build

@KumoLiu KumoLiu merged commit ab07523 into Project-MONAI:dev Feb 24, 2025
28 checks passed
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.

ControlNetLatentDiffusionInferer errors out when conditioning is not False
3 participants