From dc500946adcea7069f864d95919c7641a17c7211 Mon Sep 17 00:00:00 2001
From: Peter Baylies <pbaylies@gmail.com>
Date: Wed, 7 Sep 2022 04:50:53 -0400
Subject: [PATCH]  * Make warmup feature less aggressive to support higher
 thresholding values properly.

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

diff --git a/ldm/models/diffusion/ksampler.py b/ldm/models/diffusion/ksampler.py
index 45e1c47ba3..baf47d44a0 100644
--- a/ldm/models/diffusion/ksampler.py
+++ b/ldm/models/diffusion/ksampler.py
@@ -25,7 +25,7 @@ class CFGDenoiser(nn.Module):
         self.inner_model = model
         self.threshold = threshold
         self.warmup_max = warmup
-        self.warmup = 0
+        self.warmup = max(warmup / 10, 1)
 
     def forward(self, x, sigma, uncond, cond, cond_scale):
         x_in = torch.cat([x] * 2)