|
|
@ -107,8 +107,7 @@ class SDTurboScheduler: |
|
|
|
def get_sigmas(self, model, steps, denoise): |
|
|
|
def get_sigmas(self, model, steps, denoise): |
|
|
|
start_step = 10 - int(10 * denoise) |
|
|
|
start_step = 10 - int(10 * denoise) |
|
|
|
timesteps = torch.flip(torch.arange(1, 11) * 100 - 1, (0,))[start_step:start_step + steps] |
|
|
|
timesteps = torch.flip(torch.arange(1, 11) * 100 - 1, (0,))[start_step:start_step + steps] |
|
|
|
comfy.model_management.load_models_gpu([model]) |
|
|
|
sigmas = model.get_model_object("model_sampling").sigma(timesteps) |
|
|
|
sigmas = model.model.model_sampling.sigma(timesteps) |
|
|
|
|
|
|
|
sigmas = torch.cat([sigmas, sigmas.new_zeros([1])]) |
|
|
|
sigmas = torch.cat([sigmas, sigmas.new_zeros([1])]) |
|
|
|
return (sigmas, ) |
|
|
|
return (sigmas, ) |
|
|
|
|
|
|
|
|
|
|
|