From b5cc6f98c10c8a229ef95f4650fd71d8818cad2a Mon Sep 17 00:00:00 2001
From: andyxr <andyxr@gmail.com>
Date: Sun, 1 Oct 2023 15:29:20 +0100
Subject: [PATCH] Added 2 missing samplers

I missed two out: DPM++ 2M SDE Karras and Restart
---
 scripts/deforum_helpers/defaults.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/scripts/deforum_helpers/defaults.py b/scripts/deforum_helpers/defaults.py
index f85ebf6d2..b8aaac58e 100644
--- 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():