fix: update sampler names to include cfgpp variant for euler and eule…

…r ancestral
LightDiffusion · Mar 2, 2025 · b4d5615 · b4d5615
1 parent 2609a2f
commit b4d5615
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/modules/sample/sampling.py b/modules/sample/sampling.py
@@ -517,13 +517,13 @@ def ksampler(
     if sampler_name == "dpmpp_2m_cfgpp":
         sampler_function = samplers.sample_dpmpp_2m_cfgpp
 
-    elif sampler_name == "euler_ancestral":
+    elif sampler_name == "euler_ancestral_cfgpp":
         sampler_function = samplers.sample_euler_ancestral_dy_cfg_pp
 
     elif sampler_name == "dpmpp_sde_cfgpp":
         sampler_function = samplers.sample_dpmpp_sde_cfgpp
 
-    elif sampler_name == "euler":
+    elif sampler_name == "euler_cfgpp":
         sampler_function = samplers.sample_euler_dy_cfg_pp
 
     else:

diff --git a/modules/user/GUI.py b/modules/user/GUI.py
@@ -779,7 +779,7 @@ def _generate_image(self) -> None:
                         seed=random.randint(1, 2**64),
                         steps=10,
                         cfg=8,
-                        sampler_name="euler_ancestral",
+                        sampler_name="euler_ancestral_cfgpp",
                         scheduler="normal",
                         denoise=0.45,
                         model=hidiffoptimizer.go(
@@ -997,7 +997,7 @@ def _generate_image_flux(self) -> None:
                     seed=random.randint(1, 2**64),
                     steps=20,
                     cfg=1,
-                    sampler_name="euler",
+                    sampler_name="euler_cfgpp",
                     scheduler="beta",
                     denoise=1,
                     model=unetloadergguf_10[0],

diff --git a/modules/user/pipeline.py b/modules/user/pipeline.py
@@ -218,7 +218,7 @@ def pipeline(
                     seed=random.randint(1, 2**64),
                     steps=20,
                     cfg=1,
-                    sampler_name="euler",
+                    sampler_name="euler_cfgpp",
                     scheduler="beta",
                     denoise=1,
                     model=unetloadergguf_10[0],
@@ -313,7 +313,7 @@ def pipeline(
                         seed=random.randint(1, 2**64),
                         steps=10,
                         cfg=8,
-                        sampler_name="euler_ancestral",
+                        sampler_name="euler_ancestral_cfgpp",
                         scheduler="normal",
                         denoise=0.45,
                         model=hidiffoptimizer.go(