0xKermini
605b5875c3
Update the NODE_DISPLAY_NAME_MAPPINGS to recommend the newer RIFE 4.25+ versions instead of just 4.7 and 4.9. Affected files: __init__.py Signed-off-by: 0xKermini <67284748+lukebaze@users.noreply.github.com>
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
import os
|
|
import sys
|
|
sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))
|
|
|
|
from .other_nodes import Gradually_More_Denoise_KSampler
|
|
|
|
#Some models are commented out because the code is not completed
|
|
#from vfi_models.eisai import EISAI_VFI
|
|
from vfi_models.gmfss_fortuna import GMFSS_Fortuna_VFI
|
|
from vfi_models.ifrnet import IFRNet_VFI
|
|
from vfi_models.ifunet import IFUnet_VFI
|
|
from vfi_models.m2m import M2M_VFI
|
|
from vfi_models.rife import RIFE_VFI
|
|
from vfi_models.sepconv import SepconvVFI
|
|
from vfi_models.amt import AMT_VFI
|
|
from vfi_models.film import FILM_VFI
|
|
from vfi_models.stmfnet import STMFNet_VFI
|
|
from vfi_models.flavr import FLAVR_VFI
|
|
from vfi_models.cain import CAIN_VFI
|
|
from vfi_models.atm import ATM_VFI
|
|
from vfi_models.momo import MOMO_VFI
|
|
from vfi_utils import MakeInterpolationStateList, FloatToInt
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"KSampler Gradually Adding More Denoise (efficient)": Gradually_More_Denoise_KSampler,
|
|
# "EISAI VFI": EISAI_VFI,
|
|
"GMFSS Fortuna VFI": GMFSS_Fortuna_VFI,
|
|
"IFRNet VFI": IFRNet_VFI,
|
|
"IFUnet VFI": IFUnet_VFI,
|
|
"M2M VFI": M2M_VFI,
|
|
"RIFE VFI": RIFE_VFI,
|
|
"Sepconv VFI": SepconvVFI,
|
|
"AMT VFI": AMT_VFI,
|
|
"FILM VFI": FILM_VFI,
|
|
"Make Interpolation State List": MakeInterpolationStateList,
|
|
"STMFNet VFI": STMFNet_VFI,
|
|
"FLAVR VFI": FLAVR_VFI,
|
|
"CAIN VFI": CAIN_VFI,
|
|
"ATM VFI": ATM_VFI,
|
|
"MOMO VFI": MOMO_VFI,
|
|
"VFI FloatToInt": FloatToInt
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"RIFE VFI": "RIFE VFI (recommend rife4.25+)",
|
|
"ATM VFI": "ATM VFI (only supports 2x multiplier)",
|
|
"MOMO VFI": "MOMO VFI (only supports 2x multiplier)"
|
|
}
|