20 lines
892 B
Python
20 lines
892 B
Python
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.motion_module_ad import prepare_dinklink_motion_module_ad
|
|
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()
|
|
prepare_dinklink_motion_module_ad()
|