From bd9b9d425a355e151b43047a5df5fcead2fcdc52 Mon Sep 17 00:00:00 2001
From: Sakura-Luna <53183413+Sakura-Luna@users.noreply.github.com>
Date: Sun, 14 May 2023 13:19:11 +0800
Subject: [PATCH] Add live preview mode check

---
 modules/sd_samplers_common.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/modules/sd_samplers_common.py b/modules/sd_samplers_common.py
index d3dc130c..b1e8a780 100644
--- a/modules/sd_samplers_common.py
+++ b/modules/sd_samplers_common.py
@@ -26,8 +26,8 @@ approximation_indexes = {"Full": 0, "Tiny AE": 1, "Approx NN": 2, "Approx cheap"
 
 
 def single_sample_to_image(sample, approximation=None):
-    if approximation is None:
-        approximation = approximation_indexes.get(opts.show_progress_type, 0)
+    if approximation is None or approximation not in approximation_indexes.keys():
+        approximation = approximation_indexes.get(opts.show_progress_type, 1)
 
     if approximation == 1:
         x_sample = sd_vae_taesd.decode()(sample.to(devices.device, devices.dtype).unsqueeze(0))[0].detach()