diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index be6a9bb..375eee7 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -989,7 +989,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: + if self.resolution <= 256 or self.options['preview_samples'] == True: result = [] st = [] diff --git a/models/Model_SAEHD/config_schema.json b/models/Model_SAEHD/config_schema.json index f42cb93..6f8d262 100644 --- a/models/Model_SAEHD/config_schema.json +++ b/models/Model_SAEHD/config_schema.json @@ -165,6 +165,9 @@ "minimum": 1, "maximum": 64 }, + "force_complete_preview": { + "type": "boolean" + }, "session_name": { "type": "string" },