Revert unconditional EDMLoss to positional arg (#751)

Signed-off-by: Peter Harrington <[email protected]>
NVIDIA · Jan 8, 2025 · 263d7b1 · 263d7b1
1 parent 54ca693
commit 263d7b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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