From 413f75d20a11eeaae046580d8d9d6f537b7a0cae Mon Sep 17 00:00:00 2001
From: Jeremy Hummel <jhummel@onica.com>
Date: Mon, 12 Aug 2019 23:14:28 -0700
Subject: [PATCH] Fix input options

---
 models/Model_SAE/Model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/models/Model_SAE/Model.py b/models/Model_SAE/Model.py
index 0ce9b12..9bf8cd5 100644
--- a/models/Model_SAE/Model.py
+++ b/models/Model_SAE/Model.py
@@ -124,8 +124,8 @@ class SAEModel(ModelBase):
 
             default_apply_random_ct = ColorTransferMode.NONE if is_first_run else self.options.get('apply_random_ct', ColorTransferMode.NONE)
             self.options['apply_random_ct'] = io.input_int(
-                "Apply random color transfer to src faceset? (0) None, (1) LCT, (2) RCT, (3) RCT-masked ?:help skip:%s) : " % (
-                yn_str[default_apply_random_ct]), default_apply_random_ct,
+                "Apply random color transfer to src faceset? (0) None, (1) LCT, (2) RCT, (3) RCT-masked ?:help skip:%s) : " % (default_apply_random_ct),
+                default_apply_random_ct,
                 help_message="Increase variativity of src samples by apply LCT color transfer from random dst "
                              "samples. It is like 'face_style' learning, but more precise color transfer and without "
                              "risk of model collapse, also it does not require additional GPU resources, "