bring in attention etc.

2024-08-30 20:32:17 +00:00 · 2022-10-21 03:54:13 +02:00
parent 2e0b1c4c8b 61357e4e6e
commit 4c1267338b
8 changed files with 252 additions and 16 deletions
--- a/ldm/models/diffusion/ksampler.py
+++ b/ldm/models/diffusion/ksampler.py
@ -183,7 +183,6 @@ class KSampler(Sampler):
            )

        # sigmas are set up in make_schedule - we take the last steps items
-        total_steps = len(self.sigmas)
        sigmas = self.sigmas[-S-1:]

        # x_T is variation noise. When an init image is provided (in x0) we need to add