Add live preview mode check

This commit is contained in:
Sakura-Luna 2023-05-14 13:19:11 +08:00
parent e14b586d04
commit bd9b9d425a
1 changed files with 2 additions and 2 deletions

View File

@ -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()