From cb677f57824ab27ce0f41eb045e0b4e653cebd92 Mon Sep 17 00:00:00 2001
From: seranus <ognjen_j@yahoo.com>
Date: Thu, 9 Dec 2021 18:12:55 +0100
Subject: [PATCH] force complete preview

---
 models/Model_SAEHD/Model.py           | 2 +-
 models/Model_SAEHD/config_schema.json | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

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"
         },