Fix potential turbo scheduler model patching issue.

comfyanonymous · Jan 22, 2024 · f2d432f · f2d432f
1 parent 4871a36
commit f2d432f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/comfy_extras/nodes_custom_sampler.py b/comfy_extras/nodes_custom_sampler.py
@@ -105,9 +105,8 @@ def INPUT_TYPES(s):
     def get_sigmas(self, model, steps, denoise):
         start_step = 10 - int(10 * denoise)
         timesteps = torch.flip(torch.arange(1, 11) * 100 - 1, (0,))[start_step:start_step + steps]
-        inner_model = model.patch_model(patch_weights=False)
-        sigmas = inner_model.model_sampling.sigma(timesteps)
-        model.unpatch_model()
+        comfy.model_management.load_models_gpu([model])
+        sigmas = model.model.model_sampling.sigma(timesteps)
         sigmas = torch.cat([sigmas, sigmas.new_zeros([1])])
         return (sigmas, )