Keep sigmas on CPU

01491d30 · drhead · GitHub · ddb28b33 · 01491d30
Commit 01491d30 authored May 17, 2024 by drhead Committed by GitHub May 17, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

modules/sd_samplers_kdiffusion.py modules/sd_samplers_kdiffusion.py +1 -1

No files found.
--- a/modules/sd_samplers_kdiffusion.py
+++ b/modules/sd_samplers_kdiffusion.py
@@ -115,7 +115,7 @@ class KDiffusionSampler(sd_samplers_common.Sampler):
            if scheduler.need_inner_model:
                sigmas_kwargs['inner_model'] = self.model_wrap

-            sigmas = scheduler.function(n=steps, **sigmas_kwargs, device=shared.device)
+            sigmas = scheduler.function(n=steps, **sigmas_kwargs)

        if discard_next_to_last_sigma:
            sigmas = torch.cat([sigmas[:-2], sigmas[-1:]])