From 6aee9030d2f0472ac8ee166febac6cfd4f059edf Mon Sep 17 00:00:00 2001
From: seranus <ognjen_j@yahoo.com>
Date: Mon, 6 Dec 2021 22:38:06 +0100
Subject: [PATCH] fix

---
 models/Model_XSeg/Model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/Model_XSeg/Model.py b/models/Model_XSeg/Model.py
index b0addfd..1313e6f 100644
--- a/models/Model_XSeg/Model.py
+++ b/models/Model_XSeg/Model.py
@@ -196,7 +196,7 @@ class XSegModel(ModelBase):
         return ( ('loss', np.mean(loss) ), )
 
     #override
-    def onGetPreview(self, samples, for_history=False):
+    def onGetPreview(self, samples, for_history=False, filenames=None):
         n_samples = min(4, self.get_batch_size(), 800 // self.resolution )
         
         if self.pretrain: