vfi_models/ops/cupy_ops/__init__.py

from .costvol import *
from .sepconv import *
from .softsplat import *
from .adacof import *
from .correlation import *
from comfy.model_management import is_nvidia, get_torch_device_name, get_torch_device

def init():
    if not is_nvidia():
        raise NotImplementedError(f"CuPy ops backend only support CUDA device but found {get_torch_device_name(get_torch_device())} instead. Try Taichi ops backend by editing config.yaml")
    return