From b4f9a70120ab59d909221ea8f9f2a1f9564444ad Mon Sep 17 00:00:00 2001 From: WolframRhodium Date: Tue, 11 Apr 2023 12:47:51 +0800 Subject: [PATCH] scripts/vsmlrt.py: switch back to rife v1 implementation --- scripts/vsmlrt.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/scripts/vsmlrt.py b/scripts/vsmlrt.py index f723bbb..61b2b9f 100644 --- a/scripts/vsmlrt.py +++ b/scripts/vsmlrt.py @@ -1,4 +1,4 @@ -__version__ = "3.15.20" +__version__ = "3.15.21" __all__ = [ "Backend", "BackendV2", @@ -845,19 +845,18 @@ def RIFEMerge( multiple = int(multiple_frac.numerator) scale = float(Fraction(scale)) - # use v2 implementation by default network_path = os.path.join( models_path, "rife_v2", f"rife_v{model // 10}.{model % 10}{'_ensemble' if ensemble else ''}.onnx" ) - if _implementation != 1 and os.path.exists(network_path) and scale == 1.0: + if _implementation == 2 and os.path.exists(network_path) and scale == 1.0: implementation_version = 2 multiple = 1 # v2 implements internal padding clips = [clipa, clipb, mask] else: implementation_version = 1 - # v2 onnx not found, try v1 + network_path = os.path.join( models_path, "rife",