Skip to content

Add projection_fn to KDiffInferenceFn and skip unconditional passes when unguided. - #180

Open
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_997789931
Open

copybara-service[bot] wants to merge 1 commit into
mainfrom
test_997789931

Conversation

@copybara-service

Copy link
Copy Markdown

Add projection_fn to KDiffInferenceFn and skip unconditional passes when unguided.

  • Adds projection_fn support to KDiffInferenceFn, instantiating GuidedDiffusionInferenceFn when either guidance_fn or projection_fn is provided.
  • Adds a should_guide method to GuidanceFn implementations to avoid executing unconditional base model passes when guidance is disabled or out of interval.
  • Updates typing across projection and guidance functions to accept optional conditioning.

…hen unguided.

- Adds projection_fn support to KDiffInferenceFn, instantiating GuidedDiffusionInferenceFn when either guidance_fn or projection_fn is provided.
- Adds a should_guide method to GuidanceFn implementations to avoid executing unconditional base model passes when guidance is disabled or out of interval.
- Updates typing across projection and guidance functions to accept optional conditioning.

PiperOrigin-RevId: 997789931
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.

1 participant