Add samplers: euler, euler a

Acly · Dec 14, 2023 · 21fd767 · 21fd767
1 parent 1b3f00f
commit 21fd767
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -16,3 +16,6 @@ ai_diffusion/styles/*
 !ai_diffusion/styles/digital-artwork*.json
 krita_ai_diffusion*.zip
 TODO.md
+.idea
+venv
+.DS_Store
diff --git a/ai_diffusion/style.py b/ai_diffusion/style.py
@@ -18,6 +18,8 @@
     "DPM++ SDE Karras",
     "UniPC BH2",
     "LCM",
+    "Euler",
+    "Euler a",
 ]
 
 

diff --git a/ai_diffusion/workflow.py b/ai_diffusion/workflow.py
@@ -161,6 +161,8 @@ def _sampler_params(
         "DPM++ SDE Karras": "dpmpp_sde_gpu",
         "UniPC BH2": "uni_pc_bh2",
         "LCM": "lcm",
+        "Euler": "euler",
+        "Euler a": "euler_ancestral",
     }[config.sampler]
     sampler_scheduler = {
         "DDIM": "ddim_uniform",
@@ -171,6 +173,8 @@ def _sampler_params(
         "DPM++ SDE Karras": "karras",
         "UniPC BH2": "ddim_uniform",
         "LCM": "sgm_uniform",
+        "Euler": "normal",
+        "Euler a": "normal",
     }[config.sampler]
     params: dict[str, Any] = dict(
         sampler=sampler_name,