From e75141150fa3943a40283e67e8870b23d57861d8 Mon Sep 17 00:00:00 2001
From: maryhipp <maryhipp@gmail.com>
Date: Wed, 29 Mar 2023 06:10:36 -0700
Subject: [PATCH] use reference to sampler_name

---
 .../web/src/common/util/buildNodes.ts         | 26 ++-----------------
 1 file changed, 2 insertions(+), 24 deletions(-)

diff --git a/invokeai/frontend/web/src/common/util/buildNodes.ts b/invokeai/frontend/web/src/common/util/buildNodes.ts
index 1cac40781e..0eb26cc182 100644
--- a/invokeai/frontend/web/src/common/util/buildNodes.ts
+++ b/invokeai/frontend/web/src/common/util/buildNodes.ts
@@ -39,18 +39,7 @@ export function buildTxt2ImgNode(state: RootState): TextToImageInvocation {
     width,
     height,
     cfg_scale,
-    sampler_name: sampler as
-      | 'ddim'
-      | 'dpmpp_2'
-      | 'k_dpm_2'
-      | 'k_dpm_2_a'
-      | 'k_dpmpp_2'
-      | 'k_euler'
-      | 'k_euler_a'
-      | 'k_heun'
-      | 'k_lms'
-      | 'plms'
-      | undefined,
+    sampler_name: sampler as TextToImageInvocation['sampler_name'],
     seamless,
     model: openModel as string | undefined,
     progress_images: shouldDisplayInProgressType === 'full-res',
@@ -85,18 +74,7 @@ export function buildImg2ImgNode(state: RootState): ImageToImageInvocation {
     width,
     height,
     cfg_scale: cfgScale,
-    sampler_name: sampler as
-      | 'ddim'
-      | 'dpmpp_2'
-      | 'k_dpm_2'
-      | 'k_dpm_2_a'
-      | 'k_dpmpp_2'
-      | 'k_euler'
-      | 'k_euler_a'
-      | 'k_heun'
-      | 'k_lms'
-      | 'plms'
-      | undefined,
+    sampler_name: sampler as ImageToImageInvocation['sampler_name'],
     seamless,
     model: model as string | undefined,
     progress_images: shouldDisplayInProgressType === 'full-res',