Fix custom sigmas with euler

2025-11-03 10:12:05 +02:00
parent 5eae7087fa
commit 75109fdb79
1 changed files with 1 additions and 1 deletions
@@ -60,7 +60,7 @@ def get_scheduler(scheduler, steps, start_step, end_step, shift, device, transfo
            if flowedit_args: #seems to work better
                timesteps, _ = retrieve_timesteps(sample_scheduler, device=device, sigmas=get_sampling_sigmas(steps, shift))
            else:
-                sample_scheduler.set_timesteps(steps, device=device, sigmas=sigmas)
+                sample_scheduler.set_timesteps(steps, device=device, sigmas=sigmas[:-1].tolist() if sigmas is not None else None)
    elif 'dpm' in scheduler:
        if 'sde' in scheduler:
            algorithm_type = "sde-dpmsolver++"