Skip to content

DDPS and SDA guidance for diffusion #2

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

Open
CharlelieLrt wants to merge 15 commits into
base: diffusion-samplers-and-guidance
Choose a base branch
from
Open

DDPS and SDA guidance for diffusion #2

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
614b15a
Initial commit of guidance
CharlelieLrt Feb 6, 2026
26fca9d
Added data consistency and model consistency DPS guidance
CharlelieLrt Feb 6, 2026
308969b
Completed impl of DPS guidances
CharlelieLrt Feb 7, 2026
7091809
Added option to pass arbitrary loss functions to DPS guidance
CharlelieLrt Feb 9, 2026
9da0f76
Made norm int = 2 by default
CharlelieLrt Feb 12, 2026
896218d
Fixed license header year + missing backticks in doctsring
CharlelieLrt Feb 12, 2026
28a9232
Renamed A into observation_operator
CharlelieLrt Feb 12, 2026
d5f3c22
Moved validations to __init__ to avoid graph breaks
CharlelieLrt Feb 12, 2026
3b2ab91
Fixes on mask argument in DataConsistencyDPSGuidance
CharlelieLrt Feb 12, 2026
fe58b4b
Added automnatic dtype casting and device transfer for tensor parameters
CharlelieLrt Feb 12, 2026
04236a2
Refactored diffusion sampler (#1363)
CharlelieLrt Feb 18, 2026
c5973f2
Reformatted dps_guidance.py with Denoiser and Predictor protocols
CharlelieLrt Feb 18, 2026
0d3e2ed
Added retain_graph option in dps_guidance.py to be able to chain mult…
CharlelieLrt Feb 18, 2026
723c53a
Merge branch 'main' into diffusion-sda-guidance
CharlelieLrt Feb 18, 2026
1afbfe4
Minor variable renaming in noise_schedulers.py
CharlelieLrt Feb 18, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .importlinter
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,7 @@ containers=
layers =
generate
samplers : metrics
noise_schedulers | multi_diffusion | preconditioners | denoisers
noise_schedulers | multi_diffusion | preconditioners | guidance
utils

[importlinter:contract:physicsnemo-external-imports]
Expand Down
15 changes: 0 additions & 15 deletions physicsnemo/diffusion/denoisers/__init__.py
View file
Open in desktop

This file was deleted.

21 changes: 13 additions & 8 deletions physicsnemo/diffusion/denoisers/denoisers.py → physicsnemo/diffusion/guidance/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,13 +14,18 @@
# See the License for the specific language governing permissions and
# limitations under the License.

import warnings
"""DPS (Diffusion Posterior Sampling) guidance for diffusion models."""

from physicsnemo.core.warnings import FutureFeatureWarning

warnings.warn(
"The 'physicsnemo.diffusion.denoisers.denoisers' module is a placeholder for "
"future functionality that will be implemented in an upcoming release.",
FutureFeatureWarning,
stacklevel=2,
from .dps_guidance import (
DataConsistencyDPSGuidance,
DPSDenoiser,
DPSGuidance,
ModelConsistencyDPSGuidance,
)

__all__ = [
"DPSGuidance",
"DPSDenoiser",
"ModelConsistencyDPSGuidance",
"DataConsistencyDPSGuidance",
]
Loading