xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-04 02:07:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use sigma_max

Browse Source
This commit is contained in:
xzuyn 2025-06-18 00:52:07 -04:00 committed by GitHub
parent 36a76fd68e
commit f94aa3e95e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/samplers.py
View File

@ -464,7 +464,7 @@ def beta_scheduler(model_sampling, steps, alpha=0.6, beta=0.6):
return torch.FloatTensor(sigs) return torch.FloatTensor(sigs)
def linear_scheduler(model_sampling, steps): def linear_scheduler(model_sampling, steps):
return torch.linspace(1.0, 0.0, steps + 1) return torch.linspace(model_sampling.sigma_max, 0.0, steps + 1)
# from: https://github.com/genmoai/models/blob/main/src/mochi_preview/infer.py#L41 # from: https://github.com/genmoai/models/blob/main/src/mochi_preview/infer.py#L41
def linear_quadratic_schedule(model_sampling, steps, threshold_noise=0.025, linear_steps=None): def linear_quadratic_schedule(model_sampling, steps, threshold_noise=0.025, linear_steps=None):