Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/ComfyUI.git" did not exist on "a094b45c93e29193ba2695e834aefe7bf4635c06"
Commit c6b7a157 authored by comfyanonymous's avatar comfyanonymous
Browse files

Align simple scheduling closer to official stable cascade scheduler.

parent ec4d89ce
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
comfy/model_sampling.py
View file @ c6b7a157
...@@ -150,10 +150,10 @@ class StableCascadeSampling(ModelSamplingDiscrete): ...@@ -150,10 +150,10 @@ class StableCascadeSampling(ModelSamplingDiscrete):
self._init_alpha_cumprod = torch.cos(self.cosine_s / (1 + self.cosine_s) * torch.pi * 0.5) ** 2 self._init_alpha_cumprod = torch.cos(self.cosine_s / (1 + self.cosine_s) * torch.pi * 0.5) ** 2
#This part is just for compatibility with some schedulers in the codebase #This part is just for compatibility with some schedulers in the codebase
self.num_timesteps = 1000 self.num_timesteps = 10000
sigmas = torch.empty((self.num_timesteps), dtype=torch.float32) sigmas = torch.empty((self.num_timesteps), dtype=torch.float32)
for x in range(self.num_timesteps): for x in range(self.num_timesteps):
t = x / self.num_timesteps t = (x + 1) / self.num_timesteps
sigmas[x] = self.sigma(t) sigmas[x] = self.sigma(t)
self.set_sigmas(sigmas) self.set_sigmas(sigmas)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment