From 07998b027b0edeca718d971e9034545ed806eb85 Mon Sep 17 00:00:00 2001
From: Mr-Neutr0n <64578610+Mr-Neutr0n@users.noreply.github.com>
Date: Wed, 11 Feb 2026 23:50:25 +0530
Subject: [PATCH] fix: use .any() instead of .all() in NaN assertion

---
 ldm/models/diffusion/ddpm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ldm/models/diffusion/ddpm.py b/ldm/models/diffusion/ddpm.py
index bbedd04cf..bd2b6a5e1 100644
--- a/ldm/models/diffusion/ddpm.py
+++ b/ldm/models/diffusion/ddpm.py
@@ -166,7 +166,7 @@ def register_schedule(self, given_betas=None, beta_schedule="linear", timesteps=
         # TODO how to choose this term
         lvlb_weights[0] = lvlb_weights[1]
         self.register_buffer('lvlb_weights', lvlb_weights, persistent=False)
-        assert not torch.isnan(self.lvlb_weights).all()
+        assert not torch.isnan(self.lvlb_weights).any()
 
     @contextmanager
     def ema_scope(self, context=None):