diff --git a/__init__.py b/__init__.py index a5635e3..b939327 100644 --- a/__init__.py +++ b/__init__.py @@ -2,9 +2,11 @@ import folder_paths from .animatediff.logger import logger from .animatediff.utils_model import get_available_motion_models, Folders from .animatediff.nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS +from .animatediff import documentation 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) diff --git a/animatediff/documentation.py b/animatediff/documentation.py new file mode 100644 index 0000000..fa345e9 --- /dev/null +++ b/animatediff/documentation.py @@ -0,0 +1,75 @@ +from typing import Union + +from .logger import logger + +def image(src): + return f'' +def video(src): + return f'