Added 2 missing samplers

I missed two out: DPM++ 2M SDE Karras and Restart
deforum-art · Oct 1, 2023 · b5cc6f9 · b5cc6f9
1 parent 6e78e89
commit b5cc6f9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/scripts/deforum_helpers/defaults.py b/scripts/deforum_helpers/defaults.py
@@ -25,6 +25,7 @@ def get_samplers_list():
         'dpm++ 2s a': 'DPM++ 2S a',
         'dpm++ 2m': 'DPM++ 2M',
         'dpm++ sde': 'DPM++ SDE',
+        'dpm++ 2m sde karras': 'DPM++ 2M SDE Karras',
         'dpm fast': 'DPM fast',
         'dpm adaptive': 'DPM adaptive',
         'lms karras': 'LMS Karras',
@@ -42,7 +43,8 @@ def get_samplers_list():
         'dpm++ 3m sde exponential': 'DPM++ 3M SDE Exponential',
         'ddim': 'DDIM',
         'plms': 'PLMS',
-        'unipc': 'UniPC'
+        'unipc': 'UniPC',
+        'restart': 'Restart'
     }
 
 def DeforumAnimPrompts():