From fc1d88cf3e48f0eaaf363e57e91b682c3bba0ab9 Mon Sep 17 00:00:00 2001
From: Abdullah Alfaraj <abdullah.a.f2014@gmail.com>
Date: Thu, 23 Feb 2023 01:17:12 +0300
Subject: [PATCH] add a list of samplers for offline use

---
 utility/sampler.js | 117 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 117 insertions(+)
 create mode 100644 utility/sampler.js

diff --git a/utility/sampler.js b/utility/sampler.js
new file mode 100644
index 0000000..fd278db
--- /dev/null
+++ b/utility/sampler.js
@@ -0,0 +1,117 @@
+samplers = [
+    {
+        name: 'Euler a',
+        aliases: ['k_euler_a', 'k_euler_ancestral'],
+        options: {},
+    },
+    {
+        name: 'Euler',
+        aliases: ['k_euler'],
+        options: {},
+    },
+    {
+        name: 'LMS',
+        aliases: ['k_lms'],
+        options: {},
+    },
+    {
+        name: 'Heun',
+        aliases: ['k_heun'],
+        options: {},
+    },
+    {
+        name: 'DPM2',
+        aliases: ['k_dpm_2'],
+        options: {
+            discard_next_to_last_sigma: 'True',
+        },
+    },
+    {
+        name: 'DPM2 a',
+        aliases: ['k_dpm_2_a'],
+        options: {
+            discard_next_to_last_sigma: 'True',
+        },
+    },
+    {
+        name: 'DPM++ 2S a',
+        aliases: ['k_dpmpp_2s_a'],
+        options: {},
+    },
+    {
+        name: 'DPM++ 2M',
+        aliases: ['k_dpmpp_2m'],
+        options: {},
+    },
+    {
+        name: 'DPM++ SDE',
+        aliases: ['k_dpmpp_sde'],
+        options: {},
+    },
+    {
+        name: 'DPM fast',
+        aliases: ['k_dpm_fast'],
+        options: {},
+    },
+    {
+        name: 'DPM adaptive',
+        aliases: ['k_dpm_ad'],
+        options: {},
+    },
+    {
+        name: 'LMS Karras',
+        aliases: ['k_lms_ka'],
+        options: {
+            scheduler: 'karras',
+        },
+    },
+    {
+        name: 'DPM2 Karras',
+        aliases: ['k_dpm_2_ka'],
+        options: {
+            scheduler: 'karras',
+            discard_next_to_last_sigma: 'True',
+        },
+    },
+    {
+        name: 'DPM2 a Karras',
+        aliases: ['k_dpm_2_a_ka'],
+        options: {
+            scheduler: 'karras',
+            discard_next_to_last_sigma: 'True',
+        },
+    },
+    {
+        name: 'DPM++ 2S a Karras',
+        aliases: ['k_dpmpp_2s_a_ka'],
+        options: {
+            scheduler: 'karras',
+        },
+    },
+    {
+        name: 'DPM++ 2M Karras',
+        aliases: ['k_dpmpp_2m_ka'],
+        options: {
+            scheduler: 'karras',
+        },
+    },
+    {
+        name: 'DPM++ SDE Karras',
+        aliases: ['k_dpmpp_sde_ka'],
+        options: {
+            scheduler: 'karras',
+        },
+    },
+    {
+        name: 'DDIM',
+        aliases: [],
+        options: {},
+    },
+    {
+        name: 'PLMS',
+        aliases: [],
+        options: {},
+    },
+]
+
+module.exports = { samplers }