calculate dictionary for sampler names only once

1 files changed, 7 insertions, 0 deletions

diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py
index 43ce34eb..6f8ccf1d 100644
--- a/modules/sd_samplers.py
+++ b/modules/sd_samplers.py
@@ -52,6 +52,7 @@ all_samplers_map = {x.name: x for x in all_samplers}
 

 samplers = []

 samplers_for_img2img = []

+samplers_map = {}

 

 

 def create_sampler(name, model):

@@ -77,6 +78,12 @@ def set_samplers():
     samplers = [x for x in all_samplers if x.name not in hidden]

     samplers_for_img2img = [x for x in all_samplers if x.name not in hidden_img2img]

 

+    samplers_map.clear()

+    for sampler in all_samplers:

+        samplers_map[sampler.name.lower()] = sampler.name

+        for alias in sampler.aliases:

+            samplers_map[alias.lower()] = sampler.name

+

 

 set_samplers()