oops

2024-10-27 20:11:07 +02:00 · 2024-10-27 20:11:07 +02:00 · 0b6812671c
commit 0b6812671c
parent 3395aa8ca0
1 changed files with 2 additions and 1 deletions
--- a/nodes.py
+++ b/nodes.py
@ -373,8 +373,9 @@ class MochiSampler:

        if opt_sigmas is not None:
            sigma_schedule = opt_sigmas.tolist()
-            sigma_schedule.extend([1.0])
            steps = len(sigma_schedule)
+            sigma_schedule.extend([1.0])
+            
            logging.info(f"Using sigma_schedule: {sigma_schedule}")
        else:
            sigma_schedule = linear_quadratic_schedule(steps, 0.025)