Fix AYS sigmas (#2)

pamparamm · Jul 15, 2024 · 62eb791 · 62eb791
1 parent 32ca983
commit 62eb791
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-ppm"
 description = "Fixed AttentionCouple/NegPip(negative weights in prompts), more CFG++ samplers, etc."
-version = "1.0.1"
+version = "1.0.2"
 license = "AGPL-3.0"
 
 [project.urls]

diff --git a/schedulers.py b/schedulers.py
@@ -25,8 +25,7 @@ def _ays_scheduler(model_sampling, steps, force_sigma_min, model_type="SDXL"):
         "SDXL_30": [999, 953, 904, 850, 813, 777, 738, 695, 650, 602, 556, 510, 462, 417, 374, 331, 290, 250, 214, 182, 155, 131, 108, 85, 66, 49, 32, 20, 12, 3, 0],
     }
     indices = timestep_indices[model_type]
-    indices.reverse()
-    sigmas = simple_scheduler(model_sampling, 1000)[indices]
+    sigmas = simple_scheduler(model_sampling, 1000).flip(0)[1:][indices]
     sigmas = loglinear_interp(sigmas.tolist(), steps + 1 if not force_sigma_min else steps)
     sigmas = torch.FloatTensor(sigmas)
     sigmas = torch.cat([sigmas[:-1] if not force_sigma_min else sigmas, torch.FloatTensor([0.0])])