import folder_paths | |
from .animatediff.logger import logger | |
from .animatediff.utils_model import get_available_motion_models, Folders | |
from .animatediff.model_injection import prepare_dinklink_register_definitions | |
from .animatediff.nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS | |
from .animatediff import documentation | |
from .animatediff.dinklink import init_dinklink | |
if len(get_available_motion_models()) == 0: | |
logger.error(f"No motion models found. Please download one and place in: {folder_paths.get_folder_paths(Folders.ANIMATEDIFF_MODELS)}") | |
WEB_DIRECTORY = "./web" | |
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"] | |
documentation.format_descriptions(NODE_CLASS_MAPPINGS) | |
init_dinklink() | |
prepare_dinklink_register_definitions() | |