Skip to content

Commit

Permalink
scripts/vsmlrt.py: added support for RIFE v4.7 model
Browse files Browse the repository at this point in the history
  • Loading branch information
WolframRhodium committed Sep 25, 2023
1 parent dd6f4f5 commit 54f5d90
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion scripts/vsmlrt.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
__version__ = "3.17.7"
__version__ = "3.17.8"

__all__ = [
"Backend", "BackendV2",
Expand Down Expand Up @@ -809,6 +809,7 @@ class RIFEModel(enum.IntEnum):
v4_4 = 44
v4_5 = 45
v4_6 = 46
v4_7 = 47


def RIFEMerge(
Expand Down Expand Up @@ -871,6 +872,9 @@ def RIFEMerge(
multiple = int(multiple_frac.numerator)
scale = float(Fraction(scale))

if model >= 47 and (ensemble or scale != 1.0 or _implementation == 2):
raise ValueError("not supported")

network_path = os.path.join(
models_path,
"rife_v2",
Expand Down

0 comments on commit 54f5d90

Please sign in to comment.