Update nodes.py

2025-05-14 11:55:16 +03:00
parent 1e6a780729
commit 3cb4902ef1
1 changed files with 3 additions and 1 deletions
@@ -2394,8 +2394,10 @@ class WanVideoSampler:
            sample_scheduler = FlowMatchLCMScheduler(shift=shift, use_beta_sigmas=(scheduler == 'lcm/beta'))
            sample_scheduler.set_timesteps(steps, device=device, sigmas=sigmas.tolist() if sigmas is not None else None)
        elif 'flowmatch_causvid' in scheduler:
+            denoising_list = [999, 934, 862, 756, 603, 410, 250, 140, 74]
            sample_scheduler = FlowMatchScheduler(num_inference_steps=steps, shift=shift, sigma_min=0, extra_one_step=True)
-            sample_scheduler.timesteps = torch.tensor([999, 934, 862, 756, 603, 410, 250, 140, 74])[:steps].to(device)
+            sample_scheduler.timesteps = torch.tensor(denoising_list)[:steps].to(device)
+            sample_scheduler.sigmas = torch.cat([sample_scheduler.timesteps / 1000, torch.tensor([0.0], device=device)])
        
        if timesteps is None:
            timesteps = sample_scheduler.timesteps