nodes/kijai-ComfyUI-WanVideoWrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix longcat scheduler

Browse Source
This commit is contained in:
kijai
2025-12-08 10:26:38 +02:00
parent 00971fc121
commit e867e642d4
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
wanvideo/schedulers/__init__.py
+3 -3
View File
@@ -65,10 +65,10 @@ def get_scheduler(scheduler, steps, start_step, end_step, shift, device, transfo
sample_scheduler.set_timesteps(steps, device=device, sigmas=sigmas) sample_scheduler.set_timesteps(steps, device=device, sigmas=sigmas)
else: else:
sample_scheduler = FlowMatchEulerDiscreteScheduler(shift=shift, use_beta_sigmas=(scheduler == 'euler/beta')) sample_scheduler = FlowMatchEulerDiscreteScheduler(shift=shift, use_beta_sigmas=(scheduler == 'euler/beta'))
if sigmas is None: if sigmas is None:
sample_scheduler.set_timesteps(steps, device=device) sample_scheduler.set_timesteps(steps, device=device)
else: else:
_apply_custom_sigmas(sample_scheduler, sigmas, device) _apply_custom_sigmas(sample_scheduler, sigmas, device)
elif 'dpm' in scheduler: elif 'dpm' in scheduler:
if 'sde' in scheduler: if 'sde' in scheduler:
algorithm_type = "sde-dpmsolver++" algorithm_type = "sde-dpmsolver++"