Enabled legacy sampler settings to be imported

rhyswynn · rhyswynn · commit c0042ab2ffe6 · 2024-04-25T17:27:41.000-04:00
diff --git a/scripts/deforum_helpers/defaults.py b/scripts/deforum_helpers/defaults.py
@@ -24,6 +24,7 @@ def get_samplers_list():
         'dpm2 a': 'DPM2 a',
         'dpm++ 2s a': 'DPM++ 2S a',
         'dpm++ 2m': 'DPM++ 2M',
+        'dpm++ 2m sde': 'DPM++ 2M SDE',
         'dpm++ sde': 'DPM++ SDE',
         'dpm fast': 'DPM fast',
         'dpm adaptive': 'DPM adaptive',
diff --git a/scripts/deforum_helpers/settings.py b/scripts/deforum_helpers/settings.py
@@ -132,21 +132,9 @@ def load_all_settings(*args, ui_launch=False, **kwargs):
         if key == 'sampler' and isinstance(val, str):
             samp_val = val.split()
             scheduler_val = None
-            if samp_val[-1] == 'Uniform':
+            if samp_val[-1] in ['Uniform','SGM Uniform','Karras','Exponential','Polyexponential']:
                 scheduler_val = samp_val[-1]
-                val = (val.split(" Uniform"))[0]
-            if samp_val[-1] == 'Karras':
-                scheduler_val = samp_val[-1]
-                val = (val.split(" Karras"))[0]
-            if samp_val[-1] == 'Exponential':
-                scheduler_val = samp_val[-1]
-                val = (val.split(" Exponential"))[0]
-            if samp_val[-1] == 'Polyexponential':
-                scheduler_val = samp_val[-1]
-                val = (val.split(" Polyexponential"))[0]
-            if samp_val[-1] == 'SGM Uniform':
-                scheduler_val = samp_val[-1]
-                val = (val.split(" SGM Uniform"))[0]
+                val = (val.split(" " + samp_val[-1]))[0]
         if key == 'scheduler' and isinstance(val, str):
             if scheduler_val is not None:
                 val = scheduler_val
