scripts/vsmlrt.py: added support for SAFA v0.2-v0.4 models

AmusementClub · Dec 20, 2023 · f97dc04 · f97dc04
1 parent c2b01aa
commit f97dc04
Showing 1 changed file with 9 additions and 2 deletions.
diff --git 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,