Keep sigmas on CPU

2024-05-17 10:36:08 -04:00 · 2024-05-17 10:36:08 -04:00 · 01491d303c
parent ddb28b33a3
commit 01491d303c
1 changed files with 1 additions and 1 deletions
--- 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:]])