From 263d7b17599ec26a438cc0f3b289aa3f4f9eac93 Mon Sep 17 00:00:00 2001 From: Peter Harrington <48932392+pzharrington@users.noreply.github.com> Date: Tue, 7 Jan 2025 16:03:54 -0800 Subject: [PATCH] Revert unconditional EDMLoss to positional arg (#751) Signed-off-by: Peter Harrington --- modulus/metrics/diffusion/loss.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modulus/metrics/diffusion/loss.py b/modulus/metrics/diffusion/loss.py index 9a60f0426..001d3ece1 100644 --- a/modulus/metrics/diffusion/loss.py +++ b/modulus/metrics/diffusion/loss.py @@ -265,7 +265,7 @@ def __call__(self, net, images, condition=None, labels=None, augment_pipe=None): augment_labels=augment_labels, ) else: - D_yn = net(y + n, sigma, labels=labels, augment_labels=augment_labels) + D_yn = net(y + n, sigma, labels, augment_labels=augment_labels) loss = weight * ((D_yn - y) ** 2) return loss