From bf783a43eb7803b1ab1039abaed80d63164a11cb Mon Sep 17 00:00:00 2001 From: seranus Date: Thu, 9 Dec 2021 18:36:09 +0100 Subject: [PATCH] saehd - force full preview --- models/Model_SAEHD/Model.py | 3 ++- models/Model_SAEHD/config_schema.json | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index 3f7d4dd..5c3e3b1 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -74,6 +74,7 @@ class SAEHDModel(ModelBase): default_pretrain = self.options['pretrain'] = self.load_or_def_option('pretrain', False) default_cpu_cap = self.options['cpu_cap'] = self.load_or_def_option('cpu_cap', 8) default_preview_samples = self.options['preview_samples'] = self.load_or_def_option('preview_samples', 4) + default_full_preview = self.options['force_full_preview'] = self.load_or_def_option('force_full_preview', False) default_lr_modifier = self.options['lr_modifier'] = self.load_or_def_option('lr_modifier', 0) ask_override = False if self.read_from_conf else self.ask_override() @@ -986,7 +987,7 @@ class SAEHDModel(ModelBase): S[i] = label_face_filename(S[i], filenames[0][i]) D[i] = label_face_filename(D[i], filenames[1][i]) - if self.resolution <= 256 or self.options['preview_samples'] == True: + if self.resolution <= 256 or self.options['force_full_preview'] == True: result = [] st = [] diff --git a/models/Model_SAEHD/config_schema.json b/models/Model_SAEHD/config_schema.json index 6f8d262..ddd987a 100644 --- a/models/Model_SAEHD/config_schema.json +++ b/models/Model_SAEHD/config_schema.json @@ -165,7 +165,7 @@ "minimum": 1, "maximum": 64 }, - "force_complete_preview": { + "force_full_preview": { "type": "boolean" }, "session_name": {