From f97dc04cadf285ad91b0e00bc982971d9f113ced Mon Sep 17 00:00:00 2001
From: WolframRhodium <WolframRhodium@users.noreply.github.com>
Date: Wed, 20 Dec 2023 11:33:36 +0800
Subject: [PATCH] scripts/vsmlrt.py: added support for SAFA v0.2-v0.4 models

---
 scripts/vsmlrt.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/scripts/vsmlrt.py b/scripts/vsmlrt.py
index a8a93c2..a6e4c81 100644
--- a/scripts/vsmlrt.py
+++ b/scripts/vsmlrt.py
@@ -1,4 +1,4 @@
-__version__ = "3.18.19"
+__version__ = "3.18.20"
 
 __all__ = [
     "Backend", "BackendV2",
@@ -1243,6 +1243,9 @@ def filter_sc(n: int, f: vs.VideoFrame) -> vs.VideoNode:
 @enum.unique
 class SAFAModel(enum.IntEnum):
     v0_1 = 1
+    v0_2 = 2
+    v0_3 = 3
+    v0_4 = 4
 
 
 @enum.unique
@@ -1285,7 +1288,11 @@ def SAFA(
     else:
         overlap_w, overlap_h = overlap
 
-    multiple = 1
+    # unknown crash
+    if model <= 2:
+        multiple = 8
+    else:
+        multiple = 16
 
     (tile_w, tile_h), (overlap_w, overlap_h) = calc_tilesize(
         tiles=tiles, tilesize=tilesize,